summaryrefslogtreecommitdiffstats
path: root/browser/components/shell/ScreenshotChild.jsm
blob: 18e18b2d2de198135e2b320b47271151004250f1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

"use strict";

const EXPORTED_SYMBOLS = ["ScreenshotChild"];

class ScreenshotChild extends JSWindowActorChild {
  receiveMessage(message) {
    if (message.name == "TakeScreenshot") {
      return this.takeScreenshot(message.data);
    }
    return null;
  }

  async takeScreenshot(params) {
    if (this.document.readyState != "complete") {
      await new Promise(resolve =>
        this.contentWindow.addEventListener("load", resolve, { once: true })
      );
    }

    let { fullWidth, fullHeight } = params;
    let { contentWindow } = this;

    let canvas = contentWindow.document.createElementNS(
      "http://www.w3.org/1999/xhtml",
      "html:canvas"
    );
    let context = canvas.getContext("2d");
    let width = contentWindow.innerWidth;
    let height = contentWindow.innerHeight;
    if (fullWidth) {
      width += contentWindow.scrollMaxX - contentWindow.scrollMinX;
    }
    if (fullHeight) {
      height += contentWindow.scrollMaxY - contentWindow.scrollMinY;
    }

    canvas.width = width;
    canvas.height = height;
    context.drawWindow(
      contentWindow,
      0,
      0,
      width,
      height,
      "rgb(255, 255, 255)"
    );

    return new Promise(resolve => canvas.toBlob(resolve));
  }
}