Changes in app.window and app.runtime to support lock screen note taking

chrome/browser/chromeos/lock_screen_apps/state_controller.h

 // Copyright 2017 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.
 
 #ifndef CHROME_BROWSER_CHROMEOS_LOCK_SCREEN_APPS_STATE_CONTROLLER_H_
 #define CHROME_BROWSER_CHROMEOS_LOCK_SCREEN_APPS_STATE_CONTROLLER_H_
 
 #include <memory>
 
 #include "ash/public/interfaces/tray_action.mojom.h"
 #include "base/observer_list.h"
 #include "base/scoped_observer.h"
 #include "chrome/browser/chromeos/lock_screen_apps/app_manager.h"
 #include "chrome/browser/chromeos/lock_screen_apps/state_observer.h"
 #include "components/session_manager/core/session_manager_observer.h"
+#include "extensions/browser/app_window/app_window_registry.h"
+#include "extensions/common/api/app_runtime.h"
 #include "mojo/public/cpp/bindings/binding.h"
 
 class Profile;
 
+namespace content {
+class BrowserContext;
+}
+
+namespace extensions {
+class AppWindow;
+class Extension;
+}
+
 namespace session_manager {
 class SessionManager;
 }
 
 namespace lock_screen_apps {
 
+class AppManager;
 class StateObserver;
 
 // Manages state of lock screen action handler apps, and notifies
 // interested parties as the state changes.
 // Currently assumes single supported action - NEW_NOTE.
 class StateController : public ash::mojom::TrayActionClient,
-                        public session_manager::SessionManagerObserver {
+                        public session_manager::SessionManagerObserver,
+                        public extensions::AppWindowRegistry::Observer {
  public:
   // Returns whether the StateController is enabled - it is currently guarded by
   // a feature flag. If not enabled, |StateController| instance is not allowed
   // to be created. |Get| will still work, but it will return nullptr.
   static bool IsEnabled();
 
   // Returns the global StateController instance. Note that this can return
   // nullptr when lock screen apps are not enabled (see |IsEnabled|).
   static StateController* Get();
 
@@ skipping 21 matching lines @@
 
   // Gets current state assiciated with the lock screen note action.
   ash::mojom::TrayActionState GetLockScreenNoteState() const;
 
   // ash::mojom::TrayActionClient:
   void RequestNewLockScreenNote() override;
 
   // session_manager::SessionManagerObserver:
   void OnSessionStateChanged() override;
 
+  // extensions::AppWindowREgistry::Observer:

[benwells, 2017/06/13 10:30:46]

Nit: REgistry

[tbarzic, 2017/06/13 19:17:59]

Done.

+  void OnAppWindowRemoved(extensions::AppWindow* app_window) override;
+
+  // Returns whether an extension should be allowed to create an app window on
+  // the lock screen in order to handle the provided action.
+  bool CanCreateAppWindowForAction(

[benwells, 2017/06/13 10:30:46]

Does this have to be public?

[tbarzic, 2017/06/13 19:17:59]

It's currently used by ChromeAppWindowDelegate

+      content::BrowserContext* context,
+      const extensions::Extension* extension,
+      extensions::api::app_runtime::ActionType action) const;
+
+  // Registers the app window as action handler for the action on Chrome OS lock
+  // screen.
+  // Returns whether the app window was successfully registered.
+  bool RegisterAppWindowForAction(
+      extensions::AppWindow* window,
+      extensions::api::app_runtime::ActionType action);
+
   // If there are any active lock screen action handlers, moved their windows
   // to background, to ensure lock screen UI is visible.
   void MoveToBackground();
 
   // If there are any lock screen action handler in background, moves their
   // windows back to foreground (i.e. visible over lock screen UI).
   void MoveToForeground();
 
-  // Sets the current state - to be used in tests. Hopefully, when this class
-  // has more logic implemented, this will not be needed.
-  void SetLockScreenNoteStateForTesting(ash::mojom::TrayActionState state);
-
  private:
   // Called when app manager reports that note taking availability has changed.
   void OnNoteTakingAvailabilityChanged();
 
+  // If there is an app window registered as a handler for note taking action
+  // on lock screen, unregisters the window, and closes is if |close_window| is
+  // set. It changes the current state to kAvailable or kNotAvailable, depending
+  // on whether lock screen note taking action can still be handled.
+  void ResetNoteTakingWindowAndMoveToNextState(bool close_window);
+
   // Requests lock screen note action state change to |state|.
   // Returns whether the action state has changed.
   bool UpdateLockScreenNoteState(ash::mojom::TrayActionState state);
 
   // Notifies observers that the lock screen note action state changed.
   void NotifyLockScreenNoteStateChanged();
 
   // Lock screen note action state.
   ash::mojom::TrayActionState lock_screen_note_state_ =
       ash::mojom::TrayActionState::kNotAvailable;
 
   base::ObserverList<StateObserver> observers_;
 
   mojo::Binding<ash::mojom::TrayActionClient> binding_;
   ash::mojom::TrayActionPtr tray_action_ptr_;
 
   std::unique_ptr<AppManager> app_manager_;
 
+  extensions::AppWindow* note_app_window_ = nullptr;
+
+  ScopedObserver<extensions::AppWindowRegistry,
+                 extensions::AppWindowRegistry::Observer>
+      app_window_observer_;
   ScopedObserver<session_manager::SessionManager,
                  session_manager::SessionManagerObserver>
       session_observer_;
 
   DISALLOW_COPY_AND_ASSIGN(StateController);
 };
 
 }  // namespace lock_screen_apps
 
 #endif  // CHROME_BROWSER_CHROMEOS_LOCK_SCREEN_APPS_STATE_CONTROLLER_H_