Changeset 248736 in webkit

Timestamp:

Aug 15, 2019, 12:16:04 PM (7 years ago)

Author:

Devin Rousso

Message:

Web Inspector: support console.screenshot with detached <canvas>
​https://bugs.webkit.org/show_bug.cgi?id=200723

Reviewed by Joseph Pecoraro.

Source/WebCore:

• page/PageConsoleClient.cpp:

(WebCore::snapshotCanvas): Added.
(WebCore::PageConsoleClient::screenshot):

LayoutTests:

• inspector/console/console-screenshot.html:
• inspector/console/console-screenshot-expected.txt:

Location:

trunk

Files:

• LayoutTests/ChangeLog (modified) (1 diff)
• LayoutTests/inspector/console/console-screenshot-expected.txt (modified) (2 diffs)
• LayoutTests/inspector/console/console-screenshot.html (modified) (2 diffs)
• Source/WebCore/ChangeLog (modified) (1 diff)
• Source/WebCore/page/PageConsoleClient.cpp (modified) (3 diffs)

trunk/LayoutTests/ChangeLog

@@ +1 @@
+2019-08-15  Devin Rousso  <drousso@apple.com>
+
+        Web Inspector: support `console.screenshot` with detached <canvas>
+        https://bugs.webkit.org/show_bug.cgi?id=200723
+
+        Reviewed by Joseph Pecoraro.
+
+        * inspector/console/console-screenshot.html:
+        * inspector/console/console-screenshot-expected.txt:
+
 2019-08-15  Sihui Liu  <sihui_liu@apple.com>
 

trunk/LayoutTests/inspector/console/console-screenshot-expected.txt

@@ -3 +3 @@
 CONSOLE MESSAGE: [object HTMLImageElement]
 CONSOLE MESSAGE: [object HTMLPictureElement]
+CONSOLE MESSAGE: [object HTMLCanvasElement]
 CONSOLE MESSAGE: [object HTMLDivElement]
 CONSOLE MESSAGE: [object ImageData]
@@ -35 +36 @@
 
 -- Running test case: console.screenshot.Node.DetachedScreenshotable.Picture
+PASS: The added message should be an image.
+PASS: The image should not be empty.
+PASS: The image width should be 2px.
+PASS: The image height should be 2px.
+
+-- Running test case: console.screenshot.Node.DetachedScreenshotable.Canvas
 PASS: The added message should be an image.
 PASS: The image should not be empty.

trunk/LayoutTests/inspector/console/console-screenshot.html

@@ -30 +30 @@
         console.screenshot(picture);
     });
+}
+
+function testHTMLCanvasElement() {
+    let canvas = document.createElement("canvas");
+    canvas.width = 2;
+    canvas.height = 2;
+
+    let context = canvas.getContext("2d");
+    context.fillStyle = "red";
+    context.fillRect(0, 0, 2, 2);
+
+    console.screenshot(canvas);
 }
 
@@ -102 +114 @@
         name: "console.screenshot.Node.DetachedScreenshotable.Picture",
         expression: `testHTMLPictureElement()`,
+    });
+
+    addTest({
+        name: "console.screenshot.Node.DetachedScreenshotable.Canvas",
+        expression: `testHTMLCanvasElement()`,
+        async imageMessageAddedCallback(message) {
+            InspectorTest.expectNotEqual(message.messageText, "data:", "The image should not be empty.");
+
+            let img = await WI.ImageUtilities.promisifyLoad(message.messageText);
+            InspectorTest.expectEqual(img.width, 2, "The image width should be 2px.");
+            InspectorTest.expectEqual(img.height, 2, "The image height should be 2px.");
+        },
     });
 

trunk/Source/WebCore/ChangeLog

@@ +1 @@
+2019-08-15  Devin Rousso  <drousso@apple.com>
+
+        Web Inspector: support `console.screenshot` with detached <canvas>
+        https://bugs.webkit.org/show_bug.cgi?id=200723
+
+        Reviewed by Joseph Pecoraro.
+
+        * page/PageConsoleClient.cpp:
+        (WebCore::snapshotCanvas): Added.
+        (WebCore::PageConsoleClient::screenshot):
+
 2019-08-15  Sihui Liu  <sihui_liu@apple.com>
 

trunk/Source/WebCore/page/PageConsoleClient.cpp

@@ -282 +282 @@
 }
 
+static Optional<String> snapshotCanvas(HTMLCanvasElement& canvasElement, CanvasRenderingContext& canvasRenderingContext)
+{
+#if ENABLE(WEBGL)
+    if (is<WebGLRenderingContextBase>(canvasRenderingContext))
+        downcast<WebGLRenderingContextBase>(canvasRenderingContext).setPreventBufferClearForInspector(true);
+#endif
+
+    auto result = canvasElement.toDataURL("image/png"_s);
+
+#if ENABLE(WEBGL)
+    if (is<WebGLRenderingContextBase>(canvasRenderingContext))
+        downcast<WebGLRenderingContextBase>(canvasRenderingContext).setPreventBufferClearForInspector(false);
+#endif
+
+    if (!result.hasException())
+        return result.releaseReturnValue().string;
+
+    return WTF::nullopt;
+}
+
 void PageConsoleClient::screenshot(JSC::ExecState* state, Ref<ScriptArguments>&& arguments)
 {
@@ -322 +342 @@
                     }
 #endif
+                    else if (is<HTMLCanvasElement>(node)) {
+                        auto& canvasElement = downcast<HTMLCanvasElement>(*node);
+                        if (auto* canvasRenderingContext = canvasElement.renderingContext()) {
+                            if (auto result = snapshotCanvas(canvasElement, *canvasRenderingContext))
+                                dataURL = result.value();
+                        }
+                    }
                 }
 
-                if (!snapshot)
-                    snapshot = WebCore::snapshotNode(m_page.mainFrame(), *node);
-
-                if (snapshot)
-                    dataURL = snapshot->toDataURL("image/png"_s, WTF::nullopt, PreserveResolution::Yes);
+                if (dataURL.isEmpty()) {
+                    if (!snapshot)
+                        snapshot = WebCore::snapshotNode(m_page.mainFrame(), *node);
+
+                    if (snapshot)
+                        dataURL = snapshot->toDataURL("image/png"_s, WTF::nullopt, PreserveResolution::Yes);
+                }
             }
         } else if (auto* imageData = JSImageData::toWrapped(state->vm(), possibleTarget)) {
@@ -351 +380 @@
                 target = possibleTarget;
                 if (UNLIKELY(InspectorInstrumentation::hasFrontends())) {
-#if ENABLE(WEBGL)
-                    if (is<WebGLRenderingContextBase>(context))
-                        downcast<WebGLRenderingContextBase>(context)->setPreventBufferClearForInspector(true);
-#endif
-
-                    auto result = downcast<HTMLCanvasElement>(canvas).toDataURL("image/png"_s);
-
-#if ENABLE(WEBGL)
-                    if (is<WebGLRenderingContextBase>(context))
-                        downcast<WebGLRenderingContextBase>(context)->setPreventBufferClearForInspector(false);
-#endif
-
-                    if (!result.hasException())
-                        dataURL = result.releaseReturnValue().string;
+                    if (auto result = snapshotCanvas(downcast<HTMLCanvasElement>(canvas), *context))
+                        dataURL = result.value();
                 }
             }