Macintosh-Tools/wrp
1
0
Fork 0
mirror of https://github.com/tenox7/wrp.git synced 2025-02-16 18:30:27 +00:00
Code Issues Projects Releases Wiki Activity

fix chromedp scrollbar capture

Browse Source
This commit is contained in:
Antoni Sawicki 2022-11-06 01:47:59 -07:00
parent 08a89c6097
commit 9b15feacb2
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
wrp.go
View File

@ -219,6 +219,22 @@ func navigate(w wrpReq) {
} }
} }
// https://github.com/chromedp/chromedp/issues/979
func chromedpCaptureScreenshot(res *[]byte, h int64) chromedp.Action {
if res == nil {
panic("res cannot be nil")
}
if h == 0 {
return chromedp.CaptureScreenshot(res)
}
return chromedp.ActionFunc(func(ctx context.Context) error {
var err error
*res, err = page.CaptureScreenshot().Do(ctx)
return err
})
}
// Capture currently rendered web page to an image and fake ISMAP // Capture currently rendered web page to an image and fake ISMAP
func capture(w wrpReq) { func capture(w wrpReq) {
var err error var err error
@ -248,12 +264,12 @@ func capture(w wrpReq) {
if w.height == 0 && h > 0 { if w.height == 0 && h > 0 {
height = h + 30 height = h + 30
} }
chromedp.Run(ctx, emulation.SetDeviceMetricsOverride(int64(float64(w.width)/w.zoom), height, w.zoom, false)) chromedp.Run(
// Capture screenshot... ctx, emulation.SetDeviceMetricsOverride(int64(float64(w.width)/w.zoom), height, w.zoom, false),
err = chromedp.Run(ctx,
chromedp.Sleep(time.Second*2), chromedp.Sleep(time.Second*2),
chromedp.CaptureScreenshot(&pngcap),
) )
// Capture screenshot...
err = chromedp.Run(ctx, chromedpCaptureScreenshot(&pngcap, w.height))
if err != nil { if err != nil {
if err.Error() == "context canceled" { if err.Error() == "context canceled" {
log.Printf("%s Contex cancelled, try again", w.req.RemoteAddr) log.Printf("%s Contex cancelled, try again", w.req.RemoteAddr)