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

#include "components/download/public/common/download_save_item_data.h"
#include "components/download/public/common/download_item.h"

namespace download {
namespace {

// This memory address is used as unique key by SupportsUserData.
const void* const kSaveItemData = &kSaveItemData;

}  // namespace

DownloadSaveItemData::DownloadSaveItemData(std::vector<ItemInfo>&& item_infos)
    : item_infos_(std::move(item_infos)) {}

DownloadSaveItemData::~DownloadSaveItemData() = default;

// static
void DownloadSaveItemData::AttachItemData(DownloadItem* download_item,
                                          std::vector<ItemInfo> item_infos) {
  download_item->SetUserData(
      kSaveItemData,
      std::make_unique<DownloadSaveItemData>(std::move(item_infos)));
}

// static
std::vector<DownloadSaveItemData::ItemInfo>* DownloadSaveItemData::GetItemData(
    DownloadItem* download_item) {
  auto* item_data = static_cast<DownloadSaveItemData*>(
      download_item->GetUserData(kSaveItemData));
  return !item_data ? nullptr : &item_data->item_infos_;
}

}  // namespace download