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

#include "base/auto_reset.h"
#include "base/strings/string_util.h"
#include "base/test/scoped_feature_list.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/chrome_content_browser_client.h"
#include "chrome/browser/content_settings/host_content_settings_map_factory.h"
#include "chrome/browser/content_settings/page_specific_content_settings_delegate.h"
#include "chrome/browser/download/download_request_limiter.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/blocked_content/chrome_popup_navigation_delegate.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/content_settings/content_setting_image_model.h"
#include "chrome/browser/ui/location_bar/location_bar.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/browser/ui/test/test_browser_dialog.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/common/chrome_features.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/blocked_content/popup_blocker_tab_helper.h"
#include "components/content_settings/browser/page_specific_content_settings.h"
#include "components/content_settings/core/browser/website_settings_registry.h"
#include "components/content_settings/core/common/content_settings.h"
#include "components/content_settings/core/common/content_settings_types.h"
#include "components/content_settings/core/common/features.h"
#include "components/content_settings/core/test/content_settings_mock_provider.h"
#include "components/content_settings/core/test/content_settings_test_utils.h"
#include "components/permissions/permission_request_manager.h"
#include "components/permissions/prediction_service/permission_ui_selector.h"
#include "components/permissions/request_type.h"
#include "components/permissions/test/mock_permission_request.h"
#include "components/permissions/test/mock_permission_ui_selector.h"
#include "content/public/browser/navigation_entry.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/browser_test_utils.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "net/test/embedded_test_server/request_handler_util.h"
#include "url/gurl.h"

namespace {

using QuietUiReason = permissions::PermissionUiSelector::QuietUiReason;
using Decision = permissions::PermissionUiSelector::Decision;

// An override that returns a fake URL for every blocked popup, so the UI
// displays consistent strings for pixel tests.
class TestPopupNavigationDelegate : public ChromePopupNavigationDelegate {
 public:
  using ChromePopupNavigationDelegate::ChromePopupNavigationDelegate;

  // ChromePopupNavigationDelegate:
  GURL GetURL() override {
    return GURL("https://sub.blocked-popup-domain.com/path");
  }
};

std::unique_ptr<blocked_content::PopupNavigationDelegate>
CreateTestPopupNavigationDelegate(NavigateParams params) {
  return std::make_unique<TestPopupNavigationDelegate>(std::move(params));
}

}  // namespace

using ImageType = ContentSettingImageModel::ImageType;

class ContentSettingBubbleDialogTest
    : public DialogBrowserTest,
      public testing::WithParamInterface<content_settings::ProviderType> {
 public:
  ContentSettingBubbleDialogTest()
      : resetter_(&ChromeContentBrowserClient::
                      GetPopupNavigationDelegateFactoryForTesting(),
                  &CreateTestPopupNavigationDelegate) {
    scoped_feature_list_.InitWithFeatures(
        {features::kQuietNotificationPrompts},
        // `kLeftHandSideActivityIndicators` should be disabled as it changes
        // the UI of the camera/mic activity indicator. The new UI will be
        // tested separately.
        {content_settings::features::kLeftHandSideActivityIndicators});
  }

  ContentSettingBubbleDialogTest(const ContentSettingBubbleDialogTest&) =
      delete;
  ContentSettingBubbleDialogTest& operator=(
      const ContentSettingBubbleDialogTest&) = delete;

  void ApplyMediastreamSettings(bool mic_accessed, bool camera_accessed);
  void ApplyContentSettingsForType(ContentSettingsType content_type);
  void TriggerQuietNotificationPermissionRequest(
      QuietUiReason simulated_reason_for_quiet_ui);
  void OverrideContentSettingsProvider(
      const std::vector<ContentSettingsType>& types);
  void NavigateToContentTab();

  void ShowDialogBubble(ContentSettingImageModel::ImageType image_type);

  void ShowUi(const std::string& name) override;

 private:
  base::AutoReset<ChromeContentBrowserClient::PopupNavigationDelegateFactory>
      resetter_;
  base::test::ScopedFeatureList scoped_feature_list_;
};

