193 lines
4 KiB
TypeScript
193 lines
4 KiB
TypeScript
/**
|
|
* Misskey Entry Point!
|
|
*/
|
|
|
|
Error.stackTraceLimit = Infinity;
|
|
|
|
/**
|
|
* Module dependencies
|
|
*/
|
|
import * as fs from 'fs';
|
|
import * as os from 'os';
|
|
import * as cluster from 'cluster';
|
|
import * as debug from 'debug';
|
|
import Logger from './utils/logger';
|
|
import * as chalk from 'chalk';
|
|
import portUsed = require('tcp-port-used');
|
|
import isRoot = require('is-root');
|
|
import ProgressBar from './utils/cli/progressbar';
|
|
import LastCommitInfo from './utils/lastCommitInfo';
|
|
import EnvironmentInfo from './utils/environmentInfo';
|
|
import MachineInfo from './utils/machineInfo';
|
|
import DependencyInfo from './utils/dependencyInfo';
|
|
|
|
import { path as configPath } from './config';
|
|
import loadConfig from './config';
|
|
|
|
const clusterLog = debug('misskey:cluster');
|
|
|
|
// Init babel
|
|
require('babel-core/register');
|
|
require('babel-polyfill');
|
|
|
|
enum InitResult {
|
|
Success,
|
|
Warn,
|
|
Failure
|
|
}
|
|
|
|
process.title = 'Misskey';
|
|
|
|
// Start app
|
|
main();
|
|
|
|
/**
|
|
* Init proccess
|
|
*/
|
|
function main() {
|
|
if (cluster.isMaster) {
|
|
masterMain();
|
|
} else {
|
|
workerMain();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Init master proccess
|
|
*/
|
|
async function masterMain() {
|
|
let initResult: InitResult;
|
|
|
|
try {
|
|
// initialize app
|
|
initResult = await init();
|
|
} catch (e) {
|
|
console.error(e);
|
|
process.exit(1);
|
|
}
|
|
|
|
switch (initResult) {
|
|
case InitResult.Success:
|
|
Logger.info(chalk.green('Successfully initialized :)'));
|
|
break;
|
|
case InitResult.Warn:
|
|
Logger.warn(chalk.yellow('Initialized with some problem(s) :|'));
|
|
break;
|
|
case InitResult.Failure:
|
|
Logger.error(chalk.red('Fatal error occurred during initializing :('));
|
|
process.exit();
|
|
return;
|
|
}
|
|
|
|
spawnWorkers(() => {
|
|
Logger.info(chalk.bold.green(`Now listening on port ${loadConfig().port}`));
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Init worker proccess
|
|
*/
|
|
function workerMain() {
|
|
// start server
|
|
require('./server');
|
|
}
|
|
|
|
/**
|
|
* Init app
|
|
*/
|
|
async function init() {
|
|
let warn = false;
|
|
|
|
Logger.info('Welcome to Misskey!');
|
|
Logger.info(chalk.bold('Misskey <aoi>'));
|
|
Logger.info('Initializing...');
|
|
|
|
await LastCommitInfo.show();
|
|
EnvironmentInfo.show();
|
|
MachineInfo.show();
|
|
new DependencyInfo().showAll();
|
|
|
|
let configLogger = new Logger('Config');
|
|
if (!fs.existsSync(configPath)) {
|
|
configLogger.error('Configuration not found');
|
|
return InitResult.Failure;
|
|
}
|
|
|
|
const config = loadConfig();
|
|
|
|
configLogger.info('Successfully loaded');
|
|
configLogger.info(`maintainer: ${config.maintainer}`);
|
|
|
|
if (process.platform === 'linux' && !isRoot() && config.port < 1024) {
|
|
Logger.error('You need root privileges to listen on port below 1024 on Linux');
|
|
return InitResult.Failure;
|
|
}
|
|
|
|
// Check if a port is being used
|
|
if (await portUsed.check(config.port)) {
|
|
Logger.error(`Port ${config.port} is already used`);
|
|
return InitResult.Failure;
|
|
}
|
|
|
|
// Try to connect to MongoDB
|
|
let mongoDBLogger = new Logger('MongoDB');
|
|
try {
|
|
const db = require('./db/mongodb').default;
|
|
mongoDBLogger.info('Successfully connected');
|
|
db.close();
|
|
} catch (e) {
|
|
mongoDBLogger.error(e);
|
|
return InitResult.Failure;
|
|
}
|
|
|
|
return warn ? InitResult.Warn : InitResult.Success;
|
|
}
|
|
|
|
function spawnWorkers(onComplete: any) {
|
|
// 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
|
|
progress.on('complete', () => {
|
|
onComplete();
|
|
});
|
|
}
|
|
|
|
// 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);
|
|
|
|
// Dying away...
|
|
process.on('exit', () => {
|
|
Logger.info('The process is going exit');
|
|
});
|