chrome/browser/apps/app_service/shortcut_removal_dialog.cc

// Copyright 2023 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/apps/app_service/shortcut_removal_dialog.h"

#include "base/functional/bind.h"
#include "chrome/browser/profiles/profile.h"
#include "components/services/app_service/public/cpp/app_shortcut_image.h"
#include "ui/base/models/image_model.h"
#include "ui/chromeos/styles/cros_tokens_color_mappings.h"
#include "ui/gfx/image/image_skia_operations.h"
#include "ui/views/native_window_tracker.h"
#include "ui/views/widget/widget.h"

namespace {

// The dimensions used to create the shortcut icon.
constexpr int kShortcutIconBackgroundRadius = 24;
constexpr int kBadgeBackgroundRadius = 12;

}  // namespace

namespace apps {

ShortcutRemovalDialog::ShortcutRemovalDialog(
    Profile* profile,
    const apps::ShortcutId& shortcut_id,
    gfx::NativeWindow parent_window,
    ShortcutRemovalCallback shortcut_removal_callback)
    : profile_(profile),
      shortcut_id_(shortcut_id),
      parent_window_(parent_window),
      shortcut_removal_callback_(std::move(shortcut_removal_callback)) {
  if (parent_window) {
    parent_window_tracker_ = views::NativeWindowTracker::Create(parent_window);
  }
}

ShortcutRemovalDialog::~ShortcutRemovalDialog() = default;

void ShortcutRemovalDialog::CreateDialog(gfx::ImageSkia icon,
                                         gfx::ImageSkia badge_icon) {
  if (parent_window_ && parent_window_tracker_->WasNativeWindowDestroyed()) {
    OnDialogClosed(false);
    return;
  }

  const int icon_background_size = 2 * kShortcutIconBackgroundRadius;
  ui::ImageModel icon_with_badge = ui::ImageModel::FromImageGenerator(
      base::BindRepeating(
          [](gfx::ImageSkia icon, gfx::ImageSkia badge_icon,
             const ui::ColorProvider* color_provider) {
            return apps::AppShortcutImage::
                CreateImageWithBadgeAndTeardropBackground(
                    kShortcutIconBackgroundRadius, kBadgeBackgroundRadius,
                    kBadgeBackgroundRadius,
                    color_provider->GetColor(
                        cros_tokens::kCrosSysSystemOnBaseOpaque),
                    icon, badge_icon);
          },
          icon, badge_icon),
      gfx::Size(icon_background_size, icon_background_size));

  widget_ = Create(profile_, shortcut_id_, icon_with_badge, parent_window_,
                   weak_ptr_factory_.GetWeakPtr());
}

base::WeakPtr<views::Widget> ShortcutRemovalDialog::GetWidget() {
  return widget_;
}

void ShortcutRemovalDialog::OnDialogClosed(bool remove) {
  CHECK(shortcut_removal_callback_);
  std::move(shortcut_removal_callback_).Run(remove, this);
}

void ShortcutRemovalDialog::CloseDialog() {
  if (widget_) {
    widget_->CloseWithReason(views::Widget::ClosedReason::kUnspecified);
    return;
  }

  OnDialogClosed(false);
}

}  // namespace apps