// 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/sharing_hub/sharing_hub_bubble_controller_desktop_impl.h"

#include "base/metrics/user_metrics.h"
#include "chrome/app/chrome_command_ids.h"
#include "chrome/browser/favicon/favicon_utils.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/share/share_metrics.h"
#include "chrome/browser/sharing_hub/sharing_hub_features.h"
#include "chrome/browser/sharing_hub/sharing_hub_model.h"
#include "chrome/browser/sharing_hub/sharing_hub_service.h"
#include "chrome/browser/sharing_hub/sharing_hub_service_factory.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_commands.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/browser_window/public/global_browser_collection.h"
#include "chrome/browser/ui/color/chrome_color_id.h"
#include "chrome/browser/ui/qrcode_generator/qrcode_generator_bubble_controller.h"
#include "chrome/browser/ui/send_tab_to_self/send_tab_to_self_bubble.h"
#include "chrome/browser/ui/sharing_hub/sharing_hub_bubble_view.h"
#include "chrome/browser/ui/sharing_hub/sharing_hub_window_controller.h"
#include "chrome/grit/generated_resources.h"
#include "components/media_router/browser/media_router_dialog_controller.h"
#include "components/media_router/browser/media_router_metrics.h"
#include "content/public/browser/site_instance.h"
#include "content/public/browser/storage_partition.h"
#include "content/public/browser/web_contents.h"
#include "ui/base/l10n/l10n_util.h"
#include "ui/color/color_provider.h"
#include "ui/native_theme/native_theme.h"

