root/chrome/browser/bookmarks/enhanced_bookmarks_features.cc

DEFINITIONS

1. IsBookmarksExtensionHash
2. IsBookmarksExtensionInstalled
3. UpdateBookmarksExperiment
4. IsEnhancedBookmarksExperimentEnabled
5. IsEnableDomDistillerSet
6. IsEnableSyncArticlesSet
7. GetEnhancedBookmarksExtensionIdFromFinch

// Copyright 2013 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 "chrome/browser/bookmarks/enhanced_bookmarks_features.h"

#include "base/command_line.h"
#include "base/prefs/pref_service.h"
#include "base/prefs/scoped_user_pref_update.h"
#include "base/sha1.h"
#include "base/strings/string_number_conversions.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/pref_names.h"
#include "components/variations/variations_associated_data.h"
#include "extensions/common/features/feature.h"
#include "extensions/common/features/feature_provider.h"

namespace {

const char kFieldTrialName[] = "EnhancedBookmarks";

bool IsBookmarksExtensionHash(const std::string& sha1_hex) {
    return sha1_hex == "D5736E4B5CF695CB93A2FB57E4FDC6E5AFAB6FE2" ||
           sha1_hex == "D57DE394F36DC1C3220E7604C575D29C51A6C495";
}

};  // namespace

bool IsBookmarksExtensionInstalled(
    const extensions::ExtensionIdSet& extension_ids) {
  // Compare installed extension ids with ones we expect.
  for (extensions::ExtensionIdSet::const_iterator iter = extension_ids.begin();
       iter != extension_ids.end(); ++iter) {
    const std::string id_hash = base::SHA1HashString(*iter);
    DCHECK_EQ(id_hash.length(), base::kSHA1Length);
    std::string hash = base::HexEncode(id_hash.c_str(), id_hash.length());

    if (IsBookmarksExtensionHash(hash))
      return true;
  }
  return false;
}

void UpdateBookmarksExperiment(
    PrefService* local_state,
    BookmarksExperimentState bookmarks_experiment_state) {
  ListPrefUpdate update(local_state, prefs::kEnabledLabsExperiments);
  base::ListValue* experiments_list = update.Get();
  size_t index;
  if (bookmarks_experiment_state == kNoBookmarksExperiment) {
    experiments_list->Remove(
        base::StringValue(switches::kManualEnhancedBookmarks), &index);
    experiments_list->Remove(
        base::StringValue(switches::kManualEnhancedBookmarksOptout), &index);
  } else if (bookmarks_experiment_state == kBookmarksExperimentEnabled) {
    experiments_list->Remove(
        base::StringValue(switches::kManualEnhancedBookmarksOptout), &index);
    experiments_list->AppendIfNotPresent(
        new base::StringValue(switches::kManualEnhancedBookmarks));
  } else if (bookmarks_experiment_state ==
                 kBookmarksExperimentEnabledUserOptOut) {
    experiments_list->Remove(
        base::StringValue(switches::kManualEnhancedBookmarks), &index);
    experiments_list->AppendIfNotPresent(
        new base::StringValue(switches::kManualEnhancedBookmarksOptout));
  }
}

bool IsEnhancedBookmarksExperimentEnabled() {
  CommandLine* command_line = CommandLine::ForCurrentProcess();
  if (command_line->HasSwitch(switches::kManualEnhancedBookmarks) ||
      command_line->HasSwitch(switches::kManualEnhancedBookmarksOptout)) {
    return true;
  }

  std::string ext_id = GetEnhancedBookmarksExtensionIdFromFinch();
  extensions::FeatureProvider* feature_provider =
      extensions::FeatureProvider::GetPermissionFeatures();
  extensions::Feature* feature = feature_provider->GetFeature("metricsPrivate");
  return feature && feature->IsIdInWhitelist(ext_id);
}

bool IsEnableDomDistillerSet() {
  if (CommandLine::ForCurrentProcess()->
      HasSwitch(switches::kEnableDomDistiller)) {
    return true;
  }
  if (chrome_variations::GetVariationParamValue(
          kFieldTrialName, "enable-dom-distiller") == "1")
    return true;

  return false;
}

bool IsEnableSyncArticlesSet() {
  if (CommandLine::ForCurrentProcess()->
      HasSwitch(switches::kEnableSyncArticles)) {
    return true;
  }
  if (chrome_variations::GetVariationParamValue(
          kFieldTrialName, "enable-sync-articles") == "1")
    return true;

  return false;
}

std::string GetEnhancedBookmarksExtensionIdFromFinch() {
  return chrome_variations::GetVariationParamValue(kFieldTrialName, "id");
}