clean up
This commit is contained in:
28
src/index.ts
28
src/index.ts
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user