Diff
diff --git a/document/main.go b/document/main.go
index 185d826..9f20646 100644
--- a/document/main.go
+++ b/document/main.go
@@ -3,0 +4 @@ import (
+ "bufio"
@@ -7,0 +9,5 @@ import (
+ "net/url"
+ "os"
+ "path/filepath"
+ "regexp"
+ "strings"
@@ -103,0 +110,161 @@ func (doc Document) Open(index string, script func(*element.Node)) {
+
+// func (doc Document) CreateElement(t string) element.Node {
+// n := element.Node{TagName: t, Properties: element.Properties{Node: &html.Node{}}}
+// cstyle.InitNode(&n, doc.CSS)
+// return n
+// }
+
+func parse(path string) Window {
+ file, err := os.Open(path)
+ check(err)
+ defer file.Close()
+
+ scanner := bufio.NewScanner(file)
+ var htmlContent string
+
+ for scanner.Scan() {
+ htmlContent += scanner.Text() + "\n"
+ }
+
+ check(scanner.Err())
+ // println(encapsulateText(htmlContent))
+ doc, err := html.Parse(strings.NewReader(encapsulateText(htmlContent)))
+ check(err)
+
+ // Extract stylesheet link tags and style tags
+ stylesheets := extractStylesheets(doc, filepath.Dir(path))
+ styleTags := extractStyleTags(doc)
+
+ d := Window{
+ StyleSheets: stylesheets,
+ StyleTags: styleTags,
+ DOM: doc,
+ }
+
+ return d
+}
+
+func extractStylesheets(n *html.Node, baseDir string) []string {
+ var stylesheets []string
+
+ var dfs func(*html.Node)
+ dfs = func(node *html.Node) {
+ if node.Type == html.ElementNode && node.Data == "link" {
+ var href string
+ isStylesheet := false
+
+ for _, attr := range node.Attr {
+ if attr.Key == "rel" && attr.Val == "stylesheet" {
+ isStylesheet = true
+ } else if attr.Key == "href" {
+ href = attr.Val
+ }
+ }
+
+ if isStylesheet {
+ resolvedHref := localizePath(baseDir, href)
+ stylesheets = append(stylesheets, resolvedHref)
+ }
+ }
+
+ for c := node.FirstChild; c != nil; c = c.NextSibling {
+ dfs(c)
+ }
+ }
+
+ dfs(n)
+ return stylesheets
+}
+
+func extractStyleTags(n *html.Node) []string {
+ var styleTags []string
+
+ var dfs func(*html.Node)
+ dfs = func(node *html.Node) {
+ if node.Type == html.ElementNode && node.Data == "style" {
+ var styleContent strings.Builder
+ for c := node.FirstChild; c != nil; c = c.NextSibling {
+ if c.Type == html.TextNode {
+ styleContent.WriteString(c.Data)
+ }
+ }
+ styleTags = append(styleTags, styleContent.String())
+ }
+
+ for c := node.FirstChild; c != nil; c = c.NextSibling {
+ dfs(c)
+ }
+ }
+
+ dfs(n)
+ return styleTags
+}
+
+func localizePath(rootPath, filePath string) string {
+ // Check if the file path has a scheme, indicating it's a URL
+ u, err := url.Parse(filePath)
+ if err == nil && u.Scheme != "" {
+ return filePath
+ }
+
+ // Join the root path and the file path to create an absolute path
+ absPath := filepath.Join(rootPath, filePath)
+
+ // If the absolute path is the same as the original path, return it
+ if absPath == filePath {
+ return filePath
+ }
+
+ return "./" + absPath
+}
+
+func encapsulateText(htmlString string) string {
+ htmlString = removeHTMLComments(htmlString)
+ openOpen := regexp.MustCompile(`(<\w+[^>]*>)([^<]+)(<\w+[^>]*>)`)
+ closeOpen := regexp.MustCompile(`(]*>)([^<]+)(<\w+[^>]*>)`)
+ closeClose := regexp.MustCompile(`(]*>)([^<]+)(]*>)`)
+ a := matchFactory(openOpen)
+ t := openOpen.ReplaceAllStringFunc(htmlString, a)
+ b := matchFactory(closeOpen)
+ u := closeOpen.ReplaceAllStringFunc(t, b)
+ c := matchFactory(closeClose)
+ v := closeClose.ReplaceAllStringFunc(u, c)
+ return v
+}
+
+func matchFactory(re *regexp.Regexp) func(string) string {
+ return func(match string) string {
+ submatches := re.FindStringSubmatch(match)
+ if len(submatches) != 4 {
+ return match
+ }
+
+ // Process submatches
+ if len(removeWhitespace(submatches[2])) > 0 {
+ return submatches[1] + "" + submatches[2] + "" + submatches[3]
+ } else {
+ return match
+ }
+ }
+}
+func removeWhitespace(htmlString string) string {
+ // Remove extra white space
+ reSpaces := regexp.MustCompile(`\s+`)
+ htmlString = reSpaces.ReplaceAllString(htmlString, " ")
+
+ // Trim leading and trailing white space
+ htmlString = strings.TrimSpace(htmlString)
+
+ return htmlString
+}
+
+func removeHTMLComments(htmlString string) string {
+ re := regexp.MustCompile(``)
+ return re.ReplaceAllString(htmlString, "")
+}
+
+func check(e error) {
+ if e != nil {
+ panic(e)
+ }
+}
