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

#ifndef CHROME_BROWSER_ACTOR_UI_ACTOR_UI_TAB_CONTROLLER_H_
#define CHROME_BROWSER_ACTOR_UI_ACTOR_UI_TAB_CONTROLLER_H_

#include "base/functional/callback_helpers.h"
#include "base/memory/raw_ref.h"
#include "base/timer/timer.h"
#include "chrome/browser/actor/ui/actor_ui_tab_controller_interface.h"
#include "chrome/browser/actor/ui/handoff_button_controller.h"
#include "components/tabs/public/tab_interface.h"
#include "ui/base/unowned_user_data/scoped_unowned_user_data.h"

namespace actor {
class ActorKeyedService;
}
namespace actor::ui {

class ActorUiTabController : public ActorUiTabControllerInterface {
 public:
  ActorUiTabController(tabs::TabInterface& tab,
                       ActorKeyedService* actor_keyed_service);
  ~ActorUiTabController() override;
  DECLARE_USER_DATA(ActorUiTabController);

  // ActorUiTabControllerInterface:
  void OnUiTabStateChange(const UiTabState& ui_tab_state,
                          UiResultCallback callback) override;
  void SetActorTaskPaused() override;
  void SetActorTaskResume() override;
  base::WeakPtr<ActorUiTabControllerInterface> GetWeakPtr() override;
  UiTabState GetCurrentUiTabState() const override;

#if !BUILDFLAG(IS_ANDROID)
  void OnWebContentsAttached() override;
  void OnViewBoundsChanged() override;
  void OnOverlayHoverStatusChanged(bool is_hovering) override;
  void OnHandoffButtonHoverStatusChanged() override;
  void OnHandoffButtonFocusStatusChanged() override;
  [[nodiscard]] base::ScopedClosureRunner RegisterHandoffButtonController(
      HandoffButtonController* controller) override;

  void OnImmersiveModeChanged() override;

  [[nodiscard]] base::ScopedClosureRunner RegisterActorOverlayStateChange(
      ActorOverlayStateChangeCallback callback) override;
  [[nodiscard]] base::ScopedClosureRunner RegisterActorOverlayBackgroundChange(
      ActorOverlayBackgroundChangeCallback callback) override;
#endif

 private:
  // Called only once on startup to initialize tab subscriptions.
  void RegisterTabSubscriptions();

  // Called to propagate state and visibility changes to UI controllers.
  void UpdateUi(UiResultCallback callback);

  // Computes whether the Actor Overlay is visible based on the current state.
  bool ComputeActorOverlayVisibility();

  // Computes whether the Handoff Button is visible based on the current state.
  bool ComputeHandoffButtonVisibility();

  // Called when the tab is inserted.
  void OnTabDidInsert(tabs::TabInterface* tab);

  // Run the test callback after updates have been made.
  void OnUpdateFinished();

#if !BUILDFLAG(IS_ANDROID)
  // Called when the omnibox's popup visibility changes.
  void OnWindowOmniboxPopupVisibilityChanged() override;
  // Called when tab modal UI showing state changes.
  void OnModalUIChanged(tabs::TabInterface* tab);
#endif

  // Sets the Tab Indicator visibility.
  void SetActorTabIndicatorVisibility(TabIndicatorStatus tab_indicator_status,
                                      base::OnceClosure callback);

  // Sets the Border Glow visibility.
  void SetBorderGlowVisibility(base::OnceClosure callback);

  // Initialize and start observing ImmersiveModeController.
  void InitializeImmersiveModeObserver();

  // Updates the visibility of the scrim background. This method is debounced to
  // consolidate rapid hover events from the overlay and the handoff button. It
  // determines if the scrim background should be visible if the mouse is
  // hovering over either the overlay or the handoff button.
  void UpdateScrimBackground();

  void UnregisterActorOverlayStateChange();
  void UnregisterActorOverlayBackgroundChange();
  void UnregisterHandoffButtonController();

  // The current UiTabState.
  UiTabState current_ui_tab_state_ = {
      .actor_overlay = ActorOverlayState(),
      .handoff_button = HandoffButtonState(),
  };

  // Copy of the current tab's overlay hover status.
  bool is_overlay_hovered_ = false;

  // Determines if the scrim background should be visible. This is set to true
  // if the mouse is hovering over either the overlay or the handoff button.
  bool should_show_scrim_background_ = false;

  // Owns this class via TabModel.
  const raw_ref<tabs::TabInterface> tab_;
  // Holds subscriptions for TabInterface callbacks.
  std::vector<base::CallbackListSubscription> tab_subscriptions_;

#if !BUILDFLAG(IS_ANDROID)
  ActorOverlayStateChangeCallback on_actor_overlay_state_changed_callback_;
  ActorOverlayBackgroundChangeCallback
      actor_overlay_background_changed_callback_;
#endif

  // The Actor Keyed Service for the associated profile.
  raw_ptr<ActorKeyedService> actor_keyed_service_ = nullptr;

  // The Handoff Button controller for this tab.
  raw_ptr<HandoffButtonController> handoff_button_controller_ = nullptr;

  TabIndicatorStatus tab_indicator_ = TabIndicatorStatus::kNone;
  base::RetainingOneShotTimer update_scrim_background_debounce_timer_;

  ::ui::ScopedUnownedUserData<ActorUiTabController> scoped_unowned_user_data_;

  base::WeakPtrFactory<ActorUiTabController> weak_factory_{this};
};

}  // namespace actor::ui

#endif  // CHROME_BROWSER_ACTOR_UI_ACTOR_UI_TAB_CONTROLLER_H_