The chrome.lockScreen.data API implementation

extensions/browser/api/lock_screen_data/item_storage.cc

Index: extensions/browser/api/lock_screen_data/item_storage.cc
diff --git a/extensions/browser/api/lock_screen_data/item_storage.cc b/extensions/browser/api/lock_screen_data/item_storage.cc
new file mode 100644
index 0000000000000000000000000000000000000000..cdfe3028aff537056e6b811cf8c47ce8212073f2
--- /dev/null
+++ b/extensions/browser/api/lock_screen_data/item_storage.cc
@@ -0,0 +1,474 @@
+// Copyright 2017 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 "extensions/browser/api/lock_screen_data/item_storage.h"
+
+#include <set>
+#include <utility>
+
+#include "base/guid.h"
+#include "base/memory/ptr_util.h"
+#include "base/values.h"
+#include "components/prefs/pref_registry_simple.h"
+#include "components/prefs/pref_service.h"
+#include "components/prefs/scoped_user_pref_update.h"
+#include "content/public/browser/browser_thread.h"
+#include "extensions/browser/api/lock_screen_data/data_item.h"
+#include "extensions/browser/api/lock_screen_data/operation_result.h"
+#include "extensions/browser/api/storage/local_value_store_cache.h"
+#include "extensions/browser/event_router.h"
+#include "extensions/browser/extension_registry.h"
+#include "extensions/browser/extensions_browser_client.h"
+#include "extensions/browser/value_store/value_store.h"
+#include "extensions/browser/value_store/value_store_factory_impl.h"
+#include "extensions/common/api/lock_screen_data.h"
+
+namespace extensions {
+
+namespace lock_screen_data {
+
+namespace {
+
+const char kLockScreenDataPrefKey[] = "lockScreenDataItems";
+
+ItemStorage* g_data_item_storage = nullptr;
+
+ItemStorage::ItemFactoryCallback* g_test_item_factory_callback = nullptr;
+ItemStorage::RegisteredItemsGetter* g_test_registered_items_getter_callback =
+    nullptr;
+ItemStorage::ItemStoreDeleter* g_test_delete_all_items_callback = nullptr;
+
+std::unique_ptr<DataItem> CreateDataItem(const std::string& item_id,
+                                         const std::string& extension_id,
+                                         content::BrowserContext* context,
+                                         ValueStoreCache* value_store_cache,
+                                         base::SequencedTaskRunner* task_runner,
+                                         const std::string& crypto_key) {
+  return g_test_item_factory_callback
+             ? g_test_item_factory_callback->Run(item_id, extension_id,
+                                                 crypto_key)
+             : base::MakeUnique<DataItem>(item_id, extension_id, context,
+                                          value_store_cache, task_runner,
+                                          crypto_key);
+}
+
+void GetRegisteredItems(const std::string& extension_id,
+                        content::BrowserContext* context,
+                        ValueStoreCache* value_store_cache,
+                        base::SequencedTaskRunner* task_runner,
+                        const DataItem::RegisteredValuesCallback& callback) {
+  if (g_test_registered_items_getter_callback) {
+    g_test_registered_items_getter_callback->Run(extension_id, callback);
+    return;
+  }
+  DataItem::GetRegisteredValuesForExtension(
+      context, value_store_cache, task_runner, extension_id, callback);
+}
+
+void DeleteAllItems(const std::string& extension_id,
+                    content::BrowserContext* context,
+                    ValueStoreCache* value_store_cache,
+                    base::SequencedTaskRunner* task_runner,
+                    const base::Closure& callback) {
+  if (g_test_delete_all_items_callback) {
+    g_test_delete_all_items_callback->Run(extension_id, callback);
+    return;
+  }
+  DataItem::DeleteAllItemsForExtension(context, value_store_cache, task_runner,
+                                       extension_id, callback);
+}
+
+void ReleaseValueStoreCache(std::unique_ptr<LocalValueStoreCache> cache) {
+  // Nothing to do. Used only to defer |cache| destruction to the FILE thread.
+}
+
+}  // namespace
+
+// static
+ItemStorage* ItemStorage::Get(content::BrowserContext* context) {
+  if (g_data_item_storage && !g_data_item_storage->IsContextAllowed(context))
+    return nullptr;
+  return g_data_item_storage;
+}
+
+// static
+void ItemStorage::RegisterLocalState(PrefRegistrySimple* registry) {
+  registry->RegisterDictionaryPref(kLockScreenDataPrefKey);
+}
+
+ItemStorage::ItemStorage(content::BrowserContext* context,
+                         PrefService* local_state,
+                         const std::string& crypto_key,
+                         const base::FilePath& storage_root)
+    : context_(context),
+      user_id_(
+          ExtensionsBrowserClient::Get()->GetUserIdHashFromContext(context)),
+      crypto_key_(crypto_key),
+      local_state_(local_state),
+      storage_root_(storage_root.Append(user_id_)),
+      extension_registry_observer_(this),
+      value_store_cache_(base::MakeUnique<LocalValueStoreCache>(
+          new ValueStoreFactoryImpl(storage_root))),
+      weak_ptr_factory_(this) {
+  CHECK(!user_id_.empty());
+  extension_registry_observer_.Add(ExtensionRegistry::Get(context));
+  task_runner_ = content::BrowserThread::GetTaskRunnerForThread(
+      content::BrowserThread::FILE);
+
+  DCHECK(!g_data_item_storage);
+  g_data_item_storage = this;
+
+  ClearUninstalledAppData();
+}
+
+ItemStorage::~ItemStorage() {
+  data_item_cache_.clear();
+
+  task_runner_->PostTask(
+      FROM_HERE, base::Bind(&ReleaseValueStoreCache,
+                            base::Passed(std::move(value_store_cache_))));
+
+  DCHECK_EQ(g_data_item_storage, this);
+  g_data_item_storage = nullptr;
+}
+
+// static
+void ItemStorage::SetItemProvidersForTesting(
+    RegisteredItemsGetter* items_getter_callback,
+    ItemFactoryCallback* factory_callback,
+    ItemStoreDeleter* deleter_callback) {
+  g_test_registered_items_getter_callback = items_getter_callback;
+  g_test_item_factory_callback = factory_callback;
+  g_test_delete_all_items_callback = deleter_callback;
+}
+
+void ItemStorage::SetSessionLocked(bool session_locked) {
+  SessionLockedState new_state = session_locked
+                                     ? SessionLockedState::kLocked
+                                     : SessionLockedState::kNotLocked;
+  if (new_state == session_locked_state_)
+    return;
+
+  bool was_locked = session_locked_state_ == SessionLockedState::kLocked;
+  session_locked_state_ = new_state;
+
+  if (session_locked_state_ != SessionLockedState::kNotLocked)
+    return;
+
+  std::set<std::string> extensions = GetExtensionsWithDataItems(false);
+  for (const auto& id : extensions) {
+    // If the session state is unlocked, dispatch Item availability events to
+    // apps with available data items.
+    api::lock_screen_data::DataItemsAvailableEvent event_args;
+    event_args.was_locked = was_locked;
+
+    std::unique_ptr<Event> event = base::MakeUnique<Event>(
+        events::LOCK_SCREEN_DATA_ON_DATA_ITEMS_AVAILABLE,
+        api::lock_screen_data::OnDataItemsAvailable::kEventName,
+        api::lock_screen_data::OnDataItemsAvailable::Create(event_args));
+    EventRouter::Get(context_)->DispatchEventToExtension(id, std::move(event));
+  }
+}
+
+void ItemStorage::CreateItem(const std::string& extension_id,
+                             const CreateCallback& callback) {
+  EnsureCacheForExtensionLoaded(
+      extension_id,
+      base::Bind(&ItemStorage::CreateItemImpl, weak_ptr_factory_.GetWeakPtr(),
+                 extension_id, callback));
+}
+
+void ItemStorage::GetAllForExtension(const std::string& extension_id,
+                                     const DataItemListCallback& callback) {
+  EnsureCacheForExtensionLoaded(
+      extension_id,
+      base::Bind(&ItemStorage::GetAllForExtensionImpl,
+                 weak_ptr_factory_.GetWeakPtr(), extension_id, callback));
+}
+
+void ItemStorage::SetItemContent(const std::string& extension_id,
+                                 const std::string& item_id,
+                                 const std::vector<char>& data,
+                                 const ItemStorage::WriteCallback& callback) {
+  EnsureCacheForExtensionLoaded(
+      extension_id, base::Bind(&ItemStorage::SetItemContentImpl,
+                               weak_ptr_factory_.GetWeakPtr(), extension_id,
+                               item_id, data, callback));
+}
+
+void ItemStorage::GetItemContent(const std::string& extension_id,
+                                 const std::string& item_id,
+                                 const ItemStorage::ReadCallback& callback) {
+  EnsureCacheForExtensionLoaded(
+      extension_id, base::Bind(&ItemStorage::GetItemContentImpl,
+                               weak_ptr_factory_.GetWeakPtr(), extension_id,
+                               item_id, callback));
+}
+
+void ItemStorage::DeleteItem(const std::string& extension_id,
+                             const std::string& item_id,
+                             const WriteCallback& callback) {
+  EnsureCacheForExtensionLoaded(
+      extension_id,
+      base::Bind(&ItemStorage::DeleteItemImpl, weak_ptr_factory_.GetWeakPtr(),
+                 extension_id, item_id, callback));
+}
+
+void ItemStorage::OnExtensionUninstalled(
+    content::BrowserContext* browser_context,
+    const Extension* extension,
+    UninstallReason reason) {
+  ClearExtensionData(extension->id());
+}
+
+ItemStorage::CachedExtensionData::CachedExtensionData() = default;
+
+ItemStorage::CachedExtensionData::~CachedExtensionData() = default;
+
+bool ItemStorage::IsContextAllowed(content::BrowserContext* context) {
+  switch (session_locked_state_) {
+    case SessionLockedState::kUnknown:
+      return false;
+    case SessionLockedState::kLocked:
+      return ExtensionsBrowserClient::Get()->IsLockScreenContext(context);
+    case SessionLockedState::kNotLocked:
+      return context_ == context;
+  }
+  NOTREACHED() << "Unknown session locked state";
+  return false;
+}
+
+void ItemStorage::CreateItemImpl(const std::string& extension_id,
+                                 const CreateCallback& callback) {
+  std::unique_ptr<DataItem> item =
+      CreateDataItem(base::GenerateGUID(), extension_id, context_,
+                     value_store_cache_.get(), task_runner_.get(), crypto_key_);
+  DataItem* item_ptr = item.get();
+  item_ptr->Register(
+      base::Bind(&ItemStorage::OnItemRegistered, weak_ptr_factory_.GetWeakPtr(),
+                 base::Passed(std::move(item)), extension_id, callback));
+}
+
+void ItemStorage::GetAllForExtensionImpl(const std::string& extension_id,
+                                         const DataItemListCallback& callback) {
+  std::vector<const DataItem*> items;
+  ExtensionDataMap::iterator extension_data =
+      data_item_cache_.find(extension_id);
+  if (extension_data == data_item_cache_.end()) {
+    callback.Run(items);
+    return;
+  }
+
+  for (const auto& item : extension_data->second.data_items) {
+    if (!item.second)
+      continue;
+    items.push_back(item.second.get());
+  }
+
+  callback.Run(items);
+}
+
+void ItemStorage::SetItemContentImpl(
+    const std::string& extension_id,
+    const std::string& item_id,
+    const std::vector<char>& data,
+    const ItemStorage::WriteCallback& callback) {
+  DataItem* item = FindItem(extension_id, item_id);
+  if (!item) {
+    callback.Run(OperationResult::kNotFound);
+    return;
+  }
+
+  item->Write(data, callback);
+}
+
+void ItemStorage::GetItemContentImpl(const std::string& extension_id,
+                                     const std::string& item_id,
+                                     const ReadCallback& callback) {
+  DataItem* item = FindItem(extension_id, item_id);
+  if (!item) {
+    callback.Run(OperationResult::kNotFound, nullptr);
+    return;
+  }
+
+  item->Read(callback);
+}
+
+void ItemStorage::DeleteItemImpl(const std::string& extension_id,
+                                 const std::string& item_id,
+                                 const WriteCallback& callback) {
+  DataItem* item = FindItem(extension_id, item_id);
+  if (!item) {
+    callback.Run(OperationResult::kNotFound);
+    return;
+  }
+
+  item->Delete(base::Bind(&ItemStorage::OnItemDeleted, base::Unretained(this),
+                          extension_id, item_id, callback));
+}
+
+void ItemStorage::OnItemRegistered(std::unique_ptr<DataItem> item,
+                                   const std::string& extension_id,
+                                   const CreateCallback& callback,
+                                   OperationResult result) {
+  if (result != OperationResult::kSuccess) {
+    callback.Run(result, nullptr);
+    return;
+  }
+
+  DataItem* item_ptr = item.get();
+  data_item_cache_[extension_id].data_items.emplace(item_ptr->id(),
+                                                    std::move(item));
+
+  {
+    DictionaryPrefUpdate update(local_state_, kLockScreenDataPrefKey);
+    update->SetInteger(user_id_ + "." + extension_id,
+                       data_item_cache_[extension_id].data_items.size());
+  }
+
+  callback.Run(OperationResult::kSuccess, item_ptr);
+}
+
+void ItemStorage::OnItemDeleted(const std::string& extension_id,
+                                const std::string& item_id,
+                                const WriteCallback& callback,
+                                OperationResult result) {
+  data_item_cache_[extension_id].data_items.erase(item_id);
+  {
+    DictionaryPrefUpdate update(local_state_, kLockScreenDataPrefKey);
+    update->SetInteger(user_id_ + "." + extension_id,
+                       data_item_cache_[extension_id].data_items.size());
+  }
+
+  callback.Run(result);
+}
+
+void ItemStorage::EnsureCacheForExtensionLoaded(const std::string& extension_id,
+                                                const base::Closure& callback) {
+  CachedExtensionData* data = &data_item_cache_[extension_id];
+  if (data->state == CachedExtensionData::State::kLoaded) {
+    callback.Run();
+    return;
+  }
+
+  data->load_callbacks.push_back(callback);
+
+  if (data->state == CachedExtensionData::State::kLoading)
+    return;
+
+  data->state = CachedExtensionData::State::kLoading;
+
+  GetRegisteredItems(extension_id, context_, value_store_cache_.get(),
+                     task_runner_.get(),
+                     base::Bind(&ItemStorage::OnGotExtensionItems,
+                                weak_ptr_factory_.GetWeakPtr(), extension_id));
+}
+
+void ItemStorage::OnGotExtensionItems(
+    const std::string& extension_id,
+    OperationResult result,
+    std::unique_ptr<base::DictionaryValue> items) {
+  ExtensionDataMap::iterator data = data_item_cache_.find(extension_id);
+  if (data == data_item_cache_.end() ||
+      data->second.state != CachedExtensionData::State::kLoading) {
+    return;
+  }
+
+  if (result == OperationResult::kSuccess) {
+    for (base::DictionaryValue::Iterator item_iter(*items);
+         !item_iter.IsAtEnd(); item_iter.Advance()) {
+      std::unique_ptr<DataItem> item = CreateDataItem(
+          item_iter.key(), extension_id, context_, value_store_cache_.get(),
+          task_runner_.get(), crypto_key_);
+      data->second.data_items.emplace(item_iter.key(), std::move(item));
+    }
+  }
+
+  DictionaryPrefUpdate update(local_state_, kLockScreenDataPrefKey);
+  update->SetInteger(user_id_ + "." + extension_id,
+                     data->second.data_items.size());
+
+  data->second.state = CachedExtensionData::State::kLoaded;
+
+  std::vector<base::Closure> callbacks;
+  callbacks.swap(data->second.load_callbacks);
+  for (auto& callback : callbacks)
+    callback.Run();
+}
+
+DataItem* ItemStorage::FindItem(const std::string& extension_id,
+                                const std::string& item_id) {
+  ExtensionDataMap::iterator extension_data =
+      data_item_cache_.find(extension_id);
+  if (extension_data == data_item_cache_.end())
+    return nullptr;
+
+  if (extension_data->second.state != CachedExtensionData::State::kLoaded)
+    return nullptr;
+  DataItemMap::iterator item_it =
+      extension_data->second.data_items.find(item_id);
+  if (item_it == extension_data->second.data_items.end())
+    return nullptr;
+
+  if (!item_it->second)
+    return nullptr;
+  return item_it->second.get();
+}
+
+std::set<std::string> ItemStorage::GetExtensionsWithDataItems(
+    bool include_empty) {
+  std::set<std::string> result;
+
+  const base::DictionaryValue* user_data = nullptr;
+  const base::DictionaryValue* items =
+      local_state_->GetDictionary(kLockScreenDataPrefKey);
+  if (!items || !items->GetDictionary(user_id_, &user_data) || !user_data)
+    return result;
+
+  for (base::DictionaryValue::Iterator extension_iter(*user_data);
+       !extension_iter.IsAtEnd(); extension_iter.Advance()) {
+    if (extension_iter.value().is_int() &&
+        (include_empty || extension_iter.value().GetInt() > 0)) {
+      result.insert(extension_iter.key());
+    }
+  }
+  return result;
+}
+
+void ItemStorage::ClearUninstalledAppData() {
+  std::set<std::string> extensions =
+      GetExtensionsWithDataItems(true /* include_empty */);
+  for (const auto& id : extensions) {
+    if (!ExtensionRegistry::Get(context_)->GetInstalledExtension(id))
+      ClearExtensionData(id);
+  }
+}
+
+void ItemStorage::ClearExtensionData(const std::string& id) {
+  DeleteAllItems(id, context_, value_store_cache_.get(), task_runner_.get(),
+                 base::Bind(&ItemStorage::RemoveExtensionFromLocalState,
+                            weak_ptr_factory_.GetWeakPtr(), id));
+}
+
+void ItemStorage::RemoveExtensionFromLocalState(const std::string& id) {
+  {
+    DictionaryPrefUpdate update(local_state_, kLockScreenDataPrefKey);
+    update->Remove(user_id_ + "." + id, nullptr);
+  }
+
+  ExtensionDataMap::iterator it = data_item_cache_.find(id);
+  if (it == data_item_cache_.end())
+    return;
+
+  std::vector<base::Closure> callbacks;
+  callbacks.swap(it->second.load_callbacks);
+
+  data_item_cache_.erase(it);
+
+  for (auto& callback : callbacks)
+    callback.Run();
+}
+
+}  // namespace lock_screen_data
+}  // namespace extensions