first commit

screenshot.go

@@ -0,0 +1,175 @@
+package main
+
+import (
+	"context"
+	"io/ioutil"
+	"math"
+	"time"
+
+	"github.com/spf13/viper"
+
+	"github.com/chromedp/cdproto/emulation"
+	"github.com/chromedp/cdproto/page"
+	"github.com/chromedp/chromedp"
+)
+
+func makeScreenShot(c context.Context, page string) {
+
+	timeNow := time.Now().Format(time.RFC3339)
+
+	var buf []byte
+	var content string
+
+	// capture entire browser viewport, returning png with quality=90
+	if err := chromedp.Run(c, fullScreenshot(90, &buf, &content)); err != nil {
+		//log.Fatal(err)
+	}
+
+	if viper.GetBool("development_mode") {
+		if err := ioutil.WriteFile(timeNow+"-"+page+"-fullScreenshot.png", buf, 0644); err != nil {
+			//log.Fatal(err)
+		}
+
+		if err := ioutil.WriteFile(timeNow+"-"+page+"-content.html", []byte(content), 0644); err != nil {
+			//log.Fatal(err)
+		}
+	}
+
+}
+
+func makeScreenShotAndParsePost(c context.Context, page string) (*FBPostData, error) {
+
+	timeNow := time.Now().Format(time.RFC3339)
+
+	var buf []byte
+	var content string
+
+	// capture entire browser viewport, returning png with quality=90
+	if err := chromedp.Run(c, fullScreenshot(90, &buf, &content)); err != nil {
+		//log.Fatal(err)
+	}
+
+	if viper.GetBool("development_mode") {
+		if err := ioutil.WriteFile(timeNow+"-"+page+"-fullScreenshot.png", buf, 0644); err != nil {
+			//log.Fatal(err)
+		}
+
+		if err := ioutil.WriteFile(timeNow+"-"+page+"-content.html", []byte(content), 0644); err != nil {
+			//log.Fatal(err)
+		}
+	}
+	return ParsePost(content, "https://www.facebook.com/pg/herowarsgame/posts/?ref=page_internal")
+
+}
+
+func makeScreenShotOnly(c context.Context, page string) {
+
+	timeNow := time.Now().Format(time.RFC3339)
+
+	var buf []byte
+
+	// capture entire browser viewport, returning png with quality=90
+	if err := chromedp.Run(c, fullScreenshotOnly(90, &buf)); err != nil {
+		//log.Fatal(err)
+	}
+
+	if viper.GetBool("development_mode") {
+		if err := ioutil.WriteFile(timeNow+"-"+page+"-fullScreenshot.png", buf, 0644); err != nil {
+			//log.Fatal(err)
+		}
+	}
+
+}
+
+// fullScreenshot takes a screenshot of the entire browser viewport.
+//
+// Liberally copied from puppeteer's source.
+//
+// Note: this will override the viewport emulation settings.
+func fullScreenshot(quality int64, res *[]byte, content *string) chromedp.Tasks {
+	return chromedp.Tasks{
+		chromedp.Sleep(1 * time.Second),
+		chromedp.ActionFunc(func(ctx context.Context) error {
+			// get layout metrics
+			_, _, contentSize, err := page.GetLayoutMetrics().Do(ctx)
+			if err != nil {
+				return err
+			}
+
+			width, height := int64(math.Ceil(contentSize.Width)), int64(math.Ceil(contentSize.Height))
+
+			// force viewport emulation
+			err = emulation.SetDeviceMetricsOverride(width, height, 1, false).
+				WithScreenOrientation(&emulation.ScreenOrientation{
+					Type:  emulation.OrientationTypePortraitPrimary,
+					Angle: 0,
+				}).
+				Do(ctx)
+			if err != nil {
+				return err
+			}
+
+			// capture screenshot
+			*res, err = page.CaptureScreenshot().
+				WithQuality(quality).
+				WithClip(&page.Viewport{
+					X:      contentSize.X,
+					Y:      contentSize.Y,
+					Width:  contentSize.Width,
+					Height: contentSize.Height,
+					Scale:  1,
+				}).Do(ctx)
+			if err != nil {
+				return err
+			}
+			return nil
+		}),
+		chromedp.OuterHTML("html", content),
+	}
+}
+
+// fullScreenshotOnly takes a screenshot of the entire browser viewport.
+//
+// Liberally copied from puppeteer's source.
+//
+// Note: this will override the viewport emulation settings.
+func fullScreenshotOnly(quality int64, res *[]byte) chromedp.Tasks {
+	return chromedp.Tasks{
+		chromedp.Sleep(2 * time.Second),
+		chromedp.ActionFunc(func(ctx context.Context) error {
+			// get layout metrics
+			_, _, contentSize, err := page.GetLayoutMetrics().Do(ctx)
+			if err != nil {
+				return err
+			}
+
+			width, height := int64(math.Ceil(contentSize.Width)), int64(math.Ceil(contentSize.Height))
+
+			// force viewport emulation
+			err = emulation.SetDeviceMetricsOverride(width, height, 1, false).
+				WithScreenOrientation(&emulation.ScreenOrientation{
+					Type:  emulation.OrientationTypePortraitPrimary,
+					Angle: 0,
+				}).
+				Do(ctx)
+			if err != nil {
+				return err
+			}
+
+			// capture screenshot
+			*res, err = page.CaptureScreenshot().
+				WithQuality(quality).
+				WithClip(&page.Viewport{
+					X:      contentSize.X,
+					Y:      contentSize.Y,
+					Width:  contentSize.Width,
+					Height: contentSize.Height,
+					Scale:  1,
+				}).Do(ctx)
+			if err != nil {
+				return err
+			}
+			return nil
+		}),
+	}
+}