void ContentSettingBubbleDialogTest::ApplyMediastreamSettings(
    bool mic_accessed,
    bool camera_accessed) {
  content::WebContents* web_contents =
      browser()->GetTabStripModel()->GetActiveWebContents();

  GURL last_committed_url = web_contents->GetLastCommittedURL();
  // Default opt-in for camera PTZ permission to current tab.
  HostContentSettingsMapFactory::GetForProfile(browser()->GetProfile())
      ->SetContentSettingDefaultScope(last_committed_url, GURL(),
                                      ContentSettingsType::CAMERA_PAN_TILT_ZOOM,
                                      CONTENT_SETTING_ASK);
  content_settings::PageSpecificContentSettings::MicrophoneCameraState state;
  if (mic_accessed) {
    state.Put(
        content_settings::PageSpecificContentSettings::kMicrophoneAccessed);
  }
  if (camera_accessed) {
    state.Put(content_settings::PageSpecificContentSettings::kCameraAccessed);
  }

  content_settings::PageSpecificContentSettings* content_settings =
      content_settings::PageSpecificContentSettings::GetForFrame(
          web_contents->GetPrimaryMainFrame());
  content_settings->OnMediaStreamPermissionSet(last_committed_url, state);
}

void ContentSettingBubbleDialogTest::ApplyContentSettingsForType(
    ContentSettingsType content_type) {
  content::WebContents* web_contents =
      browser()->GetTabStripModel()->GetActiveWebContents();
  content_settings::PageSpecificContentSettings* content_settings =
      content_settings::PageSpecificContentSettings::GetForFrame(
          web_contents->GetPrimaryMainFrame());
  switch (content_type) {
    case ContentSettingsType::AUTOMATIC_DOWNLOADS: {
      // Automatic downloads are handled by DownloadRequestLimiter.
      DownloadRequestLimiter::TabDownloadState* tab_download_state =
          g_browser_process->download_request_limiter()
              ->GetOrCreateDownloadState(web_contents);
      tab_download_state->set_download_seen();
      tab_download_state->SetDownloadStatusAndNotify(
          url::Origin::Create(web_contents->GetVisibleURL()),
          DownloadRequestLimiter::DOWNLOADS_NOT_ALLOWED);
      break;
    }
    case ContentSettingsType::POPUPS: {
      ASSERT_TRUE(ui_test_utils::NavigateToURL(
          browser(),
          embedded_test_server()->GetURL("/popup_blocker/popup-many-10.html")));
      EXPECT_TRUE(content::ExecJs(web_contents, std::string()));
      auto* helper =
          blocked_content::PopupBlockerTabHelper::FromWebContents(web_contents);
      // popup-many-10.html should generate 10 blocked popups.
      EXPECT_EQ(10u, helper->GetBlockedPopupsCount());
      break;
    }
    case ContentSettingsType::PROTOCOL_HANDLERS:
      PageSpecificContentSettingsDelegate::FromWebContents(web_contents)
          ->set_pending_protocol_handler(
              custom_handlers::ProtocolHandler::CreateProtocolHandler(
                  "mailto", GURL("https://example.com/")));
      break;
    case ContentSettingsType::STORAGE_ACCESS:
      // Set a fake URL so the UI displays a consistent string for pixel tests.
      content_settings->OnTwoSitePermissionChanged(
          content_type, net::SchemefulSite(GURL("https://embedded.com")),
          CONTENT_SETTING_BLOCK);
      break;
    default:
      // For all other content_types passed in, mark them as blocked.
      content_settings->OnContentBlocked(content_type);
      break;
  }
  OverrideContentSettingsProvider({content_type});
  BrowserWindow::FromBrowser(browser())->UpdateToolbar(web_contents);
}

