Unifies the additional actions for Categories

chrome/android/java/src/org/chromium/chrome/browser/ntp/snippets/SnippetsBridge.java

Index: chrome/android/java/src/org/chromium/chrome/browser/ntp/snippets/SnippetsBridge.java
diff --git a/chrome/android/java/src/org/chromium/chrome/browser/ntp/snippets/SnippetsBridge.java b/chrome/android/java/src/org/chromium/chrome/browser/ntp/snippets/SnippetsBridge.java
index 15d272e599da19b0e3474cf2ad531d968a0a7003..c4783a7e6564f24b14d134faaf82847eecec577b 100644
--- a/chrome/android/java/src/org/chromium/chrome/browser/ntp/snippets/SnippetsBridge.java
+++ b/chrome/android/java/src/org/chromium/chrome/browser/ntp/snippets/SnippetsBridge.java
@@ -12,7 +12,10 @@ import org.chromium.chrome.browser.ntp.NewTabPageUma;
 import org.chromium.chrome.browser.ntp.cards.ActionItem;
 import org.chromium.chrome.browser.ntp.cards.SuggestionsCategoryInfo;
 import org.chromium.chrome.browser.ntp.snippets.CategoryStatus.CategoryStatusEnum;
+import org.chromium.chrome.browser.ntp.snippets.ContentSuggestionsCardLayout.ContentSuggestionsCardLayoutEnum;
 import org.chromium.chrome.browser.profiles.Profile;
+import org.chromium.chrome.browser.suggestions.ContentSuggestionsAdditionalAction;
+import org.chromium.chrome.browser.suggestions.ContentSuggestionsAdditionalAction.ContentSuggestionsAdditionalActionEnum;
 import org.chromium.chrome.browser.suggestions.DestructionObserver;
 import org.chromium.chrome.browser.suggestions.SuggestionsMetricsReporter;
 import org.chromium.chrome.browser.suggestions.SuggestionsRanker;
@@ -289,8 +292,11 @@ public class SnippetsBridge
 
     @CalledByNative
     private static SuggestionsCategoryInfo createSuggestionsCategoryInfo(int category, String title,
-            int cardLayout, boolean hasFetchAction, boolean hasViewAllAction, boolean showIfEmpty,
+            @ContentSuggestionsCardLayoutEnum int cardLayout,
+            @ContentSuggestionsAdditionalActionEnum int additionalAction, boolean showIfEmpty,
             String noSuggestionsMessage) {
+        boolean hasFetchAction = additionalAction == ContentSuggestionsAdditionalAction.FETCH;
+        boolean hasViewAllAction = additionalAction == ContentSuggestionsAdditionalAction.VIEW_ALL;
         return new SuggestionsCategoryInfo(category, title, cardLayout, hasFetchAction,
                 hasViewAllAction, showIfEmpty, noSuggestionsMessage);
     }