chrome/browser/tab/tab_state_storage_service.h

// 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_TAB_TAB_STATE_STORAGE_SERVICE_H_
#define CHROME_BROWSER_TAB_TAB_STATE_STORAGE_SERVICE_H_

#include <memory>
#include <string_view>
#include <vector>

#include "base/android/scoped_java_ref.h"
#include "base/android/token_android.h"
#include "base/functional/callback_forward.h"
#include "base/supports_user_data.h"
#include "chrome/browser/tab/tab_state_storage_backend.h"
#include "chrome/browser/tab/tab_state_storage_database.h"
#include "components/keyed_service/core/keyed_service.h"

namespace tabs {

class TabStateStorageService : public KeyedService,
                               public base::SupportsUserData {
 public:
  using LoadAllTabsCallback =
      base::OnceCallback<void(std::vector<tabs_pb::TabState>)>;

  explicit TabStateStorageService(
      std::unique_ptr<TabStateStorageBackend> tab_backend);
  ~TabStateStorageService() override;

  void SaveTab(int id,
               int parent_tab_id,
               int root_id,
               long timestamp_millis,
               const std::string* web_content_state_string,
               std::string_view opener_app_id,
               int theme_color,
               int launch_type_at_creation,
               int user_agent,
               long last_navigation_committed_timestamp_millis,
               const base::Token* tab_group_id,
               bool tab_has_sensitive_content,
               bool is_pinned);

  void LoadAllTabs(LoadAllTabsCallback callback);

  // Returns a Java object of the type TabStateStorageService. This is
  // implemented in tab_state_storage_service_android.cc
  static base::android::ScopedJavaLocalRef<jobject> GetJavaObject(
      TabStateStorageService* tab_state_storage_service);

 private:
  void OnAllTabsLoaded(LoadAllTabsCallback callback,
                       std::vector<NodeState> entries);
  std::unique_ptr<TabStateStorageBackend> tab_backend_;
};

}  // namespace tabs

#endif  // CHROME_BROWSER_TAB_TAB_STATE_STORAGE_SERVICE_H_