Lock screen for Chrome-Athena

athena/system/power_button_controller.cc

 // Copyright 2014 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 "athena/system/power_button_controller.h"
 
 #include "athena/screen/public/screen_manager.h"
+#include "athena/screen_lock/public/screen_lock_manager.h"
 #include "athena/strings/grit/athena_strings.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/gfx/font_list.h"
 #include "ui/views/background.h"
 #include "ui/views/border.h"
 #include "ui/views/controls/label.h"
 #include "ui/views/layout/box_layout.h"
 #include "ui/views/widget/widget.h"
 
@@ skipping 19 matching lines @@
 }
 
 PowerButtonController::~PowerButtonController() {
   chromeos::DBusThreadManager::Get()->GetPowerManagerClient()->RemoveObserver(
       this);
 }
 
 void PowerButtonController::ShowShutdownWarningDialog() {
   state_ = STATE_SHUTDOWN_WARNING_VISIBLE;
 
+  // Lock screen manager exists inside user session only.
+  if (athena::ScreenLockManager::Get())
+    athena::ScreenLockManager::Get()->LockScreen();
+
   shutdown_warning_message_.reset(new views::Widget);
 
   views::Widget::InitParams params(views::Widget::InitParams::TYPE_POPUP);
   params.ownership = views::Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
   params.parent = warning_message_container_;
   shutdown_warning_message_->Init(params);
 
   views::Label* label =
       new views::Label(l10n_util::GetStringUTF16(IDS_ATHENA_SHUTDOWN_WARNING));
   label->SetBackgroundColor(SK_ColorWHITE);
@@ skipping 60 matching lines @@
           ->GetPowerManagerClient()
           ->RequestSuspend();
     }
     state_ = STATE_OTHER;
     timer_.Stop();
     shutdown_warning_message_.reset();
   }
 }
 
 }  // namespace athena