Download and cache kiosk app extension when it is added via kiosk mamangement ui.

chrome/browser/chromeos/app_mode/kiosk_app_manager.cc

 // Copyright 2013 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 "chrome/browser/chromeos/app_mode/kiosk_app_manager.h"
 
 #include <map>
 #include <set>
 
 #include "base/bind.h"
+#include "base/files/file_path.h"
 #include "base/logging.h"
 #include "base/path_service.h"
 #include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/pref_service.h"
 #include "base/prefs/scoped_user_pref_update.h"
 #include "base/stl_util.h"
 #include "base/sys_info.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/app_mode/kiosk_app_data.h"
 #include "chrome/browser/chromeos/app_mode/kiosk_app_manager_observer.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h"
 #include "chrome/browser/chromeos/policy/device_local_account.h"
 #include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chrome/browser/chromeos/settings/owner_key_util.h"
+#include "chrome/browser/extensions/external_provider_impl.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/common/extensions/extension_constants.h"
+#include "chromeos/chromeos_paths.h"
 #include "chromeos/cryptohome/async_method_caller.h"
 #include "chromeos/settings/cros_settings_names.h"
 #include "content/public/browser/browser_thread.h"
 
 namespace chromeos {
 
 namespace {
 
 // Domain that is used for kiosk-app account IDs.
 const char kKioskAppAccountDomain[] = "kiosk-apps";
@@ skipping 10 matching lines @@
         << " failed, return code: " << return_code;
   }
 }
 
 // Check for presence of machine owner public key file.
 void CheckOwnerFilePresence(bool *present) {
   scoped_refptr<OwnerKeyUtil> util = OwnerKeyUtil::Create();
   *present = util->IsPublicKeyPresent();
 }
 
+scoped_refptr<base::SequencedTaskRunner> GetBackgroundTaskRunner() {
+  base::SequencedWorkerPool* pool = content::BrowserThread::GetBlockingPool();
+  CHECK(pool);
+  return pool->GetSequencedTaskRunnerWithShutdownBehavior(
+      pool->GetSequenceToken(), base::SequencedWorkerPool::SKIP_ON_SHUTDOWN);
+}
+
 }  // namespace
 
 // static
 const char KioskAppManager::kKioskDictionaryName[] = "kiosk";
 const char KioskAppManager::kKeyApps[] = "apps";
 const char KioskAppManager::kKeyAutoLoginState[] = "auto_login_state";
-const char KioskAppManager::kIconCacheDir[] = "kiosk";
+const char KioskAppManager::kIconCacheDir[] = "kiosk/icon";
+const char KioskAppManager::kCrxCacheDir[] = "kiosk/crx";
 
 // static
 static base::LazyInstance<KioskAppManager> instance = LAZY_INSTANCE_INITIALIZER;
 KioskAppManager* KioskAppManager::Get() {
   return instance.Pointer();
 }
 
 // static
 void KioskAppManager::Shutdown() {
   if (instance == NULL)
     return;
 
   instance.Pointer()->CleanUp();
 }
 
 // static
 void KioskAppManager::RegisterPrefs(PrefRegistrySimple* registry) {
   registry->RegisterDictionaryPref(kKioskDictionaryName);
 }
 
-KioskAppManager::App::App(const KioskAppData& data)
+KioskAppManager::App::App(const KioskAppData& data, bool is_extension_pending)
     : app_id(data.app_id()),
       user_id(data.user_id()),
       name(data.name()),
       icon(data.icon()),
