Apply prettier formating

[e-mobility-charging-stations-simulator.git] / test / robohydra / plugins / wsServer / index.js

const RoboHydraHead = require('robohydra').heads.RoboHydraHead;
const RoboHydraWebSocketHead = require('robohydra').heads.RoboHydraWebSocketHead;
const RoboHydraWebSocketHeadProxy = require('robohydra').heads.RoboHydraWebSocketHeadProxy;

exports.getBodyParts = function (conf) {
  let wsSocket;
  return {
    heads: [
      new RoboHydraHead({
        name: 'message',
        path: '/message',
        method: 'POST',
        handler: function (req, res) {
          const msg = JSON.stringify(req.body);
          if (wsSocket) {
            wsSocket.send(msg);
            res.send('Message sent');
          } else {
            res.send('Cannot send message, no opened websocket found');
          }
        },
      }),

      new RoboHydraHead({
        name: 'close',
        path: '/close',
        method: 'GET',
        handler: function (req, res) {
          if (wsSocket) {
            wsSocket.close();
            res.send('Websocket closed');
          } else {
            res.send('Cannot close websocket, no opened websocket found');
          }
        },
      }),

      new RoboHydraWebSocketHeadProxy({
        name: 'proxy',
        mountPath: '/proxy',
        proxyTo: 'ws://server.example.com',
        preProcessor: function (data) {
          console.log('From the client: ' + data);
        },
        postProcessor: function (data) {
          console.log('From the server: ' + data);
        },
      }),

      new RoboHydraWebSocketHead({
        name: 'WS Server',
        path: '/.*',
        handler: function (req, socket) {
          wsSocket = socket;
        },
      }),
    ],
  };
};