123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139 |
- const fs = require('fs-extra');
- const path = require('path');
- const express = require('express');
- const multer = require('multer');
- const ReaderWorker = require('./core/Reader/ReaderWorker');//singleton
- const log = new (require('./core/AppLogger'))().log;//singleton
- const c = require('./controllers');
- const utils = require('./core/utils');
- function initRoutes(app, wss, config) {
- //эксклюзив для update_checker
- if (config.mode === 'book_update_checker') {
- new c.BookUpdateCheckerController(wss, config);
- return;
- }
- initStatic(app, config);
-
- const misc = new c.MiscController(config);
- const reader = new c.ReaderController(config);
- const worker = new c.WorkerController(config);
- new c.WebSocketController(wss, config);
- //access
- const [aAll, aNormal, aSite, aReader, aOmnireader] = // eslint-disable-line no-unused-vars
- [config.mode, 'normal', 'site', 'reader', 'omnireader'];
- //multer
- const storage = multer.diskStorage({
- destination: (req, file, cb) => {
- cb(null, config.uploadDir);
- },
- filename: (req, file, cb) => {
- cb(null, utils.randomHexString(30));
- }
- });
- const upload = multer({ storage, limits: {fileSize: config.maxUploadFileSize} });
- //routes
- const routes = [
- ['POST', '/api/config', misc.getConfig.bind(misc), [aAll], {}],
- ['POST', '/api/reader/load-book', reader.loadBook.bind(reader), [aAll], {}],
- ['POST', '/api/reader/storage', reader.storage.bind(reader), [aAll], {}],
- ['POST', '/api/reader/upload-file', [upload.single('file'), reader.uploadFile.bind(reader)], [aAll], {}],
- ['POST', '/api/worker/get-state', worker.getState.bind(worker), [aAll], {}],
- ];
- //to app
- for (let route of routes) {
- let callbacks = [];
- let [httpMethod, path, controllers, access, options] = route;
- let controller = controllers;
- if (Array.isArray(controllers)) {
- controller = controllers[controllers.length - 1];
- callbacks = controllers.slice(0, -1);
- }
- access = new Set(access);
- let callback = () => {};
- if (access.has(config.mode)) {//allowed
- callback = async function(req, res) {
- try {
- const result = await controller(req, res, options);
- if (result !== false)
- res.send(result);
- } catch (e) {
- res.status(500).send({error: e.message});
- }
- };
- } else {//forbidden
- callback = async function(req, res) {
- res.status(403);
- };
- }
- callbacks.push(callback);
- switch (httpMethod) {
- case 'GET' :
- app.get(path, ...callbacks);
- break;
- case 'POST':
- app.post(path, ...callbacks);
- break;
- default:
- throw new Error(`initRoutes error: unknown httpMethod: ${httpMethod}`);
- }
- }
- }
- function initStatic(app, config) {
- const readerWorker = new ReaderWorker(config);
- //восстановление файлов в /tmp и /upload из webdav-storage, при необходимости
- app.use(async(req, res, next) => {
- if ((req.method !== 'GET' && req.method !== 'HEAD') ||
- !(req.path.indexOf('/tmp/') === 0 || req.path.indexOf('/upload/') === 0)
- ) {
- return next();
- }
- const filePath = `${config.publicDir}${req.path}`;
- //восстановим
- try {
- if (!await fs.pathExists(filePath)) {
- if (req.path.indexOf('/tmp/') === 0) {
- await readerWorker.restoreRemoteFile(req.path, '/tmp');
- } else if (req.path.indexOf('/upload/') === 0) {
- await readerWorker.restoreRemoteFile(req.path, '/upload');
- }
- }
- } catch(e) {
- log(LM_ERR, `Static.restoreRemoteFile: ${e.message}`);
- }
- return next();
- });
- const tmpDir = `${config.publicDir}/tmp`;
- app.use(express.static(config.publicDir, {
- maxAge: '30d',
- setHeaders: (res, filePath) => {
- if (path.dirname(filePath) == tmpDir) {
- res.set('Content-Type', 'application/xml');
- res.set('Content-Encoding', 'gzip');
- }
- },
- }));
- }
- module.exports = {
- initRoutes
- }
|