Add ContentSuggestionsCategoryFactory; Store categories as ints

components/ntp_snippets/offline_pages/offline_page_suggestions_provider.h

 // 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.
 
 #ifndef COMPONENTS_NTP_SNIPPETS_OFFLINE_PAGES_OFFLINE_PAGE_SUGGESTIONS_PROVIDER_H_
 #define COMPONENTS_NTP_SNIPPETS_OFFLINE_PAGES_OFFLINE_PAGE_SUGGESTIONS_PROVIDER_H_
 
 #include <string>
 #include <vector>
 
 #include "base/callback_forward.h"
 #include "components/keyed_service/core/keyed_service.h"
 #include "components/ntp_snippets/content_suggestion.h"
 #include "components/ntp_snippets/content_suggestions_category.h"
+#include "components/ntp_snippets/content_suggestions_category_factory.h"
 #include "components/ntp_snippets/content_suggestions_category_status.h"
 #include "components/ntp_snippets/content_suggestions_provider.h"
 #include "components/offline_pages/offline_page_model.h"
 #include "components/offline_pages/offline_page_types.h"
 
 namespace gfx {
 class Image;
 }
 
 namespace ntp_snippets {
 
 // Provides content suggestions from the offline pages model.
 // Currently, those are only the pages that the user last navigated to in an
 // open tab and offlined bookmarks.
 class OfflinePageSuggestionsProvider
     : public KeyedService,
       public ContentSuggestionsProvider,
       public offline_pages::OfflinePageModel::Observer {
  public:
   OfflinePageSuggestionsProvider(
+      ContentSuggestionsCategoryFactory* category_factory,
       offline_pages::OfflinePageModel* offline_page_model);
   ~OfflinePageSuggestionsProvider() override;
 
   // Inherited from KeyedService.
   void Shutdown() override;
 
  private:
   // ContentSuggestionsProvider implementation.
+  std::vector<ContentSuggestionsCategory> GetProvidedCategories() override;
   void SetObserver(ContentSuggestionsProvider::Observer* observer) override;
   ContentSuggestionsCategoryStatus GetCategoryStatus(
       ContentSuggestionsCategory category) override;
   void DismissSuggestion(const std::string& suggestion_id) override;
   void FetchSuggestionImage(const std::string& suggestion_id,
                             const ImageFetchedCallback& callback) override;
   void ClearCachedSuggestionsForDebugging() override;
   void ClearDismissedSuggestionsForDebugging() override;
 
   // OfflinePageModel::Observer implementation.
@@ skipping 11 matching lines @@
 
   // Updates the |category_status_| and notifies the |observer_|, if necessary.
   void NotifyStatusChanged(ContentSuggestionsCategoryStatus new_status);
 
   ContentSuggestionsCategoryStatus category_status_;
 
   ContentSuggestionsProvider::Observer* observer_;
 
   offline_pages::OfflinePageModel* offline_page_model_;
 
+  const ContentSuggestionsCategory provided_category_;
+
   DISALLOW_COPY_AND_ASSIGN(OfflinePageSuggestionsProvider);
 };
 
 }  // namespace ntp_snippets
 
 #endif  // COMPONENTS_NTP_SNIPPETS_OFFLINE_PAGES_OFFLINE_PAGE_SUGGESTIONS_PROVIDER_H_