src/chrome/browser/ui/startup/session_crashed_infobar_delegate.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/ui/startup/session_crashed_infobar_delegate.h"
   6
   7 #include "chrome/browser/infobars/infobar_service.h"
   8 #include "chrome/browser/profiles/profile.h"
   9 #include "chrome/browser/search/search.h"
  10 #include "chrome/browser/sessions/session_restore.h"
  11 #include "chrome/browser/ui/browser.h"
  12 #include "chrome/browser/ui/browser_finder.h"
  13 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  14 #include "chrome/common/url_constants.h"
  15 #include "chrome/grit/chromium_strings.h"
  16 #include "chrome/grit/generated_resources.h"
  17 #include "components/infobars/core/infobar.h"
  18 #include "content/public/browser/dom_storage_context.h"
  19 #include "content/public/browser/storage_partition.h"
  20 #include "grit/theme_resources.h"
  21 #include "ui/base/l10n/l10n_util.h"
  22
  23
  24 // static
  25 void SessionCrashedInfoBarDelegate::Create(Browser* browser) {
  26   // Assume that if the user is launching incognito they were previously running
  27   // incognito so that we have nothing to restore from.
  28   // Also, in ChromeBot tests, there might be a race.  This code appears to be
  29   // called during shutdown when there is no active WebContents.
  30   Profile* profile = browser->profile();
  31   content::WebContents* web_contents =
  32       browser->tab_strip_model()->GetActiveWebContents();
  33   if (profile->IsOffTheRecord() || !web_contents)
  34     return;
  35
  36   InfoBarService::FromWebContents(web_contents)->AddInfoBar(
  37       ConfirmInfoBarDelegate::CreateInfoBar(scoped_ptr<ConfirmInfoBarDelegate>(
  38           new SessionCrashedInfoBarDelegate(profile))));
  39 }
  40
  41 SessionCrashedInfoBarDelegate::SessionCrashedInfoBarDelegate(Profile* profile)
  42     : ConfirmInfoBarDelegate(),
  43       accepted_(false),
  44       profile_(profile) {
  45 }
  46
  47 SessionCrashedInfoBarDelegate::~SessionCrashedInfoBarDelegate() {
  48   // If the info bar wasn't accepted, it was either dismissed or expired. In
  49   // that case, session restore won't happen.
  50   if (!accepted_) {
  51     content::BrowserContext::GetDefaultStoragePartition(profile_)->
  52         GetDOMStorageContext()->StartScavengingUnusedSessionStorage();
  53   }
  54 }
  55
  56 int SessionCrashedInfoBarDelegate::GetIconID() const {
  57   return IDR_INFOBAR_RESTORE_SESSION;
  58 }
  59
  60 base::string16 SessionCrashedInfoBarDelegate::GetMessageText() const {
  61   return l10n_util::GetStringUTF16(IDS_SESSION_CRASHED_VIEW_MESSAGE);
  62 }
  63
  64 int SessionCrashedInfoBarDelegate::GetButtons() const {
  65   return BUTTON_OK;
  66 }
  67
  68 base::string16 SessionCrashedInfoBarDelegate::GetButtonLabel(
  69     InfoBarButton button) const {
  70   DCHECK_EQ(BUTTON_OK, button);
  71   return l10n_util::GetStringUTF16(IDS_SESSION_CRASHED_VIEW_RESTORE_BUTTON);
  72 }
  73
  74 bool SessionCrashedInfoBarDelegate::Accept() {
  75   Browser* browser = chrome::FindBrowserWithWebContents(
  76       InfoBarService::WebContentsFromInfoBar(infobar()));
  77   SessionRestore::RestoreSessionAfterCrash(browser);
  78   accepted_ = true;
  79   return true;
  80 }