// Copyright 2017 The Chromium Embedded Framework Authors.
// Portions 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 "libcef/browser/extensions/api/storage/sync_value_store_cache.h"

#include <stddef.h>

#include <limits>
#include <utility>

#include "components/value_store/value_store_factory.h"
#include "content/public/browser/browser_thread.h"
#include "extensions/browser/api/storage/backend_task_runner.h"
#include "extensions/browser/api/storage/value_store_util.h"
#include "extensions/browser/api/storage/weak_unlimited_settings_storage.h"
#include "extensions/common/api/storage.h"
#include "extensions/common/extension.h"
#include "extensions/common/permissions/permissions_data.h"

using content::BrowserThread;

namespace extensions::cef {

namespace {

// Returns the quota limit for local storage, taken from the schema in
// extensions/common/api/storage.json.
SettingsStorageQuotaEnforcer::Limits GetLocalQuotaLimits() {
  SettingsStorageQuotaEnforcer::Limits limits = {
      static_cast<size_t>(api::storage::local::QUOTA_BYTES),
      std::numeric_limits<size_t>::max(), std::numeric_limits<size_t>::max()};
  return limits;
}

}  // namespace

SyncValueStoreCache::SyncValueStoreCache(
    scoped_refptr<value_store::ValueStoreFactory> factory)
    : storage_factory_(std::move(factory)), quota_(GetLocalQuotaLimits()) {
  DCHECK_CURRENTLY_ON(BrowserThread::UI);
}

SyncValueStoreCache::~SyncValueStoreCache() {
  DCHECK(IsOnBackendSequence());
}

void SyncValueStoreCache::RunWithValueStoreForExtension(
    StorageCallback callback,
    scoped_refptr<const Extension> extension) {
  DCHECK(IsOnBackendSequence());

  value_store::ValueStore* storage = GetStorage(extension.get());

  // A neat way to implement unlimited storage; if the extension has the
  // unlimited storage permission, force through all calls to Set().
  if (extension->permissions_data()->HasAPIPermission(
          mojom::APIPermissionID::kUnlimitedStorage)) {
    WeakUnlimitedSettingsStorage unlimited_storage(storage);
    std::move(callback).Run(&unlimited_storage);
  } else {
    std::move(callback).Run(storage);
  }
}

void SyncValueStoreCache::DeleteStorageSoon(const std::string& extension_id) {
  DCHECK(IsOnBackendSequence());
  storage_map_.erase(extension_id);

  value_store_util::DeleteValueStore(settings_namespace::SYNC,
                                     value_store_util::ModelType::APP,
                                     extension_id, storage_factory_);

  value_store_util::DeleteValueStore(settings_namespace::SYNC,
                                     value_store_util::ModelType::EXTENSION,
                                     extension_id, storage_factory_);
}

value_store::ValueStore* SyncValueStoreCache::GetStorage(
    const Extension* extension) {
  auto iter = storage_map_.find(extension->id());
  if (iter != storage_map_.end()) {
    return iter->second.get();
  }

  value_store_util::ModelType model_type =
      extension->is_app() ? value_store_util::ModelType::APP
                          : value_store_util::ModelType::EXTENSION;
  std::unique_ptr<value_store::ValueStore> store =
      value_store_util::CreateSettingsStore(settings_namespace::LOCAL,
                                            model_type, extension->id(),
                                            storage_factory_);
  std::unique_ptr<SettingsStorageQuotaEnforcer> storage(
      new SettingsStorageQuotaEnforcer(quota_, std::move(store)));
  DCHECK(storage.get());

  value_store::ValueStore* storage_ptr = storage.get();
  storage_map_[extension->id()] = std::move(storage);
  return storage_ptr;
}
}  // namespace extensions::cef