加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
app.js 3.61 KB
一键复制 编辑 原始数据 按行查看 历史
iRainna 提交于 2022-12-05 17:37 . base
const Koa = require('koa')
const app = new Koa()
const views = require('koa-views')
const json = require('koa-json')
const onerror = require('koa-onerror')
const path = require('path')
const bodyparser = require('koa-bodyparser')
const logger = require('koa-logger')
const cors = require('@koa/cors')
const koajwt = require('koa-jwt')
const {
koaBody
} = require('koa-body');
const {
koaSwagger
} = require('koa2-swagger-ui');
const users = require('./src/routes/users')
const swagger = require('./src/routes/swagger')
const upload = require('./src/routes/upload')
const account = require('./src/routes/account')
const {jwt_secret} = require('./src/utils/config')
const content = require('./src/utils/content')
const mimes = require('./src/utils/mimes')
const staticPath = './static'
/*****跨域*****/
app.use(cors());
/*****跨域*****/
/*****上传*****/
app.use(koaBody({
multipart: true,
formidable: {
maxFileSize: 2 * 1024 * 1024 * 1024 // 设置上传文件大小最大限制,默认2M
}
}));
/*****上传*****/
// 错误处理
app.use((ctx, next) => {
return next().catch((err) => {
if(err.status === 401){
ctx.status = 401;
ctx.body = 'Protected resource, use Authorization header to get access\n';
}else{
throw err;
}
})
})
// 注意:放在路由前面
app.use(koajwt({
secret: jwt_secret
}).unless({ // 配置白名单
path: [/\/account\/register/, /\/account\/login/]
}))
// error handler
onerror(app)
// middlewares
app.use(bodyparser({
enableTypes: ['json', 'form', 'text']
}))
app.use(json())
app.use(logger())
app.use(require('koa-static')(__dirname + '/public'))
app.use(views(__dirname + '/views', {
extension: 'pug'
}))
// 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`)
})
// routes
/*****路由*****/
app.use(users.routes(), users.allowedMethods())
app.use(upload.routes(), upload.allowedMethods())
app.use(account.routes(), account.allowedMethods())
app.use(swagger.routes(), swagger.allowedMethods())
/*****路由*****/
/*****swagger*****/
app.use(
koaSwagger({
routePrefix: '/swagger/index.html', // 这里配置swagger的访问路径
swaggerOptions: {
url: '/swagger/swagger.json', // 这里配置swagger的文档配置URL,也就是说,我们展示的API都是通过这个接口生成的。
},
}),
)
/*****swagger*****/
/*****静态资源*****/
// 解析资源类型
function parseMime(url) {
let extName = path.extname(url)
extName = extName ? extName.slice(1).toLowerCase() : 'unknown'
return mimes[extName]
}
app.use(async (ctx) => {
if (ctx.url.indexOf('static') !== -1) {
console.log('join')
// 静态资源目录在本地的绝对路径
let fullStaticPath = path.join(__dirname, staticPath)
// 获取静态资源内容,有可能是文件内容,目录,或404
let _content = await content(ctx, fullStaticPath)
// 解析请求内容的类型
let _mime = parseMime(ctx.url)
// 如果有对应的文件类型,就配置上下文的类型
if (_mime) {
ctx.type = _mime
}
// 输出静态资源内容
if (_mime && _mime.indexOf('image/') >= 0) {
// 如果是图片,则用node原生res,输出二进制数据
console.log('_content',_content)
ctx.res.writeHead(200)
ctx.res.write(_content, 'binary')
ctx.res.end()
} else {
// 其他则输出文本
ctx.body = _content
}
}
})
/*****静态资源*****/
// error-handling
app.on('error', (err, ctx) => {
console.error('server error', err, ctx)
});
module.exports = app
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化