iceshrimp-161sh/src/server/index.ts

/**
 * Core Server
 */

import * as fs from 'fs';
import * as http from 'http';
import * as https from 'https';
import * as Koa from 'koa';
import * as Router from 'koa-router';
import * as bodyParser from 'koa-bodyparser';

import activityPub from './activitypub';
import webFinger from './webfinger';
import config from '../config';

// Init server
const app = new Koa();
app.proxy = true;
app.use(bodyParser);

// HSTS
// 6months (15552000sec)
if (config.url.startsWith('https')) {
	app.use((ctx, next) => {
		ctx.set('strict-transport-security', 'max-age=15552000; preload');
		next();
	});
}

// Init router
const router = new Router();

// Routing
router.use('/api', require('./api'));
router.use('/files', require('./file'));
router.use(activityPub.routes());
router.use(webFinger.routes());
router.use(require('./web'));

// Register router
app.use(router.routes());

function createServer() {
	if (config.https) {
		const certs = {};
		Object.keys(config.https).forEach(k => {
			certs[k] = fs.readFileSync(config.https[k]);
		});
		return https.createServer(certs, app);
	} else {
		return http.createServer(app);
	}
}

export default () => new Promise(resolve => {
	const server = createServer();

	/**
	 * Steaming
	 */
	require('./api/streaming')(server);

	/**
	 * Server listen
	 */
	server.listen(config.port, resolve);
});