forked from FoundKeyGang/FoundKey
101 lines
2 KiB
TypeScript
101 lines
2 KiB
TypeScript
/**
|
|
* Core Server
|
|
*/
|
|
|
|
import * as fs from 'fs';
|
|
import * as http from 'http';
|
|
import * as https from 'https';
|
|
import * as cluster from 'cluster';
|
|
import * as express from 'express';
|
|
import * as morgan from 'morgan';
|
|
import Accesses from 'accesses';
|
|
import vhost = require('vhost');
|
|
|
|
import log from './log-request';
|
|
import config from './conf';
|
|
|
|
function extractHostname(host) {
|
|
const index = host.indexOf(':');
|
|
return index < 0 ? host : host.substr(0, index);
|
|
}
|
|
|
|
/**
|
|
* Init app
|
|
*/
|
|
const app = express();
|
|
app.disable('x-powered-by');
|
|
app.set('trust proxy', 'loopback');
|
|
|
|
// Log
|
|
if (config.accesses && config.accesses.enable) {
|
|
const accesses = new Accesses({
|
|
appName: 'Misskey',
|
|
port: config.accesses.port
|
|
});
|
|
|
|
app.use(accesses.express);
|
|
}
|
|
|
|
app.use(morgan(process.env.NODE_ENV == 'production' ? 'combined' : 'dev', {
|
|
// create a write stream (in append mode)
|
|
stream: config.accesslog ? fs.createWriteStream(config.accesslog) : null
|
|
}));
|
|
|
|
app.use((req, res, next) => {
|
|
log(req);
|
|
next();
|
|
});
|
|
|
|
// Drop request when without 'Host' header
|
|
app.use((req, res, next) => {
|
|
if (!req.headers['host']) {
|
|
res.sendStatus(400);
|
|
} else {
|
|
next();
|
|
}
|
|
});
|
|
|
|
/**
|
|
* Register modules
|
|
*/
|
|
const hostname = extractHostname(config.host);
|
|
const secondaryHostname = extractHostname(config.secondary_host);
|
|
app.use(vhost(`api.${hostname}`, require('./api/server')));
|
|
app.use(vhost(secondaryHostname, require('./himasaku/server')));
|
|
app.use(vhost(`file.${secondaryHostname}`, require('./file/server')));
|
|
app.use(require('./web/server'));
|
|
|
|
/**
|
|
* Create server
|
|
*/
|
|
const server = (() => {
|
|
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);
|
|
}
|
|
})();
|
|
|
|
/**
|
|
* Steaming
|
|
*/
|
|
require('./api/streaming')(server);
|
|
|
|
/**
|
|
* Server listen
|
|
*/
|
|
server.listen(config.port, () => {
|
|
if (cluster.isWorker) {
|
|
// Send a 'ready' message to parent process
|
|
process.send('ready');
|
|
}
|
|
});
|
|
|
|
/**
|
|
* Export app for testing
|
|
*/
|
|
module.exports = app;
|