void ContentSettingBubbleDialogTest::TriggerQuietNotificationPermissionRequest(
    QuietUiReason simulated_reason_for_quiet_ui) {
  content::WebContents* web_contents =
      browser()->GetTabStripModel()->GetActiveWebContents();
  auto* permission_request_manager =
      permissions::PermissionRequestManager::FromWebContents(web_contents);
  permission_request_manager->set_permission_ui_selector_for_testing(
      std::make_unique<MockPermissionUiSelector>(
          Decision::UseQuietUi(simulated_reason_for_quiet_ui, std::nullopt)));

  permission_request_manager->AddRequest(
      web_contents->GetPrimaryMainFrame(),
      std::make_unique<permissions::MockPermissionRequest>(
          GURL("https://example.com"),
          permissions::RequestType::kNotifications));
  base::RunLoop().RunUntilIdle();
}

void ContentSettingBubbleDialogTest::OverrideContentSettingsProvider(
    const std::vector<ContentSettingsType>& types) {
  auto provider = std::make_unique<content_settings::MockProvider>();
  HostContentSettingsMap* map =
      HostContentSettingsMapFactory::GetForProfile(browser()->GetProfile());

  // All settings should have a default value defined.
  if (GetParam() == content_settings::ProviderType::kDefaultProvider) {
    for (auto* info :
         *content_settings::WebsiteSettingsRegistry::GetInstance()) {
      provider->SetWebsiteSetting(ContentSettingsPattern::Wildcard(),
                                  ContentSettingsPattern::Wildcard(),
                                  info->type(),
                                  info->initial_default_value().Clone(),
                                  /*constraints=*/{});
    }
  }

  // Override specified types.
  for (ContentSettingsType type : types) {
    provider->SetWebsiteSetting(
        ContentSettingsPattern::Wildcard(), ContentSettingsPattern::Wildcard(),
        type, base::Value(ContentSetting::CONTENT_SETTING_BLOCK),
        /*constraints=*/{});
  }
  content_settings::TestUtils::OverrideProvider(map, std::move(provider),
                                                GetParam());
}

void ContentSettingBubbleDialogTest::NavigateToContentTab() {
  embedded_test_server()->ServeFilesFromSourceDirectory(GetChromeTestDataDir());
  ASSERT_TRUE(embedded_test_server()->Start());

  GURL url(embedded_test_server()->GetURL(
      "/content_setting_bubble/mixed_script.html"));
  content::WebContents* web_contents =
      browser()->GetTabStripModel()->GetActiveWebContents();
  EXPECT_TRUE(content::NavigateToURL(web_contents, url));
}

void ContentSettingBubbleDialogTest::ShowDialogBubble(
    ContentSettingImageModel::ImageType image_type) {
  LocationBarTesting* location_bar_testing =
      BrowserWindow::FromBrowser(browser())
          ->GetLocationBar()
          ->GetLocationBarForTesting();
  EXPECT_TRUE(location_bar_testing->TestContentSettingImagePressed(
      ContentSettingImageModel::GetContentSettingImageModelIndexForTesting(
          image_type)));
}

