43 lines
1.3 KiB
JavaScript
43 lines
1.3 KiB
JavaScript
const Koa = require('koa')
|
||
const compose = require('koa-compose') // 组合中间件，简化写法
|
||
const http = require('http')
|
||
const { httpPort } = require('./config')
|
||
const middlewares = require('./middlewares')
|
||
const wsTerminal = require('./socket/terminal')
|
||
const wsSftp = require('./socket/sftp')
|
||
const wsClientInfo = require('./socket/clients')
|
||
const wsOnekey = require('./socket/onekey')
|
||
const { throwError } = require('./utils/tools')
|
||
|
||
const httpServer = () => {
|
||
const app = new Koa()
|
||
const server = http.createServer(app.callback())
|
||
serverHandler(app, server)
|
||
// ws一直报跨域的错误：参照官方文档使用createServer API创建服务
|
||
server.listen(httpPort, () => {
|
||
consola.success(`Server(http) is running on: http://localhost:${ httpPort }`)
|
||
})
|
||
}
|
||
|
||
// 服务
|
||
function serverHandler(app, server) {
|
||
app.proxy = true // 用于nginx反代时获取真实客户端ip
|
||
wsTerminal(server) // 终端
|
||
wsSftp(server) // sftp
|
||
wsOnekey(server) // 一键指令
|
||
wsClientInfo(server) // 客户端信息
|
||
app.context.throwError = throwError // 常用方法挂载全局ctx上
|
||
app.use(compose(middlewares))
|
||
// 捕获error.js模块抛出的服务错误
|
||
app.on('error', (err, ctx) => {
|
||
ctx.status = 500
|
||
ctx.body = {
|
||
status: ctx.status,
|
||
message: `Program Error：${ err.message }`
|
||
}
|
||
})
|
||
}
|
||
|
||
module.exports = {
|
||
httpServer
|
||
} |