package main

import (
"encoding/json"
"github.com/antchfx/htmlquery"
"github.com/tebeka/selenium"
"io/ioutil"
"log"
"strings"
"time"
)

func main() {
//设置日志输出格式
log.SetFlags(log.LstdFlags | log.Lshortfile)
//初始化1个chrome服务，端口使用4444
service, err := selenium.NewChromeDriverService("chromedriver.exe", 4444)
if err != nil {
log.Println(err)
return
}
defer service.Stop()
//启动本地chrome浏览器
caps := selenium.Capabilities{}
wd, err := selenium.NewRemote(caps, "#34;)
if err != nil {
log.Println(err)
return
}
defer wd.Quit()
//打开页面
err = wd.Get("#34;)
if err != nil {
log.Println(err)
return
}
for true {
//获取所有的handler
handles, err := wd.WindowHandles()
if err != nil {
log.Println(err)
return
}
//每次都跳转到最后一个handler
err = wd.SwitchWindow(handles[len(handles)-1])
if err != nil {
log.Println(err)
return
}
//获取当前的url地址
currentURL, err := wd.CurrentURL()
if err != nil {
log.Println(err)
return
}
log.Println(currentURL)
//获取cookies
cookies, err := wd.GetCookies()
if err != nil {
log.Println(err)
return
}
//尝试输出看看cookies的信息
for i, cookie := range cookies {
log.Println(i, cookie)
}
//通过json将cookies编码
marshalIndent, err := json.MarshalIndent(cookies, "", "\t")
if err != nil {
log.Println(err)
return
}
log.Println(marshalIndent)
//将cookies编码后的json写入到本地文件
err = ioutil.WriteFile(
"cks.dat",
marshalIndent,
0777,
)
if err != nil {
log.Println(err)
return
}
//通过xpath获取当前页面的a链接
pageSource, err := wd.PageSource()
if err != nil {
log.Println(err)
return
}
newReader := strings.NewReader(pageSource)
parse, err := htmlquery.Parse(newReader)
if err != nil {
log.Println(err)
return
}
nodes := htmlquery.Find(parse, "//a[@href]")
for i, node := range nodes {
log.Println(i, htmlquery.InnerText(node), htmlquery.SelectAttr(node, "href"))
}
time.Sleep(time.Second * 1)
break
}
}