shutdown function via /halt url

wrp.go

@@ -18,6 +18,7 @@ import (
 	"log"
 	"net/http"
 	"net/url"
+	"os"
 	"strconv"
 	"time"
 
@@ -61,6 +62,15 @@ func imgServer(out http.ResponseWriter, req *http.Request) {
 	out.Write(gifbuf.Bytes())
 }
 
+func haltServer(out http.ResponseWriter, req *http.Request) {
+	log.Printf("%s Shutdown request received [%s]\n", req.RemoteAddr, req.URL.Path)
+	out.Header().Set("Content-Type", "text/plain")
+	fmt.Fprintf(out, "WRP Shutdown")
+	out.(http.Flusher).Flush()
+	cancel()
+	os.Exit(0)
+}
+
 func capture(gourl string, out http.ResponseWriter) {
 	var nodes []*cdp.Node
 	ctxx := chromedp.FromContext(ctx)
@@ -122,6 +132,7 @@ func main() {
 	http.HandleFunc("/", pageServer)
 	http.HandleFunc("/wrp.gif", imgServer)
 	http.HandleFunc("/favicon.ico", http.NotFound)
+	http.HandleFunc("/halt", haltServer)
 	log.Printf("Starting http server on %s\n", addr)
 	http.ListenAndServe(addr, nil)
 }