使用 Go 操作 Chrome 瀏覽器

前言

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

做法

建立新專案,並啟用 Go module。

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

程式碼

參考資料