Files
iblog2/app.js
2019-06-23 17:52:14 +08:00

148 lines
3.8 KiB
JavaScript

const express = require('express');
const fs = require('fs');
const path = require('path');
const favicon = require('serve-favicon');
const cookieParser = require('cookie-parser');
const session = require('express-session');
const bodyParser = require('body-parser');
const helmet = require('helmet');
const route = require('./routes/index');
const blog = require('./routes/blog');
const misc = require('./routes/misc');
const auth = require('./routes/auth');
const admin = require('./routes/admin');
const locale = require('./routes/locale');
const ue = require('./routes/ue');
const tool = require('./utility/tool');
const passport = require('passport');
const i18n = require('./models/i18n');
const serverlog = require('serverlog-node');
serverlog.config({
extension: {
enable: true,
key: 'iblog2_server_log_key'
}
});
const logger = serverlog.getLogger('app');
const log = require('./proxy/log');
const app = express();
/**
* 记录未捕获异常
*/
process.on('uncaughtException', err => {
log.store('Error', err);
logger.error(err);
});
/**
* 记录未处理的Promise失败
*/
process.on('unhandledRejection', reason => {
log.store('Error', reason);
logger.error(reason);
});
// 设置模板引擎
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'pug');
// 在模板页面可以使用 staticPrefix 变量以区别本地开发环境和线上环境引入不同的资源文件
app.locals.staticPrefix = app.get('env') === 'production' ? '/static/dist' : '/static/src';
// 增加安全性头部
app.use(helmet());
// 注册 ServerLog 中间件以注入 req 对象给日志
app.use(serverlog.middleware());
// 记录所有请求
app.use((req, res, next) => {
logger.info(`${req.method.toUpperCase()} ${tool.getFullUrl(req)}`);
next();
});
// 网站 Icon
app.use(favicon(path.join(__dirname, 'public', 'favicon.ico')));
// parse body
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: false }));
// parse cookie
app.use(cookieParser());
// 多语言
app.use(i18n.init);
// 设置 Session
app.use(session({
secret: 'iblog-exp-session',
cookie: {
maxAge: 24 * 60 * 60 * 1000
},
resave: false,
saveUninitialized: false
}));
app.use(passport.initialize());
app.use(passport.session());
// 静态文件
app.use('/static', express.static(path.join(__dirname, 'public')));
app.use('/uploads', express.static(path.join(__dirname, 'public', 'uploads')));
app.use('/nodeModules', express.static(path.join(__dirname, 'node_modules')));
// Service Worker
app.get('/sw.js', (req, res) => {
fs.readFile(path.resolve(__dirname, './sw.js'), (err, data) => {
if (err) {
logger.error(err);
throw err;
}
res.writeHead(200, {
'Content-type': 'text/javascript'
});
res.end(data);
});
});
// 前台站点路由,无需登录
app.use('/', route);
app.use('/', locale);
app.use('/', misc);
app.use('/', auth);
app.use('/blog', blog);
app.use('/ue/controller', ue);
// 后台站点路由,需要身份验证
app.use('/admin', require('connect-ensure-login')
.ensureLoggedIn('/login'), admin);
// 捕获 404
app.use((req, res) => {
const err = new Error(`Not Found! URL: ${req.originalUrl}`);
err.status = 404;
log.store('Warn', err);
logger.warn(`Not Found! URL: ${req.originalUrl}`);
res.status(404)
.render('./shared/error', {
code: 404,
message: res.__('error.404_1')
});
});
// 捕获 500
app.use((err, req, res) => {
const code = err.status || 500;
err.status = code;
log.store('Error', err);
logger.error(err);
res.status(code)
.render('./shared/error', {
code,
message: res.__('error.404_2')
});
});
module.exports = app;