// Copyright 2021 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/screenshot/screenshot_captured_bubble_controller.h"

#include "base/feature_list.h"
#include "base/notimplemented.h"
#include "chrome/browser/image_editor/screenshot_flow.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/sharing_hub/sharing_hub_window_controller.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "content/public/browser/web_contents.h"
#include "ui/accessibility/platform/ax_platform.h"
#include "ui/base/clipboard/scoped_clipboard_writer.h"

namespace sharing_hub {

ScreenshotCapturedBubbleController::~ScreenshotCapturedBubbleController() {
  HideBubble();
}

// static
ScreenshotCapturedBubbleController* ScreenshotCapturedBubbleController::Get(
    content::WebContents* web_contents) {
  ScreenshotCapturedBubbleController::CreateForWebContents(web_contents);
  ScreenshotCapturedBubbleController* controller =
      ScreenshotCapturedBubbleController::FromWebContents(web_contents);
  return controller;
}

void ScreenshotCapturedBubbleController::ShowBubble(
    const image_editor::ScreenshotCaptureResult& image) {
  const gfx::Image& captured_image = image.image;
  ui::ScopedClipboardWriter(ui::ClipboardBuffer::kCopyPaste)
      .WriteImage(*captured_image.ToSkBitmap());

  BrowserWindowInterface* browser =
      GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(
          &GetWebContents());
  sharing_hub::SharingHubWindowController::From(browser)
      ->ShowScreenshotCapturedBubble(&GetWebContents(), captured_image);
}

void ScreenshotCapturedBubbleController::HideBubble() {
  NOTIMPLEMENTED();
}

void ScreenshotCapturedBubbleController::OnBubbleClosed() {
  NOTIMPLEMENTED();
}

void ScreenshotCapturedBubbleController::Capture(
    BrowserWindowInterface* browser) {
  content::WebContents* web_contents =
      browser->GetTabStripModel()->GetActiveWebContents();
  screenshot_flow_ =
      std::make_unique<image_editor::ScreenshotFlow>(web_contents);

  base::OnceCallback<void(const image_editor::ScreenshotCaptureResult&)>
      callback = base::BindOnce(
          [](base::WeakPtr<content::WebContents> web_contents,
             const image_editor::ScreenshotCaptureResult& image) {
            if (image.image.IsEmpty() || !web_contents) {
              return;
            }
            sharing_hub::ScreenshotCapturedBubbleController* controller =
                sharing_hub::ScreenshotCapturedBubbleController::Get(
                    web_contents.get());
            controller->ShowBubble(image);
          },
          web_contents->GetWeakPtr());

  if (ui::AXPlatform::GetInstance().IsScreenReaderActive()) {
    screenshot_flow_->StartFullscreenCapture(std::move(callback));
  } else {
    screenshot_flow_->Start(std::move(callback));
  }
}

ScreenshotCapturedBubbleController::ScreenshotCapturedBubbleController(
    content::WebContents* web_contents)
    : content::WebContentsUserData<ScreenshotCapturedBubbleController>(
          *web_contents) {}

WEB_CONTENTS_USER_DATA_KEY_IMPL(ScreenshotCapturedBubbleController);

}  // namespace sharing_hub