-      is_loading(data.IsLoading()) {
+      is_loading(data.IsLoading() || is_extension_pending) {
 }
 
 KioskAppManager::App::App() : is_loading(false) {}
 KioskAppManager::App::~App() {}
 
 std::string KioskAppManager::GetAutoLaunchApp() const {
   return auto_launch_app_id_;
 }
 
 void KioskAppManager::SetAutoLaunchApp(const std::string& app_id) {
@@ skipping 185 matching lines @@
 
   policy::SetDeviceLocalAccounts(CrosSettings::Get(), device_local_accounts);
 }
 
 void KioskAppManager::GetApps(Apps* apps) const {
   apps->clear();
   apps->reserve(apps_.size());
   for (size_t i = 0; i < apps_.size(); ++i) {
     const KioskAppData& app_data = *apps_[i];
     if (app_data.status() != KioskAppData::STATUS_ERROR)
-      apps->push_back(App(app_data));
+      apps->push_back(App(
+          app_data, external_cache_->IsExtensionPending(app_data.app_id())));
   }
 }
 
 bool KioskAppManager::GetApp(const std::string& app_id, App* app) const {
   const KioskAppData* data = GetAppData(app_id);
   if (!data)
     return false;
 
-  *app = App(*data);
+  *app = App(*data, external_cache_->IsExtensionPending(app_id));
   return true;
 }
 
 const base::RefCountedString* KioskAppManager::GetAppRawIcon(
     const std::string& app_id) const {
   const KioskAppData* data = GetAppData(app_id);
   if (!data)
     return NULL;
 
   return data->raw_icon();
@@ skipping 30 matching lines @@
 
 void KioskAppManager::AddObserver(KioskAppManagerObserver* observer) {
   observers_.AddObserver(observer);
 }
 
 void KioskAppManager::RemoveObserver(KioskAppManagerObserver* observer) {
   observers_.RemoveObserver(observer);
 }
 
 KioskAppManager::KioskAppManager() : ownership_established_(false) {
+  base::FilePath cache_dir;
+  GetKioskAppCrxCacheDir(&cache_dir);
+  external_cache_.reset(
+      new ExternalCache(cache_dir,
+                        g_browser_process->system_request_context(),
+                        GetBackgroundTaskRunner(),
+                        this,
+                        true /* always_check_updates */,
+                        false /* wait_for_cache_initialization */));
+
   UpdateAppData();
   local_accounts_subscription_ =
       CrosSettings::Get()->AddSettingsObserver(
           kAccountsPrefDeviceLocalAccounts,
           base::Bind(&KioskAppManager::UpdateAppData, base::Unretained(this)));
   local_account_auto_login_id_subscription_ =
       CrosSettings::Get()->AddSettingsObserver(
           kAccountsPrefDeviceLocalAccountAutoLoginId,
           base::Bind(&KioskAppManager::UpdateAppData, base::Unretained(this)));
 }
 
 KioskAppManager::~KioskAppManager() {}
 
 void KioskAppManager::CleanUp() {
   local_accounts_subscription_.reset();
   local_account_auto_login_id_subscription_.reset();
   apps_.clear();
+  external_cache_.reset();
 }
 
 const KioskAppData* KioskAppManager::GetAppData(
     const std::string& app_id) const {
   for (size_t i = 0; i < apps_.size(); ++i) {
     const KioskAppData* data = apps_[i];
     if (data->app_id() == app_id)
       return data;
   }
 
@@ skipping 37 matching lines @@
       old_apps.erase(old_it);
     } else {
       KioskAppData* new_app =
           new KioskAppData(this, it->kiosk_app_id, it->user_id);
       apps_.push_back(new_app);  // Takes ownership of |new_app|.
       new_app->Load();
     }
   }
 
   // Clears cache and deletes the remaining old data.
+  std::vector<std::string> apps_to_remove;
   for (std::map<std::string, KioskAppData*>::iterator it = old_apps.begin();
        it != old_apps.end(); ++it) {
     it->second->ClearCache();
     cryptohome::AsyncMethodCaller::GetInstance()->AsyncRemove(
         it->second->user_id(),
         base::Bind(&OnRemoveAppCryptohomeComplete, it->first));
+    apps_to_remove.push_back(it->second->app_id());
   }
   STLDeleteValues(&old_apps);
+  external_cache_->RemoveExtensions(apps_to_remove);
+
+  // Request external_cache_ to download new apps and update the existing
+  // apps.
+  scoped_ptr<base::DictionaryValue> prefs(new base::DictionaryValue);
+  for (size_t i = 0; i < apps_.size(); ++i)
+    prefs->Set(apps_[i]->app_id(), new base::DictionaryValue);
+  external_cache_->UpdateExtensionsList(prefs.Pass());
 
   FOR_EACH_OBSERVER(KioskAppManagerObserver, observers_,
                     OnKioskAppsSettingsChanged());
 }
 
 void KioskAppManager::GetKioskAppIconCacheDir(base::FilePath* cache_dir) {
   base::FilePath user_data_dir;
   CHECK(PathService::Get(chrome::DIR_USER_DATA, &user_data_dir));
   *cache_dir = user_data_dir.AppendASCII(kIconCacheDir);
 }
 
 void KioskAppManager::OnKioskAppDataChanged(const std::string& app_id) {
   FOR_EACH_OBSERVER(KioskAppManagerObserver,
                     observers_,
                     OnKioskAppDataChanged(app_id));
 }
 
 void KioskAppManager::OnKioskAppDataLoadFailure(const std::string& app_id) {
   FOR_EACH_OBSERVER(KioskAppManagerObserver,
                     observers_,
                     OnKioskAppDataLoadFailure(app_id));
 }
 
+void KioskAppManager::OnExtensionListsUpdated(
+    const base::DictionaryValue* prefs) {
+}
+
+void KioskAppManager::OnExtensionLoadedInCache(const std::string& id) {
+  KioskAppData* app_data = GetAppDataMutable(id);
+  if (!app_data)
+    return;
+  OnKioskAppDataChanged(id);
+}
+
+void KioskAppManager::OnExtensionDownloadFailed(
+    const std::string& id,
+    extensions::ExtensionDownloaderDelegate::Error error) {
+  KioskAppData* app_data = GetAppDataMutable(id);
+  if (!app_data)
+    return;
+  OnKioskAppDataLoadFailure(id);
+}
+
 KioskAppManager::AutoLoginState KioskAppManager::GetAutoLoginState() const {
   PrefService* prefs = g_browser_process->local_state();
   const base::DictionaryValue* dict =
       prefs->GetDictionary(KioskAppManager::kKioskDictionaryName);
   int value;
   if (!dict->GetInteger(kKeyAutoLoginState, &value))
     return AUTOLOGIN_NONE;
 
   return static_cast<AutoLoginState>(value);
 }
 
 void KioskAppManager::SetAutoLoginState(AutoLoginState state) {
   PrefService* prefs = g_browser_process->local_state();
   DictionaryPrefUpdate dict_update(prefs,
                                    KioskAppManager::kKioskDictionaryName);
   dict_update->SetInteger(kKeyAutoLoginState, state);
   prefs->CommitPendingWrite();
 }
 
+void KioskAppManager::GetKioskAppCrxCacheDir(base::FilePath* cache_dir) {
+  base::FilePath user_data_dir;
+  CHECK(PathService::Get(chrome::DIR_USER_DATA, &user_data_dir));
+  *cache_dir = user_data_dir.AppendASCII(kCrxCacheDir);
+}
+
 }  // namespace chromeos