Add bookmark provider for content suggestions

chrome/browser/android/ntp/bookmark_suggestions_provider_factory.cc

+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/android/ntp/bookmark_suggestions_provider_factory.h"
+
+#include "base/feature_list.h"
+#include "base/memory/singleton.h"
+#include "chrome/browser/android/chrome_feature_list.h"
+#include "chrome/browser/bookmarks/bookmark_model_factory.h"
+#include "chrome/browser/ntp_snippets/content_suggestions_service_factory.h"
+#include "chrome/browser/profiles/profile.h"
+#include "components/bookmarks/browser/bookmark_model.h"
+#include "components/keyed_service/content/browser_context_dependency_manager.h"
+#include "components/ntp_snippets/bookmarks/bookmark_suggestions_provider.h"
+#include "components/ntp_snippets/content_suggestions_service.h"
+#include "content/public/browser/browser_context.h"
+
+using ntp_snippets::ContentSuggestionsService;
+using ntp_snippets::BookmarkSuggestionsProvider;
+using bookmarks::BookmarkModel;
+
+// static
+BookmarkSuggestionsProviderFactory*
+BookmarkSuggestionsProviderFactory::GetInstance() {
+  return base::Singleton<BookmarkSuggestionsProviderFactory>::get();
+}
+
+// static
+BookmarkSuggestionsProvider* BookmarkSuggestionsProviderFactory::GetForProfile(
+    Profile* profile) {
+  DCHECK(!profile->IsOffTheRecord());
+  return static_cast<BookmarkSuggestionsProvider*>(
+      GetInstance()->GetServiceForBrowserContext(profile, true));
+}
+
+BookmarkSuggestionsProviderFactory::BookmarkSuggestionsProviderFactory()
+    : BrowserContextKeyedServiceFactory(
+          "BookmarkSuggestionsProvider",
+          BrowserContextDependencyManager::GetInstance()) {
+  DependsOn(ContentSuggestionsServiceFactory::GetInstance());
+  DependsOn(BookmarkModelFactory::GetInstance());
+}
+
+BookmarkSuggestionsProviderFactory::~BookmarkSuggestionsProviderFactory() {}
+
+KeyedService* BookmarkSuggestionsProviderFactory::BuildServiceInstanceFor(
+    content::BrowserContext* context) const {
+  Profile* profile = Profile::FromBrowserContext(context);
+
+  ContentSuggestionsService* content_suggestions_service =
+      ContentSuggestionsServiceFactory::GetForProfile(profile);
+  if (content_suggestions_service->state() ==
+      ContentSuggestionsService::State::DISABLED) {
+    return nullptr;
+  }
+
+  if (!base::FeatureList::IsEnabled(
+          chrome::android::kNTPBookmarkSuggestionsFeature)) {
+    return nullptr;
+  }
+
+  BookmarkModel* bookmark_model = BookmarkModelFactory::GetForProfile(profile);
+
+  BookmarkSuggestionsProvider* bookmark_suggestions_provider =
+      new BookmarkSuggestionsProvider(bookmark_model);
+  content_suggestions_service->RegisterProvider(bookmark_suggestions_provider);
+  return bookmark_suggestions_provider;
+}