1 // Command click is a chromedp example demonstrating how to use a selector to
 2 // click on an element.
 3 package main
 4 
 5 import (
 6     "context"
 7     "fmt"
 8     "github.com/chromedp/cdproto/page"
 9     "github.com/chromedp/chromedp"
10     "github.com/chromedp/chromedp/runner"
11     "log"
12     "os"
13     "time"
14 )
15 
16 func main() {
17     var err error
18 
19     // create context
20     ctxt, cancel := context.WithCancel(context.Background())
21     defer cancel()
22 
23     // create chrome instance
24     //c, err := chromedp.New(ctxt, chromedp.WithLog(log.Printf))
25     //if err != nil {
26     //    log.Fatal(err)
27     //}
28 
29     c, err := NewHeadless(ctxt, `https://xxx.xx`)
30     if err != nil {
31         log.Fatalf("new headless: %s",err)
32     }
33 
34     // run task list
35     err = c.Run(ctxt, click())
36     if err != nil {
37         log.Fatalf("task err: %s",err)
38     }
39 
40 
41     // shutdown chrome
42     err = c.Shutdown(ctxt)
43     if err != nil {
44         log.Fatalf("shutdown err: %s",err)
45     }
46 
47     // wait for chrome to finish
48     err = c.Wait()
49     if err != nil {
50         log.Fatalf("wait err: %s",err)
51     }
52 }
53 
54 func click() chromedp.Tasks {
55     //url := `https://free-ss.site`
56     //sect := "div.main > div > table > tbody >tr:nth-child(2) > td:nth-child(9) > a"
57     sect := `//a[@href="v/443.json"]`
58     wd,_ := os.Getwd()
59     fmt.Println(wd)
60     return chromedp.Tasks{
61         //chromedp.Navigate(url),
62         chromedp.WaitVisible(sect),
63         page.SetDownloadBehavior(page.SetDownloadBehaviorBehaviorAllow).WithDownloadPath(wd),
64         chromedp.Click(sect, chromedp.NodeVisible),
65         chromedp.Sleep(3 * time.Second),
66     }
67 }
68 
69 
70 // NewHeadless 创建headless chrome实例
71 func NewHeadless(ctx context.Context, starturl string) (*chromedp.CDP, error) {
72     // runner.Flag设置启动headless chrome时的命令行参数
73     // Windows用户需要设置runner.Flag("disable-gpu", true)，具体信息参见文档的FAQ
74     run, err := runner.New(runner.Flag("headless", false),runner.StartURL(starturl))
75 
76     if err != nil {
77         return nil, err
78     }
79 
80     // run.Start启动实例
81     err = run.Start(ctx)
82     if err != nil {
83         return nil, err
84     }
85 
86     // 默认情况chromedp会输出大量log，因为是示例所以选择屏蔽，dropChromeLogs为自定义函数，形式为func(string, ...interface{}){}
87     // 使用runner初始化chromedp实例
88     // 实例在使用完毕后需要调用c.Shutdown()来释放资源
89     //c, err := chromedp.New(ctx, chromedp.WithRunner(run), chromedp.WithErrorf(dropChromeLogs))
90     c, err := chromedp.New(ctx, chromedp.WithRunner(run), chromedp.WithLog(log.Printf))
91     if err != nil {
92         return nil, err
93     }
94 
95     return c, nil
96 }