chrome/browser/tab/storage_loaded_data.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_STORAGE_LOADED_DATA_H_
#define CHROME_BROWSER_TAB_STORAGE_LOADED_DATA_H_

#include <memory>
#include <optional>
#include <string>
#include <utility>
#include <vector>

#include "base/functional/callback_forward.h"
#include "chrome/browser/tab/payload.h"
#include "chrome/browser/tab/protocol/tab_state.pb.h"
#include "chrome/browser/tab/restore_id_associator.h"
#include "chrome/browser/tab/tab_group_collection_data.h"
#include "components/tabs/public/tab_interface.h"
#include "third_party/abseil-cpp/absl/container/flat_hash_set.h"

namespace tabs {

using OnTabInterfaceCreation = base::OnceCallback<void(const TabInterface*)>;
using LoadedTabState = std::pair<tabs_pb::TabState, OnTabInterfaceCreation>;

// Represents data loaded from the database.
class StorageLoadedData {
 public:
  StorageLoadedData(
      std::vector<LoadedTabState> loaded_tabs,
      std::vector<std::unique_ptr<TabGroupCollectionData>> loaded_groups,
      std::unique_ptr<RestoreIdAssociator> associator,
      std::optional<int> active_tab_index);
  ~StorageLoadedData();

  StorageLoadedData(const StorageLoadedData&) = delete;
  StorageLoadedData& operator=(const StorageLoadedData&) = delete;

  StorageLoadedData(StorageLoadedData&&);
  StorageLoadedData& operator=(StorageLoadedData&&);

  RestoreIdAssociator* GetNodeAssociator() const;
  std::vector<LoadedTabState>& GetLoadedTabs();
  std::vector<std::unique_ptr<TabGroupCollectionData>>& GetLoadedGroups();
  std::optional<int> GetActiveTabIndex() const;

 private:
  std::vector<LoadedTabState> loaded_tabs_;
  std::vector<std::unique_ptr<TabGroupCollectionData>> loaded_groups_;
  std::unique_ptr<RestoreIdAssociator> node_associator_;
  std::optional<int> active_tab_index_;
};

}  // namespace tabs

#endif  // CHROME_BROWSER_TAB_STORAGE_LOADED_DATA_H_