Add other variations (with ids) for NTPSnippets to chrome://flags

components/variations/variations_http_header_provider.cc

 // Copyright 2014 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/variations/variations_http_header_provider.h"
 
 #include <stddef.h>
 
 #include <set>
 #include <string>
@@ skipping 38 matching lines @@
   {
     base::AutoLock scoped_lock(lock_);
     for (VariationID id : GetAllVariationIds()) {
       ids_string.append(base::IntToString(id));
       ids_string.push_back(' ');
     }
   }
   return ids_string;
 }
 
+bool VariationsHttpHeaderProvider::ForceVariationIds(
+    const std::string& command_line_variation_ids,
+    std::vector<std::string>* variation_ids) {
+  if (!command_line_variation_ids.empty()) {
+    // Combine |variation_ids| with |command_line_variation_ids|.
+    std::vector<std::string> variation_ids_flags =
+        base::SplitString(command_line_variation_ids, ",",
+                          base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
+    variation_ids->insert(variation_ids->end(), variation_ids_flags.begin(),
+                          variation_ids_flags.end());
+  }
+
+  if (!variation_ids->empty()) {
+    // Create default variation ids which will always be included in the
+    // X-Client-Data request header.
+    return SetDefaultVariationIds(*variation_ids);
+  }
+  return true;
+}
+
+
 bool VariationsHttpHeaderProvider::SetDefaultVariationIds(
-    const std::string& variation_ids) {
+    const std::vector<std::string>& variation_ids) {
   default_variation_ids_set_.clear();
   default_trigger_id_set_.clear();
-  for (const base::StringPiece& entry : base::SplitStringPiece(
-           variation_ids, ",", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL)) {
+  for (const std::string& entry : variation_ids) {
     if (entry.empty()) {
       default_variation_ids_set_.clear();
       default_trigger_id_set_.clear();
       return false;
     }
     bool trigger_id =
         base::StartsWith(entry, "t", base::CompareCase::SENSITIVE);
     // Remove the "t" prefix if it's there.
-    base::StringPiece trimmed_entry = trigger_id ? entry.substr(1) : entry;
+    std::string trimmed_entry = trigger_id ? entry.substr(1) : entry;
 
     int variation_id = 0;
     if (!base::StringToInt(trimmed_entry, &variation_id)) {
       default_variation_ids_set_.clear();
       default_trigger_id_set_.clear();
       return false;
     }
     if (trigger_id)
       default_trigger_id_set_.insert(variation_id);
     else
@@ skipping 140 matching lines @@
 
   std::set<VariationID> all_variation_ids_set = default_variation_ids_set_;
   for (VariationID id : variation_ids_set_)
     all_variation_ids_set.insert(id);
   for (VariationID id : synthetic_variation_ids_set_)
     all_variation_ids_set.insert(id);
   return all_variation_ids_set;
 }
 
 }  // namespace variations