From 936cb97bc0abfa48a6cca102411abc9d15ff6e0d Mon Sep 17 00:00:00 2001
From: Antoni Sawicki <tenox@google.com>
Date: Mon, 3 Jun 2019 17:50:16 -0700
Subject: [PATCH] added headed mode and chromedp debug output

---
 wrp.go | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/wrp.go b/wrp.go
index 4256032..55d72e3 100644
--- a/wrp.go
+++ b/wrp.go
@@ -270,12 +270,30 @@ func capture(gourl string, w int64, h int64, s float64, co int, p int64, i bool,
 }
 
 func main() {
-	ctx, cancel = chromedp.NewContext(context.Background())
-	defer cancel()
 	var addr string
+	var head,headless bool
+	var debug bool
 	flag.StringVar(&addr, "l", ":8080", "Listen address:port, default :8080")
-	flag.StringVar(&iaddr, "i", "", "Address:port prefix for ISMAP Redirect")
+	flag.StringVar(&iaddr, "i", "", "Address:port prefix for ISMAP Redirect, otherwise http-equiv/redirect will be used")
+	flag.BoolVar(&head, "h", false, "Headed mode - display browser window")
+	flag.BoolVar(&debug, "d", false, "Debug ChromeDP")
 	flag.Parse()
+	if head {
+		headless = false
+	} else {
+		headless = true
+	}
+	opts := append(chromedp.DefaultExecAllocatorOptions[:],
+        chromedp.Flag("headless", headless),
+	)
+	actx, cancel := chromedp.NewExecAllocator(context.Background(), opts...)
+	defer cancel()
+	if debug {
+		ctx, cancel = chromedp.NewContext(actx, chromedp.WithDebugf(log.Printf))
+	} else {
+		ctx, cancel = chromedp.NewContext(actx)
+	}
+	defer cancel()
 	rand.Seed(time.Now().UnixNano())
 	http.HandleFunc("/", pageServer)
 	http.HandleFunc("/img/", imgServer)