typescript-assistant/src/server.ts

Combines and integrates professional Typescript tools into your project

import { Bus, Report } from './bus';
import * as WebSocket from 'ws';
import * as http from 'http';
import * as fs from 'fs';
import { Logger } from './logger';
import { Formatter } from './code-style/formatter';
import { Linter } from './code-style/linter';

export interface Server {
  start(): void;
}

let indexHtml = fs.readFileSync(`${__dirname}/../public/index.html`, { encoding: 'UTF8' });

export let createServer = (deps: { bus: Bus, logger: Logger, linter: Linter, formatter: Formatter }): Server => {
  let { logger, bus, linter, formatter } = deps;

  let lastReports: { [tool: string]: Report } = {};

  return {
    start: () => {
      let processReport = (report: Report) => {
        lastReports[report.tool] = report;
        let data = JSON.stringify(report, undefined, 2);
        wss.clients.forEach(client => client.send(data));
      };

      bus.register('report', processReport);

      let server = http.createServer((req, res) => {
        indexHtml = fs.readFileSync(`${__dirname}/../public/index.html`, { encoding: 'UTF8' });

        res.writeHead(200, { type: 'text/html' });
        res.write(indexHtml);
        res.end();
      });
      const wss = new WebSocket.Server({ server, path: '/ws' });

      server.on('listening', () => {
        logger.log('server', 'Experimental server listening on http://localhost:4551');
      });
      server.listen(4551);

      wss.on('connection', function connection(ws) {
        ws.on('message', function incoming(message) {
          logger.log('server', `received: ${message}`);
          if (message === 'lint-fix') {
            linter.lintOnce(true)
              .catch((err: any) => logger.error('server', `Error during lint-fix ${err}`));
          }
          if (message === 'format-fix') {
            formatter.formatFiles(undefined)
              .catch((err: any) => logger.error('server', `Error during format-fix ${err}`));
          }
        });
        Object.keys(lastReports).forEach((tool) => {
          ws.send(JSON.stringify(lastReports[tool], undefined, 2));
        });
      });
    }
  };
};