Refactor ContinueWindow::Show() to accept a callback parameter.

remoting/host/continue_window_win.cc

 // Copyright (c) 2011 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 "remoting/host/continue_window.h"
 
 #include <windows.h>
 
 #include "base/compiler_specific.h"
 #include "base/logging.h"
@@ skipping 14 matching lines @@
 //   SimpleHost: simple_host_process.cc
 extern HMODULE g_hModule;
 
 namespace remoting {
 
 class ContinueWindowWin : public ContinueWindow {
  public:
   ContinueWindowWin();
   virtual ~ContinueWindowWin();
 
-  virtual void Show(remoting::ChromotingHost* host) OVERRIDE;
+  virtual void Show(remoting::ChromotingHost* host,
+                    const ContinueSessionCallback& callback) OVERRIDE;
   virtual void Hide() OVERRIDE;
 
  private:
   static BOOL CALLBACK DialogProc(HWND hwmd, UINT msg, WPARAM wParam,
                                   LPARAM lParam);
 
   BOOL OnDialogMessage(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam);
 
   void EndDialog();
   void SetStrings(const UiStrings& strings);
 
   remoting::ChromotingHost* host_;
+  ContinueSessionCallback callback_;
   HWND hwnd_;
 
   DISALLOW_COPY_AND_ASSIGN(ContinueWindowWin);
 };
 
 ContinueWindowWin::ContinueWindowWin()
     : host_(NULL),
       hwnd_(NULL) {
 }
 
@@ skipping 23 matching lines @@
     case WM_CLOSE:
       // Ignore close messages.
       return TRUE;
     case WM_DESTROY:
       // Ensure we don't try to use the HWND anymore.
       hwnd_ = NULL;
       return TRUE;
     case WM_COMMAND:
       switch (LOWORD(wParam)) {
         case IDC_CONTINUE_DEFAULT:
-          CHECK(host_);
-          host_->PauseSession(false);
+          callback_.Run(true);
           ::EndDialog(hwnd, LOWORD(wParam));
           hwnd_ = NULL;
           return TRUE;
         case IDC_CONTINUE_CANCEL:
-          CHECK(host_);
-          host_->Shutdown(base::Closure());
+          callback_.Run(false);
           ::EndDialog(hwnd, LOWORD(wParam));
           hwnd_ = NULL;
           return TRUE;
       }
   }
   return FALSE;
 }
 
-void ContinueWindowWin::Show(ChromotingHost* host) {
+void ContinueWindowWin::Show(ChromotingHost* host,
+                             const ContinueSessionCallback& callback) {
   host_ = host;
+  callback_ = callback;
 
   CHECK(!hwnd_);
   hwnd_ = CreateDialogParam(g_hModule, MAKEINTRESOURCE(IDD_CONTINUE), NULL,
       (DLGPROC)DialogProc, (LPARAM)this);
   if (!hwnd_) {
     LOG(ERROR) << "Unable to create Disconnect dialog for remoting.";
     return;
   }
 
   SetStrings(host->ui_strings());
@@ skipping 25 matching lines @@
   HWND hwndCancel = GetDlgItem(hwnd_, IDC_CONTINUE_CANCEL);
   CHECK(hwndCancel);
   SetWindowText(hwndCancel, strings.stop_sharing_button_text.c_str());
 }
 
 ContinueWindow* ContinueWindow::Create() {
   return new ContinueWindowWin();
 }
 
 }  // namespace remoting