// 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_UI_WEBUI_TAB_STRIP_INTERNALS_TAB_STRIP_INTERNALS_HANDLER_H_
#define CHROME_BROWSER_UI_WEBUI_TAB_STRIP_INTERNALS_TAB_STRIP_INTERNALS_HANDLER_H_

#include <memory>
#include <vector>

#include "base/memory/raw_ptr.h"
#include "chrome/browser/ui/webui/tab_strip_internals/tab_strip_internals.mojom.h"
#include "content/public/browser/web_contents_observer.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "mojo/public/cpp/bindings/remote.h"

class Profile;
class SessionID;
class TabStripInternalsObserver;

namespace content {
class WebContents;
}  // namespace content

namespace sessions {
struct SessionWindow;
}  // namespace sessions

// Browser side handler for requests from `chrome://tab-strip-internals` WebUI.
class TabStripInternalsPageHandler
    : public tab_strip_internals::mojom::PageHandler,
      public content::WebContentsObserver {
 public:
  TabStripInternalsPageHandler(
      content::WebContents* web_contents,
      mojo::PendingReceiver<tab_strip_internals::mojom::PageHandler> receiver,
      mojo::PendingRemote<tab_strip_internals::mojom::Page> page);

  TabStripInternalsPageHandler(const TabStripInternalsPageHandler&) = delete;
  TabStripInternalsPageHandler& operator=(const TabStripInternalsPageHandler&) =
      delete;
  ~TabStripInternalsPageHandler() override;

  // Fetch the current state of all tabstrip models.
  void GetTabStripData(GetTabStripDataCallback callback) override;

  // content::WebContentsObserver:
  void OnVisibilityChanged(content::Visibility visibility) override;

 private:
  // Build a snapshot of the current state of all tabstrip models.
  tab_strip_internals::mojom::ContainerPtr BuildSnapshot();
  // Push live updates to the webui.
  void NotifyTabStripUpdated();
  // Callback invoked when session data is read from disk.
  void OnGotSavedSession(
      std::vector<std::unique_ptr<sessions::SessionWindow>> windows,
      SessionID active_window_id,
      bool read_error);

  mojo::Receiver<tab_strip_internals::mojom::PageHandler> receiver_;
  mojo::Remote<tab_strip_internals::mojom::Page> page_;

  raw_ptr<Profile> profile_;
  std::unique_ptr<TabStripInternalsObserver> observer_;

  // Callback for pending Mojo request.
  GetTabStripDataCallback pending_callback_;
  // Cache of saved session restore data.
  std::vector<std::unique_ptr<sessions::SessionWindow>> saved_session_windows_;

  base::WeakPtrFactory<TabStripInternalsPageHandler> weak_ptr_factory_{this};
};

#endif  // CHROME_BROWSER_UI_WEBUI_TAB_STRIP_INTERNALS_TAB_STRIP_INTERNALS_HANDLER_H_