[Update] 修改ws使用的库

package.json

@@ -26,6 +26,7 @@
     "@swimlane/ngx-datatable": "^11.3.2",
     "@swimlane/ngx-ui": "^20.2.1",
     "@types/jquery": "^3.3.6",
+    "@types/neffos.js": "^0.1.1",
     "@types/socket.io-client": "^1.4.32",
     "ajv": "^6.5.0",
     "animate.css": "^3.6.1",
@@ -49,6 +50,7 @@
     "material-design-icons": "^3.0.1",
     "materialize-css": "^0.100.2",
     "metismenu": "^2.7.9",
+    "neffos.js": "^0.1.19",
     "ng2-charts": "^1.5.0",
     "ngx-bootstrap": "^1.6.6",
     "ngx-cookie-service": "^1.0.10",
@@ -63,8 +65,8 @@
     "roboto-fontface": "^0.8.0",
     "rxjs": "5.5.6",
     "sass-math": "^1.0.0",
-    "socket.io": "^2.1.0",
-    "socket.io-client": "^2.1.0",
+    "socket.io": "^1.4.32",
+    "socket.io-client": "^1.4.32",
     "ssh-keygen": "^0.4.1",
     "tether": "^1.4.4",
     "tslib": "^1.9.0",

src/app/elements/ssh-term/ssh-term.component.ts

-import {AfterViewInit, Component, Input, OnInit } from '@angular/core';
+import {AfterViewInit, Component, Input, OnInit, OnDestroy } from '@angular/core';
 import {Terminal} from 'xterm';
+import * as neffos from 'neffos.js';
 import {NavList} from '../../pages/control/control/control.component';
 import {UUIDService} from '../../app.service';
 import {CookieService} from 'ngx-cookie-service';
-import {TermWS} from '../../globals';
+import {Socket, Room} from '../../utils/socket';
+import {getWsSocket} from '../../globals';
 
-const ws = TermWS;
 
 @Component({
   selector: 'elements-ssh-term',
   templateUrl: './ssh-term.component.html',
   styleUrls: ['./ssh-term.component.scss']
 })
