mirror of
https://github.com/DIYgod/RSSHub.git
synced 2025-12-04 02:58:08 +08:00
feat: support package
This commit is contained in:
91
lib/index.js
91
lib/index.js
@@ -1,95 +1,16 @@
|
||||
require('dotenv').config();
|
||||
require('module-alias/register');
|
||||
const app = require('./app');
|
||||
const config = require('./config');
|
||||
const Koa = require('koa');
|
||||
const fs = require('fs');
|
||||
const logger = require('./utils/logger');
|
||||
require('./utils/request-wrapper');
|
||||
|
||||
const onerror = require('./middleware/onerror');
|
||||
const header = require('./middleware/header');
|
||||
const utf8 = require('./middleware/utf8');
|
||||
const cache = require('./middleware/cache');
|
||||
const parameter = require('./middleware/parameter');
|
||||
const template = require('./middleware/template');
|
||||
const favicon = require('koa-favicon');
|
||||
const debug = require('./middleware/debug');
|
||||
const accessControl = require('./middleware/access-control');
|
||||
|
||||
const router = require('./router');
|
||||
const protected_router = require('./protected_router');
|
||||
const mount = require('koa-mount');
|
||||
|
||||
// API related
|
||||
|
||||
const apiTemplate = require('./middleware/api-template');
|
||||
const api_router = require('./api_router');
|
||||
const apiResponseHandler = require('./middleware/api-response-handler');
|
||||
|
||||
const cluster = require('cluster');
|
||||
const numCPUs = require('os').cpus().length;
|
||||
|
||||
if (cluster.isMaster && config.enableCluster && process.env.NODE_ENV !== 'test' && process.env.NODE_ENV !== 'dev') {
|
||||
if (config.enableCluster && cluster.isMaster && process.env.NODE_ENV !== 'test' && process.env.NODE_ENV !== 'dev') {
|
||||
for (let i = 0; i < numCPUs; i++) {
|
||||
cluster.fork();
|
||||
}
|
||||
} else {
|
||||
process.on('uncaughtException', (e) => {
|
||||
logger.error('uncaughtException: ' + e);
|
||||
});
|
||||
|
||||
logger.info('🎉 RSSHub start! Cheers!');
|
||||
|
||||
const app = new Koa();
|
||||
app.proxy = true;
|
||||
|
||||
// favicon
|
||||
app.use(favicon(__dirname + '/favicon.png'));
|
||||
|
||||
// global error handing
|
||||
app.use(onerror);
|
||||
|
||||
// 1 set header
|
||||
app.use(header);
|
||||
|
||||
app.use(accessControl);
|
||||
|
||||
// 6 debug
|
||||
app.context.debug = {
|
||||
hitCache: 0,
|
||||
request: 0,
|
||||
paths: [],
|
||||
routes: [],
|
||||
ips: [],
|
||||
errorPaths: [],
|
||||
errorRoutes: [],
|
||||
};
|
||||
app.use(debug);
|
||||
|
||||
// 5 fix incorrect `utf-8` characters
|
||||
app.use(utf8);
|
||||
|
||||
app.use(apiTemplate);
|
||||
app.use(apiResponseHandler());
|
||||
|
||||
// 4 generate body
|
||||
app.use(template);
|
||||
// 3 filter content
|
||||
app.use(parameter);
|
||||
|
||||
// 2 cache
|
||||
app.use(cache(app));
|
||||
|
||||
// router
|
||||
app.use(mount('/', router.routes())).use(router.allowedMethods());
|
||||
|
||||
// routes the require authentication
|
||||
app.use(mount('/protected', protected_router.routes())).use(protected_router.allowedMethods());
|
||||
|
||||
// API router
|
||||
app.use(mount('/api', api_router.routes())).use(api_router.allowedMethods());
|
||||
|
||||
// connect
|
||||
let server;
|
||||
if (config.connect.socket) {
|
||||
if (fs.existsSync(config.connect.socket)) {
|
||||
@@ -107,9 +28,7 @@ if (cluster.isMaster && config.enableCluster && process.env.NODE_ENV !== 'test'
|
||||
logger.info('Listening Port ' + config.connect.port);
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
server: server,
|
||||
app: app,
|
||||
cache: app.context.cache.client,
|
||||
};
|
||||
logger.info('🎉 RSSHub start! Cheers!');
|
||||
|
||||
module.exports = server;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user