ref: 314a17cf6d0deaf40f45786fe5d6db27a892c0a3
dir: /nodes/nodes.go/
package nodes import ( "bytes" "fmt" "golang.org/x/net/html" "github.com/chris-ramon/douceur/css" "github.com/psilva261/opossum/logger" "github.com/psilva261/opossum/style" "strings" ) var log *logger.Logger func SetLogger(l *logger.Logger) { log = l } // Node represents a node at the render stage. It // represents a subTree or just a single html node. type Node struct { DomSubtree *html.Node Text string Wrappable bool Attrs []html.Attribute style.Map Children []*Node Parent *Node } // NewNodeTree propagates the cascading styles to the leaves // // First applies the parent style and at the end the local style attribute's style is attached. func NewNodeTree(doc *html.Node, ps style.Map, nodeMap map[*html.Node]style.Map, parent *Node) (n *Node) { //fmt.Printf("<%v %v>\n", doc.Data, doc.Attr) //fmt.Printf(" nodeMap=%+v\n", nodeMap) ncs := style.Map{ Declarations: make(map[string]css.Declaration), } //fmt.Printf(" ps=%+v\n", ps) ncs = ps.ApplyChildStyle(ncs, false) //fmt.Printf(" ncs=%+v\n", ncs) // add from matching selectors // (keep only inheriting properties from parent node) if m, ok := nodeMap[doc]; ok { //fmt.Printf(" m=%+v\n", m) ncs = ncs.ApplyChildStyle(m, false) } else { //fmt.Printf(" no nodeMap entry\n") } //fmt.Printf(" ncs=%+v\n", ncs) // add style attribute // (keep all properties that already match) styleAttr := style.NewMap(doc) ncs = ncs.ApplyChildStyle(styleAttr, true) //fmt.Printf(" ncs=%+v\n", ncs) data := doc.Data if doc.Type == html.ElementNode { data = strings.ToLower(data) } n = &Node{ //Data: data, //Type: doc.Type, DomSubtree: doc, Attrs: doc.Attr, Map: ncs, Children: make([]*Node, 0, 2), Parent: parent, } n.Wrappable = doc.Type == html.TextNode || doc.Data == "span" // TODO: probably this list needs to be extended if doc.Type == html.TextNode { n.Text = filterText(doc.Data) n.Map = style.Map{ Declarations: make(map[string]css.Declaration), } n.Map.Declarations["display"] = css.Declaration{ Property: "display", Value: "inline", } } i := 0 for c := doc.FirstChild; c != nil; c = c.NextSibling { if c.Type != html.CommentNode { n.Children = append(n.Children, NewNodeTree(c, ncs, nodeMap, n)) i++ } } return } // filterText removes line break runes (TODO: add this later but handle properly) func filterText(t string) (text string) { return strings.ReplaceAll(t, "", "") } func (n Node) Type() html.NodeType { return n.DomSubtree.Type } func (n Node) Data() string { if n.DomSubtree == nil { return "" } return n.DomSubtree.Data } // Ancestor of tag func (n *Node) Ancestor(tag string) *Node { if n.DomSubtree == nil { return nil } log.Printf("<%v>.ParentForm()", n.DomSubtree.Data) if n.DomSubtree.Data == tag { log.Printf(" I'm a %v :-)", tag) return n } if n.Parent != nil { log.Printf(" go to my parent") return n.Parent.Ancestor(tag) } return nil } func (n *Node) Find(tag string) (c *Node) { for _, cc := range n.Children { if cc.Data() == tag { return cc } else if f := cc.Find(tag); f != nil { return f } } return } func (n *Node) Attr(k string) string { for _, a := range n.Attrs { if a.Key == k { return a.Val } } return "" } func (n *Node) HasAttr(k string) bool { for _, a := range n.Attrs { if a.Key == k { return true } } return false } // QueryRef relative to html > body func (n *Node) QueryRef() string { nRef, ok := n.queryRef() if ok && strings.Contains(nRef, "#") { return nRef } path := make([]string, 0, 5) if n.Type() != html.TextNode { if ok { path = append(path, nRef) } } for p := n.Parent; p != nil; p = p.Parent { if part := p.Data(); part != "html" && part != "body" { if pRef, ok := p.queryRef(); ok { path = append([]string{pRef}, path...) if strings.Contains(pRef, "#") { break } } } } return strings.TrimSpace(strings.Join(path, " > ")) } func (n *Node) queryRef() (ref string, ok bool) { if n.DomSubtree == nil || n.Type() != html.ElementNode { return } if id := n.Attr("id"); id != "" { // https://stackoverflow.com/questions/605630/how-to-select-html-nodes-by-id-with-jquery-when-the-id-contains-a-dot id = strings.ReplaceAll(id, `.`, `\\.`) return "#" + id, true } ref = n.Data() if n.Parent == nil { return ref, true } i := 1 for _, c := range n.Parent.Children { if c == n { break } if c.Type() == html.ElementNode { i++ } } ref += fmt.Sprintf(":nth-child(%v)", i) return ref, true } func IsPureTextContent(n Node) bool { if n.Text != "" { return true } for _, c := range n.Children { if c.Text == "" { return false } } return true } func ContentFrom(n Node) string { var content string if n.Text != "" && n.Type() == html.TextNode && !n.Map.IsDisplayNone() { content += n.Text } for _, c := range n.Children { if !c.Map.IsDisplayNone() { content += ContentFrom(*c) } } return strings.TrimSpace(content) } func (n *Node) Serialized() (string, error) { var b bytes.Buffer err := html.Render(&b, n.DomSubtree) return b.String(), err } // SetText by replacing child nodes with a TextNode containing t. func (n *Node) SetText(t string) { d := n.DomSubtree if len(n.Children) != 1 || n.Type() != html.TextNode { for d.FirstChild != nil { d.RemoveChild(d.FirstChild) } c := &html.Node{ Parent: d, Type: html.TextNode, } d.FirstChild = c n.Children = []*Node{ &Node{ DomSubtree: c, Wrappable: true, Parent: n, }, } } n.Children[0].Text = t n.Children[0].DomSubtree.Data = t return }