// Copyright 2018 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/download/download_open_prompt.h"

#include "base/callback.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/ui/browser_dialogs.h"
#include "chrome/browser/ui/tab_modal_confirm_dialog.h"
#include "chrome/browser/ui/tab_modal_confirm_dialog_delegate.h"
#include "chrome/grit/generated_resources.h"
#include "content/public/browser/web_contents.h"
#include "ui/base/l10n/l10n_util.h"

namespace {

// Dialog class for prompting users about opening a download.
class DownloadOpenConfirmationDialog : public DownloadOpenPrompt,
                                       public TabModalConfirmDialogDelegate {
 public:
  DownloadOpenConfirmationDialog(
      content::WebContents* web_contents,
      const std::string& extension_name,
      const base::FilePath& file_path,
      DownloadOpenPrompt::OpenCallback open_callback);
  ~DownloadOpenConfirmationDialog() override;

  base::string16 GetTitle() override;
  base::string16 GetDialogMessage() override;
  base::string16 GetAcceptButtonTitle() override;
  base::string16 GetCancelButtonTitle() override;

 private:
  void OnAccepted() override;
  void OnCanceled() override;
  void OnClosed() override;

  DownloadOpenPrompt::OpenCallback open_callback_;

  std::string extension_name_;

  base::FilePath file_path_;

  DISALLOW_COPY_AND_ASSIGN(DownloadOpenConfirmationDialog);
};

DownloadOpenConfirmationDialog::DownloadOpenConfirmationDialog(
    content::WebContents* web_contents,
    const std::string& extension_name,
    const base::FilePath& file_path,
    DownloadOpenPrompt::OpenCallback open_callback)
    : TabModalConfirmDialogDelegate(web_contents),
      open_callback_(std::move(open_callback)),
      extension_name_(extension_name),
      file_path_(file_path) {
  chrome::RecordDialogCreation(
      chrome::DialogIdentifier::DOWNLOAD_OPEN_CONFIRMATION);
}

DownloadOpenConfirmationDialog::~DownloadOpenConfirmationDialog() = default;

base::string16 DownloadOpenConfirmationDialog::GetTitle() {
  return l10n_util::GetStringUTF16(IDS_DOWNLOAD_OPEN_CONFIRMATION_DIALOG_TITLE);
}

base::string16 DownloadOpenConfirmationDialog::GetDialogMessage() {
  return l10n_util::GetStringFUTF16(
      IDS_DOWNLOAD_OPEN_CONFIRMATION_DIALOG_MESSAGE,
      base::UTF8ToUTF16(extension_name_),
      file_path_.BaseName().AsUTF16Unsafe());
}

base::string16 DownloadOpenConfirmationDialog::GetAcceptButtonTitle() {
  return l10n_util::GetStringUTF16(IDS_CONFIRM_MESSAGEBOX_YES_BUTTON_LABEL);
}

base::string16 DownloadOpenConfirmationDialog::GetCancelButtonTitle() {
  return l10n_util::GetStringUTF16(IDS_CONFIRM_MESSAGEBOX_NO_BUTTON_LABEL);
}

void DownloadOpenConfirmationDialog::OnAccepted() {
  std::move(open_callback_).Run(true);
}

void DownloadOpenConfirmationDialog::OnCanceled() {
  std::move(open_callback_).Run(false);
}

void DownloadOpenConfirmationDialog::OnClosed() {
  std::move(open_callback_).Run(false);
}

}  // namespace

DownloadOpenPrompt::DownloadOpenPrompt() = default;

DownloadOpenPrompt::~DownloadOpenPrompt() = default;

DownloadOpenPrompt* DownloadOpenPrompt::CreateDownloadOpenConfirmationDialog(
    content::WebContents* web_contents,
    const std::string& extension_name,
    const base::FilePath& file_path,
    DownloadOpenPrompt::OpenCallback open_callback) {
  DownloadOpenConfirmationDialog* prompt = new DownloadOpenConfirmationDialog(
      web_contents, extension_name, file_path, std::move(open_callback));
  TabModalConfirmDialog::Create(prompt, web_contents);
  return prompt;
}

void DownloadOpenPrompt::AcceptConfirmationDialogForTesting(
    DownloadOpenPrompt* download_open_prompt) {
  static_cast<DownloadOpenConfirmationDialog*>(download_open_prompt)->Accept();
}