// 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 "chrome/browser/ui/download/download_bubble_row_list_view_info.h"

DownloadBubbleRowListViewInfoObserver::DownloadBubbleRowListViewInfoObserver() =
    default;

DownloadBubbleRowListViewInfoObserver::
    ~DownloadBubbleRowListViewInfoObserver() {
  CHECK(!IsInObserverList());
}

DownloadBubbleRowListViewInfo::DownloadBubbleRowListViewInfo(
    std::vector<DownloadUIModel::DownloadUIModelPtr> models) {
  CHECK(!models.empty());

  for (DownloadUIModel::DownloadUIModelPtr& model : models) {
    AddRow(std::move(model));
  }
}

DownloadBubbleRowListViewInfo::~DownloadBubbleRowListViewInfo() = default;

void DownloadBubbleRowListViewInfo::OnDownloadDestroyed(
    const offline_items_collection::ContentId& id) {
  RemoveRow(id);
}

const DownloadBubbleRowViewInfo* DownloadBubbleRowListViewInfo::GetRowInfo(
    const offline_items_collection::ContentId& id) const {
  auto it = row_list_iter_map_.find(id);
  if (it == row_list_iter_map_.end()) {
    return nullptr;
  }
  return &(*it->second);
}

void DownloadBubbleRowListViewInfo::AddRow(
    DownloadUIModel::DownloadUIModelPtr model,
    std::optional<size_t> index) {
  offline_items_collection::ContentId id = model->GetContentId();

  // Find the insert position.
  size_t insert_index = index.value_or(rows_.size());
  CHECK_LE(insert_index, rows_.size());
  auto it = (insert_index == rows_.size())
                ? rows_.end()
                : std::next(rows_.begin(), insert_index);

  auto new_it = rows_.emplace(it, std::move(model));
  new_it->AddObserver(this);
  row_list_iter_map_.emplace(id, new_it);
  NotifyObservers(&DownloadBubbleRowListViewInfoObserver::OnRowAdded, id,
                  insert_index);
}

void DownloadBubbleRowListViewInfo::RemoveRow(
    const offline_items_collection::ContentId& id) {
  auto it = row_list_iter_map_.find(id);
  if (it == row_list_iter_map_.end()) {
    return;
  }

  NotifyObservers(&DownloadBubbleRowListViewInfoObserver::OnRowWillBeRemoved,
                  id);
  rows_.erase(it->second);
  row_list_iter_map_.erase(it);
  NotifyObservers(&DownloadBubbleRowListViewInfoObserver::OnAnyRowRemoved);
}