[DevTools] Enable full-size screenshots

third_party/WebKit/Source/devtools/front_end/emulation/DeviceModeModel.js

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 /**
  * @implements {SDK.TargetManager.Observer}
  * @unrestricted
  */
 Emulation.DeviceModeModel = class {
   /**
    * @param {function()} updateCallback
@@ skipping 530 matching lines @@
       scale,
       deviceScaleFactor,
       mobile,
       screenOrientation,
       resetPageScaleFactor) {
     screenSize.width = Math.max(1, Math.floor(screenSize.width));
     screenSize.height = Math.max(1, Math.floor(screenSize.height));
 
     var pageWidth = screenSize.width - insets.left - insets.right;
     var pageHeight = screenSize.height - insets.top - insets.bottom;
-    this._emulatedPageSize = new UI.Size(Math.floor(pageWidth * scale), Math.floor(pageHeight * scale));
+    this._emulatedPageSize = new UI.Size(pageWidth, pageHeight);
 
     var positionX = insets.left;
     var positionY = insets.top;
     var screenOrientationAngle =
         screenOrientation === Protocol.Emulation.ScreenOrientationType.LandscapePrimary ? 90 : 0;
 
     this._appliedDeviceSize = screenSize;
     this._appliedDeviceScaleFactor = deviceScaleFactor || window.devicePixelRatio;
     this._screenRect = new UI.Rect(
         Math.max(0, (this._availableSize.width - screenSize.width * scale) / 2), outline.top * scale,
@@ skipping 55 matching lines @@
           params.screenOrientation = {type: screenOrientation, angle: screenOrientationAngle};
         setDevicePromise = this._target.emulationAgent().invoke_setDeviceMetricsOverride(
             params, this._deviceMetricsOverrideAppliedForTest.bind(this));
       }
       allPromises.push(setDevicePromise);
       return Promise.all(allPromises);
     }
   }
 
   /**
-   * @param {function(?string)} callback
+   * @param {boolean} fullSize
+   * @return {!Promise<?string>}
    */
-  captureFullSizeScreenshot(callback) {
+  async captureScreenshot(fullSize) {
     var screenCaptureModel = this._target ? this._target.model(SDK.ScreenCaptureModel) : null;
-    if (!screenCaptureModel) {
-      callback(null);
-      return;
+    if (!screenCaptureModel)
+      return null;
+
+    var metrics = await screenCaptureModel.fetchLayoutMetrics();
+    if (!metrics)
+      return null;
+
+    var pageSize = fullSize ? new UI.Size(metrics.contentWidth, metrics.contentHeight) : this._emulatedPageSize;
+    var promises = [];
+    promises.push(
+        this._target.emulationAgent().setVisibleSize(Math.floor(pageSize.width), Math.floor(pageSize.height)));
+    if (fullSize) {
+      promises.push(this._target.emulationAgent().forceViewport(0, 0, 1));
+    } else {
+      promises.push(this._target.emulationAgent().forceViewport(
+          Math.floor(metrics.viewportX), Math.floor(metrics.viewportY), metrics.viewportScale));
     }
+    promises.push(this._target.emulationAgent().invoke_setDeviceMetricsOverride({
+      width: 0,
+      height: 0,
+      deviceScaleFactor: this._appliedDeviceScaleFactor,
+      mobile: this._isMobile(),
+      fitWindow: false,
+      scale: 1,
+    }));
+    await Promise.all(promises);
 
-    screenCaptureModel.fetchContentSize().then(contentSize => {
-      if (!contentSize) {
-        callback(null);
-        return;
-      }
-
-      var scaledPageSize = new UI.Rect(0, 0, contentSize.width, contentSize.height).scale(this._scale);
-      var promises = [];
-      promises.push(this._target.emulationAgent().forceViewport(0, 0, 1));
-      promises.push(this._target.emulationAgent().invoke_setDeviceMetricsOverride({
-        width: 0,
-        height: 0,
-        deviceScaleFactor: this._appliedDeviceScaleFactor,
-        mobile: this._isMobile(),
-        fitWindow: false,
-        scale: this._scale,
-      }));
-      promises.push(this._target.emulationAgent().setVisibleSize(
-          Math.floor(scaledPageSize.width), Math.floor(scaledPageSize.height)));
-      Promise.all(promises).then(() => screenCaptureModel.captureScreenshot('png', 100)).then(content => {
-        this._target.emulationAgent().setVisibleSize(this._emulatedPageSize.width, this._emulatedPageSize.height);
-        this._target.emulationAgent().resetViewport();
-        callback(content);
-      });
-    });
+    var screenshot = await screenCaptureModel.captureScreenshot('png', 100);
+    this._target.emulationAgent().setVisibleSize(
+        Math.floor(this._emulatedPageSize.width * this._scale),
+        Math.floor(this._emulatedPageSize.height * this._scale));
+    this._target.emulationAgent().resetViewport();
+    this._calculateAndEmulate(false);
+    return screenshot;
   }
 
   _deviceMetricsOverrideAppliedForTest() {
     // Used for sniffing in tests.
   }
 
   /**
    * @param {boolean} touchEnabled
    * @param {boolean} mobile
    */
@@ skipping 19 matching lines @@
 
 Emulation.DeviceModeModel.MinDeviceSize = 50;
 Emulation.DeviceModeModel.MaxDeviceSize = 9999;
 
 
 Emulation.DeviceModeModel._defaultMobileUserAgent =
     'Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/%s Mobile Safari/537.36';
 Emulation.DeviceModeModel._defaultMobileUserAgent =
     SDK.MultitargetNetworkManager.patchUserAgentWithChromeVersion(Emulation.DeviceModeModel._defaultMobileUserAgent);
 Emulation.DeviceModeModel.defaultMobileScaleFactor = 2;