blob: 6b996d60fd8ac9f8a2b03c63d889037eab61982c [file] [log] [blame]
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/sessions/session_tab_helper.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/sessions/session_service.h"
#include "chrome/browser/sessions/session_service_factory.h"
#include "content/public/browser/web_contents.h"
#if defined(ENABLE_EXTENSIONS)
#include "extensions/common/extension_messages.h"
#endif
DEFINE_WEB_CONTENTS_USER_DATA_KEY(SessionTabHelper);
SessionTabHelper::SessionTabHelper(content::WebContents* contents)
: content::WebContentsObserver(contents) {
}
SessionTabHelper::~SessionTabHelper() {
}
void SessionTabHelper::SetWindowID(const SessionID& id) {
window_id_ = id;
#if defined(ENABLE_EXTENSIONS)
// Extension code in the renderer holds the ID of the window that hosts it.
// Notify it that the window ID changed.
web_contents()->SendToAllFrames(
new ExtensionMsg_UpdateBrowserWindowId(MSG_ROUTING_NONE, id.id()));
#endif
}
// static
SessionID::id_type SessionTabHelper::IdForTab(const content::WebContents* tab) {
const SessionTabHelper* session_tab_helper =
tab ? SessionTabHelper::FromWebContents(tab) : NULL;
return session_tab_helper ? session_tab_helper->session_id().id() : -1;
}
// static
SessionID::id_type SessionTabHelper::IdForWindowContainingTab(
const content::WebContents* tab) {
const SessionTabHelper* session_tab_helper =
tab ? SessionTabHelper::FromWebContents(tab) : NULL;
return session_tab_helper ? session_tab_helper->window_id().id() : -1;
}
void SessionTabHelper::UserAgentOverrideSet(const std::string& user_agent) {
#if defined(ENABLE_SESSION_SERVICE)
Profile* profile =
Profile::FromBrowserContext(web_contents()->GetBrowserContext());
SessionService* session = SessionServiceFactory::GetForProfile(profile);
if (session)
session->SetTabUserAgentOverride(window_id(), session_id(), user_agent);
#endif
}