[Easy Unlock] Update handling of the trial easy unlock/signin run

chrome/browser/extensions/api/screenlock_private/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/extensions/api/screenlock_private/screenlock_private_api.h"
 
 #include "base/lazy_instance.h"
 #include "base/values.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/signin/easy_unlock_service.h"
 #include "chrome/common/extensions/api/screenlock_private.h"
+#include "chrome/common/extensions/extension_constants.h"
 #include "extensions/browser/event_router.h"
 
 namespace screenlock = extensions::api::screenlock_private;
 
 namespace extensions {
 
 namespace {
 
 screenlock::AuthType FromLockHandlerAuthType(
     ScreenlockBridge::LockHandler::AuthType auth_type) {
@@ skipping 32 matching lines @@
 }
 
 ScreenlockPrivateSetLockedFunction::ScreenlockPrivateSetLockedFunction() {}
 
 ScreenlockPrivateSetLockedFunction::~ScreenlockPrivateSetLockedFunction() {}
 
 bool ScreenlockPrivateSetLockedFunction::RunAsync() {
   scoped_ptr<screenlock::SetLocked::Params> params(
       screenlock::SetLocked::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
-  if (params->locked)
+  if (params->locked) {
+    if (extension()->id() == extension_misc::kEasyUnlockAppId) {
+      // Mark the Easy Unlock behaviour on the lock screen as the one initiated
+      // by the Easy Unlock setup app as a trial one.
+      // TODO(tbarzic): Move this logic to a new easyUnlockPrivate function.
+      EasyUnlockService* service = EasyUnlockService::Get(GetProfile());
+      if (service)
+        service->SetTrialRun();
+    }
     ScreenlockBridge::Get()->Lock(GetProfile());
-  else
+  } else {
     ScreenlockBridge::Get()->Unlock(GetProfile());
+  }
   SendResponse(error_.empty());
   return true;
 }
 
 ScreenlockPrivateAcceptAuthAttemptFunction::
     ScreenlockPrivateAcceptAuthAttemptFunction() {}
 
 ScreenlockPrivateAcceptAuthAttemptFunction::
     ~ScreenlockPrivateAcceptAuthAttemptFunction() {}
 
 bool ScreenlockPrivateAcceptAuthAttemptFunction::RunSync() {
   scoped_ptr<screenlock::AcceptAuthAttempt::Params> params(
       screenlock::AcceptAuthAttempt::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
   Profile* profile = Profile::FromBrowserContext(browser_context());
-  EasyUnlockService::Get(profile)->FinalizeUnlock(params->accept);
+  EasyUnlockService* service = EasyUnlockService::Get(profile);
+  if (service)
+    service->FinalizeUnlock(params->accept);
   return true;
 }
 
 ScreenlockPrivateEventRouter::ScreenlockPrivateEventRouter(
     content::BrowserContext* context)
     : browser_context_(context) {
   ScreenlockBridge::Get()->AddObserver(this);
 }
 
 ScreenlockPrivateEventRouter::~ScreenlockPrivateEventRouter() {}
@@ skipping 48 matching lines @@
   args->AppendString(screenlock::ToString(FromLockHandlerAuthType(auth_type)));
   args->AppendString(value);
 
   scoped_ptr<extensions::Event> event(new extensions::Event(
       screenlock::OnAuthAttempted::kEventName, args.Pass()));
   router->BroadcastEvent(event.Pass());
   return true;
 }
 
 }  // namespace extensions