Implement ElementHandle.contentFrame() (#371)

Author: kblok

lib/PuppeteerSharp.Tests/ElementHandleTests/ContentFrameTests.cs

@@ -0,0 +1,24 @@
+using System.Threading.Tasks;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace PuppeteerSharp.Tests.ElementHandleTests
+{
+    [Collection("PuppeteerLoaderFixture collection")]
+    public class ContentFrameTests : PuppeteerPageBaseTest
+    {
+        public ContentFrameTests(ITestOutputHelper output) : base(output)
+        {
+        }
+
+        [Fact]
+        public async Task ShouldWork()
+        {
+            await Page.GoToAsync(TestConstants.EmptyPage);
+            await FrameUtils.AttachFrameAsync(Page, "frame1", TestConstants.EmptyPage);
+            var elementHandle = await Page.QuerySelectorAsync("#frame1");
+            var frame = await elementHandle.ContentFrameAsync();
+            Assert.Equal(Page.Frames[1], frame);
+        }
+    }
+}

lib/PuppeteerSharp/ElementHandle.cs

@@ -19,10 +19,19 @@ public class ElementHandle : JSHandle
     {
         internal Page Page { get; }
 
-        internal ElementHandle(ExecutionContext context, CDPSession client, object remoteObject, Page page) :
+        private readonly FrameManager _frameManager;
+
+        internal ElementHandle(
+            ExecutionContext
+            context,
+            CDPSession client,
+            object remoteObject,
+            Page page,
+            FrameManager frameManager) :
             base(context, client, remoteObject)
         {
             Page = page;
+            _frameManager = frameManager;
         }
 
         /// <summary>
@@ -354,5 +363,23 @@ public async Task<BoundingBox> BoundingBoxAsync()
 
             return new BoundingBox(x, y, width, height);
         }
+
+        /// <summary>
+        ///Content frame for element handles referencing iframe nodes, or null otherwise.
+        /// </summary>
+        /// <returns>Resolves to the content frame</returns>
+        public async Task<Frame> ContentFrameAsync()
+        {
+            var nodeInfo = await Client.SendAsync<DomDescribeNodeResponse>("DOM.describeNode", new
+            {
+                RemoteObject.objectId
+            });
+
+            if (string.IsNullOrEmpty(nodeInfo.Node.FrameId))
+            {
+                return null;
+            }
+            return _frameManager.Frames[nodeInfo.Node.FrameId];
+        }
     }
 }

lib/PuppeteerSharp/FrameManager.cs

@@ -49,7 +49,7 @@ internal JSHandle CreateJSHandle(int contextId, dynamic remoteObject)
 
             if (remoteObject.subtype == "node")
             {
-                return new ElementHandle(storedContext, _client, remoteObject, _page);
+                return new ElementHandle(storedContext, _client, remoteObject, _page, this);
             }
 
             return new JSHandle(storedContext, _client, remoteObject);

lib/PuppeteerSharp/Messaging/DomDescribeNodeResponse.cs

@@ -0,0 +1,17 @@
+using System;
+using Newtonsoft.Json;
+
+namespace PuppeteerSharp.Messaging
+{
+    internal class DomDescribeNodeResponse
+    {
+        [JsonProperty("node")]
+        public DomNode Node { get; set; }
+
+        internal class DomNode
+        {
+            [JsonProperty("frameId")]
+            public string FrameId { get; set; }
+        }
+    }
+}