ntp_snippets: default to staging API on dev/canary

components/ntp_snippets/remote/remote_suggestions_fetcher.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 "components/ntp_snippets/remote/remote_suggestions_fetcher.h"
 
 #include <cstdlib>
 #include <utility>
 
 #include "base/files/file_path.h"
@@ skipping 92 matching lines @@
     case FetchResult::INVALID_SNIPPET_CONTENT_ERROR:
     case FetchResult::JSON_PARSE_ERROR:
       return Status(StatusCode::TEMPORARY_ERROR, FetchResultToString(result));
     case FetchResult::RESULT_MAX:
       break;
   }
   NOTREACHED();
   return Status(StatusCode::PERMANENT_ERROR, std::string());
 }
 
-std::string GetFetchEndpoint() {
-  std::string endpoint = variations::GetVariationParamValueByFeature(
-      ntp_snippets::kArticleSuggestionsFeature, kContentSuggestionsBackend);
-  return endpoint.empty() ? kContentSuggestionsServer : endpoint;
-}
-
 bool UsesChromeContentSuggestionsAPI(const GURL& endpoint) {
   if (endpoint == kChromeReaderServer) {
     return false;
   }
 
   if (endpoint != kContentSuggestionsServer &&
       endpoint != kContentSuggestionsStagingServer &&
       endpoint != kContentSuggestionsAlphaServer) {
     LOG(WARNING) << "Unknown value for " << kContentSuggestionsBackend << ": "
                  << endpoint << "; assuming chromecontentsuggestions-style API";
@@ skipping 64 matching lines @@
     categories->clear();
     return;
   }
   RemoteSuggestionsFetcher::FetchedCategory category = std::move(*category_it);
   categories->clear();
   categories->push_back(std::move(category));
 }
 
 }  // namespace
 
+std::string GetFetchEndpoint(bool is_stable_or_beta_channel) {
+  std::string endpoint = variations::GetVariationParamValueByFeature(
+      ntp_snippets::kArticleSuggestionsFeature, kContentSuggestionsBackend);
+  if (endpoint.empty()) {
+    endpoint = is_stable_or_beta_channel ? kContentSuggestionsServer
+                                         : kContentSuggestionsStagingServer;
+  }
+  return endpoint;
+}
+
 CategoryInfo BuildArticleCategoryInfo(
     const base::Optional<base::string16>& title) {
   return CategoryInfo(
       title.has_value() ? title.value()
                         : l10n_util::GetStringUTF16(
                               IDS_NTP_ARTICLE_SUGGESTIONS_SECTION_HEADER),
       ContentSuggestionsCardLayout::FULL_CARD,
       /*has_fetch_action=*/true,
       /*has_view_all_action=*/false,
       /*show_if_empty=*/true,
@@ skipping 26 matching lines @@
 RemoteSuggestionsFetcher::FetchedCategory::operator=(FetchedCategory&&) =
     default;
 
 RemoteSuggestionsFetcher::RemoteSuggestionsFetcher(
     SigninManagerBase* signin_manager,
     OAuth2TokenService* token_service,
     scoped_refptr<URLRequestContextGetter> url_request_context_getter,
     PrefService* pref_service,
     LanguageModel* language_model,
     const ParseJSONCallback& parse_json_callback,
+    const std::string& api_endpoint,
     const std::string& api_key,
     const UserClassifier* user_classifier)
     : OAuth2TokenService::Consumer("ntp_snippets"),
       signin_manager_(signin_manager),
       token_service_(token_service),
       url_request_context_getter_(std::move(url_request_context_getter)),
       language_model_(language_model),
       parse_json_callback_(parse_json_callback),
-      fetch_url_(GetFetchEndpoint()),
+      fetch_url_(api_endpoint),
       fetch_api_(UsesChromeContentSuggestionsAPI(fetch_url_)
                      ? FetchAPI::CHROME_CONTENT_SUGGESTIONS_API
                      : FetchAPI::CHROME_READER_API),
       api_key_(api_key),
       clock_(new base::DefaultClock()),
       user_classifier_(user_classifier),
       request_throttler_rare_ntp_user_(
           pref_service,
           RequestThrottler::RequestType::
               CONTENT_SUGGESTION_FETCHER_RARE_NTP_USER),
@@ skipping 313 matching lines @@
           interactive_request);
     case UserClassifier::UserClass::ACTIVE_SUGGESTIONS_CONSUMER:
       return request_throttler_active_suggestions_consumer_
           .DemandQuotaForRequest(interactive_request);
   }
   NOTREACHED();
   return false;
 }
 
 }  // namespace ntp_snippets