// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/file_system_access/file_system_access_dangerous_file_dialog.h"

#include "base/functional/callback_helpers.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/dialogs/browser_dialogs.h"
#include "chrome/browser/ui/file_system_access/file_system_access_ui_helpers.h"
#include "chrome/grit/generated_resources.h"
#include "components/safe_browsing/content/common/file_type_policies.h"
#include "content/public/browser/file_system_access_permission_context.h"
#include "content/public/browser/web_contents.h"
#include "ui/base/l10n/l10n_util.h"
#include "ui/base/models/dialog_model.h"
#include "ui/base/models/dialog_model_field.h"
#include "ui/base/mojom/dialog_button.mojom.h"

namespace {

using DangerousFileResult =
    content::FileSystemAccessPermissionContext::SensitiveEntryResult;

std::unique_ptr<ui::DialogModel> CreateFileSystemAccessDangerousFileDialog(
    content::WebContents* web_contents,
    const url::Origin& origin,
    const content::PathInfo& path_info,
    base::OnceCallback<
        void(content::FileSystemAccessPermissionContext::SensitiveEntryResult)>
        callback) {
  auto split_callback = base::SplitOnceCallback(std::move(callback));
  auto accept_callback = base::BindOnce(std::move(split_callback.first),
                                        DangerousFileResult::kAllowed);
  // Further split the cancel callback, which we need to pass to two different
  // builder methods.
  auto cancel_callbacks = base::SplitOnceCallback(base::BindOnce(
      std::move(split_callback.second), DangerousFileResult::kAbort));

  Profile* profile =
      web_contents
          ? Profile::FromBrowserContext(web_contents->GetBrowserContext())
          : nullptr;
  std::u16string origin_identity_name =
      file_system_access_ui_helper::GetUrlIdentityName(profile,
                                                       origin.GetURL());

#if BUILDFLAG(IS_WIN)
  std::u16string file_extension = base::WideToUTF16(
      safe_browsing::FileTypePolicies::GetFileExtension(path_info.path));
#else
  std::u16string file_extension = base::ASCIIToUTF16(
      safe_browsing::FileTypePolicies::GetFileExtension(path_info.path));
#endif  // BUILDFLAG(IS_WIN)

  ui::DialogModel::Builder dialog_builder;
  dialog_builder
      .SetTitle(l10n_util::GetStringFUTF16(
          IDS_FILE_SYSTEM_ACCESS_DANGEROUS_FILE_TITLE,
          file_system_access_ui_helper::GetElidedPathForDisplayAsTitle(
              path_info)))
      .AddParagraph(ui::DialogModelLabel::CreateWithReplacements(
          IDS_FILE_SYSTEM_ACCESS_DANGEROUS_FILE_TEXT,
          {ui::DialogModelLabel::CreateEmphasizedText(file_extension),
           ui::DialogModelLabel::CreateEmphasizedText(origin_identity_name)}))
      .AddOkButton(
          std::move(accept_callback),
          ui::DialogModel::Button::Params().SetLabel(l10n_util::GetStringUTF16(
              IDS_FILE_SYSTEM_ACCESS_DANGEROUS_FILE_SAVE)))
      .AddCancelButton(
          std::move(cancel_callbacks.first),
          ui::DialogModel::Button::Params().SetLabel(l10n_util::GetStringUTF16(
              IDS_FILE_SYSTEM_ACCESS_DANGEROUS_FILE_DONT_SAVE)))
      .SetCloseActionCallback(std::move(cancel_callbacks.second))
      .OverrideDefaultButton(ui::mojom::DialogButton::kCancel);
  return dialog_builder.Build();
}

}  // namespace

void ShowFileSystemAccessDangerousFileDialog(
    const url::Origin& origin,
    const content::PathInfo& path_info,
    base::OnceCallback<
        void(content::FileSystemAccessPermissionContext::SensitiveEntryResult)>
        callback,
    content::WebContents* web_contents) {
  chrome::ShowTabModal(
      CreateFileSystemAccessDangerousFileDialog(web_contents, origin, path_info,
                                                std::move(callback)),
      web_contents);
}

std::unique_ptr<ui::DialogModel>
CreateFileSystemAccessDangerousFileDialogForTesting(  // IN-TEST
    const url::Origin& origin,
    const content::PathInfo& path_info,
    base::OnceCallback<
        void(content::FileSystemAccessPermissionContext::SensitiveEntryResult)>
        callback) {
  return CreateFileSystemAccessDangerousFileDialog(
      /*web_contents=*/nullptr, origin, path_info, std::move(callback));
}