// 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 "chrome/browser/ui/browser_window/public/profile_browser_collection.h"

#include "base/check_deref.h"
#include "build/build_config.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/browser_window/public/global_browser_collection.h"

#if BUILDFLAG(IS_ANDROID)
#include "chrome/browser/ui/android/android_profile_browser_collection_service.h"
#include "chrome/browser/ui/android/android_profile_browser_collection_service_factory.h"
#else
#include "chrome/browser/ui/browser_manager_service.h"
#include "chrome/browser/ui/browser_manager_service_factory.h"
#include "chrome/browser/ui/browser_window.h"
#endif  // BUILDFLAG(IS_ANDROID)

ProfileBrowserCollection::ProfileBrowserCollection(Profile* profile)
    : profile_(CHECK_DEREF(profile)) {}

ProfileBrowserCollection::~ProfileBrowserCollection() = default;

BrowserWindowInterface* ProfileBrowserCollection::FindTabbedBrowser(
    bool match_original_profiles) {
  Profile* original =
      match_original_profiles ? profile_->GetOriginalProfile() : nullptr;
  BrowserWindowInterface* match = nullptr;

  auto find = [&match, original](BrowserWindowInterface* browser) {
    if (browser->GetType() != BrowserWindowInterface::TYPE_NORMAL ||
        browser->IsDeleteScheduled()) {
      return true;
    }
    if (original && browser->GetProfile()->GetOriginalProfile() != original) {
      return true;
    }

#if !BUILDFLAG(IS_ANDROID)
    BrowserWindow* browser_window = BrowserWindow::FromBrowser(browser);
    if (!browser_window || !browser_window->IsOnCurrentWorkspace()) {
      return true;
    }
#endif

    match = browser;
    return false;  // stop iterating
  };

  if (match_original_profiles) {
    GlobalBrowserCollection::GetInstance()->ForEach(find, Order::kActivation);
  } else {
    ForEach(find, Order::kActivation);
  }
  return match;
}

size_t ProfileBrowserCollection::GetOffTheRecordBrowserCount() {
  size_t count = 0;
  auto count_browsers = [&count](BrowserWindowInterface* browser) {
  // TODO(crbug.com/511561378): Explore removing this exception for
  // TYPE_DEVTOOLS. Inherited from the prior BrowserList implementation,
  // but similar simplifications elsewhere suggest this exception is
  // unnecessary and is an unexpected exception to clients of this API.
#if !BUILDFLAG(IS_ANDROID)
    if (browser->GetType() == BrowserWindowInterface::Type::TYPE_DEVTOOLS) {
      return true;
    }
#endif
    ++count;
    return true;
  };
  for (Profile* otr : profile_->GetAllOffTheRecordProfiles()) {
    if (ProfileBrowserCollection* otr_collection = GetForProfile(otr)) {
      otr_collection->ForEach(count_browsers);
    }
  }
  return count;
}

// static
ProfileBrowserCollection* ProfileBrowserCollection::GetForProfile(
    Profile* profile) {
#if BUILDFLAG(IS_ANDROID)
  return AndroidProfileBrowserCollectionServiceFactory::GetForProfile(profile);
#else
  return BrowserManagerServiceFactory::GetForProfile(profile);
#endif  // BUILDFLAG(IS_ANDROID)
}