// Copyright 2015 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/offline_pages/offline_page_feature.h"

#include <string>

#include "base/command_line.h"
#include "base/metrics/field_trial.h"
#include "build/build_config.h"
#include "components/offline_pages/offline_page_switches.h"

#if defined(OS_ANDROID)

namespace offline_pages {

namespace {
const char kOfflinePagesFieldTrialName[] = "OfflinePages";
// The old experiment has only one mode to enable offline pages.
const char kEnabledGroupName[] = "Enabled";
// The new experiment supports two modes for offline pages.
const char kEnabledAsBookmarksGroupName[] = "EnabledAsBookmarks";
const char kEnabledAsSavedPagesGroupName[] = "EnabledAsSavedPages";
}  // namespace

FeatureMode GetOfflinePageFeatureMode() {
  // The old experiment 'Enabled' defaults to showing saved page.
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kEnableOfflinePages)) {
    return FeatureMode::ENABLED_AS_SAVED_PAGES;
  }
  // The new experiment can control showing either bookmark or saved page.
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kEnableOfflinePagesAsBookmarks)) {
    return FeatureMode::ENABLED_AS_BOOKMARKS;
  }
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kEnableOfflinePagesAsSavedPages)) {
    return FeatureMode::ENABLED_AS_SAVED_PAGES;
  }
  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
          switches::kDisableOfflinePages)) {
    return FeatureMode::DISABLED;
  }

  std::string group_name =
      base::FieldTrialList::FindFullName(kOfflinePagesFieldTrialName);
  // The old experiment 'Enabled' defaults to showing saved page.
  if (group_name == kEnabledGroupName)
    return FeatureMode::ENABLED_AS_SAVED_PAGES;
  // The new experiment can control showing either bookmark or saved page.
  if (group_name == kEnabledAsBookmarksGroupName)
    return FeatureMode::ENABLED_AS_BOOKMARKS;
  if (group_name == kEnabledAsSavedPagesGroupName)
    return FeatureMode::ENABLED_AS_SAVED_PAGES;
  return FeatureMode::DISABLED;
}

bool IsOfflinePagesEnabled() {
  FeatureMode mode = GetOfflinePageFeatureMode();
  return mode == FeatureMode::ENABLED_AS_BOOKMARKS ||
         mode == FeatureMode::ENABLED_AS_SAVED_PAGES;
}

}  // namespace offline_pages

#endif