[Chromoting] Share code that updates the UI when clientSession changes.

remoting/webapp/crd/js/client_session.js

 // Copyright (c) 2012 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.
 
 /**
  * @fileoverview
  * Class handling creation and teardown of a remoting client session.
  *
  * The ClientSession class controls lifetime of the client plugin
  * object and provides the plugin with the functionality it needs to
@@ skipping 572 matching lines @@
     this.plugin_ = null;
   }
 
   // Leave full-screen mode, and stop listening for related events.
   var listener = this.callOnFullScreenChanged_;
   remoting.fullscreen.activate(
       false,
       function() {
         remoting.fullscreen.removeListener(listener);
       });
-  if (remoting.windowFrame) {
-    remoting.windowFrame.setClientSession(null);
-  } else {
-    remoting.toolbar.setClientSession(null);
-  }
-  remoting.optionsMenu.setClientSession(null);
-  document.body.classList.remove('connected');
+  this.updateClientSessionUi_(null);
 
   // Remove mediasource-rendering class from the container - this will also
   // hide the <video> element.
   this.container_.classList.remove('mediasource-rendering');
 
   this.container_.removeEventListener('mousemove',
                                       this.updateMouseCursorPosition_,
                                       true);
 };
 
 /**
+ * @param {remoting.ClientSession} clientSession The active session, or null if
+ *     there is no connection.
+ */
+remoting.ClientSession.prototype.updateClientSessionUi_ = function(
+    clientSession) {
+  if (remoting.windowFrame) {
+    remoting.windowFrame.setClientSession(clientSession);
+  }
+  if (remoting.toolbar) {
+    remoting.toolbar.setClientSession(clientSession);
+  }
+  if (remoting.optionsMenu) {
+    remoting.optionsMenu.setClientSession(clientSession);
+  }
+
+  if (clientSession == null) {
+    document.body.classList.remove('connected');
+  } else {
+    document.body.classList.add('connected');
+  }
+}
+
+/**
  * Disconnect the current session with a particular |error|.  The session will
  * raise a |stateChanged| event in response to it.  The caller should then call
  * |cleanup| to remove and destroy the <embed> element.
  *
  * @param {remoting.Error} error The reason for the disconnection.  Use
  *    remoting.Error.NONE if there is no error.
  * @return {void} Nothing.
  */
 remoting.ClientSession.prototype.disconnect = function(error) {
   var state = (error == remoting.Error.NONE) ?
@@ skipping 327 matching lines @@
     this.setFocusHandlers_();
     this.onDesktopSizeChanged_();
     if (this.resizeToClient_) {
       var clientArea = this.getClientArea_();
       this.plugin_.notifyClientResolution(clientArea.width,
                                           clientArea.height,
                                           window.devicePixelRatio);
     }
     // Activate full-screen related UX.
     remoting.fullscreen.addListener(this.callOnFullScreenChanged_);
-    if (remoting.windowFrame) {
-      remoting.windowFrame.setClientSession(this);
-    } else {
-      remoting.toolbar.setClientSession(this);
-    }
-    remoting.optionsMenu.setClientSession(this);
-    document.body.classList.add('connected');
-
+    this.updateClientSessionUi_(this);
     this.container_.addEventListener('mousemove',
                                      this.updateMouseCursorPosition_,
                                      true);
 
   } else if (status == remoting.ClientSession.State.FAILED) {
     switch (error) {
       case remoting.ClientSession.ConnectionError.HOST_IS_OFFLINE:
         this.error_ = remoting.Error.HOST_IS_OFFLINE;
         break;
       case remoting.ClientSession.ConnectionError.SESSION_REJECTED:
@@ skipping 658 matching lines @@
  * @param {string} data Contents of the extension message.
  * @return {boolean} True if the message was recognized, false otherwise.
  */
 remoting.ClientSession.prototype.handleExtensionMessage =
     function(type, data) {
   if (this.videoFrameRecorder_) {
     return this.videoFrameRecorder_.handleMessage(type, data);
   }
   return false;
 }