-export class ElementSshTermComponent implements OnInit, AfterViewInit {
+export class ElementSshTermComponent implements OnInit, AfterViewInit, OnDestroy {
   @Input() host: any;
   @Input() userid: any;
   @Input() index: number;
@@ -20,12 +21,20 @@ export class ElementSshTermComponent implements OnInit, AfterViewInit {
 
   term: Terminal;
   secret: string;
+  ws: Socket;
+  room: Room;
 
   constructor(private _uuid: UUIDService, private _cookie: CookieService) {
   }
 
   ngOnInit() {
     this.secret = this._uuid.gen();
+    getWsSocket().then(sock => {
+      this.ws = sock;
+      console.log('Connect ok');
+      console.log(this.ws);
+      this.joinRoom();
+    });
     const fontSize = localStorage.getItem('fontSize') || '14';
     this.term = new Terminal({
       fontFamily: 'monaco, Consolas, "Lucida Console", monospace',
@@ -38,57 +47,66 @@ export class ElementSshTermComponent implements OnInit, AfterViewInit {
   }
 
   ngAfterViewInit() {
-    this.joinRoom();
   }
 
   changeWinSize(size: Array<number>) {
-    ws.emit('resize', {'cols': size[0], 'rows': size[1]});
+    if (this.ws) {
+      this.ws.emit('resize', {'cols': size[0], 'rows': size[1]});
+    }
   }
 
   joinRoom() {
     NavList.List[this.index].Term = this.term;
-    console.log(this.term);
-    console.log('Col: ', this.term.cols, 'rows', this.term.rows);
     if (this.host) {
-      ws.emit('host', {
-        'uuid': this.host.id,
-        'userid': this.userid,
-        'secret': this.secret,
-        'size': [this.term.cols, this.term.rows]
-      });
+      const data = {
+        uuid: this.host.id,
+        userid: this.userid,
+        secret: this.secret,
+        size: [this.term.cols, this.term.rows]
+      };
+      this.ws.emit('host', data);
     }
     if (this.token) {
-      ws.emit('token', {
+      const data = {
         'token': this.token, 'secret': this.secret,
         'size': [this.term.cols, this.term.rows]
-      });
+      };
+      this.ws.emit('token', data);
     }
     const that = this;
 
-    this.term.on('data', function (data) {
-      ws.emit('data', {'data': data, 'room': NavList.List[that.index].room});
+    this.term.on('data', data => {
+      const d = {'data': data};
+      this.room.emit('data', d);
     });
 
-    ws.on('data', data => {
+    this.ws.on('data', (roomName, msg) => {
+      const data = msg.unmarshal();
       const view = NavList.List[that.index];
-      if (view && data['room'] === view.room) {
+      if (view && roomName === view.room) {
         that.term.write(data['data']);
       }
     });
 
-    ws.on('disconnect', () => {
+    this.ws.on('disconnect', () => {
       that.close();
     });
 
-    ws.on('logout', (data) => {
-      if (data['room'] === NavList.List[that.index].room) {
+    this.ws.on('logout', (roomName, msg) => {
+      const data = msg.unmarshal();
+      if (data.room === NavList.List[that.index].room) {
         NavList.List[that.index].connected = false;
       }
     });
 
-    ws.on('room', data => {
-      if (data['secret'] === this.secret) {
-        NavList.List[that.index].room = data['room'];
+    this.ws.on('room', (roomName, msg) => {
+      const data = msg.unmarshal();
+      console.log('On room', roomName, data);
+      if (data.secret === this.secret && data.room) {
+        this.ws.JoinRoom(data.room).then(room => {
+          this.room = room;
+        });
+        NavList.List[that.index].room = data.room;
       }
     });
   }
@@ -97,11 +115,16 @@ export class ElementSshTermComponent implements OnInit, AfterViewInit {
     const view = NavList.List[this.index];
     if (view) {
       NavList.List[this.index].connected = false;
-      ws.emit('logout', NavList.List[this.index].room);
+      console.log(NavList.List[this.index].room);
+      this.room.emit('logout', {'room': NavList.List[this.index].room});
     }
   }
 
   active() {
     this.term.focus();
   }
+
+  ngOnDestroy(): void {
+    this.close();
+  }
 }

src/app/globals.ts

 'use strict';
-import * as io from 'socket.io-client';
+import {EventEmitter} from 'events/events';
+// import * as io from 'socket.io-client';
+import * as neffos from 'neffos.js';
 import {Terminal} from 'xterm';
+// const abc = io.connect('/ssh');
+import {getWsSock, Socket} from './utils/socket';
 
-export const TermWS = io.connect('/ssh');
+const scheme = document.location.protocol === 'https:' ? 'wss' : 'ws';
+const port = document.location.port ? ':' + document.location.port : '';
+const wsURL = scheme + '://' + document.location.hostname + port + '/socket.io/';
+export let TermWS = null;
+
+export const emitter = new(EventEmitter);
 export const sep = '/';
 export let Video: {
   id: string,
@@ -93,7 +102,7 @@ export let DataStore: {
   guacamole_token: string;
   guacamole_token_time: number;
 } = {
-  socket: io.connect(),
+  socket: TermWS,
   Nav: [{}],
   NavShow: true,
   Path: {},
@@ -136,3 +145,14 @@ export let wsEvent: {
 };
 
 export const i18n = new Map();
+
+export async function getWsSocket() {
+  if (TermWS) {
+    return TermWS;
+  }
+  TermWS = await getWsSock(wsURL, 'ssh');
+  DataStore.socket = TermWS;
+  return TermWS;
+}
+
+

src/app/pages/control/control/control.component.ts

@@ -52,7 +52,7 @@ export class ControlComponent implements OnInit {
     if (NavList.List[id].connected) {
       NavList.List[id].connected = false;
       NavList.List[id].Term.write('\r\n\x1b[31mBye Bye!\x1b[m\r\n');
-      TermWS.emit('logout', NavList.List[id].room);
+      TermWS.emit('logout', {room: NavList.List[id].room});
     }
   }
 

src/app/utils/socket.ts

+import {EventEmitter} from 'events/events';
+import {NSConn, marshal} from 'neffos.js';
+import * as neffos from 'neffos.js';
+
+export class Room {
+  room: neffos.Room;
+
+  constructor(room: neffos.Room) {
+    this.room = room;
+  }
+
+  emit(event: string, obj: object): boolean {
+    const msg = marshal(obj);
+    return this.room.emit(event, msg);
+  }
+}
+
+export class Socket {
+  conn: NSConn;
+  emitter: EventEmitter;
+
+  constructor(conn: NSConn, emitter: EventEmitter) {
+    this.conn = conn;
+    this.emitter = emitter;
+  }
+
+  emit(type: string, obj: any) {
+    const msg = marshal(obj);
+    this.conn.emit(type, msg);
+  }
+
+  on(type: string, fn: Function, opt_scope?: any, opt_oneshot?: boolean) {
+    this.emitter.on(type, fn, opt_scope, opt_oneshot);
+  }
+
+  async JoinRoom(roomName: string): Promise<Room> {
+    console.log('Join room: ', roomName);
+    const room = await this.conn.joinRoom(roomName);
+    const wrapper = new Room(room);
+    return wrapper;
+  }
+
+  getRoom(roomName: string) {
+    const room = this.conn.rooms[roomName];
+    if (room) {
+      return new Room(room);
+    }
+    return null;
+  }
+}
+
+export async function getWsSock(url: string, namespace: string): Promise<Socket> {
+  const emitter = new EventEmitter();
+  const events = {};
+  events[namespace] = {
+    _OnNamespaceConnected: function (ns, msg) {
+       emitter.emit('connect', ns);
+    },
+
+    _OnNamespaceDisconnect: function (ns, msg) {
+      emitter.emit('disconnect', ns);
+    },
+
+    _OnRoomJoined: function (ns, msg) {
+      emitter.emit('roomJoined', msg.Room);
+    },
+
+    _OnRoomLeft: function (ns, msg) {
+      emitter.emit('roomLeft', msg.Room);
+    },
+
+    _OnAnyEvent: function (ns, msg) {
+      emitter.emit(msg.Event, msg.Room, msg);
+    }
+  };
+  const options = {reconnnect: 5};
+  const conn = <neffos.Conn>await neffos.dial(url, events, options);
+  const nsConn = <neffos.NSConn>await conn.connect(namespace);
+  const sock = new Socket(nsConn, emitter);
+  return sock;
+}