// Copyright 2015 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/history/chrome_history_backend_client.h"

#include "components/bookmarks/browser/history_bookmark_model.h"
#include "components/bookmarks/browser/model_loader.h"
#include "components/bookmarks/browser/url_and_title.h"
#include "content/public/browser/child_process_security_policy.h"
#include "url/gurl.h"

ChromeHistoryBackendClient::ChromeHistoryBackendClient(
    scoped_refptr<bookmarks::ModelLoader> model_loader)
    : model_loader_(std::move(model_loader)) {}

ChromeHistoryBackendClient::~ChromeHistoryBackendClient() = default;

bool ChromeHistoryBackendClient::IsPinnedURL(const GURL& url) {
  if (!model_loader_)
    return false;

  // HistoryBackendClient is used to determine if an URL is bookmarked. The data
  // is loaded on a separate thread and may not be done when this method is
  // called, therefore blocks until the bookmarks have finished loading.
  model_loader_->BlockTillLoaded();
  return model_loader_->history_bookmark_model()->IsBookmarked(url);
}

std::vector<history::URLAndTitle> ChromeHistoryBackendClient::GetPinnedURLs() {
  std::vector<history::URLAndTitle> result;
  if (!model_loader_)
    return result;

  // HistoryBackendClient is used to determine the set of bookmarked URLs. The
  // data is loaded on a separate thread and may not be done when this method is
  // called, therefore blocks until the bookmarks have finished loading.
  model_loader_->BlockTillLoaded();
  std::vector<bookmarks::UrlAndTitle> url_and_titles =
      model_loader_->history_bookmark_model()->GetUniqueUrls();

  result.reserve(url_and_titles.size());
  for (const auto& url_and_title : url_and_titles) {
    result.push_back(
        history::URLAndTitle{url_and_title.url, url_and_title.title});
  }
  return result;
}

bool ChromeHistoryBackendClient::IsWebSafe(const GURL& url) {
  return content::ChildProcessSecurityPolicy::GetInstance()->IsWebSafeScheme(
      url.GetScheme());
}