// Copyright 2020 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/enterprise/reporting/extension_request/extension_request_observer_factory.h"

#include "chrome/browser/browser_process.h"
#include "chrome/browser/profiles/profile_manager.h"

namespace enterprise_reporting {

ExtensionRequestObserverFactory::ExtensionRequestObserverFactory(
    Profile* profile)
    : profile_(profile) {
  if (profile) {
    OnProfileAdded(profile);
  } else {
    ProfileManager* profile_manager = g_browser_process->profile_manager();
    profile_manager->AddObserver(this);
    for (Profile* loaded_profile : profile_manager->GetLoadedProfiles())
      OnProfileAdded(loaded_profile);
  }
}

ExtensionRequestObserverFactory::~ExtensionRequestObserverFactory() {
  // Remove any pending observers.
  for (const auto& entry : observers_) {
    Profile* profile = entry.first;
    profile->RemoveObserver(this);
  }

  if (!profile_ && g_browser_process->profile_manager()) {
    g_browser_process->profile_manager()->RemoveObserver(this);
  }
}

bool ExtensionRequestObserverFactory::IsReportEnabled() {
  return !report_trigger_.is_null();
}

void ExtensionRequestObserverFactory::EnableReport(
    ExtensionRequestObserver::ReportTrigger trigger) {
  report_trigger_ = trigger;
  for (const auto& observerItem : observers_) {
    observerItem.second->EnableReport(trigger);
  }
}

void ExtensionRequestObserverFactory::DisableReport() {
  report_trigger_.Reset();
  for (const auto& observerItem : observers_) {
    observerItem.second->DisableReport();
  }
}

void ExtensionRequestObserverFactory::OnProfileAdded(Profile* profile) {
  if (profile->IsSystemProfile() || profile->IsGuestSession() ||
      profile->IsOffTheRecord()) {
    return;
  }

  if (profile_ && (profile_ != profile || !observers_.empty()))
    return;

  // Listen for OnProfileWillBeDestroyed() on this profile.
  profile->AddObserver(this);
  auto observer = std::make_unique<ExtensionRequestObserver>(profile);
  if (report_trigger_)
    observer->EnableReport(report_trigger_);
  observers_.emplace(profile, std::move(observer));
}

void ExtensionRequestObserverFactory::OnProfileMarkedForPermanentDeletion(
    Profile* profile) {
  profile->RemoveObserver(this);
  observers_.erase(profile);
}

void ExtensionRequestObserverFactory::OnProfileWillBeDestroyed(
    Profile* profile) {
  profile->RemoveObserver(this);
  observers_.erase(profile);
}

ExtensionRequestObserver*
ExtensionRequestObserverFactory::GetObserverByProfileForTesting(
    Profile* profile) {
  auto it = observers_.find(profile);
  return it == observers_.end() ? nullptr : it->second.get();
}

int ExtensionRequestObserverFactory::GetNumberOfObserversForTesting() {
  return observers_.size();
}

}  // namespace enterprise_reporting