Add chrome.screenlockPrivate API functions to handle alternate auth types.

chrome/browser/chromeos/extensions/screenlock_private_api.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/extensions/screenlock_private_api.h"
 
 #include "base/lazy_instance.h"
 #include "base/values.h"
 #include "chrome/browser/chromeos/login/screen_locker.h"
 #include "chrome/browser/extensions/image_loader.h"
 #include "chrome/common/extensions/api/screenlock_private.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "extensions/browser/event_router.h"
 #include "extensions/browser/extension_system.h"
 
 namespace screenlock = extensions::api::screenlock_private;
 
 namespace extensions {
 
+namespace {
+
+const char kNotLockedError[] = "Screen is not currently locked.";
+
+chromeos::LoginDisplay::AuthType ToLoginDisplayAuthType(
+    screenlock::AuthType auth_type) {
+  switch (auth_type) {
+    case screenlock::AUTH_TYPE_SYSTEMPASSWORD:
+      return chromeos::LoginDisplay::OFFLINE_PASSWORD;
+    case screenlock::AUTH_TYPE_NUMERICPIN:
+      return chromeos::LoginDisplay::NUMERIC_PIN;
+    case screenlock::AUTH_TYPE_USERCLICK:
+      return chromeos::LoginDisplay::USER_CLICK;
+    default:
+      NOTREACHED();
+      return chromeos::LoginDisplay::OFFLINE_PASSWORD;
+  }
+}
+
+screenlock::AuthType ToScreenlockPrivateAuthType(
+    chromeos::LoginDisplay::AuthType auth_type) {
+  switch (auth_type) {
+    case chromeos::LoginDisplay::OFFLINE_PASSWORD:
+      return screenlock::AUTH_TYPE_SYSTEMPASSWORD;
+    case chromeos::LoginDisplay::NUMERIC_PIN:
+      return screenlock::AUTH_TYPE_NUMERICPIN;
+    case chromeos::LoginDisplay::USER_CLICK:
+      return screenlock::AUTH_TYPE_USERCLICK;
+    case chromeos::LoginDisplay::ONLINE_SIGN_IN:
+      // Apps should treat forced online sign in same as system password.
+      return screenlock::AUTH_TYPE_SYSTEMPASSWORD;
+    default:
+      NOTREACHED();
+      return screenlock::AUTH_TYPE_SYSTEMPASSWORD;
+  }
+}
+
+}  // namespace
+
 ScreenlockPrivateGetLockedFunction::ScreenlockPrivateGetLockedFunction() {}
 
 ScreenlockPrivateGetLockedFunction::~ScreenlockPrivateGetLockedFunction() {}
 
 bool ScreenlockPrivateGetLockedFunction::RunImpl() {
   bool locked = false;
   chromeos::ScreenLocker* locker =
       chromeos::ScreenLocker::default_screen_locker();
   if (locker)
     locked = locker->locked();
@@ skipping 48 matching lines @@
 ScreenlockPrivateShowButtonFunction::
   ~ScreenlockPrivateShowButtonFunction() {}
 
 bool ScreenlockPrivateShowButtonFunction::RunImpl() {
   scoped_ptr<screenlock::ShowButton::Params> params(
       screenlock::ShowButton::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
   chromeos::ScreenLocker* locker =
       chromeos::ScreenLocker::default_screen_locker();
   if (!locker) {
-    SendResponse(error_.empty());
+    SetError(kNotLockedError);
+    SendResponse(false);
     return true;
   }
   extensions::ImageLoader* loader = extensions::ImageLoader::Get(GetProfile());
   loader->LoadImageAsync(
       GetExtension(), GetExtension()->GetResource(params->icon),
       gfx::Size(kMaxButtonIconSize, kMaxButtonIconSize),
       base::Bind(&ScreenlockPrivateShowButtonFunction::OnImageLoaded, this));
   return true;
 }
 
 void ScreenlockPrivateShowButtonFunction::OnImageLoaded(
     const gfx::Image& image) {
   chromeos::ScreenLocker* locker =
       chromeos::ScreenLocker::default_screen_locker();
   ScreenlockPrivateEventRouter* router =
     ScreenlockPrivateEventRouter::GetFactoryInstance()->GetForProfile(
         GetProfile());
+  const chromeos::User* user =
+      chromeos::UserManager::Get()->GetUserByProfile(GetProfile());
   locker->ShowUserPodButton(
-      GetProfile()->GetProfileName(), image,
+      user->email(),
+      image,
       base::Bind(&ScreenlockPrivateEventRouter::OnButtonClicked,
                  base::Unretained(router)));
   SendResponse(error_.empty());
 }
 
+ScreenlockPrivateHideButtonFunction::ScreenlockPrivateHideButtonFunction() {}
+
+ScreenlockPrivateHideButtonFunction::~ScreenlockPrivateHideButtonFunction() {}
+
+bool ScreenlockPrivateHideButtonFunction::RunImpl() {
+  chromeos::ScreenLocker* locker =
+      chromeos::ScreenLocker::default_screen_locker();
+  if (locker) {
+    const chromeos::User* user =
+        chromeos::UserManager::Get()->GetUserByProfile(GetProfile());
+    locker->HideUserPodButton(user->email());
+  } else {
+    SetError(kNotLockedError);
+  }
+  SendResponse(error_.empty());
+  return true;
+}
+
+ScreenlockPrivateSetAuthTypeFunction::ScreenlockPrivateSetAuthTypeFunction() {}
+
+ScreenlockPrivateSetAuthTypeFunction::~ScreenlockPrivateSetAuthTypeFunction() {}
+
+bool ScreenlockPrivateSetAuthTypeFunction::RunImpl() {
+  scoped_ptr<screenlock::SetAuthType::Params> params(
+      screenlock::SetAuthType::Params::Create(*args_));
+  EXTENSION_FUNCTION_VALIDATE(params.get());
+
+  chromeos::ScreenLocker* locker =
+      chromeos::ScreenLocker::default_screen_locker();
+  if (locker) {
+    std::string initial_value =
+        params->initial_value.get() ? *(params->initial_value.get()) : "";
+    const chromeos::User* user =
+        chromeos::UserManager::Get()->GetUserByProfile(GetProfile());
+    locker->SetAuthType(user->email(),
+                        ToLoginDisplayAuthType(params->auth_type),
+                        initial_value);
+  } else {
+    SetError(kNotLockedError);
+  }
+  SendResponse(error_.empty());
+  return true;
+}
+
+ScreenlockPrivateGetAuthTypeFunction::ScreenlockPrivateGetAuthTypeFunction() {}
+
+ScreenlockPrivateGetAuthTypeFunction::~ScreenlockPrivateGetAuthTypeFunction() {}
+
+bool ScreenlockPrivateGetAuthTypeFunction::RunImpl() {
+  chromeos::ScreenLocker* locker =
+      chromeos::ScreenLocker::default_screen_locker();
+  if (locker) {
+    const chromeos::User* user =
+        chromeos::UserManager::Get()->GetUserByProfile(GetProfile());
+    chromeos::LoginDisplay::AuthType auth_type =
+        locker->GetAuthType(user->email());
+    std::string auth_type_name =
+        screenlock::ToString(ToScreenlockPrivateAuthType(auth_type));
+    SetResult(new base::StringValue(auth_type_name));
+  } else {
+    SetError(kNotLockedError);
+  }
+  SendResponse(error_.empty());
+  return true;
+}
+
+ScreenlockPrivateAcceptAuthAttemptFunction::
+    ScreenlockPrivateAcceptAuthAttemptFunction() {}
+
+ScreenlockPrivateAcceptAuthAttemptFunction::
+    ~ScreenlockPrivateAcceptAuthAttemptFunction() {}
+
+bool ScreenlockPrivateAcceptAuthAttemptFunction::RunImpl() {
+  scoped_ptr<screenlock::AcceptAuthAttempt::Params> params(
+      screenlock::AcceptAuthAttempt::Params::Create(*args_));
+  EXTENSION_FUNCTION_VALIDATE(params.get());
+
+  chromeos::ScreenLocker* locker =
+      chromeos::ScreenLocker::default_screen_locker();
+  if (locker) {
+    if (params->accept)
+      locker->Hide();

[xiyuan, 2014/02/20 17:44:44]

Isn't Hide() a static method?

[Tim Song, 2014/02/20 21:06:08]

Done.

+    else
+      locker->EnableInput();
+  } else {
+    SetError(kNotLockedError);
+  }
+  SendResponse(error_.empty());
+  return true;
+}
+
 ScreenlockPrivateEventRouter::ScreenlockPrivateEventRouter(Profile* profile)
     : profile_(profile) {
   chromeos::SessionManagerClient* session_manager =
       chromeos::DBusThreadManager::Get()->GetSessionManagerClient();
   if (!session_manager->HasObserver(this))
     session_manager->AddObserver(this);
 }
 
 ScreenlockPrivateEventRouter::~ScreenlockPrivateEventRouter() {}
 
@@ skipping 33 matching lines @@
   chromeos::SessionManagerClient* session_manager =
       chromeos::DBusThreadManager::Get()->GetSessionManagerClient();
   if (session_manager->HasObserver(this))
     session_manager->RemoveObserver(this);
 }
 
 void ScreenlockPrivateEventRouter::OnButtonClicked() {
   DispatchEvent(screenlock::OnButtonClicked::kEventName, NULL);
 }
 
+void ScreenlockPrivateEventRouter::OnAuthAttempted(
+    chromeos::LoginDisplay::AuthType auth_type,
+    const std::string& value) {
+  scoped_ptr<base::ListValue> args(new base::ListValue());
+  args->AppendString(
+      screenlock::ToString(ToScreenlockPrivateAuthType(auth_type)));
+  args->AppendString(value);
+
+  scoped_ptr<extensions::Event> event(new extensions::Event(
+      screenlock::OnAuthAttempted::kEventName, args.Pass()));
+  extensions::ExtensionSystem::Get(profile_)->event_router()->BroadcastEvent(
+      event.Pass());
+}
+
 }  // namespace extensions