spee.ch/routes/sockets-routes.js

const logger = require('winston');
const fs = require('fs');
const publishController = require('../controllers/publishController.js');
const publishHelpers = require('../helpers/libraries/publishHelpers.js');
const errorHandlers = require('../helpers/libraries/errorHandlers.js');
const { postToStats } = require('../controllers/statsController.js');

module.exports = (app, siofu, hostedContentPath) => {
  const http = require('http');
  const server = http.Server(app);
  const io = require('socket.io')(server);

  io.on('connection', socket => {
    logger.silly('a user connected via sockets');
    // attach upload listeners
    const uploader = new siofu();
    uploader.dir = hostedContentPath;
    uploader.listen(socket);
    // listener for when file upload starts
    uploader.on('start', ({ file }) => {
      // log
      logger.info('client started an upload:', file.name);
      // server side test to make sure file is not a bad file type
      if (/\.exe$/.test(file.name)) {
        uploader.abort(file.id, socket);
      }
    });
    // listener for when file upload encounters an error
    uploader.on('error', ({ error }) => {
      logger.error('an error occured while uploading', error);
      postToStats('publish', '/', null, error);
      socket.emit('publish-status', error);
    });
    // listener for when file has been uploaded
    uploader.on('saved', ({ file }) => {
      if (file.success) {
        logger.debug(`Client successfully uploaded ${file.name}`);
        socket.emit('publish-status', 'file upload successfully completed. Your image is being published (this might take a second)...');
        // prepare the publish parameters
        const publishParams = publishHelpers.createPublishParams(file.meta.name, file.pathName, file.meta.license, file.meta.nsfw);
        // publish the file
        publishController
          .publish(publishParams, file.name, file.meta.type)
          .then(result => {
            postToStats('publish', '/', null, 'success');
            socket.emit('publish-complete', { name: publishParams.name, result });
          })
          .catch(error => {
            error = errorHandlers.handlePublishError(error);
            postToStats('publish', '/', null, error);
            socket.emit('publish-failure', error);
          });
      } else {
        logger.error(`An error occurred in uploading the client's file`);
        socket.emit('publish-failure', 'file uploaded, but with errors');
        postToStats('publish', '/', null, 'file uploaded, but with errors');
        // to-do: remove the file if not done automatically
      }
    });
    // handle asset requests
    socket.on('asset-request', filePath => {
      logger.debug('received a request for the following file', filePath);
      fs.readFile(filePath, (err, assetBuffer) => {
        if (err) {
          logger.error('fs.readFile error', err);
          return;
        };
        // set the data
        const fileExtension = filePath.substring(filePath.lastIndexOf('.'));
        let data = {
          type  : null,
          buffer: assetBuffer.toString('base64'),
        };
        switch (fileExtension) {
          case 'jpeg' || 'jpg':
            data['type'] = 'image/jpeg';
            break;
          case 'gif':
            data['type'] = 'image/gif';
            break;
          case 'png':
            data['type'] = 'image/png';
            break;
          case 'mp4':
            data['type'] = 'video/mp4';
            break;
          default:
            data['type'] = 'image/jpeg';
            logger.warn('showing file with unknown type as jpeg');
            break;
        }
        // send the asset
        socket.emit('asset-transfer', data);
      });
    });
    // handle disconnect
    socket.on('disconnect', () => {
      logger.silly('a user disconnected via sockets');
    });
  });

  return server;
};