Make ContentSuggestionsService recognize new/removed categories

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 <map>
 #include <string>
 #include <vector>
@@ skipping 108 matching lines @@
  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 Category& left, const Category& right) const;
   };
 
   // Implementation of ContentSuggestionsProvider::Observer.
-  void OnNewSuggestions(Category changed_category,
+  void OnNewSuggestions(ContentSuggestionsProvider* provider,
+                        Category category,
                         std::vector<ContentSuggestion> suggestions) override;
-  void OnCategoryStatusChanged(Category changed_category,
+  void OnCategoryStatusChanged(ContentSuggestionsProvider* provider,
+                               Category category,
                                CategoryStatus new_status) override;
   void OnProviderShutdown(ContentSuggestionsProvider* provider) override;
 
-  // Checks whether a provider for the given |category| is registered.
-  bool IsCategoryRegistered(Category category) const;
+  // Ensures the given |provider| is is registered for the given |category|.
+  // Returns true if the category was newly registered, false if it was present
+  // before.
+  bool EnsureCategoryRegistered(ContentSuggestionsProvider* provider,

[Marc Treib, 2016/08/01 14:23:38]

EnsureCategoryIsRegistered?

[Philipp Keck, 2016/08/01 14:59:04]

Acknowledged.

+                                Category category);
 
   // Fires the OnCategoryStatusChanged event for the given |category|.
   void NotifyCategoryStatusChanged(Category category);
 
   // Whether the content suggestions feature is enabled.
   State state_;
 
   // Provides new and existing categories and an order for them.
   CategoryFactory category_factory_;
 
@@ skipping 23 matching lines @@
   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_