Updating forecast example

Author: kenshaw

forecast/main.go

@@ -22,12 +22,15 @@ import (
 
 	"github.com/chromedp/cdproto/cdp"
 	"github.com/chromedp/cdproto/dom"
+	"github.com/chromedp/cdproto/runtime"
 	"github.com/chromedp/chromedp"
 	"github.com/kenshaw/rasterm"
 	"golang.org/x/exp/maps"
 )
 
-const dataSel = `div[data-ve-view]`
+const hdrSel = `#taw`
+
+const dataSel = `#wob_wc`
 
 const svgSel = `#wob_d svg path`
 
@@ -67,9 +70,9 @@ func run(ctx context.Context, verbose bool, timeout time.Duration, query, lang,
 		return fmt.Errorf("invalid unit %q", unit)
 	}
 	switch typ = strings.ToLower(typ); typ {
-	case "temp":
-		typ = ""
-	case "", "rain", "wind":
+	case "":
+		typ = "temp"
+	case "temp", "rain", "wind":
 	default:
 		return fmt.Errorf("invalid type %q", typ)
 	}
@@ -104,31 +107,44 @@ func run(ctx context.Context, verbose bool, timeout time.Duration, query, lang,
 	ctx, cancel = context.WithTimeout(ctx, timeout)
 	defer cancel()
 
-	// get nodes
-	var nodes []*cdp.Node
+	// get
+	var nodes, dataNodes []*cdp.Node
 	if err := chromedp.Run(ctx,
 		chromedp.Navigate("https://www.google.com/search?"+v.Encode()),
-		chromedp.WaitVisible(dataSel, chromedp.ByQuery),
-		chromedp.WaitVisible(svgSel, chromedp.ByQuery),
-		chromedp.Nodes(dataSel, &nodes, chromedp.ByQuery, chromedp.NodeVisible),
+		chromedp.QueryAfter(hdrSel, func(ctx context.Context, id runtime.ExecutionContextID, n ...*cdp.Node) error {
+			nodes = append(nodes, n[0])
+			return nil
+		}, chromedp.ByQuery, chromedp.NodeVisible),
+		chromedp.QueryAfter(dataSel, func(ctx context.Context, id runtime.ExecutionContextID, n ...*cdp.Node) error {
+			nodes = append(nodes, n[0])
+			return nil
+		}, chromedp.ByQuery, chromedp.NodeVisible),
+		chromedp.Nodes(dataSel, &dataNodes, chromedp.ByQuery, chromedp.NodeVisible),
 		chromedp.ActionFunc(func(ctx context.Context) error {
-			return dom.RequestChildNodes(nodes[0].NodeID).WithDepth(-1).Do(ctx)
+			return dom.RequestChildNodes(dataNodes[0].NodeID).WithDepth(-1).Do(ctx)
 		}),
 	); err != nil {
 		return err
 	}
 
 	// click on unit
 	if unit != "" {
-		if node := findNode(`°`+unit, nodes); node != nil {
+		if node := findNode(`°`+unit, dataNodes); node != nil {
 			_ = chromedp.Run(ctx, chromedp.MouseClickNode(node))
 		}
 	}
 
 	// click on type
-	if typ != "" {
+	if typ != "temp" {
 		_ = chromedp.Run(ctx, chromedp.Click("wob_"+typ, chromedp.ByID))
 	}
+	// hide other types
+	_ = chromedp.Run(ctx, chromedp.QueryAfter(`#wob_d > div:first-child > *:not(#wob_`+typ+`)`, func(ctx context.Context, id runtime.ExecutionContextID, nodes ...*cdp.Node) error {
+		for _, n := range nodes {
+			_ = dom.SetAttributeValue(n.NodeID, "style", "display:none;").Do(ctx)
+		}
+		return nil
+	}))
 
 	// click on day
 	if day != 0 {
@@ -137,7 +153,7 @@ func run(ctx context.Context, verbose bool, timeout time.Duration, query, lang,
 
 	// capture screenshot
 	var buf []byte
-	if err := chromedp.Run(ctx, chromedp.ScreenshotScale(dataSel, scale, &buf, chromedp.ByQuery)); err != nil {
+	if err := chromedp.Run(ctx, chromedp.ScreenshotNodes(nodes, scale, &buf)); err != nil {
 		return err
 	}
 	// decode png

go.mod

@@ -5,8 +5,8 @@ go 1.21
 toolchain go1.21.6
 
 require (
-	github.com/chromedp/cdproto v0.0.0-20231011050154-1d073bb38998
-	github.com/chromedp/chromedp v0.9.3
+	github.com/chromedp/cdproto v0.0.0-20240127002248-bd7a66284627
+	github.com/chromedp/chromedp v0.9.4
 	github.com/kenshaw/rasterm v0.1.8
 	golang.org/x/exp v0.0.0-20240119083558-1b970713d09a
 )
@@ -15,7 +15,7 @@ require (
 	github.com/chromedp/sysutil v1.0.0 // indirect
 	github.com/gobwas/httphead v0.1.0 // indirect
 	github.com/gobwas/pool v0.2.1 // indirect
-	github.com/gobwas/ws v1.3.0 // indirect
+	github.com/gobwas/ws v1.3.2 // indirect
 	github.com/josharian/intern v1.0.0 // indirect
 	github.com/mailru/easyjson v0.7.7 // indirect
 	github.com/mattn/go-sixel v0.0.5 // indirect

go.sum

@@ -1,15 +1,15 @@
-github.com/chromedp/cdproto v0.0.0-20231011050154-1d073bb38998 h1:2zipcnjfFdqAjOQa8otCCh0Lk1M7RBzciy3s80YAKHk=
-github.com/chromedp/cdproto v0.0.0-20231011050154-1d073bb38998/go.mod h1:GKljq0VrfU4D5yc+2qA6OVr8pmO/MBbPEWqWQ/oqGEs=
-github.com/chromedp/chromedp v0.9.3 h1:Wq58e0dZOdHsxaj9Owmfcf+ibtpYN1N0FWVbaxa/esg=
-github.com/chromedp/chromedp v0.9.3/go.mod h1:NipeUkUcuzIdFbBP8eNNvl9upcceOfWzoJn6cRe4ksA=
+github.com/chromedp/cdproto v0.0.0-20240127002248-bd7a66284627 h1:L5rJ/yzLfSU3kcjsjq11xYDqAdianisL21CXQ/08Zag=
+github.com/chromedp/cdproto v0.0.0-20240127002248-bd7a66284627/go.mod h1:GKljq0VrfU4D5yc+2qA6OVr8pmO/MBbPEWqWQ/oqGEs=
+github.com/chromedp/chromedp v0.9.4 h1:HWAQlqLbqvJ6MkHo+e6kNEE18RRr7OBPB/UQjMX2EBA=
+github.com/chromedp/chromedp v0.9.4/go.mod h1:6wuMtDoRuOCeYF3gP7E+hxen58OsvlloigZZtHMHERk=
 github.com/chromedp/sysutil v1.0.0 h1:+ZxhTpfpZlmchB58ih/LBHX52ky7w2VhQVKQMucy3Ic=
 github.com/chromedp/sysutil v1.0.0/go.mod h1:kgWmDdq8fTzXYcKIBqIYvRRTnYb9aNS9moAV0xufSww=
 github.com/gobwas/httphead v0.1.0 h1:exrUm0f4YX0L7EBwZHuCF4GDp8aJfVeBrlLQrs6NqWU=
 github.com/gobwas/httphead v0.1.0/go.mod h1:O/RXo79gxV8G+RqlR/otEwx4Q36zl9rqC5u12GKvMCM=
 github.com/gobwas/pool v0.2.1 h1:xfeeEhW7pwmX8nuLVlqbzVc7udMDrwetjEv+TZIz1og=
 github.com/gobwas/pool v0.2.1/go.mod h1:q8bcK0KcYlCgd9e7WYLm9LpyS+YeLd8JVDW6WezmKEw=
-github.com/gobwas/ws v1.3.0 h1:sbeU3Y4Qzlb+MOzIe6mQGf7QR4Hkv6ZD0qhGkBFL2O0=
-github.com/gobwas/ws v1.3.0/go.mod h1:hRKAFb8wOxFROYNsT1bqfWnhX+b5MFeJM9r2ZSwg/KY=
+github.com/gobwas/ws v1.3.2 h1:zlnbNHxumkRvfPWgfXu8RBwyNR1x8wh9cf5PTOCqs9Q=
+github.com/gobwas/ws v1.3.2/go.mod h1:hRKAFb8wOxFROYNsT1bqfWnhX+b5MFeJM9r2ZSwg/KY=
 github.com/josharian/intern v1.0.0 h1:vlS4z54oSdjm0bgjRigI+G1HpF+tI+9rE5LLzOg8HmY=
 github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y=
 github.com/kenshaw/rasterm v0.1.8 h1:XHaCBNB0H9HE20z4g2rUrSsObW/PJe1Hn4Ya2iZDi5Y=