Macintosh-Tools/wrp
1
0
Fork 0
mirror of https://github.com/tenox7/wrp.git synced 2024-06-06 04:29:31 +00:00
Code Issues Projects Releases Wiki Activity

add width height and scale input boxes

Browse Source
This commit is contained in:
Antoni Sawicki 2019-05-30 18:08:48 -07:00
parent d382c38547
commit d905704a2a
1 changed files with 28 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
wrp.go
View File

@ -36,22 +36,34 @@ var (
gifmap = make(map[string]bytes.Buffer) gifmap = make(map[string]bytes.Buffer)
) )
func pageServer(out http.ResponseWriter, req *http.Request) { func pageServer(out http.ResponseWriter, r *http.Request) {
req.ParseForm() r.ParseForm()
furl := req.Form["url"] u := r.FormValue("url")
var gourl string w, _ := strconv.ParseInt(r.FormValue("w"), 10, 64)
if len(furl) >= 1 && len(furl[0]) > 4 { if w < 10 {
gourl = furl[0] w = 1024
} else {
gourl = "https://www.bbc.com/news"
} }
log.Printf("%s Page Reqest for %s [%s]\n", req.RemoteAddr, gourl, req.URL.Path) h, _ := strconv.ParseInt(r.FormValue("h"), 10, 64)
if h < 10 {
h = 768
}
s, _ := strconv.ParseFloat(r.FormValue("s"), 64)
if s < 0.1 {
s = 1.0
}
log.Printf("%s Page Reqest for url=\"%s\" [%s]\n", r.RemoteAddr, u, r.URL.Path)
out.Header().Set("Content-Type", "text/html") out.Header().Set("Content-Type", "text/html")
fmt.Fprintf(out, "<HTML>\n<HEAD><TITLE>WRP %s</TITLE>\n<BODY BGCOLOR=\"#F0F0F0\">", gourl) fmt.Fprintf(out, "<HTML>\n<HEAD><TITLE>WRP %s</TITLE>\n<BODY BGCOLOR=\"#F0F0F0\">", u)
fmt.Fprintf(out, "<FORM ACTION=\"/\">URL: <INPUT TYPE=\"TEXT\" NAME=\"url\" VALUE=\"%s\">", gourl) fmt.Fprintf(out, "<FORM ACTION=\"/\">URL: <INPUT TYPE=\"TEXT\" NAME=\"url\" VALUE=\"%s\" SIZE=\"40\">", u)
fmt.Fprintf(out, "<INPUT TYPE=\"SUBMIT\" VALUE=\"Go\"></FORM><P>\n") fmt.Fprintf(out, "<INPUT TYPE=\"SUBMIT\" VALUE=\"Go\"><P>\n")
if len(gourl) > 4 { fmt.Fprintf(out, "Width:<INPUT TYPE=\"TEXT\" NAME=\"w\" VALUE=\"%d\" SIZE=\"5\"> \n", w)
capture(gourl, out) fmt.Fprintf(out, "Height:<INPUT TYPE=\"TEXT\" NAME=\"h\" VALUE=\"%d\" SIZE=\"5\"> \n", h)
fmt.Fprintf(out, "Scale:<INPUT TYPE=\"TEXT\" NAME=\"s\" VALUE=\"%0.1f\" SIZE=\"4\"> \n", s)
fmt.Fprintf(out, "</FORM><P>")
if len(u) > 4 {
capture(u, w, h, s, out)
} else {
fmt.Fprintf(out, "No URL specified")
} }
fmt.Fprintf(out, "</BODY>\n</HTML>\n") fmt.Fprintf(out, "</BODY>\n</HTML>\n")
} }
@ -75,7 +87,7 @@ func haltServer(out http.ResponseWriter, req *http.Request) {
os.Exit(0) os.Exit(0)
} }
func capture(gourl string, out http.ResponseWriter) { func capture(gourl string, w int64, h int64, s float64, out http.ResponseWriter) {
var nodes []*cdp.Node var nodes []*cdp.Node
ctxx := chromedp.FromContext(ctx) ctxx := chromedp.FromContext(ctx)
var pngbuf []byte var pngbuf []byte
@ -86,7 +98,7 @@ func capture(gourl string, out http.ResponseWriter) {
// Run ChromeDP Magic // Run ChromeDP Magic
chromedp.Run(ctx, chromedp.Run(ctx,
emulation.SetDeviceMetricsOverride(1024, 768, 1.0, false), emulation.SetDeviceMetricsOverride(w, h, s, false),
chromedp.Navigate(gourl), chromedp.Navigate(gourl),
chromedp.Sleep(time.Second*2), chromedp.Sleep(time.Second*2),
chromedp.CaptureScreenshot(&pngbuf), chromedp.CaptureScreenshot(&pngbuf),