1 files changed, 4 insertions, 8 deletions
diff --git a/remoting/host/desktop_environment.h b/remoting/host/desktop_environment.h
index e0ed196..31335f5 100644
--- a/remoting/host/desktop_environment.h
+++ b/remoting/host/desktop_environment.h
@@ -58,11 +58,7 @@ class DesktopEnvironment {
   void OnPause(bool pause);
 
  private:
-  enum ContinueTimerState {
-    INACTIVE,      // The timer is not running or has been cancelled.
-    SHOW_DIALOG,   // Show the continue dialog when the timer expires.
-    SHUTDOWN_HOST  // Shutdown the Chromoting host when the timer expires.
-  };
+  class TimerTask;
 
   void ProcessOnConnect(const std::string& username);
   void ProcessOnLastDisconnect();
@@ -79,7 +75,8 @@ class DesktopEnvironment {
 
   void StartContinueWindowTimer(bool start);
 
-  void ContinueWindowTimerFunc();
+  void OnContinueWindowTimer();
+  void OnShutdownHostTimer();
 
   // The host that owns this DesktopEnvironment.
   ChromotingHost* host_;
@@ -111,8 +108,7 @@ class DesktopEnvironment {
   bool is_monitoring_local_inputs_;
 
   // Timer controlling the "continue session" dialog.
-  ContinueTimerState continue_timer_state_;
-  base::Time continue_timer_target_time_;
+  scoped_ptr<TimerTask> timer_task_;
 
   ScopedThreadProxy ui_thread_proxy_;