fox-send/server/portal_server.js

const express = require('express');
const busboy = require('connect-busboy');
const path = require('path');
const fs = require('fs-extra');
const bodyParser = require('body-parser');
const crypto = require('crypto');
const conf = require('./config.js');
const stream = require('stream');
const fetch = require('node-fetch');
const storage = require('./storage.js');

let isProduction =
  conf.env === 'production' &&
  conf.s3_bucket !== 'localhost' &&
  conf.bitly_key !== 'localhost';

const AWS = require('aws-sdk');
const s3 = new AWS.S3();

const app = express();
const redis = require('redis');
const redis_client = redis.createClient();

redis_client.on('error', err => {
  console.log(err);
});

app.use(busboy());
app.use(bodyParser.json());
app.use(express.static(path.join(__dirname, '../public')));

app.get('/download/:id', (req, res) => {
  res.sendFile(path.join(__dirname + '/../public/download.html'));
});

app.get('/assets/download/:id', (req, res) => {
  let id = req.params.id;
  if (!validateID(id)) {
    res.sendStatus(404);
    return;
  }

  redis_client.hget(id, 'filename', (err, reply) => {
    if (!reply) {
      res.sendStatus(404);
    } else {
      storage.length(id).then(contentLength => {
        res.writeHead(200, {
          'Content-Disposition': 'attachment; filename=' + reply,
          'Content-Type': 'application/octet-stream',
          'Content-Length': contentLength
        });
      });

      let file_stream = storage.get(id);

      file_stream.on('close', () => {
        storage.forceDelete(id, redis_client).then(err => {
          if (!err) {
            console.log('Deleted.');
          }
        });
      });

      file_stream.pipe(res);
    }
  });
});

app.post('/delete/:id', (req, res) => {
  let id = req.params.id;

  if (!validateID(id)) {
    res.send(404);
    return;
  }

  let delete_token = req.body.delete_token;

  if (!delete_token) {
    res.sendStatus(404);
  }

  storage
    .delete(id, redis_client, delete_token)
    .then(err => {
      if (!err) {
        console.log('Deleted off s3.');
      }
    })
    .catch(err => res.sendStatus(404));
});

app.post('/upload/:id', (req, res, next) => {
  if (!validateID(req.params.id)) {
    res.send(404);
    return;
  }

  req.pipe(req.busboy);
  req.busboy.on('file', (fieldname, file, filename) => {
    console.log('Uploading: ' + filename);
    let url = `${req.protocol}://${req.get('host')}/download/${req.params.id}/`;

    storage
      .set(req.params.id, file, filename, redis_client, url)
      .then(linkAndID => {
        res.json(linkAndID);
      });
  });
});

let server = app.listen(conf.listen_port, () => {
  console.log(`Portal app listening on port ${conf.listen_port}!`);
});

let validateID = route_id => {
  return route_id.match(/^[0-9a-fA-F]{32}$/) !== null;
};