simple-api-server.js

'use strict';

const Path = require('path');
const PassThrough = require('stream').PassThrough;

const express = require('express');
const bodyParser = require('body-parser');
const cookieParser = require('cookie-parser');
const session = require('express-session');

const Unifile = require('../lib/');

const app = express();
const unifile = new Unifile();

app.use(bodyParser.json());
app.use(cookieParser());
app.use(session({
	secret: 'unifile',
	resave: false,
	saveUninitialized: false
}));

// Configure connectors
const ghconnector = new Unifile.GitHubConnector({
	clientId: 'b4e46028bf36d871f68d',
	clientSecret: 'c39806c4d0906cfeaac932012996a1919475cc78'
});
const dbxconnector = new Unifile.DropboxConnector({
	clientId: '37mo489tld3rdi2',
	clientSecret: 'kqfzd11vamre6xr',
	redirectUri: 'http://localhost:6805/dropbox/oauth-callback'
});
const ftpconnector = new Unifile.FtpConnector({redirectUri: 'http://localhost:6805/ftp/signin'});
//const wdconnector = new Unifile.WebDavConnector({redirectUri: 'http://localhost:6805/webdav/signin'});
const rsconnector = new Unifile.RemoteStorageConnector({redirectUri: 'http://localhost:6805/remotestorage/callback'});
const fsconnector = new Unifile.FsConnector({showHiddenFile: true});
const sftpconnector = new Unifile.SftpConnector({redirectUri: 'http://localhost:6805/sftp/signin'});

// Register connectors
unifile.use(ghconnector);
unifile.use(dbxconnector);
unifile.use(ftpconnector);
unifile.use(rsconnector);
//unifile.use(wdconnector);
unifile.use(fsconnector);
unifile.use(sftpconnector);

// Expose connector methods
app.post('/:connector/authorize', function(req, res) {
	if(req.body != null) {
		if(req.session.unifile.remotestorage)
			req.session.unifile.remotestorage.userAddress = req.body.userAddress;
		else
			req.session.unifile.remotestorage = req.body;
	}
	unifile.getAuthorizeURL(req.session.unifile, req.params.connector)
	.catch((err) => {
		console.error('Error while authorizing Unifile', err);
		res.statusCode = 400;
		res.end();
	})
	.then((result) => res.end(result));
});

// Search for a old session token in the cookies
app.get('/', function(req, res) {
	// Init unifile session in Express
	req.session.unifile = req.session.unifile || {};

	let response;
	if(req.cookies.unifile_github)
		response = unifile.setAccessToken(req.session.unifile, 'github', req.cookies.unifile_github);
	if(req.cookies.unifile_dropbox)
		response = unifile.setAccessToken(req.session.unifile, 'dropbox', req.cookies.unifile_dropbox);

	if(response)
		response.then(() => res.sendFile(Path.join(__dirname, 'public', 'index.html')));
	else res.sendFile(Path.join(__dirname, 'public', 'index.html'));
});

// List files and folders
app.get(/\/(.*)\/ls\/(.*)/, function(req, res) {
	unifile.readdir(req.session.unifile, req.params[0], req.params[1])
	.then(function(result) {
		res.send(result);
	})
	.catch(function(err) {
		console.error(err);
		res.status(400).send(err);
	});
});

app.put(/\/(.*)\/mkdir\/(.*)/, function(req, res) {
	unifile.mkdir(req.session.unifile, req.params[0], req.params[1])
	.then(function(result) {
		res.send(result);
	})
	.catch(function(err) {
		console.error(err);
		res.status(400).send(err);
	});
});

app.put(/\/(.*)\/put\/(.*)/, function(req, res) {
	unifile.writeFile(req.session.unifile, req.params[0], req.params[1], req.body.content)
	.then(function(result) {
		res.send(result);
	})
	.catch(function(err) {
		console.error(err);
		res.status(400).send(err);
	});
});

app.get(/\/(.*)\/get\/(.*)/, function(req, res) {
	unifile.readFile(req.session.unifile, req.params[0], req.params[1])
	.then(function(result) {
		res.send(result);
	})
	.catch(function(err) {
		console.error(err);
		res.status(400).send(err);
	});
});

app.patch(/\/(.*)\/mv\/(.*)/, function(req, res) {
	unifile.rename(req.session.unifile, req.params[0], req.params[1], req.body.destination)
	.then(function(result) {
		res.send(result);
	})
	.catch(function(err) {
		console.error(err);
		res.status(400).send(err);
	});
});

app.delete(/\/(.*)\/rm\/(.*)/, function(req, res) {
	unifile.unlink(req.session.unifile, req.params[0], req.params[1])
	.then(function(result) {
		res.send(result);
	})
	.catch(function(err) {
		console.error(err);
		res.status(400).send(err);
	});
});

app.delete(/\/(.*)\/rmdir\/(.*)/, function(req, res) {
	unifile.rmdir(req.session.unifile, req.params[0], req.params[1])
	.then(function(result) {
		res.send(result);
	})
	.catch(function(err) {
		console.error(err);
		res.status(400).send(err);
	});
});

app.post(/\/(.*)\/cp\/(.*)/, function(req, res) {
	let stream = unifile.createReadStream(req.session.unifile, req.params[0], req.params[1]);
	// Use PassThrough to prevent request from copying headers between requests
	if(req.params[0] !== 'webdav' && req.params[0] !== 'fs') stream = stream.pipe(new PassThrough());
	stream.pipe(unifile.createWriteStream(req.session.unifile, req.params[0], req.body.destination))
	.pipe(res);
});

app.post(/\/(.*)\/batch\/(.*)/, function(req, res) {
	const path = req.params[1];
	const batch = [
		{name: 'mkdir', path: path},
		{name: 'writeFile', path: path + '/test.txt', content: 'Hello world'},
		{name: 'writeFile', path: path + '/test2.txt', content: 'Hello world too'},
		{name: 'rename', path: path + '/test.txt', destination: path + '/test_old.txt'},
		{name: 'unlink', path: path + '/test2.txt'},
		{name: 'rmdir', path: path}
	];
	unifile.batch(req.session.unifile, req.params[0], batch)
	.then((result) => {
		res.send(result);
	})
	.catch((err) => {
		console.error(err);
		res.status(400).send(err);
	});
});

app.get(/\/(.*)\/stat\/(.*)/, function(req, res) {
	unifile.stat(req.session.unifile, req.params[0], req.params[1])
	.then((result) => {
		res.send(result);
	})
	.catch((err) => {
		res.status(400).send(err.message);
	});
});

// register callback url
app.get('/:connector/oauth-callback', function(req, res) {
	if('error' in req.query) {
		res.status(500).send(req.query);
	} else {
		unifile.login(req.session.unifile, req.params.connector, req.query)
		.then(function(result) {
			res.cookie('unifile_' + req.params.connector, result);
			res.end('<script>window.close();</script>');
		})
		.catch(function(err) {
			console.error('ERROR', err);
			res.status(500).send(err);
		});
	}
});

app.get('/remotestorage/callback', function(req, res) {
	// Return a script that get the hash and redirect to oauth-callback
	res.end('<script>' +
				'var token = location.hash.substr(1).split("=")[1];location="/remotestorage/oauth-callback?token="+token' +
				'</script>');
});

app.get('/:connector/signin', function(req, res) {
	res.sendFile(Path.join(__dirname, 'public', req.params.connector + '_login.html'));
});

// server 'loop'
const port = process.env.PORT || 6805; // 6805 is the date of sexual revolution started in paris france 8-)
app.listen(port, function() {
	console.log('Listening on ' + port);
});