// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/extensions/extension_migrator.h"

#include <utility>

#include "base/values.h"
#include "chrome/browser/defaults.h"
#include "chrome/browser/extensions/external_provider_impl.h"
#include "chrome/browser/profiles/profile.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/common/extension_urls.h"
#include "extensions/common/manifest.h"

namespace extensions {

ExtensionMigrator::ExtensionMigrator(Profile* profile,
                                     const std::string& old_id,
                                     const std::string& new_id)
    : profile_(profile), old_id_(old_id), new_id_(new_id) {}

ExtensionMigrator::~ExtensionMigrator() = default;

void ExtensionMigrator::StartLoading() {
  auto prefs = base::Value::Dict();

  const bool should_have_extension =
      IsAppPresent(old_id_) || IsAppPresent(new_id_);
  if (should_have_extension) {
    base::Value::Dict entry;
    entry.Set(ExternalProviderImpl::kExternalUpdateUrl,
              extension_urls::GetWebstoreUpdateUrl().spec());

    prefs.Set(new_id_, std::move(entry));
  }

  LoadFinished(std::move(prefs));
}

bool ExtensionMigrator::IsAppPresent(const std::string& app_id) {
  const Extension* extension =
      ExtensionRegistry::Get(profile_)->GetInstalledExtension(app_id);

  // If the extension was previously force-installed by policy, don't migrate it
  // or keep it installed. This prevents the extension getting stuck in a
  // "non-uninstallable" state. crbug.com/1416682
  return extension && !Manifest::IsPolicyLocation(extension->location());
}

}  // namespace extensions