misskey/src/index.ts

169 lines
3.5 KiB
TypeScript
Raw Normal View History

2016-12-28 23:49:51 +01:00
/**
2017-01-02 22:09:17 +01:00
* Misskey Entry Point!
2016-12-28 23:49:51 +01:00
*/
Error.stackTraceLimit = Infinity;
import * as fs from 'fs';
import * as os from 'os';
import * as cluster from 'cluster';
2017-01-24 02:28:14 +01:00
import * as debug from 'debug';
2016-12-28 23:49:51 +01:00
import * as chalk from 'chalk';
2017-04-14 13:45:37 +02:00
// import portUsed = require('tcp-port-used');
2017-01-02 21:49:37 +01:00
import isRoot = require('is-root');
2017-04-05 02:58:29 +02:00
import { master } from 'accesses';
import Logger from './utils/logger';
2016-12-28 23:49:51 +01:00
import ProgressBar from './utils/cli/progressbar';
2016-12-30 19:35:19 +01:00
import EnvironmentInfo from './utils/environmentInfo';
2016-12-30 19:19:59 +01:00
import MachineInfo from './utils/machineInfo';
import DependencyInfo from './utils/dependencyInfo';
2016-12-28 23:49:51 +01:00
2017-04-05 02:58:29 +02:00
import { Config, path as configPath } from './config';
2017-01-17 00:19:34 +01:00
import loadConfig from './config';
2017-01-24 02:28:14 +01:00
const clusterLog = debug('misskey:cluster');
2016-12-28 23:49:51 +01:00
process.title = 'Misskey';
// Start app
main();
/**
2017-02-27 08:11:49 +01:00
* Init process
2016-12-28 23:49:51 +01:00
*/
2017-01-24 02:28:14 +01:00
function main() {
2016-12-28 23:49:51 +01:00
if (cluster.isMaster) {
2017-01-02 21:17:21 +01:00
masterMain();
2017-01-02 21:15:50 +01:00
} else {
2017-01-02 21:17:21 +01:00
workerMain();
2016-12-28 23:49:51 +01:00
}
}
/**
2017-02-27 08:11:49 +01:00
* Init master process
2016-12-28 23:49:51 +01:00
*/
2017-01-24 02:28:14 +01:00
async function masterMain() {
2017-04-05 02:58:29 +02:00
let config: Config;
2016-12-28 23:49:51 +01:00
try {
// initialize app
2017-04-05 02:58:29 +02:00
config = await init();
2016-12-28 23:49:51 +01:00
} catch (e) {
console.error(e);
2017-04-05 02:58:29 +02:00
Logger.error(chalk.red('Fatal error occurred during initializing :('));
2017-04-23 08:40:13 +02:00
process.exit(1);
2017-04-05 02:58:29 +02:00
}
Logger.info(chalk.green('Successfully initialized :)'));
// Init accesses
if (config.accesses && config.accesses.enable) {
master();
2016-12-28 23:49:51 +01:00
}
2017-01-02 21:18:03 +01:00
spawnWorkers(() => {
2017-04-23 08:40:13 +02:00
Logger.info(chalk.bold.green(`Now listening on port ${config.port}`));
2016-12-28 23:49:51 +01:00
});
}
/**
2017-02-27 08:11:49 +01:00
* Init worker process
2016-12-28 23:49:51 +01:00
*/
2017-01-24 02:28:14 +01:00
function workerMain() {
2017-01-17 01:12:33 +01:00
// start server
require('./server');
2016-12-28 23:49:51 +01:00
}
/**
* Init app
*/
2017-04-05 02:58:29 +02:00
async function init(): Promise<Config> {
2016-12-29 15:09:21 +01:00
Logger.info('Welcome to Misskey!');
2017-01-19 01:53:29 +01:00
Logger.info(chalk.bold('Misskey <aoi>'));
2016-12-29 15:09:21 +01:00
Logger.info('Initializing...');
2016-12-28 23:49:51 +01:00
2016-12-30 19:35:19 +01:00
EnvironmentInfo.show();
2016-12-30 19:19:59 +01:00
MachineInfo.show();
new DependencyInfo().showAll();
2016-12-30 19:14:38 +01:00
2016-12-30 18:54:08 +01:00
let configLogger = new Logger('Config');
2017-01-17 00:19:34 +01:00
if (!fs.existsSync(configPath)) {
2017-04-23 08:40:13 +02:00
throw 'Configuration not found';
2016-12-28 23:49:51 +01:00
}
2017-01-17 00:19:34 +01:00
const config = loadConfig();
2016-12-29 15:09:21 +01:00
configLogger.info('Successfully loaded');
configLogger.info(`maintainer: ${config.maintainer}`);
2016-12-28 23:49:51 +01:00
2016-12-29 17:35:25 +01:00
if (process.platform === 'linux' && !isRoot() && config.port < 1024) {
2017-04-23 08:40:13 +02:00
throw 'You need root privileges to listen on port below 1024 on Linux';
2016-12-29 17:35:25 +01:00
}
2016-12-28 23:49:51 +01:00
// Check if a port is being used
2017-03-13 10:22:58 +01:00
/* https://github.com/stdarg/tcp-port-used/issues/3
2016-12-28 23:49:51 +01:00
if (await portUsed.check(config.port)) {
2017-04-23 08:40:13 +02:00
throw `Port ${config.port} is already used`;
2016-12-28 23:49:51 +01:00
}
2017-03-13 10:22:58 +01:00
*/
2016-12-28 23:49:51 +01:00
// Try to connect to MongoDB
2016-12-29 15:09:21 +01:00
let mongoDBLogger = new Logger('MongoDB');
2016-12-28 23:49:51 +01:00
try {
2017-01-17 01:12:33 +01:00
const db = require('./db/mongodb').default;
2016-12-29 15:09:21 +01:00
mongoDBLogger.info('Successfully connected');
2016-12-28 23:49:51 +01:00
db.close();
} catch (e) {
2017-04-23 08:40:13 +02:00
throw e;
2016-12-28 23:49:51 +01:00
}
2017-04-05 02:58:29 +02:00
return config;
2016-12-28 23:49:51 +01:00
}
2017-04-23 08:55:37 +02:00
function spawnWorkers(onComplete: Function) {
2016-12-28 23:49:51 +01:00
// Count the machine's CPUs
const cpuCount = os.cpus().length;
const progress = new ProgressBar(cpuCount, 'Starting workers');
// Create a worker for each CPU
for (let i = 0; i < cpuCount; i++) {
const worker = cluster.fork();
worker.on('message', message => {
if (message === 'ready') {
progress.increment();
}
});
}
// On all workers started
2017-04-23 08:49:17 +02:00
progress.on('complete', onComplete);
2016-12-28 23:49:51 +01:00
}
2017-01-24 02:28:14 +01:00
// Listen new workers
cluster.on('fork', worker => {
clusterLog(`Process forked: [${worker.id}]`);
});
// Listen online workers
cluster.on('online', worker => {
clusterLog(`Process is now online: [${worker.id}]`);
});
// Listen for dying workers
cluster.on('exit', worker => {
// Replace the dead worker,
// we're not sentimental
clusterLog(chalk.red(`[${worker.id}] died :(`));
cluster.fork();
});
// Display detail of unhandled promise rejection
process.on('unhandledRejection', console.dir);
2016-12-28 23:49:51 +01:00
// Dying away...
process.on('exit', () => {
2016-12-29 17:12:49 +01:00
Logger.info('The process is going exit');
2016-12-28 23:49:51 +01:00
});