// Package main ...
package main
import (
"fmt"
"io"
"log"
"net/http"
"net/http/httptest"
"os"
"strings"
"github.com/go-rod/rod"
"github.com/go-rod/rod/lib/proto"
)
func main() {
// create a test server to serve the page
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
_, _ = fmt.Fprint(w, `
Title
`,
)
}))
defer ts.Close()
page := rod.New().MustConnect().MustPage(ts.URL)
node, err := page.MustElement("body").Describe(-1, true)
if err != nil {
log.Fatal(err)
}
printNodes(os.Stdout, node.Children, "", " ")
}
func printNodes(w io.Writer, nodes []*proto.DOMNode, padding, indent string) {
for _, node := range nodes {
switch {
case node.NodeName == "#text":
fmt.Fprintf(w, "%s#text: %q\n", padding, node.NodeValue)
default:
fmt.Fprintf(w, "%s%s:\n", padding, strings.ToLower(node.NodeName))
if n := len(node.Attributes); n > 0 {
fmt.Fprintf(w, "%sattributes:\n", padding+indent)
for i := 0; i < n; i += 2 {
fmt.Fprintf(w, "%s%s: %q\n", padding+indent+indent, node.Attributes[i], node.Attributes[i+1])
}
}
}
if node.ChildNodeCount != nil {
fmt.Fprintf(w, "%schildren:\n", padding+indent)
printNodes(w, node.Children, padding+indent+indent, indent)
}
}
}