Screensaver at login screen.

chrome/browser/chromeos/ui/screensaver_extension_dialog.cc

Index: chrome/browser/chromeos/ui/screensaver_extension_dialog.cc
diff --git a/chrome/browser/chromeos/ui/screensaver_extension_dialog.cc b/chrome/browser/chromeos/ui/screensaver_extension_dialog.cc
new file mode 100644
index 0000000000000000000000000000000000000000..117d6747edd5ace6486761fe111ee8fa73fc2b42
--- /dev/null
+++ b/chrome/browser/chromeos/ui/screensaver_extension_dialog.cc
@@ -0,0 +1,151 @@
+// Copyright (c) 2012 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/ui/screensaver_extension_dialog.h"
+
+#include "base/bind.h"
+#include "base/logging.h"
+#include "base/memory/ref_counted.h"
+#include "chrome/browser/extensions/extension_service.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/profiles/profile_manager.h"
+#include "chrome/browser/sessions/restore_tab_helper.h"
+#include "chrome/browser/ui/views/extensions/extension_dialog.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/extensions/extension.h"
+#include "chrome/common/extensions/extension_file_util.h"
+
+using content::BrowserThread;
+
+namespace {
+
+ScreensaverExtensionDialog* g_instance = NULL;
+
+}  // namespace
+
+namespace browser {
+
+void ShowScreensaverDialog() {
+  ScreensaverExtensionDialog::ShowScreensaverDialog();
+}
+
+void CloseScreensaverDialog() {
+  ScreensaverExtensionDialog::CloseScreensaverDialog();
+}
+
+}  // namespace browser
+
+// static
+void ScreensaverExtensionDialog::ShowScreensaverDialog() {
+  if (!g_instance)
+    g_instance = new ScreensaverExtensionDialog();
+  g_instance->Show();
+}
+
+// static
+void ScreensaverExtensionDialog::CloseScreensaverDialog() {
+  if (g_instance)
+    g_instance->Close();
+}
+
+ScreensaverExtensionDialog::ScreensaverExtensionDialog()
+    : screensaver_extension_(NULL), loading_extension_(false) {

[sky, 2012/02/27 15:52:43]

nit: when you wrap, each param on its own line.

[rkc, 2012/02/27 22:24:43]

Done.

+}
+
+void ScreensaverExtensionDialog::LoadExtension() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+  std::string error;
+  std::string extension_path = CommandLine::ForCurrentProcess()->
+      GetSwitchValueASCII(switches::kKioskModeScreensaverPath);
+
+  scoped_refptr<Extension> screensaver_extension =
+      extension_file_util::LoadExtension(
+          FilePath(extension_path),
+          Extension::COMPONENT,
+          Extension::NO_FLAGS,
+          &error);
+
+  if (!screensaver_extension) {
+    LOG(ERROR) << "Could not load screensaver extension from: " <<
+        extension_path;
+    return;
+  }
+
+  BrowserThread::PostTask(BrowserThread::UI,
+                          FROM_HERE,
+                          base::Bind(
+                              &ScreensaverExtensionDialog::SetExtensionAndShow,
+                              base::Unretained(this),
+                              screensaver_extension));
+}
+
+void ScreensaverExtensionDialog::SetExtensionAndShow(
+    scoped_refptr<Extension> extension) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  screensaver_extension_ = extension;
+  loading_extension_ = false;
+  Show();
+}
+
+void ScreensaverExtensionDialog::Show() {
+  // Whenever we're loading the extension, Show() will
+  // be called after the load finishes, so return.
+  if (loading_extension_) return;

[sky, 2012/02/27 15:52:43]

Don't use single line ifs.

[rkc, 2012/02/27 22:24:43]

Done.

+
+  if (!screensaver_extension_) {
+    loading_extension_ = true;
+    BrowserThread::PostTask(BrowserThread::FILE,
+                            FROM_HERE,
+                            base::Bind(
+                                &ScreensaverExtensionDialog::LoadExtension,
+                                base::Unretained(this)));
+    return;
+  }
+
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  Profile* default_profile = ProfileManager::GetDefaultProfile();
+  if (default_profile->GetExtensionService()->AddExtension(
+          screensaver_extension_)) {
+    extension_dialog_ = ExtensionDialog::ShowFullscreen(
+        screensaver_extension_->GetFullLaunchURL(),
+        default_profile,
+        string16(),
+        this);
+  } else {
+    LOG(ERROR) << "Couldn't add screensaver extension to profile.";
+  }
+}
+
+void ScreensaverExtensionDialog::Close() {
+  if (extension_dialog_) {

[sky, 2012/02/27 15:52:43]

Is it possible for close to be invoked while you're loading the extension?

[rkc, 2012/02/27 22:24:43]

Yep, but in that case extension_dialog will be null so this code will do
nothing.

+    extension_dialog_->Close();
+    extension_dialog_ = NULL;
+  }
+}
+
+ScreensaverExtensionDialog::~ScreensaverExtensionDialog() {
+  if (extension_dialog_)
+    extension_dialog_->ObserverDestroyed();
+}
+
+void ScreensaverExtensionDialog::ExtensionDialogClosing(
+    ExtensionDialog* dialog) {
+  // Release our reference to the dialog to allow it to close.
+  extension_dialog_ = NULL;
+}
+
+void ScreensaverExtensionDialog::ExtensionTerminated(
+    ExtensionDialog* dialog) {
+  MessageLoop::current()->PostTask(FROM_HERE,

[sky, 2012/02/27 15:52:43]

Add comment as to why you need to delay this.

[rkc, 2012/02/27 22:24:43]

Done.

+      base::Bind(&ScreensaverExtensionDialog::ReloadAndShow,
+          base::Unretained(this)));
+  dialog->Close();
+}
+
+void ScreensaverExtensionDialog::ReloadAndShow() {
+  ProfileManager::GetDefaultProfile()->GetExtensionService()->ReloadExtension(
+      screensaver_extension_->id());
+
+  Show();
+}