cros: Explicitly manage app session lifetime.

chrome/browser/chromeos/app_mode/startup_app_launcher.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/app_mode/startup_app_launcher.h"
 
 #include "ash/shell.h"
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/json/json_file_value_serializer.h"
 #include "base/path_service.h"
 #include "base/prefs/pref_service.h"
 #include "base/time.h"
 #include "base/values.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/app_mode/kiosk_app_manager.h"
 #include "chrome/browser/chromeos/app_mode/kiosk_app_update_service.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/chromeos/ui/app_launch_view.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.h"
+#include "chrome/browser/extensions/shell_window_registry.h"
 #include "chrome/browser/extensions/webstore_startup_installer.h"
 #include "chrome/browser/lifetime/application_lifetime.h"
 #include "chrome/browser/policy/browser_policy_connector.h"
 #include "chrome/browser/signin/token_service.h"
 #include "chrome/browser/signin/token_service_factory.h"
 #include "chrome/browser/ui/extensions/application_launch.h"
+#include "chrome/browser/ui/extensions/shell_window.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/manifest_handlers/kiosk_enabled_info.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/browser_thread.h"
 #include "google_apis/gaia/gaia_auth_consumer.h"
 #include "google_apis/gaia/gaia_constants.h"
 
 using content::BrowserThread;
 using extensions::Extension;
 using extensions::WebstoreStartupInstaller;
+using extensions::ShellWindowRegistry;
 
 namespace chromeos {
 
 namespace {
 
 
 const char kOAuthRefreshToken[] = "refresh_token";
 const char kOAuthClientId[] = "client_id";
 const char kOAuthClientSecret[] = "client_secret";
 
 const base::FilePath::CharType kOAuthFileName[] =
     FILE_PATH_LITERAL("kiosk_auth");
 
 // Application install splash screen minimum show time in milliseconds.
 const int kAppInstallSplashScreenMinTimeMS = 3000;
 
-// Initial delay that gives an app 30 seconds during which the main app window
-// must be created. If app fails to create the main window in this timeframe,
-// chrome will exit.
-// TODO(xiyuan): Find a nicer way to trace process lifetime management at
-// startup. This just fixes a race that happens on faster machines.
-const int kInitialEndKeepAliveDelayinSec = 30;
-
 bool IsAppInstalled(Profile* profile, const std::string& app_id) {
   return extensions::ExtensionSystem::Get(profile)->extension_service()->
       GetInstalledExtension(app_id);
 }
 
 }  // namespace
 
+////////////////////////////////////////////////////////////////////////////////
+// StartupAppLauncher::AppWindowWatcher watches for a ShellWindow creation
+// for the given app id then runs the callback to notify its host.
+
+class StartupAppLauncher::AppWindowWatcher
+    : public ShellWindowRegistry::Observer {
+ public:
+  AppWindowWatcher(Profile* profile,
+                   const std::string& app_id,
+                   const base::Closure& cb)
+      : window_registry_(ShellWindowRegistry::Get(profile)),
+        app_id_(app_id),
+        cb_(cb) {
+    if (window_registry_)
+      window_registry_->AddObserver(this);
+  }
+
+  virtual ~AppWindowWatcher() {
+    if (window_registry_)
+      window_registry_->RemoveObserver(this);
+  }
+
+ private:
+  // extensions::ShellWindowRegistry::Observer overrides:
+  virtual void OnShellWindowAdded(ShellWindow* shell_window) OVERRIDE {
+    if (shell_window->extension()->id() == app_id_)

[zel, 2013/04/03 22:17:50]

are you sure that extension() isn't NULL?

what if identity API window pops up first?

+      cb_.Run();
+  }
+  virtual void OnShellWindowIconChanged(ShellWindow* shell_window) OVERRIDE {}
+  virtual void OnShellWindowRemoved(ShellWindow* shell_window) OVERRIDE {}
+
+  ShellWindowRegistry* window_registry_;  // Not owned.
+  const std::string app_id_;
+  base::Closure cb_;
+
+  DISALLOW_COPY_AND_ASSIGN(AppWindowWatcher);
+};
+
+////////////////////////////////////////////////////////////////////////////////
+// StartupAppLauncher
+
 StartupAppLauncher::StartupAppLauncher(Profile* profile,
                                        const std::string& app_id)
     : profile_(profile),
       app_id_(app_id),
       launch_splash_start_time_(0) {
   DCHECK(profile_);
   DCHECK(Extension::IdIsValid(app_id_));
   DCHECK(ash::Shell::HasInstance());
   ash::Shell::GetInstance()->AddPreTargetHandler(this);
 }
@@ skipping 134 matching lines @@
   }
 }
 
 void StartupAppLauncher::Cleanup() {
   chromeos::CloseAppLaunchSplashScreen();
 
   // Ends OpenAsh() keep alive since the session should either be bound with
   // the just launched app on success or should be ended on failure.
   // Invoking it via a PostNonNestableTask because Cleanup() could be called
   // before main message loop starts.
-  BrowserThread::PostNonNestableDelayedTask(
+  BrowserThread::PostNonNestableTask(
       BrowserThread::UI,
       FROM_HERE,
-      base::Bind(&chrome::EndKeepAlive),
-      base::TimeDelta::FromSeconds(kInitialEndKeepAliveDelayinSec));
+      base::Bind(&chrome::EndKeepAlive));
 
   delete this;
 }
 
 void StartupAppLauncher::OnLaunchSuccess() {
   const int64 time_taken_ms = (base::TimeTicks::Now() -
       base::TimeTicks::FromInternalValue(launch_splash_start_time_)).
       InMilliseconds();
 
   // Enforce that we show app install splash screen for some minimum amount
@@ skipping 42 matching lines @@
   if (!g_browser_process->browser_policy_connector()->IsEnterpriseManaged()) {
     PrefService* local_state = g_browser_process->local_state();
     local_state->SetBoolean(prefs::kRebootAfterUpdate, true);
   }
 
   // Always open the app in a window.
   chrome::OpenApplication(chrome::AppLaunchParams(profile_,
                                                   extension,
                                                   extension_misc::LAUNCH_WINDOW,
                                                   NEW_WINDOW));
-  OnLaunchSuccess();
+
+  // Waits until app window is created to report OnLaunchSuccess.
+  DCHECK(extension->is_platform_app());
+  app_window_watcher_.reset(new AppWindowWatcher(
+      profile_,
+      app_id_,
+      base::Bind(&StartupAppLauncher::OnLaunchSuccess, AsWeakPtr())));

[zel, 2013/04/03 22:17:50]

what happens if app never creates a window? are we going to be just hanging
there indefinitely?

 }
 
 void StartupAppLauncher::BeginInstall() {
   DVLOG(1) << "BeginInstall... connection = "
            <<  net::NetworkChangeNotifier::GetConnectionType();
 
   chromeos::UpdateAppLaunchSplashScreenState(
       chromeos::APP_LAUNCH_STATE_INSTALLING_APPLICATION);
 
   if (IsAppInstalled(profile_, app_id_)) {
@@ skipping 58 matching lines @@
   if (event->key_code() != ui::VKEY_S ||
       !(event->flags() & ui::EF_CONTROL_DOWN) ||
       !(event->flags() & ui::EF_ALT_DOWN)) {
     return;
   }
 
   OnLaunchFailure(KioskAppLaunchError::USER_CANCEL);
 }
 
 }   // namespace chromeos