Sharkey/src/server/index.ts

111 lines
2.4 KiB
TypeScript
Raw Normal View History

2016-12-29 00:49:51 +02:00
/**
* Core Server
*/
import * as fs from 'fs';
import * as http from 'http';
2018-05-24 16:21:10 +03:00
import * as http2 from 'http2';
2018-04-13 08:28:09 +03:00
import * as zlib from 'zlib';
2018-04-12 18:51:55 +03:00
import * as Koa from 'koa';
import * as Router from 'koa-router';
2018-04-13 00:17:14 +03:00
import * as mount from 'koa-mount';
2018-04-13 08:28:09 +03:00
import * as compress from 'koa-compress';
2018-04-19 12:03:46 +03:00
import * as logger from 'koa-logger';
2018-09-14 23:40:58 +03:00
const requestStats = require('request-stats');
2018-04-29 15:21:32 +03:00
//const slow = require('koa-slow');
2016-12-29 00:49:51 +02:00
import activityPub from './activitypub';
2018-04-01 08:12:07 +03:00
import webFinger from './webfinger';
2018-04-02 07:15:53 +03:00
import config from '../config';
2018-10-21 01:10:35 +03:00
import { coreChart } from '../services/stats';
2018-10-16 00:37:21 +03:00
import apiServer from './api';
2017-01-17 01:06:39 +02:00
2018-04-13 00:06:18 +03:00
// Init app
2018-04-12 18:51:55 +03:00
const app = new Koa();
app.proxy = true;
2017-11-13 12:58:29 +02:00
2018-04-19 12:03:46 +03:00
if (process.env.NODE_ENV != 'production') {
// Logger
app.use(logger());
2018-04-26 05:46:42 +03:00
// Delay
2018-04-29 11:17:15 +03:00
//app.use(slow({
// delay: 1000
//}));
2018-04-19 12:03:46 +03:00
}
// Compress response
2018-04-13 08:28:09 +03:00
app.use(compress({
flush: zlib.constants.Z_SYNC_FLUSH
}));
2018-04-12 18:51:55 +03:00
// HSTS
// 6months (15552000sec)
if (config.url.startsWith('https') && !config.disableHsts) {
2018-04-13 01:34:27 +03:00
app.use(async (ctx, next) => {
2018-04-12 18:51:55 +03:00
ctx.set('strict-transport-security', 'max-age=15552000; preload');
2018-04-13 01:34:27 +03:00
await next();
});
}
2018-10-16 00:37:21 +03:00
app.use(mount('/api', apiServer));
2018-04-13 00:17:14 +03:00
app.use(mount('/files', require('./file')));
2018-04-12 18:51:55 +03:00
// Init router
const router = new Router();
2018-04-12 18:51:55 +03:00
// Routing
router.use(activityPub.routes());
router.use(webFinger.routes());
2018-04-13 00:17:14 +03:00
2018-04-12 18:51:55 +03:00
// Register router
app.use(router.routes());
2016-12-29 00:49:51 +02:00
2018-04-13 01:34:27 +03:00
app.use(mount(require('./web')));
2018-03-28 19:20:40 +03:00
function createServer() {
2017-11-25 01:11:58 +02:00
if (config.https) {
2018-06-18 03:54:53 +03:00
const certs: any = {};
2017-11-25 01:11:58 +02:00
Object.keys(config.https).forEach(k => {
certs[k] = fs.readFileSync(config.https[k]);
});
2018-04-13 19:50:43 +03:00
certs['allowHTTP1'] = true;
2018-05-24 16:21:10 +03:00
return http2.createSecureServer(certs, app.callback());
2017-11-25 01:11:58 +02:00
} else {
return http.createServer(app.callback());
2017-11-25 01:11:58 +02:00
}
2018-03-28 19:20:40 +03:00
}
2016-12-29 00:49:51 +02:00
2018-03-28 19:20:40 +03:00
export default () => new Promise(resolve => {
const server = createServer();
2016-12-29 00:49:51 +02:00
2018-04-13 19:54:54 +03:00
// Init stream server
2018-03-28 19:20:40 +03:00
require('./api/streaming')(server);
2017-01-17 00:51:27 +02:00
2018-04-13 19:54:54 +03:00
// Listen
2018-03-28 19:20:40 +03:00
server.listen(config.port, resolve);
2018-09-14 23:40:58 +03:00
//#region Network stats
let queue: any[] = [];
requestStats(server, (stats: any) => {
if (stats.ok) {
queue.push(stats);
}
});
// Bulk write
setInterval(() => {
if (queue.length == 0) return;
const requests = queue.length;
const time = queue.reduce((a, b) => a + b.time, 0);
const incomingBytes = queue.reduce((a, b) => a + b.req.bytes, 0);
const outgoingBytes = queue.reduce((a, b) => a + b.res.bytes, 0);
queue = [];
2018-10-21 01:10:35 +03:00
coreChart.updateNetworkStats(requests, time, incomingBytes, outgoingBytes);
2018-09-14 23:40:58 +03:00
}, 5000);
//#endregion
2018-03-28 19:20:40 +03:00
});