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

223 lines
5.0 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';
const prominence = require('prominence');
import { log } from './utils/logger';
2016-12-29 07:49:51 +09: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 19:06:41 +09:00
log('Error', chalk.red('Fatal error occurred :('));
2016-12-29 07:49:51 +09:00
process.exit();
return;
case State.warn:
2016-12-29 19:06:41 +09:00
log('Warn', chalk.yellow('Some problem(s) :|'));
2016-12-29 07:49:51 +09:00
break;
case State.success:
2016-12-29 19:06:41 +09:00
log('Info', chalk.green('OK :)'));
2016-12-29 07:49:51 +09:00
break;
}
// Spawn workers
spawn(() => {
2016-12-29 20:58:36 +09:00
console.log(chalk.bold.green(`\nMisskey is now running. [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
*/
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 20:39:46 +09:00
let warn = false;
2016-12-29 07:49:51 +09:00
2016-12-29 20:39:46 +09:00
log('Info', 'Welcome to Misskey!');
2016-12-29 17:50:57 +09:00
log('Info', chalk.bold('Misskey Core <aoi>'));
2016-12-29 20:39:46 +09:00
log('Info', 'Initializing...');
2016-12-29 07:49:51 +09:00
// Get commit info
try {
const commit = await prominence(git).getLastCommit();
2016-12-29 20:03:34 +09: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');
log('Info', `${shortHash}${chalk.gray(hash.substr(shortHash.length))}`, 'LastCommit');
log('Info', `${commit.subject} ${chalk.green(`(${commitDate} ${commitTime})`)} ${chalk.blue(`<${commit.author.name}>`)}`, 'LastCommit');
2016-12-29 07:49:51 +09:00
} catch (e) {
2016-12-29 20:03:34 +09:00
log('Info', `No commit information found`, 'LastCommit');
2016-12-29 07:49:51 +09:00
}
2016-12-29 20:39:46 +09:00
log('Info', typeof env == 'undefined' ? 'NODE_ENV is not set' : `NODE_ENV: ${env}`, 'Env');
2016-12-29 07:49:51 +09:00
if (IS_DEBUG) {
2016-12-29 20:39:46 +09:00
log('Warn', 'The environment is not in production mode', 'Env');
2016-12-29 20:45:50 +09:00
log('Warn', 'Do not use for production purpose', 'Env');
2016-12-29 07:49:51 +09: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 20:07:05 +09:00
log('Info', `${os.hostname()}`, 'Machine');
log('Info', `CPU: ${os.cpus().length}core`, 'Machine');
log('Info', `MEM: ${totalmem}GB (available: ${freemem}GB)`, 'Machine');
2016-12-29 07:49:51 +09:00
if (!fs.existsSync(`${__dirname}/../.config/config.yml`)) {
log('Error', 'Configuration not found');
2016-12-29 07:49:51 +09:00
return State.failed;
}
2016-12-29 20:14:51 +09:00
log('Info', 'Successfully loaded', 'Config');
log('Info', `maintainer: ${config.maintainer}`, 'Config');
2016-12-29 07:49:51 +09:00
checkDependencies();
// Check if a port is being used
if (await portUsed.check(config.port)) {
log('Error', `Port: ${config.port} is already used!`);
2016-12-29 07:49:51 +09:00
return State.failed;
}
// Try to connect to MongoDB
try {
const db = await initdb(config);
2016-12-29 20:22:54 +09:00
log('Info', 'Successfully connected', 'MongoDB');
2016-12-29 07:49:51 +09:00
db.close();
} catch (e) {
2016-12-29 20:22:54 +09:00
log('Error', `${e}`, 'MongoDB');
2016-12-29 07:49:51 +09: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 20:26:05 +09:00
log('Info', 'Misskey is going down');
2016-12-29 07:49:51 +09:00
});