mizar/src/main/libs/Server.ts

import { ServerPort } from 'common/interfaces';
import * as net from 'net';

interface ServerParams {
   trace: boolean;
   echo: boolean;
   alertReset: boolean;
 }

class Server {
   process: NodeJS.Process;
   trace: boolean;
   echo: boolean;
   alertReset: boolean;
   ports: ServerPort[];
   server: net.Server[];
   nBytes: number[];
   nMsgs: number[];

   constructor (process: NodeJS.Process) {
      this.process = process;
      this.trace = false;
      this.echo = false;
      this.alertReset = false;
      this.ports = [];
      this.server = [];
      this.nBytes = [];
      this.nMsgs = [];
   }

   /**
    * Setta le porte
    *
    * @param {*} ports
    * @memberof Server
    */
   setPorts (ports: ServerPort[]) {
      this.ports = ports;
   }

   /**
    * Invia i log al render process
    *
    * @param {string} message Messaggio del log
    * @param {string} [color=''] Colore del log (green, yellow, red)
    * @memberof Server
    */
   sendLog (message: string, color = '') {
      const log = {
         event: 'log',
         content: { message, color }
      };
      this.process.send(log);
   }

   startServer (params: ServerParams) {
      this.trace = params.trace;
      this.echo = params.echo;
      this.alertReset = params.alertReset;

      for (let i = 0; i < this.ports.length; i++) {
         const port = this.ports[i].port;

         this.server[i] = net.createServer();
         this.nBytes[i] = 0;
         this.nMsgs[i] = 0;

         this.server[i].on('connection', (socket: net.Socket) => {
            if (this.trace) this.sendLog(`Client connesso su porta ${port}`);

            socket.on('data', (msg: Buffer) => {
               const msgString = msg.toString();
               if (this.echo) socket.write(msg);
               this.nBytes[i] += msg.length;
               this.nMsgs[i]++;

               if (this.trace) this.sendLog(`Messaggio ricevuto su porta ${port}: ${msgString}`);
            });// <- socket data

            socket.on('end', () => {
               if (this.trace) this.sendLog(`Client disconnesso su porta ${port}`);
            });

            socket.on('error', (err: Error & { code: string }) => {
               switch (err.code) {
                  case 'ECONNRESET':
                     if (this.alertReset)
                        this.sendLog(`Errore client su porta ${port}: \n${err}`, 'yellow');
                     else
                     if (this.trace) this.sendLog(`Client disconnesso su porta ${port}`);
                     break;
                  default:
                     this.sendLog(`Errore client su porta ${port}: \n${err}`, 'red');
               }
            });
         });// <- server

         this.server[i].on('error', (err: Error) => {
            this.sendLog(`Errore server su porta ${port}: \n${err}`, 'red');
         });

         this.server[i].listen(port, () => {
            this.sendLog(`In ascolto sulla porta ${port}`);
         });
      }
   }

   stopServer (callback: () => void) {
      (async () => {
         for (let i = 0; i < this.server.length; i++) {
            this.server[i].close(() => {
               this.server[i].unref();
            });
         }
         callback();
      })();
   }

   getReports () {
      const reportList: {port: number; sockets: number; data: number; messages: number}[] = [];
      for (let i = 0; i < this.server.length; i++) {
         const report = {
            port: (this.server[i].address() as net.AddressInfo).port,
            sockets: null as number,
            data: this.nBytes[i],
            messages: this.nMsgs[i]
         };

         this.server[i].getConnections((err: Error, nSockets: number) => {
            if (err) this.sendLog(`Errore report: \n${err}`, 'red');
            report.sockets = nSockets;
            reportList.push(report);

            if ((i + 1) === this.server.length) {
               const rep = {
                  event: 'report',
                  content: reportList
               };
               this.process.send(rep);
            }
         });
      }
   }

   resetReports () {
      for (let i = 0; i < this.server.length; i++) {
         this.nBytes[i] = 0;
         this.nMsgs[i] = 0;
      }
   }
}
export { Server, ServerParams };