OLD | NEW |
---|---|
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/sessions/session_tab_helper.h" | 5 #include "chrome/browser/sessions/session_tab_helper.h" |
6 | 6 |
7 #include "chrome/browser/profiles/profile.h" | 7 #include "chrome/browser/profiles/profile.h" |
8 #include "chrome/browser/sessions/session_service.h" | 8 #include "chrome/browser/sessions/session_service.h" |
9 #include "chrome/browser/sessions/session_service_factory.h" | 9 #include "chrome/browser/sessions/session_service_factory.h" |
10 #include "chrome/common/chrome_notification_types.h" | 10 #include "chrome/common/chrome_notification_types.h" |
11 #include "chrome/common/extensions/extension_messages.h" | 11 #include "chrome/common/extensions/extension_messages.h" |
12 #include "content/public/browser/notification_service.h" | 12 #include "content/public/browser/notification_service.h" |
13 #include "content/public/browser/render_view_host.h" | 13 #include "content/public/browser/render_view_host.h" |
14 #include "content/public/browser/web_contents.h" | 14 #include "content/public/browser/web_contents.h" |
15 | 15 |
16 DEFINE_WEB_CONTENTS_USER_DATA_KEY(SessionTabHelper); | 16 DEFINE_WEB_CONTENTS_USER_DATA_KEY(SessionTabHelper); |
17 | 17 |
18 SessionTabHelper::SessionTabHelper(content::WebContents* contents) | 18 SessionTabHelper::SessionTabHelper(content::WebContents* contents) |
19 : content::WebContentsObserver(contents) { | 19 : content::WebContentsObserver(contents) { |
20 } | 20 } |
21 | 21 |
22 SessionTabHelper::~SessionTabHelper() { | 22 SessionTabHelper::SessionTabHelper(content::WebContents* contents, |
23 const SessionID& id) | |
24 : content::WebContentsObserver(contents), session_id_(id) {} | |
sky
2013/05/22 00:46:36
nit: since the constructor wrapped one param on ea
| |
25 | |
26 // static | |
27 void SessionTabHelper::CreateForWebContentsWithId( | |
28 content::WebContents* contents, | |
29 const SessionID& id) { | |
30 DCHECK(contents); | |
31 if (FromWebContents(contents)) { | |
32 DCHECK_EQ(id.id(), FromWebContents(contents)->session_id().id()); | |
33 return; | |
34 } | |
35 contents->SetUserData(UserDataKey(), new SessionTabHelper(contents, id)); | |
23 } | 36 } |
24 | 37 |
38 SessionTabHelper::~SessionTabHelper() {} | |
39 | |
25 void SessionTabHelper::SetWindowID(const SessionID& id) { | 40 void SessionTabHelper::SetWindowID(const SessionID& id) { |
26 window_id_ = id; | 41 window_id_ = id; |
27 | 42 |
28 // Extension code in the renderer holds the ID of the window that hosts it. | 43 // Extension code in the renderer holds the ID of the window that hosts it. |
29 // Notify it that the window ID changed. | 44 // Notify it that the window ID changed. |
30 web_contents()->GetRenderViewHost()->Send( | 45 web_contents()->GetRenderViewHost()->Send( |
31 new ExtensionMsg_UpdateBrowserWindowId( | 46 new ExtensionMsg_UpdateBrowserWindowId( |
32 web_contents()->GetRenderViewHost()->GetRoutingID(), id.id())); | 47 web_contents()->GetRenderViewHost()->GetRoutingID(), id.id())); |
33 } | 48 } |
34 | 49 |
35 void SessionTabHelper::RenderViewCreated( | 50 void SessionTabHelper::RenderViewCreated( |
36 content::RenderViewHost* render_view_host) { | 51 content::RenderViewHost* render_view_host) { |
37 render_view_host->Send( | 52 render_view_host->Send( |
38 new ExtensionMsg_UpdateBrowserWindowId(render_view_host->GetRoutingID(), | 53 new ExtensionMsg_UpdateBrowserWindowId(render_view_host->GetRoutingID(), |
39 window_id_.id())); | 54 window_id_.id())); |
40 } | 55 } |
41 | 56 |
42 void SessionTabHelper::UserAgentOverrideSet(const std::string& user_agent) { | 57 void SessionTabHelper::UserAgentOverrideSet(const std::string& user_agent) { |
43 #if defined(ENABLE_SESSION_SERVICE) | 58 #if defined(ENABLE_SESSION_SERVICE) |
44 Profile* profile = | 59 Profile* profile = |
45 Profile::FromBrowserContext(web_contents()->GetBrowserContext()); | 60 Profile::FromBrowserContext(web_contents()->GetBrowserContext()); |
46 SessionService* session = SessionServiceFactory::GetForProfile(profile); | 61 SessionService* session = SessionServiceFactory::GetForProfile(profile); |
47 if (session) | 62 if (session) |
48 session->SetTabUserAgentOverride(window_id(), session_id(), user_agent); | 63 session->SetTabUserAgentOverride(window_id(), session_id(), user_agent); |
49 #endif | 64 #endif |
50 } | 65 } |
OLD | NEW |