1
0
This commit is contained in:
2022-10-03 00:01:13 +02:00
parent 7f44f304a2
commit 45a781ea66

View File

@@ -340,6 +340,7 @@ export default class Router {
url, url,
handlers handlers
}) })
return this return this
} }
@@ -353,23 +354,33 @@ export default class Router {
private getRoute(request: RouterRequest): Route | undefined { private getRoute(request: RouterRequest): Route | undefined {
const url = new URL(request.url) const url = new URL(request.url)
const pathArr = url.pathname.split('/').filter(i => i) const pathArr = url.pathname.split('/').filter(i => i)
return this.routes.find(r => { return this.routes.find(r => {
const routeArr = r.url.split('/').filter(i => i) const routeArr = r.url.split('/').filter(i => i)
if (![request.method, '*'].includes(r.method) || routeArr.length !== pathArr.length) if (![request.method, '*'].includes(r.method) || routeArr.length !== pathArr.length)
return false return false
const params: RouterRequestParams = {} const params: RouterRequestParams = {}
for (let i = 0; i < routeArr.length; i++) { for (let i = 0; i < routeArr.length; i++) {
if (routeArr[i] !== pathArr[i] && routeArr[i][0] !== ':') if (routeArr[i] !== pathArr[i] && routeArr[i][0] !== ':')
return false return false
if (routeArr[i][0] === ':') if (routeArr[i][0] === ':')
params[routeArr[i].substring(1)] = pathArr[i] params[routeArr[i].substring(1)] = pathArr[i]
} }
request.params = params request.params = params
const query: any = {} const query: any = {}
for (const [k, v] of url.searchParams.entries()) { for (const [k, v] of url.searchParams.entries()) {
query[k] = v query[k] = v
} }
request.query = query request.query = query
return true return true
}) || this.routes.find(r => r.url === '*' && [request.method, '*'].includes(r.method)) }) || this.routes.find(r => r.url === '*' && [request.method, '*'].includes(r.method))
} }
@@ -380,9 +391,9 @@ export default class Router {
* @param {any} env * @param {any} env
* @param {Request} request * @param {Request} request
* @param {any} [extend] * @param {any} [extend]
* @returns {Response} * @returns {Promise<Response>}
*/ */
public async handle(env: any, request: Request, extend: any = {}) { public async handle(env: any, request: Request, extend: any = {}): Promise<Response> {
try { try {
const req: RouterRequest = { const req: RouterRequest = {
...extend, ...extend,
@@ -394,8 +405,10 @@ export default class Router {
query: {}, query: {},
body: '' body: ''
} }
const headers = new Headers() const headers = new Headers()
const route = this.getRoute(req) const route = this.getRoute(req)
if (this.corsEnabled) { if (this.corsEnabled) {
if (this.corsConfig.allowOrigin) if (this.corsConfig.allowOrigin)
headers.set('Access-Control-Allow-Origin', this.corsConfig.allowOrigin) headers.set('Access-Control-Allow-Origin', this.corsConfig.allowOrigin)
@@ -413,8 +426,10 @@ export default class Router {
}) })
} }
} }
if (!route) if (!route)
return new Response(this.debugMode ? 'Route not found!' : null, { status: 404 }) return new Response(this.debugMode ? 'Route not found!' : null, { status: 404 })
if (['POST', 'PUT', 'PATCH'].includes(req.method)) { if (['POST', 'PUT', 'PATCH'].includes(req.method)) {
if (req.headers.has('Content-Type') && req.headers.get('Content-Type')!.includes('json')) { if (req.headers.has('Content-Type') && req.headers.get('Content-Type')!.includes('json')) {
try { try {
@@ -430,24 +445,33 @@ export default class Router {
} }
} }
} }
const res: RouterResponse = { headers } const res: RouterResponse = { headers }
const handlers = [...this.globalHandlers, ...route.handlers] const handlers = [...this.globalHandlers, ...route.handlers]
let prevIndex = -1 let prevIndex = -1
const runner = async (index: number) => { const runner = async (index: number) => {
if (index === prevIndex) if (index === prevIndex)
throw new Error('next() called multiple times') throw new Error('next() called multiple times')
prevIndex = index prevIndex = index
if (typeof handlers[index] === 'function') if (typeof handlers[index] === 'function')
await handlers[index]({ env, req, res, next: async () => await runner(index + 1) }) await handlers[index]({ env, req, res, next: async () => await runner(index + 1) })
} }
await runner(0) await runner(0)
if (typeof res.body === 'object') { if (typeof res.body === 'object') {
if (!res.headers.has('Content-Type')) if (!res.headers.has('Content-Type'))
res.headers.set('Content-Type', 'application/json') res.headers.set('Content-Type', 'application/json')
res.body = JSON.stringify(res.body) res.body = JSON.stringify(res.body)
} }
if (res.raw) if (res.raw)
return res.raw return res.raw
return new Response([101, 204, 205, 304].includes(res.status || (res.body ? 200 : 204)) ? null : res.body, { status: res.status, headers: res.headers, webSocket: res.webSocket || null }) return new Response([101, 204, 205, 304].includes(res.status || (res.body ? 200 : 204)) ? null : res.body, { status: res.status, headers: res.headers, webSocket: res.webSocket || null })
} catch(err) { } catch(err) {
console.error(err) console.error(err)