void ContentSettingBubbleDialogTest::ShowUi(const std::string& name) {
  // Open a tab for which we will invoke the dialog bubble.
  NavigateToContentTab();
  auto scoped_display_url =
      ContentSettingBubbleModel::CreateScopedDisplayURLOverrideForTesting();

  if (base::StartsWith(name, "mediastream", base::CompareCase::SENSITIVE)) {
    bool apply_mic_settings = name.find("mic") != std::string::npos;
    bool apply_camera_settings = name.find("camera") != std::string::npos;
    std::vector<ContentSettingsType> types;
    if (apply_mic_settings) {
      types.push_back(ContentSettingsType::MEDIASTREAM_MIC);
    }
    if (apply_camera_settings) {
      types.push_back(ContentSettingsType::MEDIASTREAM_CAMERA);
    }
    OverrideContentSettingsProvider(types);
    ApplyMediastreamSettings(apply_mic_settings, apply_camera_settings);
    ShowDialogBubble(ImageType::kMediaStream);
    return;
  }

  if (base::StartsWith(name, "notifications_quiet",
                       base::CompareCase::SENSITIVE)) {
    QuietUiReason reason = QuietUiReason::kEnabledInPrefs;
    if (base::StartsWith(name, "notifications_quiet_crowd_deny",
                         base::CompareCase::SENSITIVE)) {
      reason = QuietUiReason::kTriggeredByCrowdDeny;
    } else if (base::StartsWith(name, "notifications_quiet_abusive",
                                base::CompareCase::SENSITIVE)) {
      reason = QuietUiReason::kTriggeredDueToAbusiveRequests;
    } else if (base::StartsWith(name, "notifications_quiet_abusive_content",
                                base::CompareCase::SENSITIVE)) {
      reason = QuietUiReason::kTriggeredDueToAbusiveContent;
    } else if (base::StartsWith(name,
                                "notifications_quiet_predicted_very_unlikely",
                                base::CompareCase::SENSITIVE)) {
      reason = QuietUiReason::kServicePredictedVeryUnlikelyGrant;
    }
    TriggerQuietNotificationPermissionRequest(reason);
    ShowDialogBubble(ImageType::kNotifications);
    return;
  }

  constexpr struct {
    const char* name;
    ContentSettingsType content_type;
    ContentSettingImageModel::ImageType image_type;
  } content_settings_values[] = {
      {"cookies", ContentSettingsType::COOKIES, ImageType::kCookies},
      {"images", ContentSettingsType::IMAGES, ImageType::kImages},
      {"javascript", ContentSettingsType::JAVASCRIPT, ImageType::kJavaScript},
      {"popups", ContentSettingsType::POPUPS, ImageType::kPopups},
      {"geolocation", ContentSettingsType::GEOLOCATION,
       ImageType::kGeolocation},
      {"mixed_script", ContentSettingsType::MIXEDSCRIPT,
       ImageType::kMixedScript},
      {"protocol_handlers", ContentSettingsType::PROTOCOL_HANDLERS,
       ImageType::kProtocolHandlers},
      {"automatic_downloads", ContentSettingsType::AUTOMATIC_DOWNLOADS,
       ImageType::kAutomaticDownloads},
      {"midi_sysex", ContentSettingsType::MIDI_SYSEX, ImageType::kMidiSysex},
      {"ads", ContentSettingsType::ADS, ImageType::kAds},
      {"storage_access", ContentSettingsType::STORAGE_ACCESS,
       ImageType::kStorageAccess},
  };
  for (auto content_settings : content_settings_values) {
    if (base::StartsWith(name, content_settings.name,
                         base::CompareCase::SENSITIVE)) {
      ApplyContentSettingsForType(content_settings.content_type);
      ShowDialogBubble(content_settings.image_type);
      return;
    }
  }
  ADD_FAILURE() << "Unknown dialog type " << name;
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest, InvokeUi_cookies) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest, InvokeUi_images) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest, InvokeUi_javascript) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest, InvokeUi_popups) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest, InvokeUi_geolocation) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest, InvokeUi_mixed_script) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest,
                       InvokeUi_mediastream_mic) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest,
                       InvokeUi_mediastream_camera) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest,
                       InvokeUi_mediastream_mic_and_camera) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest,
                       InvokeUi_protocol_handlers) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest,
                       InvokeUi_automatic_downloads) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest, InvokeUi_midi_sysex) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest, InvokeUi_ads) {
  ShowAndVerifyUi();
}

IN_PROC_BROWSER_TEST_P(ContentSettingBubbleDialogTest,
                       InvokeUi_storage_access) {
  set_baseline("79537604");
  ShowAndVerifyUi();
}

INSTANTIATE_TEST_SUITE_P(
    ,
    ContentSettingBubbleDialogTest,
    testing::Values(content_settings::ProviderType::kSupervisedProvider,
                    content_settings::ProviderType::kDefaultProvider));