Chromium Code Reviews| 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 { |