// Copyright (c) 2012 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/autocomplete/autocomplete_field_trial.h"

#include <string>

#include "base/metrics/field_trial.h"

namespace {

// Field trial names.
static const char kAggressiveHUPFieldTrialName[] =
    "OmniboxAggressiveHistoryURLProvider";
static const char kDisallowInlineHQPFieldTrialName[] =
    "OmniboxDisallowInlineHQP";

// Field trial experiment probabilities.

// For aggressive History URL Provider field trial, put 50% ( = 50/100 )
// of the users in the aggressive experiment group.
const base::FieldTrial::Probability kAggressiveHUPFieldTrialDivisor = 100;
const base::FieldTrial::Probability
    kAggressiveHUPFieldTrialExperimentFraction = 50;

// For inline History Quick Provider field trial, put 10% ( = 10/100 )
// of the users in the disallow-inline experiment group.
const base::FieldTrial::Probability kDisallowInlineHQPFieldTrialDivisor = 100;
const base::FieldTrial::Probability
    kDisallowInlineHQPFieldTrialExperimentFraction = 10;

// Field trial IDs.
// Though they are not literally "const", they are set only once, in
// Activate() below.

// Field trial ID for the aggressive History URL Provider experiment group.
int aggressive_hup_experiment_group = 0;

// Field trial ID for the disallow-inline History Quick Provider
// experiment group.
int disallow_inline_hqp_experiment_group = 0;

}


void AutocompleteFieldTrial::Activate() {
  // Because users tend to use omnibox without attention to it--habits
  // get ingrained, users tend to learn that a particular suggestion is
  // at a particular spot in the drop-down--we're going to make these
  // field trials sticky.  We want users to stay in them once assigned
  // so they have a better experience and also so we don't get weird
  // effects as omnibox ranking keeps changing and users learn they can't
  // trust the omnibox.  Hence, to create the field trials we require
  // that field trials can be made sticky.
  if (base::FieldTrialList::IsOneTimeRandomizationEnabled()) {  // sticky trials
    // Create aggressive History URL Provider field trial.
    // Make it expire on August 1, 2012.
    scoped_refptr<base::FieldTrial> trial(new base::FieldTrial(
        kAggressiveHUPFieldTrialName, kAggressiveHUPFieldTrialDivisor,
        "Standard", 2012, 8, 1));
    trial->UseOneTimeRandomization();
    aggressive_hup_experiment_group = trial->AppendGroup("Aggressive",
        kAggressiveHUPFieldTrialExperimentFraction);

    // Create inline History Quick Provider field trial.
    // Make it expire on November 8, 2012.
    trial = new base::FieldTrial(
        kDisallowInlineHQPFieldTrialName, kDisallowInlineHQPFieldTrialDivisor,
        "Standard", 2012, 11, 8);
    trial->UseOneTimeRandomization();
    disallow_inline_hqp_experiment_group = trial->AppendGroup("DisallowInline",
        kDisallowInlineHQPFieldTrialExperimentFraction);
  }
}

bool AutocompleteFieldTrial::InAggressiveHUPFieldTrial() {
  return base::FieldTrialList::TrialExists(kAggressiveHUPFieldTrialName);
}

bool AutocompleteFieldTrial::InAggressiveHUPFieldTrialExperimentGroup() {
  if (!base::FieldTrialList::TrialExists(kAggressiveHUPFieldTrialName))
    return false;

  // Return true if we're in the aggressive experiment group.
  const int group = base::FieldTrialList::FindValue(
      kAggressiveHUPFieldTrialName);
  return group == aggressive_hup_experiment_group;
}

bool AutocompleteFieldTrial::InDisallowInlineHQPFieldTrial() {
  return base::FieldTrialList::TrialExists(kDisallowInlineHQPFieldTrialName);
}

bool AutocompleteFieldTrial::InDisallowInlineHQPFieldTrialExperimentGroup() {
  if (!base::FieldTrialList::TrialExists(kDisallowInlineHQPFieldTrialName))
    return false;

  // Return true if we're in the experiment group.
  const int group = base::FieldTrialList::FindValue(
      kDisallowInlineHQPFieldTrialName);
  return group == disallow_inline_hqp_experiment_group;
}