Index: headless/lib/browser/headless_browser_impl.cc |
diff --git a/headless/lib/browser/headless_browser_impl.cc b/headless/lib/browser/headless_browser_impl.cc |
index 57605807d53a9488123ae1ca0d000934b2ff4067..c3c2f2570a1a7be848ee33247e82cbf9dc690588 100644 |
--- a/headless/lib/browser/headless_browser_impl.cc |
+++ b/headless/lib/browser/headless_browser_impl.cc |
@@ -200,6 +200,19 @@ HeadlessWebContents* HeadlessBrowserImpl::GetWebContentsForDevToolsAgentHostId( |
return nullptr; |
} |
+HeadlessWebContentsImpl* HeadlessBrowserImpl::GetWebContentsForWindowId( |
+ const int window_id) { |
+ for (HeadlessBrowserContext* context : GetAllBrowserContexts()) { |
+ for (HeadlessWebContents* web_contents : context->GetAllWebContents()) { |
+ auto* contents = HeadlessWebContentsImpl::From(web_contents); |
+ if (contents->window_id() == window_id) { |
+ return contents; |
+ } |
+ } |
+ } |
+ return nullptr; |
+} |
+ |
HeadlessBrowserContext* HeadlessBrowserImpl::GetBrowserContextForId( |
const std::string& id) { |
auto find_it = browser_contexts_.find(id); |