// Copyright (c) 2010 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/blocked_plugin_manager.h"

#include "app/l10n_util.h"
#include "app/resource_bundle.h"
#include "base/command_line.h"
#include "base/utf_string_conversions.h"
#include "chrome/browser/host_content_settings_map.h"
#include "chrome/browser/metrics/user_metrics.h"
#include "chrome/browser/profile.h"
#include "chrome/browser/renderer_host/render_view_host.h"
#include "chrome/browser/tab_contents/tab_contents.h"
#include "chrome/common/chrome_switches.h"
#include "grit/generated_resources.h"
#include "grit/theme_resources.h"

BlockedPluginManager::BlockedPluginManager(TabContents* tab_contents)
    : ConfirmInfoBarDelegate(tab_contents),
      tab_contents_(tab_contents) { }

void BlockedPluginManager::OnNonSandboxedPluginBlocked(
    const std::string& plugin,
    const string16& name) {
  plugin_ = plugin;
  name_ = name;
  UserMetrics::RecordAction(UserMetricsAction("ClickToPlay_InfobarShown"));
  tab_contents_->AddInfoBar(this);
}

void BlockedPluginManager::OnBlockedPluginLoaded() {
  tab_contents_->RemoveInfoBar(this);
}

int BlockedPluginManager::GetButtons() const {
  int buttons = BUTTON_OK;
  if (CommandLine::ForCurrentProcess()->HasSwitch(
      switches::kEnableResourceContentSettings))
    buttons |= BUTTON_CANCEL;
  return buttons;
}

string16 BlockedPluginManager::GetButtonLabel(InfoBarButton button) const {
  if (button == BUTTON_OK)
    return l10n_util::GetStringUTF16(IDS_PLUGIN_LOAD_SHORT);
  if (button == BUTTON_CANCEL)
    return l10n_util::GetStringUTF16(IDS_BLOCKED_PLUGINS_UNBLOCK_SHORT);
  return ConfirmInfoBarDelegate::GetButtonLabel(button);
}

string16 BlockedPluginManager::GetMessageText() const {
  return l10n_util::GetStringFUTF16(IDS_PLUGIN_BLOCKED_PROMPT, name_);
}

string16 BlockedPluginManager::GetLinkText() {
  return l10n_util::GetStringUTF16(IDS_LEARN_MORE);
}

SkBitmap* BlockedPluginManager::GetIcon() const {
  return ResourceBundle::GetSharedInstance().GetBitmapNamed(
      IDR_INFOBAR_PLUGIN_INSTALL);
}

bool BlockedPluginManager::Accept() {
  UserMetrics::RecordAction(UserMetricsAction("ClickToPlay_LoadAll_Infobar"));
  tab_contents_->render_view_host()->LoadBlockedPlugins();
  return true;
}

bool BlockedPluginManager::Cancel() {
  DCHECK(CommandLine::ForCurrentProcess()->HasSwitch(
      switches::kEnableResourceContentSettings));
  UserMetrics::RecordAction(UserMetricsAction("ClickToPlay_AllowAlways"));
  tab_contents_->profile()->GetHostContentSettingsMap()->AddExceptionForURL(
      tab_contents_->GetURL(), CONTENT_SETTINGS_TYPE_PLUGINS, plugin_,
      CONTENT_SETTING_ALLOW);
  tab_contents_->render_view_host()->LoadBlockedPlugins();
  return true;
}

void BlockedPluginManager::InfoBarDismissed() {
  UserMetrics::RecordAction(UserMetricsAction("ClickToPlay_Dismiss_Infobar"));
}

bool BlockedPluginManager::LinkClicked(WindowOpenDisposition disposition) {
  // TODO(bauerb): Navigate to a help page explaining why we blocked the plugin,
  // once we have one.
  return false;
}