namespace sharing_hub {

// static
// SharingHubBubbleController:
SharingHubBubbleController*
SharingHubBubbleController::CreateOrGetFromWebContents(
    content::WebContents* web_contents) {
  SharingHubBubbleControllerDesktopImpl::CreateForWebContents(web_contents);
  SharingHubBubbleControllerDesktopImpl* controller =
      SharingHubBubbleControllerDesktopImpl::FromWebContents(web_contents);
  return controller;
}

SharingHubBubbleControllerDesktopImpl::
    ~SharingHubBubbleControllerDesktopImpl() {
  if (sharing_hub_bubble_view_) {
    sharing_hub_bubble_view_->Hide();
  }
}

void SharingHubBubbleControllerDesktopImpl::HideBubble() {
  if (sharing_hub_bubble_view_) {
    sharing_hub_bubble_view_->Hide();
    sharing_hub_bubble_view_ = nullptr;
  }
}

void SharingHubBubbleControllerDesktopImpl::ShowBubble(
    share::ShareAttempt attempt) {
  ui::ImageModel preview_image = GetPreviewImage();
  if (!preview_image.IsEmpty()) {
    attempt.preview_image = preview_image;
  }

  BrowserWindowInterface* browser =
      GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(
          web_contents());

  sharing_hub_bubble_view_ =
      sharing_hub::SharingHubWindowController::From(browser)
          ->ShowSharingHubBubble(attempt);

  share::LogShareSourceDesktop(share::ShareSourceDesktop::kOmniboxSharingHub);
}

SharingHubBubbleView*
SharingHubBubbleControllerDesktopImpl::sharing_hub_bubble_view() const {
  return sharing_hub_bubble_view_;
}

bool SharingHubBubbleControllerDesktopImpl::ShouldOfferOmniboxIcon() {
  return SharingHubOmniboxEnabled(GetWebContents().GetBrowserContext());
}

std::u16string SharingHubBubbleControllerDesktopImpl::GetWindowTitle() const {
  return l10n_util::GetStringUTF16(IDS_SHARING_HUB_TITLE);
}

Profile* SharingHubBubbleControllerDesktopImpl::GetProfile() const {
  return Profile::FromBrowserContext(web_contents()->GetBrowserContext());
}

std::vector<SharingHubAction>
SharingHubBubbleControllerDesktopImpl::GetFirstPartyActions() {
  std::vector<SharingHubAction> actions;

  SharingHubModel* model = GetSharingHubModel();
  if (model) {
    actions = model->GetFirstPartyActionList(&GetWebContents());
  }

  return actions;
}

base::WeakPtr<SharingHubBubbleController>
SharingHubBubbleControllerDesktopImpl::GetWeakPtr() {
  return weak_factory_.GetWeakPtr();
}

void SharingHubBubbleControllerDesktopImpl::OnActionSelected(
    const SharingHubAction& action) {
  BrowserWindowInterface* browser =
      GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(
          &GetWebContents());
  // Can be null in tests.
  if (!browser) {
    return;
  }

  base::RecordComputedAction(action.feature_name_for_metrics);

  // Show a back button for 1P dialogs anchored to the sharing hub icon.
  if (action.command_id == IDC_QRCODE_GENERATOR) {
    qrcode_generator::QRCodeGeneratorBubbleController* qrcode_controller =
        qrcode_generator::QRCodeGeneratorBubbleController::Get(
            &GetWebContents());
    qrcode_controller->ShowBubble(GetWebContents().GetLastCommittedURL(), true);
  } else if (action.command_id == IDC_SEND_TAB_TO_SELF) {
    send_tab_to_self::ShowBubble(
        &GetWebContents(), send_tab_to_self::ShareEntryPoint::kToolbarIcon,
        /*show_back_button=*/true);
  } else if (action.command_id == IDC_ROUTE_MEDIA) {
    media_router::MediaRouterDialogController* dialog_controller =
        media_router::MediaRouterDialogController::GetOrCreateForWebContents(
            browser->GetTabStripModel()->GetActiveWebContents());
    if (!dialog_controller) {
      return;
    }

    dialog_controller->ShowMediaRouterDialog(
        media_router::MediaRouterDialogActivationLocation::SHARING_HUB);
  } else {
    chrome::ExecuteCommand(browser, action.command_id);
  }
}

void SharingHubBubbleControllerDesktopImpl::OnBubbleClosed() {
  sharing_hub_bubble_view_ = nullptr;
}

SharingHubModel* SharingHubBubbleControllerDesktopImpl::GetSharingHubModel() {
  if (!sharing_hub_model_) {
    SharingHubService* const service =
        SharingHubServiceFactory::GetForProfile(GetProfile());
    if (!service) {
      return nullptr;
    }
    sharing_hub_model_ = service->GetSharingHubModel();
  }
  return sharing_hub_model_;
}

ui::ImageModel SharingHubBubbleControllerDesktopImpl::GetPreviewImage() {
  content::WebContents* web_contents = &GetWebContents();
  gfx::Image favicon = favicon::TabFaviconFromWebContents(web_contents);
  if (favicon.IsEmpty()) {
    return {};
  }

  content::NavigationController& controller = web_contents->GetController();
  content::NavigationEntry* entry = controller.GetLastCommittedEntry();
  // Select chrome URLs show themified icons in dark mode.
  if (favicon::ShouldThemifyFaviconForEntry(entry) &&
      ui::NativeTheme::GetInstanceForNativeUi()->preferred_color_scheme() ==
          ui::NativeTheme::PreferredColorScheme::kDark) {
    const ui::ColorProvider& color_provider = web_contents->GetColorProvider();
    favicon = gfx::Image(favicon::ThemeFavicon(
        *favicon.ToImageSkia(),
        color_provider.GetColor(kColorToolbarButtonIcon),
        color_provider.GetColor(kColorTabBackgroundActiveFrameActive),
        color_provider.GetColor(kColorTabBackgroundInactiveFrameActive)));
  }

  return ui::ImageModel::FromImage(favicon);
}

SharingHubBubbleControllerDesktopImpl::SharingHubBubbleControllerDesktopImpl(
    content::WebContents* web_contents)
    : content::WebContentsObserver(web_contents),
      content::WebContentsUserData<SharingHubBubbleControllerDesktopImpl>(
          *web_contents) {}

WEB_CONTENTS_USER_DATA_KEY_IMPL(SharingHubBubbleControllerDesktopImpl);

}  // namespace sharing_hub