Add ContentSuggestionsCategoryFactory; Store categories as ints

components/ntp_snippets/content_suggestions_service.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_CONTENT_SUGGESTIONS_SERVICE_H_
 #define COMPONENTS_NTP_SNIPPETS_CONTENT_SUGGESTIONS_SERVICE_H_
 
-#include <stddef.h>
-
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/callback_forward.h"
 #include "base/observer_list.h"
 #include "components/keyed_service/core/keyed_service.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"
 
 namespace gfx {
 class Image;
 }
 
 namespace ntp_snippets {
 
 // Retrieves suggestions from a number of ContentSuggestionsProviders and serves
@@ skipping 85 matching lines @@
   // same results when they fetch the next time. In particular, calling this
   // method will not mark any suggestions as dismissed.
   void ClearCachedSuggestionsForDebugging();
 
   // Only for debugging use through the internals page. Some providers
   // internally store a list of dismissed suggestions to prevent them from
   // reappearing. This function clears all such lists in all providers, making
   // dismissed suggestions reappear (only for certain providers).
   void ClearDismissedSuggestionsForDebugging();
 
+  ContentSuggestionsCategoryFactory* category_factory() {
+    return &category_factory_;
+  }
+
  private:
   friend class ContentSuggestionsServiceTest;
 
+  // This is just an arbitrary ordering by ID, used by the maps in this class,
+  // because the ordering needs to be constant for maps.
+  struct CompareCategoriesByID {
+    bool operator()(const ContentSuggestionsCategory& left,
+                    const ContentSuggestionsCategory& right) const;
+  };
+
   // Implementation of ContentSuggestionsProvider::Observer.
   void OnNewSuggestions(ContentSuggestionsCategory changed_category,
                         std::vector<ContentSuggestion> suggestions) override;
   void OnCategoryStatusChanged(
       ContentSuggestionsCategory changed_category,
       ContentSuggestionsCategoryStatus new_status) override;
   void OnProviderShutdown(ContentSuggestionsProvider* provider) override;
 
   // Checks whether a provider for the given |category| is registered.
   bool IsCategoryRegistered(ContentSuggestionsCategory category) const;
 
   // Fires the OnCategoryStatusChanged event for the given |category|.
   void NotifyCategoryStatusChanged(ContentSuggestionsCategory category);
 
   // Whether the content suggestions feature is enabled.
   State state_;
 
+  // Provides new and existing categories and an order for them.
+  ContentSuggestionsCategoryFactory category_factory_;
+
   // All registered providers. A provider may be contained multiple times, if it
   // provides multiple categories. The keys of this map are exactly the entries
   // of |categories_|.
-  std::map<ContentSuggestionsCategory, ContentSuggestionsProvider*> providers_;
+  std::map<ContentSuggestionsCategory,
+           ContentSuggestionsProvider*,
+           CompareCategoriesByID>
+      providers_;
 
-  // All current suggestion categories, in an order determined by the service.
-  // Currently, this is simply the order in which the providers were registered.
-  // This vector contains exactly the same categories as |providers_|.
-  // TODO(pke): Implement a useful and consistent ordering for categories.
+  // All current suggestion categories, in an order determined by the
+  // |category_factory_|. This vector contains exactly the same categories as
+  // |providers_|.
   std::vector<ContentSuggestionsCategory> categories_;
 
   // All current suggestions grouped by category. This contains an entry for
   // every category in |categories_| whose status is an available status. It may
   // contain an empty vector if the category is available but empty (or still
   // loading).
-  std::map<ContentSuggestionsCategory, std::vector<ContentSuggestion>>
+  std::map<ContentSuggestionsCategory,
+           std::vector<ContentSuggestion>,
+           CompareCategoriesByID>
       suggestions_by_category_;
 
   // Map used to determine the category of a suggestion (of which only the ID
   // is available). This also determines the provider that delivered the
   // suggestion.
   std::map<std::string, ContentSuggestionsCategory> id_category_map_;
 
   base::ObserverList<Observer> observers_;
 
   const std::vector<ContentSuggestion> no_suggestions_;
 
   DISALLOW_COPY_AND_ASSIGN(ContentSuggestionsService);
 };
 
 }  // namespace ntp_snippets
 
 #endif  // COMPONENTS_NTP_SNIPPETS_CONTENT_SUGGESTIONS_SERVICE_H_