95 lines
1.7 KiB
Go
95 lines
1.7 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"errors"
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
cdp "github.com/knq/chromedp"
|
||
|
|
cdptypes "github.com/knq/chromedp/cdp"
|
||
|
|
"github.com/knq/chromedp/cdp/network"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
flagPort = flag.Int("port", 8544, "port")
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
var err error
|
||
|
|
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
// setup http server
|
||
|
|
mux := http.NewServeMux()
|
||
|
|
mux.HandleFunc("/", func(res http.ResponseWriter, req *http.Request) {
|
||
|
|
buf, err := json.Marshal(req.Cookies())
|
||
|
|
if err != nil {
|
||
|
|
http.Error(res, err.Error(), http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
fmt.Fprintf(res, indexHTML, string(buf))
|
||
|
|
})
|
||
|
|
go http.ListenAndServe(fmt.Sprintf(":%d", *flagPort), mux)
|
||
|
|
|
||
|
|
// create context
|
||
|
|
ctxt, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
// create chrome instance
|
||
|
|
c, err := cdp.New(ctxt, cdp.WithLog(log.Printf))
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// run task list
|
||
|
|
var res string
|
||
|
|
err = c.Run(ctxt, setcookies(fmt.Sprintf("http://localhost:%d", *flagPort), &res))
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// shutdown chrome
|
||
|
|
err = c.Shutdown(ctxt)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// wait for chrome to finish
|
||
|
|
err = c.Wait()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Printf("passed cookies: %s", res)
|
||
|
|
}
|
||
|
|
|
||
|
|
func setcookies(host string, res *string) cdp.Tasks {
|
||
|
|
return cdp.Tasks{
|
||
|
|
cdp.ActionFunc(func(ctxt context.Context, h cdptypes.Handler) error {
|
||
|
|
success, err := network.SetCookie(host, "cookiename", "cookievalue").Do(ctxt, h)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if !success {
|
||
|
|
return errors.New("could not set cookie")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}),
|
||
|
|
cdp.Navigate(host),
|
||
|
|
cdp.Text(`#result`, res, cdp.ByID, cdp.NodeVisible),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const (
|
||
|
|
indexHTML = `<!doctype html>
|
||
|
|
<html>
|
||
|
|
<body>
|
||
|
|
<div id="result">%s</div>
|
||
|
|
</body>
|
||
|
|
</html>`
|
||
|
|
)
|