1
0
mirror of https://github.com/hotomoe/hotomoe synced 2024-12-05 10:18:09 +09:00
hotomoe/src/index.ts

196 lines
4.1 KiB
TypeScript
Raw Normal View History

2016-12-29 07:49:51 +09:00
/**
2016-12-29 21:00:22 +09:00
* Misskey Entory Point!
2016-12-29 07:49:51 +09:00
*/
Error.stackTraceLimit = Infinity;
/**
* Module dependencies
*/
import * as fs from 'fs';
import * as os from 'os';
import * as cluster from 'cluster';
2016-12-29 23:09:21 +09:00
import Logger from './utils/logger';
2016-12-29 07:49:51 +09:00
import * as chalk from 'chalk';
const portUsed = require('tcp-port-used');
2016-12-30 01:35:25 +09:00
const isRoot = require('is-root');
2016-12-29 07:49:51 +09:00
import ProgressBar from './utils/cli/progressbar';
import initdb from './db/mongodb';
2016-12-31 03:29:24 +09:00
import LastCommitInfo from './utils/lastCommitInfo';
2016-12-31 03:35:19 +09:00
import EnvironmentInfo from './utils/environmentInfo';
2016-12-31 03:19:59 +09:00
import MachineInfo from './utils/machineInfo';
import DependencyInfo from './utils/dependencyInfo';
2016-12-29 07:49:51 +09:00
// Init babel
require('babel-core/register');
require('babel-polyfill');
global.config = require('./config').default(`${__dirname}/../.config/config.yml`);
2017-01-03 05:15:01 +09:00
enum InitResult {
Success,
Warn,
Failure
2016-12-29 07:49:51 +09:00
}
process.title = 'Misskey';
// Start app
main();
/**
* Init proccess
*/
function main(): void {
if (cluster.isMaster) {
2017-01-03 05:17:21 +09:00
masterMain();
2017-01-03 05:15:50 +09:00
} else {
2017-01-03 05:17:21 +09:00
workerMain();
2016-12-29 07:49:51 +09:00
}
}
/**
* Init master proccess
*/
2017-01-03 05:17:21 +09:00
async function masterMain(): Promise<void> {
2017-01-03 05:15:01 +09:00
let initResult: InitResult;
2016-12-29 07:49:51 +09:00
try {
// initialize app
2017-01-03 05:15:01 +09:00
initResult = await init();
2016-12-29 07:49:51 +09:00
} catch (e) {
console.error(e);
process.exit(1);
}
2017-01-03 05:15:01 +09:00
switch (initResult) {
2017-01-03 05:15:27 +09:00
case InitResult.Success:
Logger.info(chalk.green('Successfully initialized :)'));
break;
case InitResult.Warn:
Logger.warn(chalk.yellow('Initialized with some problem(s) :|'));
break;
2017-01-03 05:15:01 +09:00
case InitResult.Failure:
2016-12-30 02:28:51 +09:00
Logger.error(chalk.red('Fatal error occurred during initializing :('));
2016-12-29 07:49:51 +09:00
process.exit();
return;
}
2017-01-03 05:18:03 +09:00
spawnWorkers(() => {
2016-12-30 02:26:37 +09:00
Logger.info(chalk.bold.green(`Now listening on port ${config.port}`));
2016-12-29 07:49:51 +09:00
// Listen new workers
cluster.on('fork', worker => {
console.log(`Process forked: [${worker.id}]`);
});
// Listen online workers
cluster.on('online', worker => {
console.log(`Process is now online: [${worker.id}]`);
});
// Listen for dying workers
cluster.on('exit', worker => {
// Replace the dead worker,
// we're not sentimental
console.log(chalk.red(`[${worker.id}] died :(`));
cluster.fork();
});
});
}
/**
* Init worker proccess
*/
2017-01-03 05:17:21 +09:00
function workerMain(): void {
2016-12-29 07:49:51 +09:00
// Register config
global.config = config;
// Init mongo
initdb().then(db => {
global.db = db;
// start server
require('./server');
}, err => {
console.error(err);
process.exit(0);
});
}
/**
* Init app
*/
2017-01-03 05:15:01 +09:00
async function init(): Promise<InitResult> {
2016-12-29 20:39:46 +09:00
let warn = false;
2016-12-29 07:49:51 +09:00
2016-12-29 23:09:21 +09:00
Logger.info('Welcome to Misskey!');
Logger.info(chalk.bold('Misskey Core <aoi>'));
Logger.info('Initializing...');
2016-12-29 07:49:51 +09:00
2016-12-31 03:29:24 +09:00
await LastCommitInfo.show();
2016-12-31 03:35:19 +09:00
EnvironmentInfo.show();
2016-12-31 03:19:59 +09:00
MachineInfo.show();
new DependencyInfo().showAll();
2016-12-31 03:14:38 +09:00
2016-12-31 02:54:08 +09:00
let configLogger = new Logger('Config');
2016-12-29 07:49:51 +09:00
if (!fs.existsSync(`${__dirname}/../.config/config.yml`)) {
2016-12-31 02:54:08 +09:00
configLogger.error('Configuration not found');
2017-01-03 05:15:01 +09:00
return InitResult.Failure;
2016-12-29 07:49:51 +09:00
}
2016-12-29 23:09:21 +09:00
configLogger.info('Successfully loaded');
configLogger.info(`maintainer: ${config.maintainer}`);
2016-12-29 07:49:51 +09:00
2016-12-30 01:35:25 +09:00
if (process.platform === 'linux' && !isRoot() && config.port < 1024) {
2016-12-30 08:23:03 +09:00
Logger.error('You need root privileges to listen on port below 1024 on Linux');
2017-01-03 05:15:01 +09:00
return InitResult.Failure;
2016-12-30 01:35:25 +09:00
}
2016-12-29 07:49:51 +09:00
// Check if a port is being used
if (await portUsed.check(config.port)) {
2016-12-30 02:56:37 +09:00
Logger.error(`Port ${config.port} is already used`);
2017-01-03 05:15:01 +09:00
return InitResult.Failure;
2016-12-29 07:49:51 +09:00
}
// Try to connect to MongoDB
2016-12-29 23:09:21 +09:00
let mongoDBLogger = new Logger('MongoDB');
2016-12-29 07:49:51 +09:00
try {
const db = await initdb(config);
2016-12-29 23:09:21 +09:00
mongoDBLogger.info('Successfully connected');
2016-12-29 07:49:51 +09:00
db.close();
} catch (e) {
2017-01-03 05:31:08 +09:00
mongoDBLogger.error(e);
2017-01-03 05:15:01 +09:00
return InitResult.Failure;
2016-12-29 07:49:51 +09:00
}
2017-01-03 05:15:01 +09:00
return warn ? InitResult.Warn : InitResult.Success;
2016-12-29 07:49:51 +09:00
}
2017-01-03 05:22:25 +09:00
function spawnWorkers(onComplete: any): void {
2016-12-29 07:49:51 +09: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
progress.on('complete', () => {
2017-01-03 05:22:25 +09:00
onComplete();
2016-12-29 07:49:51 +09:00
});
}
// Dying away...
process.on('exit', () => {
2016-12-30 01:12:49 +09:00
Logger.info('The process is going exit');
2016-12-29 07:49:51 +09:00
});