forked from lincenying/mmf-blog-vue2-ssr
-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.js
174 lines (154 loc) · 5.62 KB
/
server.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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
process.env.VUE_ENV = 'server'
const isProd = process.env.NODE_ENV === 'production'
const fs = require('fs')
const path = require('path')
const favicon = require('serve-favicon')
const express = require('express')
const compression = require('compression')
// const HTMLStream = require('vue-ssr-html-stream')
const logger = require('morgan')
const cookieParser = require('cookie-parser')
const bodyParser = require('body-parser')
const { createBundleRenderer } = require('vue-server-renderer')
const config = require('./src/api/config-server')
const resolve = file => path.resolve(__dirname, file)
const serverInfo =
`express/${require('express/package.json').version} ` +
`vue-server-renderer/${require('vue-server-renderer/package.json').version}`
// 引入 mongoose 相关模型
require('./server/models/admin')
require('./server/models/article')
require('./server/models/category')
require('./server/models/comment')
// require('./server/models/like')
require('./server/models/user')
// 引入 api 路由
const routes = require('./server/routes/index')
function createRenderer(bundle, template) {
// https://github.com/vuejs/vue/blob/dev/packages/vue-server-renderer/README.md#why-use-bundlerenderer
return createBundleRenderer(bundle, {
template,
cache: require('lru-cache')({
max: 1000,
maxAge: 1000 * 60 * 15
})
})
}
const app = express()
// 由 html-webpack-plugin 生成
let frontend
let backend
// 创建来自 webpack 生成的服务端包
let renderer
if (isProd) {
// 生产模式: 从 fs 创建服务器 HTML 渲染器和索引
const bundle = require('./dist/vue-ssr-server-bundle.json')
frontend = fs.readFileSync(resolve('./dist/server.html'), 'utf-8')
renderer = createRenderer(bundle, frontend)
} else {
// 开发模式: 设置带有热重新加载的 dev 服务器,并在文件更改时更新渲染器和索引 HTML
require('./build/setup-dev-server')(app, (bundle, _template) => {
frontend = _template.frontend
backend = _template.backend
renderer = createRenderer(bundle, frontend)
})
}
// 设置静态文件缓存时间
const serve = (path, cache) => express.static(resolve(path), { maxAge: cache && isProd ? 60 * 60 * 24 * 30 : 0 })
// 引用 esj 模板引擎
app.set('views', path.join(__dirname, 'dist'))
app.engine('.html', require('ejs').__express)
app.set('view engine', 'ejs')
app.use(favicon('./favicon.ico'))
app.use(compression({ threshold: 0 }))
// 日志
app.use(logger('":method :url" :status :res[content-length] ":referrer" ":user-agent"'))
// body 解析中间件
app.use(bodyParser.json())
app.use(bodyParser.urlencoded({ extended: false }))
// cookie 解析中间件
app.use(cookieParser())
// 设置 express 根目录
app.use(express.static(path.join(__dirname, 'dist')))
app.use('/server', serve('./dist/server', true))
app.use('/static', serve('./dist/static', true))
app.use('/manifest.json', serve('./manifest.json'))
app.use('/service-worker.js', serve('./dist/service-worker.js'))
// api 路由
app.use('/api', routes)
// 前台路由, ssr 渲染
app.get(
['/', '/category/:id', '/search/:qs', '/article/:id', '/about', '/trending/:by', '/user/account', '/user/password'],
(req, res) => {
if ((req.originalUrl === '/user/account' || req.originalUrl === '/user/password') && !req.cookies.user) {
return res.redirect('/')
}
if (!renderer) {
return res.end('waiting for compilation... refresh in a moment.')
}
const s = Date.now()
res.setHeader('Content-Type', 'text/html')
res.setHeader('Server', serverInfo)
const errorHandler = err => {
if (err && err.code === 404) {
res.status(404).end('404 | Page Not Found')
} else {
// Render Error Page or Redirect
res.status(500).end('Internal Error 500')
console.error(`error during render : ${req.url}`)
console.error(err)
}
}
const context = {
title: 'M.M.F 小屋',
description: 'M.M.F 小屋',
url: req.url,
cookies: req.cookies
}
renderer.renderToString(context, (err, html) => {
if (err) {
return errorHandler(err)
}
res.end(html)
console.log(`whole request: ${Date.now() - s}ms`)
})
// const htmlStream = new HTMLStream({ template: frontend, context })
// htmlStream.on('beforeStart', () => {
// const meta = context.meta.inject()
// context.head = (context.head || '') + meta.title.text()
// })
// renderer.renderToStream(context)
// .on('error', errorHandler)
// .pipe(htmlStream)
// .on('end', () => console.log(`whole request: ${Date.now() - s}ms`))
// .pipe(res)
}
)
// 后台渲染
app.get(['/backend', '/backend/*'], (req, res) => {
if (req.originalUrl !== '/backend' && req.originalUrl !== '/backend/' && !req.cookies.b_user) {
return res.redirect('/backend')
}
if (isProd) {
res.render('admin.html', { title: '登录' })
} else {
res.send(backend)
}
})
// 404 页面
app.get('*', (req, res) => {
res.send('HTTP STATUS: 404')
})
app.use(function(req, res, next) {
var err = new Error(req.originalUrl + ' Not Found')
err.status = 404
next(err)
})
app.use(function(err, req, res) {
res.status(err.status || 500)
res.send(err.message)
})
const port = process.env.PORT || config.port || 8080
app.listen(port, () => {
console.log(`server started at localhost:${port}`)
})