// 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.

#include "components/contextual_search/contextual_search_session_entry.h"

#include "components/contextual_search/contextual_search_context_controller.h"
#include "components/contextual_search/contextual_search_metrics_recorder.h"

namespace contextual_search {

ContextualSearchSessionEntry::ContextualSearchSessionEntry(
    ContextualSearchSessionEntry&& other) noexcept
    : controller_(std::move(other.controller_)),
      metrics_recorder_(std::move(other.metrics_recorder_)),
      ref_count_(other.ref_count_) {
  if (controller_) {
    context_upload_status_observer_.Observe(controller_.get());
  }
}

ContextualSearchSessionEntry& ContextualSearchSessionEntry::operator=(
    ContextualSearchSessionEntry&& other) noexcept {
  if (this != &other) {
    controller_ = std::move(other.controller_);
    metrics_recorder_ = std::move(other.metrics_recorder_);
    ref_count_ = other.ref_count_;
    context_upload_status_observer_.Reset();
    if (controller_) {
      context_upload_status_observer_.Observe(controller_.get());
    }
  }
  return *this;
}

ContextualSearchSessionEntry::ContextualSearchSessionEntry(
    std::unique_ptr<ContextualSearchContextController> controller,
    std::unique_ptr<ContextualSearchMetricsRecorder> metrics_recorder)
    : controller_(std::move(controller)),
      metrics_recorder_(std::move(metrics_recorder)) {
  if (controller_) {
    context_upload_status_observer_.Observe(controller_.get());
  }
}

ContextualSearchSessionEntry::~ContextualSearchSessionEntry() = default;

void ContextualSearchSessionEntry::OnContextUploadStatusChanged(
    const base::UnguessableToken& context_token,
    lens::MimeType mime_type,
    contextual_search::ContextUploadStatus context_upload_status,
    const std::optional<contextual_search::ContextUploadErrorType>&
        error_type) {
  if (metrics_recorder_) {
    metrics_recorder_->OnContextUploadStatusChanged(
        mime_type, context_upload_status, error_type);
  }
}

}  // namespace contextual_search