chromium/chrome/browser/ui/ash/birch/birch_most_visited_provider.cc

// Copyright 2024 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/ash/birch/birch_most_visited_provider.h"

#include "ash/birch/birch_item.h"
#include "ash/birch/birch_model.h"
#include "ash/shell.h"
#include "chrome/browser/history/history_service_factory.h"
#include "chrome/grit/chrome_unscaled_resources.h"
#include "components/history/core/browser/history_service.h"
#include "components/history/core/browser/history_types.h"
#include "components/keyed_service/core/service_access_type.h"
#include "ui/base/resource/resource_bundle.h"

namespace ash {

BirchMostVisitedProvider::BirchMostVisitedProvider(Profile* profile)
    : profile_(profile),
      history_service_(HistoryServiceFactory::GetForProfile(
          profile_,
          ServiceAccessType::EXPLICIT_ACCESS)) {}

BirchMostVisitedProvider::~BirchMostVisitedProvider() = default;

void BirchMostVisitedProvider::RequestBirchDataFetch() {
  // `history_service_` can be null in some tests, so check that here.
  if (!history_service_) {
    Shell::Get()->birch_model()->SetMostVisitedItems({});
    return;
  }

  // Get the most frequently accessed URL.
  history_service_->QueryMostVisitedURLs(
      /*result_count=*/1,
      base::BindOnce(&BirchMostVisitedProvider::OnGotMostVisitedURLs,
                     weak_factory_.GetWeakPtr()),
      &cancelable_task_tracker_);
}

void BirchMostVisitedProvider::OnGotMostVisitedURLs(
    history::MostVisitedURLList urls) {
  if (urls.empty()) {
    Shell::Get()->birch_model()->SetMostVisitedItems({});
    return;
  }
  // Birch only shows the most frequent URL.
  const auto& most_visited_url = urls[0];

  std::vector<BirchMostVisitedItem> most_visited_items;
  most_visited_items.emplace_back(most_visited_url.title, most_visited_url.url);
  Shell::Get()->birch_model()->SetMostVisitedItems(
      std::move(most_visited_items));
}

}  // namespace ash