// 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 "component_cld_data_harness.h"

#include "base/base_paths.h"
#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/path_service.h"
#include "chrome/browser/component_updater/cld_component_installer.h"
#include "chrome/common/chrome_paths.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

// This has to match what's in cld_component_installer.cc.
const base::FilePath::CharType kComponentDataFileName[] =
    FILE_PATH_LITERAL("cld2_data.bin");

}  // namespace

namespace test {

ComponentCldDataHarness::ComponentCldDataHarness() {
  // Constructor does nothing in all cases. See Init() for initialization.
}

ComponentCldDataHarness::~ComponentCldDataHarness() {
  DVLOG(1) << "Tearing down CLD data harness";
  // Dynamic data mode is enabled and we are using the component updater.
  component_updater::CldComponentInstallerTraits::SetLatestCldDataFile(
      base::FilePath());
  ClearComponentDataFileState();
  DeleteComponentTree();
}

void ComponentCldDataHarness::Init() {
  DVLOG(1) << "Initializing CLD data harness";
  // Dynamic data mode is enabled and we are using the component updater.
  ASSERT_NO_FATAL_FAILURE(CopyComponentTree());
  base::FilePath data_file;
  GetComponentDataFileDestination(&data_file);
  component_updater::CldComponentInstallerTraits::SetLatestCldDataFile(
      data_file);
}

void ComponentCldDataHarness::ClearComponentDataFileState() {
  DVLOG(1) << "Clearing component CLD data file state";
  base::FilePath nothing;
  component_updater::CldComponentInstallerTraits::SetLatestCldDataFile(nothing);
}

// DIR_COMPONENT_CLD2 is also defined as being relative to USER_DATA_DIR, so
// like GetStandaloneDataFileDestination, this is safe to run in multiple
// parallel test processes.
void ComponentCldDataHarness::GetExtractedComponentDestination(
    base::FilePath* out_path) {
  ASSERT_TRUE(PathService::Get(chrome::DIR_COMPONENT_CLD2, out_path));
}

void ComponentCldDataHarness::GetComponentDataFileDestination(
    base::FilePath* out_path) {
  GetExtractedComponentDestination(out_path);
  *out_path = out_path->Append(CldDataHarness::GetTestDataSourceCrxVersion())
                  .Append(FILE_PATH_LITERAL("_platform_specific"))
                  .Append(FILE_PATH_LITERAL("all"))
                  .Append(kComponentDataFileName);
}

void ComponentCldDataHarness::DeleteComponentTree() {
  base::FilePath tree_path;
  ASSERT_NO_FATAL_FAILURE(GetExtractedComponentDestination(&tree_path));
  DVLOG(1) << "Deleting CLD component test files from " << tree_path.value();
  base::DeleteFile(tree_path, true);
}

void ComponentCldDataHarness::CopyComponentTree() {
  DeleteComponentTree();  // sanity: blow away any old copies.
  base::FilePath target_dir;
  GetExtractedComponentDestination(&target_dir);
  base::FilePath source_dir;
  CldDataHarness::GetTestDataSourceDirectory(&source_dir);
  DVLOG(1) << "Copying CLD component test files from " << source_dir.value()
           << " to " << target_dir.value();
  ASSERT_TRUE(base::CreateDirectoryAndGetError(target_dir, NULL));
  ASSERT_TRUE(base::CopyDirectory(source_dir, target_dir, true));
  ASSERT_TRUE(base::PathExists(target_dir));
  base::FilePath check_path;
  GetComponentDataFileDestination(&check_path);
  ASSERT_TRUE(base::PathExists(check_path));
}

scoped_ptr<CldDataHarness> CreateCldDataHarness() {
  scoped_ptr<CldDataHarness> result(new ComponentCldDataHarness());
  return result.Pass();
}

}  // namespace test