使用 chromedp 操作 chrome

go get -u github.com/chromedp/chromedp

import "github.com/chromedp/chromedp"

ctx, cancel := chromedp.NewContext(context.Background())
defer cancel()

// 指定 Chrome 路径
opts := append(chromedp.DefaultExecAllocatorOptions[:],
	//chromedp.ExecPath("C:/Program Files/Google/Chrome/Application/chrome.exe"),        // Windows
	chromedp.ExecPath("/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"), // MacOS
	chromedp.Flag("headless", false),
)

allocCtx, cancel := chromedp.NewExecAllocator(context.Background(), opts...)
defer cancel()

ctx, cancel := chromedp.NewContext(allocCtx)
defer cancel()

allocatorContext, _ := chromedp.NewRemoteAllocator(context.Background(), "ws://10.20.30.1:9222/")
defer cancel()

chromedp.Run(ctx,
    chromedp.Navigate("https://www.example.com"),
)

chromedp.Run(ctx,
	chromedp.Emulate(device.IPhone15ProMax),
	chromedp.Navigate("https://m.example.com"),
)

// 同步执行示例
var title string
chromedp.Run(ctx,
	chromedp.Evaluate(`document.title`, &title),
)

// 异步 Promise 处理
chromedp.Run(ctx,
	chromedp.Evaluate(
		`someReturnPromiseFunc()`,
		&result,
		func(p *runtime.EvaluateParams) *runtime.EvaluateParams {
			return p.WithAwaitPromise(true) // 等待 Promise 完成
		}),
)

// 获取元素的 text
var res string
chromedp.Run(ctx,
	chromedp.Navigate(`https://pkg.go.dev/time`),
	chromedp.Text(`.Documentation-overview`, &res, chromedp.NodeVisible),
)

// 打开一个页面, 等待一个元素渲染完成, 点击
var example string
chromedp.Run(ctx,
	chromedp.Navigate(`https://pkg.go.dev/time`),
	// 等待页面的 footer 元素可见 (即，页面已经加载完成后)
	chromedp.WaitVisible(`body > footer`),
	// 找到并点击页面上的 "Example" 连接
	chromedp.Click(`#example-After`, chromedp.NodeVisible),
	// 获取 textarea 区域的文本
	chromedp.Value(`#example-After textarea`, &example),
)

func doGithubSearch() {
	// create context
	ctx, cancel := chromedp.NewContext(context.Background(), chromedp.WithDebugf(log.Printf))
	defer cancel()

	// run task list
	var res string
	err := chromedp.Run(ctx, submit(`https://github.com/search`, `//input[@name="q"]`, `chromedp`, &res))
	if err != nil {
		log.Fatal(err)
	}

	log.Printf("got: `%s`", strings.TrimSpace(res))
}

func submit(urlStr, sel, q string, res *string) chromedp.Tasks {
	return chromedp.Tasks{
		chromedp.Navigate(urlStr),
		chromedp.WaitVisible(sel),
		chromedp.SendKeys(sel, q),
		chromedp.Submit(sel),
		chromedp.WaitVisible(`//*[contains(., 'repository results')]`),
		chromedp.Text(`(//*//ul[contains(@class, "repo-list")]/li[1]//p)[1]`, res),
	}
}

package main

import (
	"context"
	"log"
	"os"

	"github.com/chromedp/chromedp"
)

func main() {
	// 新建 context
	ctx, cancel := chromedp.NewContext(
		context.Background(),
		chromedp.WithDebugf(log.Printf), // 打印 debug 日志
	)
	defer cancel()

	// 捕获网页的一个元素进行屏幕截图
	var buf []byte
	if err := chromedp.Run(ctx, elementScreenshot(`https://pkg.go.dev/`, `img.Homepage-logo`, &buf)); err != nil {
		log.Fatal(err)
	}
	if err := os.WriteFile("elementScreenshot.png", buf, 0o644); err != nil {
		log.Fatal(err)
	}

	// 捕获整个浏览器视口，返回质量为 90 的 png
	if err := chromedp.Run(ctx, fullScreenshot(`https://brank.as/`, 90, &buf)); err != nil {
		log.Fatal(err)
	}
	if err := os.WriteFile("fullScreenshot.png", buf, 0o644); err != nil {
		log.Fatal(err)
	}

	log.Printf("wrote elementScreenshot.png and fullScreenshot.png")
}

// elementScreenshot 截取特定元素的屏幕截图。
func elementScreenshot(urlStr, sel string, res *[]byte) chromedp.Tasks {
	return chromedp.Tasks{
		chromedp.Navigate(urlStr),
		chromedp.Screenshot(sel, res, chromedp.NodeVisible),
	}
}

// fullScreenshot 会截取整个浏览器视口的屏幕截图。
//
// 注意: chromedp.FullScreenshot 会覆盖设备的emulation 设置。 Use
// 使用 device.Reset 重置 emulation 和 viewport 设置。
func fullScreenshot(urlstr string, quality int, res *[]byte) chromedp.Tasks {
	return chromedp.Tasks{
		chromedp.Navigate(urlstr),
		chromedp.FullScreenshot(res, quality),
	}
}

package main

import (
	"context"
	"fmt"
	"log"
	"os"

	"github.com/chromedp/cdproto/page"
	"github.com/chromedp/chromedp"
)

func main() {
	// 新建 context
	ctx, cancel := chromedp.NewContext(context.Background())
	defer cancel()

	// 抓取屏幕为 PDF
	var buf []byte
	if err := chromedp.Run(ctx, printToPDF(`https://www.baidu.com/`, &buf)); err != nil {
		log.Fatal(err)
	}

	// 保存为本地的 sample.pdf 文件
	if err := os.WriteFile("sample.pdf", buf, 0o644); err != nil {
		log.Fatal(err)
	}
	fmt.Println("wrote sample.pdf")
}

// 打印特定的 pdf 页面.
func printToPDF(urlStr string, res *[]byte) chromedp.Tasks {
	return chromedp.Tasks{
		chromedp.Navigate(urlStr),
		chromedp.ActionFunc(func(ctx context.Context) error {
			buf, _, err := page.PrintToPDF().WithPrintBackground(false).Do(ctx)
			if err != nil {
				return err
			}
			*res = buf
			return nil
		}),
	}
}