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

#include "ash/webui/graduation/graduation_ui_handler.h"

#include <memory>
#include <utility>

#include "ash/webui/graduation/graduation_state_tracker.h"
#include "ash/webui/graduation/mojom/graduation_ui.mojom-shared.h"
#include "ash/webui/graduation/mojom/graduation_ui.mojom.h"
#include "ash/webui/graduation/webview_auth_handler.h"
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "components/user_manager/user.h"
#include "mojo/public/cpp/bindings/callback_helpers.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "ui/base/webui/web_ui_util.h"
#include "ui/gfx/image/image.h"
#include "ui/gfx/image/image_skia.h"
#include "ui/gfx/image/image_skia_rep.h"

namespace ash::graduation {

GraduationUiHandler::TestApi::TestApi(GraduationUiHandler* handler)
    : handler_(handler) {
  CHECK(handler_);
}

GraduationUiHandler::TestApi::~TestApi() = default;

WebviewAuthHandler* GraduationUiHandler::TestApi::GetWebviewAuthHandler() {
  return handler_->auth_handler_.get();
}

GraduationUiHandler::GraduationUiHandler(
    mojo::PendingReceiver<graduation_ui::mojom::GraduationUiHandler> receiver,
    std::unique_ptr<WebviewAuthHandler> auth_handler,
    const user_manager::User& user)
    : receiver_(this, std::move(receiver)),
      auth_handler_(std::move(auth_handler)),
      user_(user) {
  CHECK(auth_handler_);
}

GraduationUiHandler::~GraduationUiHandler() = default;

void GraduationUiHandler::AuthenticateWebview(
    AuthenticateWebviewCallback callback) {
  auth_handler_->AuthenticateWebview(
      mojo::WrapCallbackWithDefaultInvokeIfNotRun(
          base::BindOnce(&GraduationUiHandler::OnAuthenticationFinished,
                         base::Unretained(this), std::move(callback)),
          false));
}

void GraduationUiHandler::GetProfileInfo(GetProfileInfoCallback callback) {
  std::move(callback).Run(graduation_ui::mojom::ProfileInfo::New(
      user_->GetDisplayEmail(),
      webui::GetBitmapDataUrl(
          user_->GetImage().GetRepresentation(1.0f).GetBitmap())));
}

void GraduationUiHandler::OnScreenSwitched(
    graduation_ui::mojom::GraduationScreen screen) {
  switch (screen) {
    case graduation_ui::mojom::GraduationScreen::kWelcome:
      state_tracker_.set_flow_state(
          GraduationStateTracker::FlowState::kWelcome);
      break;
    case graduation_ui::mojom::GraduationScreen::kTakeoutUi:
      state_tracker_.set_flow_state(
          GraduationStateTracker::FlowState::kTakeoutUi);
      break;
    case graduation_ui::mojom::GraduationScreen::kError:
      state_tracker_.set_flow_state(GraduationStateTracker::FlowState::kError);
      break;
  }
}

void GraduationUiHandler::OnTransferComplete() {
  state_tracker_.set_flow_state(
      GraduationStateTracker::FlowState::kTakeoutTransferComplete);
}

void GraduationUiHandler::OnAuthenticationFinished(
    AuthenticateWebviewCallback callback,
    bool is_success) {
  std::move(callback).Run(is_success
                              ? graduation_ui::mojom::AuthResult::kSuccess
                              : graduation_ui::mojom::AuthResult::kError);
}

}  // namespace ash::graduation