使用 Go 操作 Chrome 瀏覽器

Others
, , ,

前言

以下範例讓 Go 開啟一個 Chrome 瀏覽器，到 The Go Playground 輸入一些程式碼，並點選執行按鈕。

做法

建立專案。

1
2
mkdir go-selenium-example
cd go-selenium-example

初始化 Go Modules。

1
go mod init github.com/memochou1993/go-selenium-example

下載 tebeka/selenium 套件。

1
go get -t -d github.com/tebeka/selenium

到 ChromeDriver 下載 chromedriver 執行檔，並將執行檔加入至環境變數。

新增一個 main.go 檔：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package main

import (
	"fmt"
	"log"
	"os"
	"strings"
	"time"

	"github.com/tebeka/selenium"
)

const (
	chromeDriverPath = "chromedriver"
	port             = 8080
)

func main() {
	// 啟動一個 WebDriver 實體
	opts := []selenium.ServiceOption{
		selenium.Output(os.Stderr),
	}
	service, err := selenium.NewChromeDriverService(chromeDriverPath, port, opts...)
	if err != nil {
		log.Fatalln(err.Error())
	}
	defer service.Stop()

	// 連線到運作在本地的 WebDriver 實體
	caps := selenium.Capabilities{"browserName": "chrome"}
	wd, err := selenium.NewRemote(caps, fmt.Sprintf("http://localhost:%d/wd/hub", port))
	if err != nil {
		log.Fatalln(err.Error())
	}
	defer wd.Quit()

	// 進入指定頁面
	if err := wd.Get("http://play.golang.org"); err != nil {
		log.Fatalln(err.Error())
	}

	// 取得輸入欄位節點
	elem, err := wd.FindElement(selenium.ByCSSSelector, "#code")
	if err != nil {
		log.Fatalln(err.Error())
	}
	if err := elem.Clear(); err != nil {
		log.Fatalln(err.Error())
	}

	code := `package main
import "fmt"
func main() { fmt.Println("Hello World!") }
`

	// 輸入程式碼
	if err = elem.SendKeys(code); err != nil {
		log.Fatalln(err.Error())
	}

	// 執行程式碼
	btn, err := wd.FindElement(selenium.ByCSSSelector, "#run")
	if err != nil {
		log.Fatalln(err.Error())
	}
	if err := btn.Click(); err != nil {
		log.Fatalln(err.Error())
	}

	// 取得輸出欄位節點
	outputDiv, err := wd.FindElement(selenium.ByCSSSelector, "#output")
	if err != nil {
		log.Fatalln(err.Error())
	}

	// 等待執行結果
	var output string
	for {
		output, err = outputDiv.Text()
		if err != nil {
			log.Fatalln(err.Error())
		}
		if output != "Waiting for remote server..." {
			break
		}
		time.Sleep(time.Millisecond * 100)
	}

	// 印出執行結果
	fmt.Printf("%s", strings.Replace(output, "\n\n", "\n", -1))

	time.Sleep(5 * time.Second)
}

執行。

1
go run main.go

程式碼

參考資料