package main import ( "context" "fmt" "log" "net/http" "strconv" "strings" "time" "github.com/chromedp/cdproto/cdp" "github.com/chromedp/cdproto/dom" "github.com/chromedp/chromedp" ) var ( ctx context.Context cancel context.CancelFunc scrcap []byte ) func pageServer(out http.ResponseWriter, req *http.Request) { req.ParseForm() furl := req.Form["url"] var url string if len(furl) >= 1 && len(furl[0]) > 4 { url = furl[0] } else { url = "https://www.google.com/" } log.Printf("%s Page Reqest for %s URL=%s\n", req.RemoteAddr, req.URL.Path, url) out.Header().Set("Content-Type", "text/html") fmt.Fprintf(out, "\nWRP %s\n", url) fmt.Fprintf(out, "
URL: ", url) fmt.Fprintf(out, "

\n") if len(url) > 4 { capture(url, out) } fmt.Fprintf(out, "\n\n") } func imgServer(out http.ResponseWriter, req *http.Request) { log.Printf("%s Img Reqest for %s\n", req.RemoteAddr, req.URL.Path) out.Header().Set("Content-Type", "image/png") out.Header().Set("Content-Length", strconv.Itoa(len(scrcap))) out.Write(scrcap) } func capture(url string, out http.ResponseWriter) { var nodes []*cdp.Node ctxx := chromedp.FromContext(ctx) var target string log.Printf("Caputure Request for %s\n", url) chromedp.Run(ctx, chromedp.Navigate(url), chromedp.Sleep(time.Second*2), chromedp.CaptureScreenshot(&scrcap), chromedp.Nodes("a", &nodes, chromedp.ByQueryAll)) fmt.Fprintf(out, "\"wrp\"\n\n") for _, n := range nodes { b, err := dom.GetBoxModel().WithNodeID(n.NodeID).Do(cdp.WithExecutor(ctx, ctxx.Target)) if strings.HasPrefix(n.AttributeValue("href"), "/") { target = fmt.Sprintf("/?url=%s%s", url, n.AttributeValue("href")) } else { target = fmt.Sprintf("/?url=%s", n.AttributeValue("href")) } if err == nil && len(b.Content) > 6 { fmt.Fprintf(out, "\"%s\"\n", b.Content[0], b.Content[1], b.Content[4], b.Content[5], n.AttributeValue("href"), n.AttributeValue("href"), target) } } fmt.Fprintf(out, "\n") log.Printf("Done with caputure for %s\n", url) } func main() { ctx, cancel = chromedp.NewContext(context.Background()) defer cancel() http.HandleFunc("/", pageServer) http.HandleFunc("/wrp.png", imgServer) log.Printf("Starting http server on :8080\n") http.ListenAndServe(":8080", nil) }