2016-12-28 22:49:51 +00:00
|
|
|
/**
|
2016-12-29 12:00:22 +00:00
|
|
|
* Misskey Entory Point!
|
2016-12-28 22:49:51 +00:00
|
|
|
*/
|
|
|
|
|
|
|
|
Error.stackTraceLimit = Infinity;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Module dependencies
|
|
|
|
*/
|
|
|
|
import * as fs from 'fs';
|
|
|
|
import * as os from 'os';
|
|
|
|
import * as cluster from 'cluster';
|
|
|
|
const prominence = require('prominence');
|
2016-12-29 14:09:21 +00:00
|
|
|
import Logger from './utils/logger';
|
2016-12-28 22:49:51 +00:00
|
|
|
import * as chalk from 'chalk';
|
|
|
|
const git = require('git-last-commit');
|
|
|
|
const portUsed = require('tcp-port-used');
|
|
|
|
import ProgressBar from './utils/cli/progressbar';
|
|
|
|
import initdb from './db/mongodb';
|
|
|
|
import checkDependencies from './utils/check-dependencies';
|
|
|
|
|
|
|
|
// Init babel
|
|
|
|
require('babel-core/register');
|
|
|
|
require('babel-polyfill');
|
|
|
|
|
|
|
|
const env = process.env.NODE_ENV;
|
|
|
|
const IS_PRODUCTION = env === 'production';
|
|
|
|
const IS_DEBUG = !IS_PRODUCTION;
|
|
|
|
|
|
|
|
global.config = require('./config').default(`${__dirname}/../.config/config.yml`);
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Initialize state
|
|
|
|
*/
|
|
|
|
enum State {
|
|
|
|
success,
|
|
|
|
warn,
|
|
|
|
failed
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set process title
|
|
|
|
process.title = 'Misskey';
|
|
|
|
|
|
|
|
// Start app
|
|
|
|
main();
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Init proccess
|
|
|
|
*/
|
|
|
|
function main(): void {
|
|
|
|
// Master
|
|
|
|
if (cluster.isMaster) {
|
|
|
|
master();
|
|
|
|
} else { // Workers
|
|
|
|
worker();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Init master proccess
|
|
|
|
*/
|
|
|
|
async function master(): Promise<void> {
|
|
|
|
let state: State;
|
|
|
|
|
|
|
|
try {
|
|
|
|
// initialize app
|
|
|
|
state = await init();
|
|
|
|
} catch (e) {
|
|
|
|
console.error(e);
|
|
|
|
process.exit(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
switch (state) {
|
|
|
|
case State.failed:
|
2016-12-29 14:09:21 +00:00
|
|
|
Logger.error(chalk.red('Fatal error occurred :('));
|
2016-12-28 22:49:51 +00:00
|
|
|
process.exit();
|
|
|
|
return;
|
|
|
|
case State.warn:
|
2016-12-29 14:09:21 +00:00
|
|
|
Logger.warn(chalk.yellow('Some problem(s) :|'));
|
2016-12-28 22:49:51 +00:00
|
|
|
break;
|
|
|
|
case State.success:
|
2016-12-29 14:09:21 +00:00
|
|
|
Logger.info(chalk.green('OK :)'));
|
2016-12-28 22:49:51 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Spawn workers
|
|
|
|
spawn(() => {
|
2016-12-29 11:58:36 +00:00
|
|
|
console.log(chalk.bold.green(`\nMisskey is now running. [port:${config.port}]`));
|
2016-12-28 22:49:51 +00: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
|
|
|
|
*/
|
|
|
|
function worker(): void {
|
|
|
|
// 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
|
|
|
|
*/
|
|
|
|
async function init(): Promise<State> {
|
2016-12-29 11:39:46 +00:00
|
|
|
let warn = false;
|
2016-12-28 22:49:51 +00:00
|
|
|
|
2016-12-29 14:09:21 +00:00
|
|
|
Logger.info('Welcome to Misskey!');
|
|
|
|
Logger.info(chalk.bold('Misskey Core <aoi>'));
|
|
|
|
Logger.info('Initializing...');
|
2016-12-28 22:49:51 +00:00
|
|
|
|
|
|
|
// Get commit info
|
2016-12-29 14:09:21 +00:00
|
|
|
let lastCommitLogger = new Logger('LastCommit');
|
2016-12-28 22:49:51 +00:00
|
|
|
try {
|
|
|
|
const commit = await prominence(git).getLastCommit();
|
2016-12-29 11:03:34 +00:00
|
|
|
const shortHash: string = commit.shortHash;
|
|
|
|
const hash: string = commit.hash;
|
|
|
|
const commitDate = new Date(parseInt(commit.committedOn, 10) * 1000).toLocaleDateString('ja-JP');
|
|
|
|
const commitTime = new Date(parseInt(commit.committedOn, 10) * 1000).toLocaleTimeString('ja-JP');
|
2016-12-29 14:09:21 +00:00
|
|
|
lastCommitLogger.info(`${shortHash}${chalk.gray(hash.substr(shortHash.length))}`);
|
|
|
|
lastCommitLogger.info(`${commit.subject} ${chalk.green(`(${commitDate} ${commitTime})`)} ${chalk.blue(`<${commit.author.name}>`)}`);
|
2016-12-28 22:49:51 +00:00
|
|
|
} catch (e) {
|
2016-12-29 14:09:21 +00:00
|
|
|
lastCommitLogger.info('No commit information found')
|
2016-12-28 22:49:51 +00:00
|
|
|
}
|
|
|
|
|
2016-12-29 14:09:21 +00:00
|
|
|
let envLogger = new Logger('Env');
|
|
|
|
envLogger.info(typeof env == 'undefined' ? 'NODE_ENV is not set' : `NODE_ENV: ${env}`);
|
2016-12-28 22:49:51 +00:00
|
|
|
if (IS_DEBUG) {
|
2016-12-29 14:09:21 +00:00
|
|
|
envLogger.warn('The environment is not in production mode');
|
|
|
|
envLogger.warn('Do not use for production purpose');
|
2016-12-28 22:49:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Get machine info
|
|
|
|
const totalmem = (os.totalmem() / 1024 / 1024 / 1024).toFixed(1);
|
|
|
|
const freemem = (os.freemem() / 1024 / 1024 / 1024).toFixed(1);
|
2016-12-29 14:09:21 +00:00
|
|
|
let machineLogger = new Logger('Machine');
|
|
|
|
machineLogger.info(os.hostname());
|
|
|
|
machineLogger.info(`CPU: ${os.cpus().length}core`);
|
|
|
|
machineLogger.info(`MEM: ${totalmem}GB (available: ${freemem}GB)`);
|
2016-12-28 22:49:51 +00:00
|
|
|
|
|
|
|
if (!fs.existsSync(`${__dirname}/../.config/config.yml`)) {
|
2016-12-29 14:09:21 +00:00
|
|
|
Logger.error('Configuration not found');
|
2016-12-28 22:49:51 +00:00
|
|
|
return State.failed;
|
|
|
|
}
|
|
|
|
|
2016-12-29 14:09:21 +00:00
|
|
|
let configLogger = new Logger('Config');
|
|
|
|
configLogger.info('Successfully loaded');
|
|
|
|
configLogger.info(`maintainer: ${config.maintainer}`);
|
2016-12-28 22:49:51 +00:00
|
|
|
|
|
|
|
checkDependencies();
|
|
|
|
|
|
|
|
// Check if a port is being used
|
|
|
|
if (await portUsed.check(config.port)) {
|
2016-12-29 14:09:21 +00:00
|
|
|
Logger.error(`Port: ${config.port} is already used!`);
|
2016-12-28 22:49:51 +00:00
|
|
|
return State.failed;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Try to connect to MongoDB
|
2016-12-29 14:09:21 +00:00
|
|
|
let mongoDBLogger = new Logger('MongoDB');
|
2016-12-28 22:49:51 +00:00
|
|
|
try {
|
|
|
|
const db = await initdb(config);
|
2016-12-29 14:09:21 +00:00
|
|
|
mongoDBLogger.info('Successfully connected');
|
2016-12-28 22:49:51 +00:00
|
|
|
db.close();
|
|
|
|
} catch (e) {
|
2016-12-29 14:09:21 +00:00
|
|
|
mongoDBLogger.error(`${e}`);
|
2016-12-28 22:49:51 +00:00
|
|
|
return State.failed;
|
|
|
|
}
|
|
|
|
|
|
|
|
return warn ? State.warn : State.success;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Spawn workers
|
|
|
|
*/
|
|
|
|
function spawn(callback: any): void {
|
|
|
|
// 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', () => {
|
|
|
|
callback();
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
// Dying away...
|
|
|
|
process.on('exit', () => {
|
2016-12-29 16:12:49 +00:00
|
|
|
Logger.info('The process is going exit');
|
2016-12-28 22:49:51 +00:00
|
|
|
});
|