forked from BNDS-Programmers/Blog
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.js
142 lines (130 loc) · 3.69 KB
/
app.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
const Koa = require('koa')
const app = new Koa()
const views = require('koa-views')
const json = require('koa-json')
const onerror = require('koa-onerror')
const bodyparser = require('koa-bodyparser')
const logger = require('koa-logger')
const csrf = require('koa-csrf')
const session = require('koa-session')
const index = require('./routes/index')
const users = require('./routes/users')
const manage = require('./routes/manage')
global.config = require('./config')
// error handler
onerror(app)
// Config session
app.keys = [global.config.session_sec]
let sessionConf = {
secret: global.config.session_sec,
cookie: {},
rolling: false,
saveUninitialized: true,
resave: true
}
app.use(session(sessionConf, app));
// middlewares
app.use(bodyparser({
enableTypes:['json', 'form', 'text'],
jsonLimit: '50mb',
formLimit: '50mb',
}))
app.use(json())
app.use(logger())
app.use(new csrf({
invalidSessionSecretMessage: 'Invalid session secret',
invalidSessionSecretStatusCode: 403,
invalidTokenMessage: 'Invalid CSRF token',
invalidTokenStatusCode: 403,
excludedMethods: ['GET', 'HEAD', 'OPTIONS'],
disableQuery: false
}));
app.use(require('koa-static')(__dirname + '/static'))
app.use(views(__dirname + '/views', {
extension: 'ejs'
}))
// logger
app.use(async (ctx, next) => {
const start = new Date()
await next()
const ms = new Date() - start
console.log(`${ctx.method} ${ctx.url} - ${ms}ms from ${ctx.ip}`)
})
global.blog = {
log(msg){
console.log('IBlog => ' + msg)
},
db: null,
async connectDB(){
const Sequlize = require('sequelize');
this.db = new Sequlize(global.config.db.database, global.config.db.username, global.config.db.password, {
host: global.config.db.host,
dialect: global.config.db.dialect,
logging: global.config.mode === 'dev' ? this.log: null
});
global.Promise = Sequlize.Promise;
this.loadModel('article');
this.loadModel('user');
await this.db.sync();
},
loadModel(name){
return require(`./models/${name}`);
},
loadApp(name){
return require(`./app/${name}`);
},
loadModule(name){
return require(`./modules/${name}`);
},
init(){
this.connectDB();
},
run(){
app.listen(global.config.port, global.config.host, () => {
console.log("Blog is running @ port " + (global.config.port))
});
}
}
blog.init();
blog.run();
// error page
app.use(async (ctx, next) => {
try {
await next();
if (ctx.status === 404) {
ctx.throw(404);
}
} catch (err) {
let dict_render = blog.loadModule('user_agent_snap').response(ctx, 'error', 'Error', err.status);
console.error(err.stack);
dict_render.error = {};
const status = err.status || 500;
dict_render.error.status = err.status;
ctx.status = status;
if (status === 404) {
dict_render.error.message = '何もいませんです QwQ';
} else if (status === 500) {
dict_render.error.message = 'まじやばくね?Server Internal Error!';
} else if (status === 403) {
dict_render.error.message = '立入禁止!!出でいけ!';
}
await ctx.render("error", dict_render);
}
})
// routes
app.use(index.routes(), index.allowedMethods())
app.use(users.routes(), users.allowedMethods())
app.use(manage.routes(), manage.allowedMethods())
// error-handling
app.on('error', async (err, ctx) => {
console.error('server error', err, ctx);
let dict_render = blog.loadModule('user_agent_snap').response(ctx, 'error', 'Error', toString(err.status));
dict_render.error = {};
if(err.status === 404) {
dict_render.error.message = '何もいませんです';
} else if(err.status === 500) {
dict_render.error.message = 'Server Error QuQ!!!';
}
await ctx.render('error', dict_render);
});
module.exports = app