Index: extensions/browser/api/web_view/web_view_internal_api.cc |
diff --git a/extensions/browser/api/web_view/web_view_internal_api.cc b/extensions/browser/api/web_view/web_view_internal_api.cc |
index 4b041e911764efabd574034f0e175b015ffebd65..29b43a3461dd5812d988204ac44f4ae3720d61dd 100644 |
--- a/extensions/browser/api/web_view/web_view_internal_api.cc |
+++ b/extensions/browser/api/web_view/web_view_internal_api.cc |
@@ -102,7 +102,22 @@ bool WebViewInternalExecuteCodeFunction::Init() { |
return false; |
details_ = details.Pass(); |
- return true; |
+ |
+ if (extension()) { |
+ set_host_id(HostID(HostID::EXTENSIONS, extension()->id())); |
+ return true; |
+ } |
+ |
+ const content::RenderViewHost* rvh = render_view_host(); |
Devlin
2015/02/27 17:11:36
This can actually be simplified to:
content::WebCo
Xi Han
2015/02/27 21:07:46
Done.
|
+ if (rvh) { |
+ WebContents* web_contents = WebContents::FromRenderViewHost(rvh); |
+ if (web_contents && web_contents->GetWebUI()) { |
+ const GURL& url = rvh->GetSiteInstance()->GetSiteURL(); |
+ set_host_id(HostID(HostID::WEBUI, url.spec())); |
+ return true; |
+ } |
+ } |
+ return false; |
} |
bool WebViewInternalExecuteCodeFunction::ShouldInsertCSS() const { |