// Copyright (c) 2009 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/extensions/extension_disabled_infobar_delegate.h"

#include "app/l10n_util.h"
#include "chrome/browser/chrome_thread.h"
#include "chrome/browser/extensions/crx_installer.h"
#include "chrome/browser/extensions/extension_install_ui.h"
#include "chrome/browser/extensions/extensions_service.h"
#include "chrome/browser/tab_contents/infobar_delegate.h"
#include "chrome/browser/tab_contents/tab_contents.h"
#include "chrome/browser/browser_list.h"
#include "chrome/common/notification_registrar.h"
#include "chrome/common/notification_service.h"
#include "grit/generated_resources.h"

class ExtensionDisabledDialogDelegate
    : public ExtensionInstallUI::Delegate,
      public base::RefCountedThreadSafe<ExtensionDisabledDialogDelegate> {
 public:
  ExtensionDisabledDialogDelegate(Profile* profile,
                                  ExtensionsService* service,
                                  Extension* extension)
        : profile_(profile), service_(service), extension_(extension),
          ui_loop_(MessageLoop::current()) {
    AddRef();  // balanced in ContinueInstall or AbortInstall.

    // Do this now because we can't touch extension on the file loop.
    install_icon_path_ =
        extension_->GetIconPath(Extension::EXTENSION_ICON_LARGE);

    ChromeThread::GetMessageLoop(ChromeThread::FILE)->PostTask(FROM_HERE,
        NewRunnableMethod(this, &ExtensionDisabledDialogDelegate::Start));
  }

  // ExtensionInstallUI::Delegate
  virtual void ContinueInstall() {
    service_->EnableExtension(extension_->id());
    Release();
  }
  virtual void AbortInstall() {
    // Do nothing. The extension will remain disabled.
    Release();
  }

 private:
  void Start() {
    // We start on the file thread so we can decode the install icon.
    CrxInstaller::DecodeInstallIcon(install_icon_path_, &install_icon_);
    // Then we display the UI on the UI thread.
    ui_loop_->PostTask(FROM_HERE,
        NewRunnableMethod(this,
                          &ExtensionDisabledDialogDelegate::ConfirmInstall));
  }

  void ConfirmInstall() {
    DCHECK(MessageLoop::current() == ui_loop_);
    ExtensionInstallUI ui(profile_);
    ui.ConfirmInstall(this, extension_, install_icon_.get());
  }

  Profile* profile_;
  ExtensionsService* service_;
  Extension* extension_;
  FilePath install_icon_path_;
  scoped_ptr<SkBitmap> install_icon_;
  MessageLoop* ui_loop_;
};

class ExtensionDisabledInfobarDelegate
    : public ConfirmInfoBarDelegate,
      public NotificationObserver {
 public:
  ExtensionDisabledInfobarDelegate(TabContents* tab_contents,
                                   ExtensionsService* service,
                                   Extension* extension)
      : ConfirmInfoBarDelegate(tab_contents),
        tab_contents_(tab_contents),
        service_(service),
        extension_(extension) {
    // The user might re-enable the extension in other ways, so watch for that.
    registrar_.Add(this, NotificationType::EXTENSION_LOADED,
                   Source<ExtensionsService>(service));
    registrar_.Add(this, NotificationType::EXTENSION_UNLOADED_DISABLED,
                   Source<ExtensionsService>(service));
  }
  virtual void InfoBarClosed() {
    delete this;
  }
  virtual std::wstring GetMessageText() const {
    return l10n_util::GetStringF(IDS_EXTENSION_DISABLED_INFOBAR_LABEL,
                                 UTF8ToWide(extension_->name()));
  }
  virtual SkBitmap* GetIcon() const {
    return NULL;
  }
  virtual int GetButtons() const {
    return BUTTON_OK;
  }
  virtual std::wstring GetButtonLabel(
      ConfirmInfoBarDelegate::InfoBarButton button) const {
    return l10n_util::GetString(IDS_EXTENSION_DISABLED_INFOBAR_ENABLE_BUTTON);
  }
  virtual bool Accept() {
    // This object manages its own lifetime.
    new ExtensionDisabledDialogDelegate(tab_contents_->profile(),
                                        service_, extension_);
    return true;
  }

  virtual void Observe(NotificationType type,
                       const NotificationSource& source,
                       const NotificationDetails& details) {
    // TODO(mpcomplete): RemoveInfoBar doesn't seem to always result in us
    // getting deleted.
    switch (type.value) {
      case NotificationType::EXTENSION_LOADED:
      case NotificationType::EXTENSION_UNLOADED_DISABLED: {
        Extension* extension = Details<Extension>(details).ptr();
        if (extension == extension_)
          tab_contents_->RemoveInfoBar(this);
        break;
      }
      default:
        NOTREACHED();
    }
  }

 private:
  NotificationRegistrar registrar_;
  TabContents* tab_contents_;
  ExtensionsService* service_;
  Extension* extension_;
};

void ShowExtensionDisabledUI(ExtensionsService* service, Profile* profile,
                             Extension* extension) {
  Browser* browser = BrowserList::GetLastActiveWithProfile(profile);
  if (!browser)
    return;

  TabContents* tab_contents = browser->GetSelectedTabContents();
  if (!tab_contents)
    return;

  tab_contents->AddInfoBar(new ExtensionDisabledInfobarDelegate(
      tab_contents, service, extension));
}