video-processor.js

const http = require('http');
const shell = require('shelljs');
const uuidv4 = require('uuid/v4');
const formidable = require('formidable');
var cmds = require('./video-processor-cmds.js');
var config = require('./config.js')

// variable to assure only one upload request happens
var reqhappened = false;

// generated token
const genToken = uuidv4();

http.createServer(function (req, res) {

	res.setHeader('access-control-allow-headers', 'DNT,X-CustomHeader,Keep-Alive,User-Agent,X-Requested-With,If-Modified-Since,Cache-Control,Content-Type,Content-Range,Range');
	res.setHeader('access-control-allow-origin', config.cors);
	res.setHeader('access-control-allow-credentials', 'true');
	res.setHeader('Content-Type', 'application/json; charset=utf-8');
	res.setHeader('access-control-max-age', '600');

	if (req.url == '/getStatus') {
		console.log("getStatus Hit")
		res.statusCode = 200;
		if (!reqhappened)
			res.end('{"version":"'+config.version+'","currentWaitingInQueue":{"audioCpuToEncode":0,"videoGpuToEncode":0,"audioVideoCpuToEncode":0,"spriteToCreate":0,"ipfsToAdd":0}}');
		else
			res.end('{"version":"'+config.version+'","currentWaitingInQueue":{"audioCpuToEncode":0,"videoGpuToEncode":0,"audioVideoCpuToEncode":99,"spriteToCreate":0,"ipfsToAdd":99}}');
	}

	// sending encoder progress to user
	if (req.url.match(/\/getProgressByToken.*/)) {
		console.log("getProgressByToken Hit")
		res.end(JSON.stringify(cmds.encoderResponse));
	}

	// file upload
	if (req.url == '/uploadVideo?videoEncodingFormats=240p,480p,720p,1080p&sprite=true' && !reqhappened) {

		if (req.method === 'OPTIONS') {
			console.log("uploadVideo Options Hit")
			res.statusCode = 204;
			res.end();

		} else {
			var fileUploadLimit = setTimeout(() => {
				console.log("Upload timed out")
				console.log("Killing container")
				process.exit();
			}, 3600000);

			res.statusCode = 200;
			console.log("uploadVideo Non-Options Hit")

			var form = new formidable.IncomingForm();

			//Sane Form options
			form.maxFields = 1;
			form.encoding = 'utf-8';
			form.maxFileSize = '4096000000';

			form.parse(req, function (err, fields, files) {
			});

			// file is moved to upload folder and renamed to uuid
			form.on('fileBegin', function (name, file) {

				reqhappened = true;
				console.log("File upload Began")
				file.path = "./upload/" + genToken;
			});


			form.on('file', function (name, file) {
				console.log("File upload Finished, setting reqhappened")
				clearTimeout(fileUploadLimit)
				//frontend needs to know if upload was successful and receive the token
				var successResponse = { success: "", token: "" };

				// check if file is valid
				shell.exec(cmds.ffprobe_cmds.createCmdString(file.path), function (code, stdout, stderr) {

					// code isn't 0 if error occurs
					if (code) {
						// if error, success is false, no token, end process
						console.log(stderr);
						console.log("Bad File, killing.")
						successResponse.success = "false";
						res.end(JSON.stringify(successResponse));
						process.exit();
					}

					var fileData = JSON.parse(stdout);

					// if file is valid, success is true and provide token
					successResponse.success = "true";
					successResponse.token = genToken;
					console.log(successResponse + " sent to user")
					res.end(JSON.stringify(successResponse));

					var videoHeight = null
					for (let i = 0; i < fileData.streams.length; i++) {
						if (fileData.streams[i].height) {
							videoHeight = fileData.streams[i].height;
							break
						}
					}
					var fileDuration = fileData.format.duration;

					//create sprite and upload it to ipfs (ipfs upload function is called within sprite function)
					console.log("Making the sprite")
					cmds.sprite_cmds.sprite(file.path, fileDuration, "./sprite");

					// find ipfs hash of source file
					cmds.ipfs_cmds.ipfsUpload(file.path, "ipfsAddSourceVideo", true);

					// videos under 240 res or longer than 10 minutes not encoded
					if (videoHeight <= 240 || fileDuration > config.maxEncodedDuration) {
						console.log("No encoding")
					} else if (videoHeight > 240 && videoHeight <= 480) {
						console.log("Encoding 240p")
						// adds data about encoded video to encoder response
						cmds.addEncodedVideoData(["240p"]);

						cmds.encoder_cmds.encode(cmds.encoder_cmds.changeSettings(file.path, "fileres240.mp4", 426, 240), 0);

						// videos over 480 res get encoded to 240 and 480 res
					} else if (videoHeight > 480) {
						cmds.encoderResponse.sourceStored = false
						console.log("encoding 240p and 480p")
						cmds.addEncodedVideoData(["240p", "480p"]);

						cmds.encoder_cmds.encode(cmds.encoder_cmds.changeSettings(file.path, "fileres240.mp4", 426, 240), 0, function () {
							cmds.encoder_cmds.encode(cmds.encoder_cmds.changeSettings(file.path, "fileres480.mp4", 854, 480), 1);
						});
					}

					// checks if all procedures are done so that finish status can be set
					cmds.checkIfFinished(file.path);

				});

			});

			form.on('error', function (err) {
				console.error('Error', err);
				process.exit();
			});

		}

	} else {
		res.statusCode = 503;
		res.end("There's nothing here for you");
	}

}).listen(config.port, () => {
	console.log("listening on port "+config.port);
});