import { Value } from 'https://esm.sh/@sinclair/typebox@0.25.21/value'import { ErrorMessage } from './ErrorMessage.d.ts'import type { DarkflareContext } from './DarkflareContext.d.ts'import type { TSchema } from 'https://esm.sh/@sinclair/typebox@0.25.21'
export const route = < B = unknown, Q = unknown, H = unknown, C = unknown, P = unknown,>( schema: { body?: B query?: Q headers?: H cookies?: C parameters?: P }, handler: ( ctx: DarkflareContext<B, Q, H, C, P>, ) => unknown | Promise<unknown>,) => { return { schema,
async handle( __darkflare: { onError: (err: ErrorMessage, request: Request) => Response cors?: string cache?: number }, request: Request, parameters: Record<string, string>, query: Record<string, unknown>, cf: Record<string, string>, waitUntil: (promise: Promise<unknown>) => void, env: darkflare.Environment, ) { const headers: { [key: string]: string } = {} let body = {}, cookies = {}
const h = new Headers() let p, c: number | undefined
if ( schema.parameters && !Value.Check((schema.parameters as unknown) as TSchema, parameters) ) { return __darkflare.onError('MALFORMED PARAMETERS', request) }
if ( schema.query && !Value.Check((schema.query as unknown) as TSchema, query) ) { return __darkflare.onError('MALFORMED QUERY', request) }
for (const [key, value] of request.headers) { headers[key.toLowerCase()] = value }
if ( schema.headers && !Value.Check((schema.headers as unknown) as TSchema, headers) ) { return __darkflare.onError('MALFORMED HEADERS', request) }
try { const type = headers['content-type']
if (type?.includes('application/json')) { body = await request.json() } else if (type?.includes('text/plain')) { body = await request.text() } } catch (_err) { body = {} }
if ( schema.body && !Value.Check((schema.body as unknown) as TSchema, body) ) { return __darkflare.onError('MALFORMED BODY', request) }
try { cookies = (headers.cookies || '') .split(/;\s*/) .map((pair) => pair.split(/=(.+)/)) .reduce((acc: Record<string, string>, [key, value]) => { acc[key] = value
return acc }, {}) } catch (_err) { cookies = {} }
if ( schema.cookies && !Value.Check((schema.cookies as unknown) as TSchema, cookies) ) { return __darkflare.onError('MALFORMED COOKIES', request) }
if (__darkflare.cors) { h.set('access-control-allow-origin', __darkflare.cors) }
if (request.method === 'GET') { h.set('cache-control', `max-age=${__darkflare.cache ?? 0}`) }
const ctx: DarkflareContext = { env,
req: { raw: request, body, query, headers, cookies, parameters, geo: { ip: headers['cf-connecting-ip'], city: cf.city, region: cf.region, country: cf.country, continent: cf.continent, regionCode: cf.regionCode, latitude: cf.latitude, longitude: cf.longitude, postalCode: cf.postalCode, timezone: cf.timezone, datacenter: cf.colo, }, buffer: request.arrayBuffer, blob: request.blob, formData: request.formData, stream: request.body, },
res: { code: (code) => { c = code }, cookie: (name, value, options) => { let cookie = `${name}=${value};`
h.set( 'set-cookie', ( options?.expiresAt && (cookie += ` expires=${options.expiresAt.toUTCString()};`), options?.maxAge && (cookie += ` max-age=${options.maxAge};`), options?.domain && (cookie += ` domain=${options.domain};`), options?.path && (cookie += ` path=${options.path};`), options?.secure && (cookie += ' secure;'), options?.httpOnly && (cookie += ' httpOnly;'), options?.sameSite && (cookie += ` sameSite=${ options.sameSite.charAt(0).toUpperCase() + options.sameSite.slice(1) };`), cookie ), ) }, header(name, value) { h.set(name, value) }, html(payload, code) { h.set('content-type', 'text/html; charset=utf-8;')
p = payload c = code }, json(payload, code) { h.set('content-type', 'application/json; charset=utf-8;')
p = payload c = code }, text(payload, code) { h.set('content-type', 'text/plain; charset=utf-8;')
p = payload c = code }, redirect(destination, code) { h.set('location', destination)
c = code ?? 307 }, blob(payload, code) { p = payload c = code }, stream(payload, code) { p = payload c = code }, buffer(payload, code) { p = payload c = code }, formData(payload, code) { p = payload c = code }, },
waitUntil, }
try { const result = await handler(ctx as DarkflareContext<B, Q, H, C, P>)
if (!c) { c = 200 }
if (c !== 200) { h.delete('cache-control') }
if (h.has('location')) { return new Response(null, { headers: h, status: c, }) }
if (!p) { p = result }
if (!p) { return new Response(null, { headers: h, status: c, }) }
if (typeof p === 'string') { h.set('content-length', p.length.toString())
if (!h.has('content-type')) { h.set('content-type', 'text/plain; charset=utf-8;') } } else if (p instanceof Uint8Array || p instanceof ArrayBuffer) { h.set('content-length', p.byteLength.toString())
} else if (p instanceof Blob) { h.set('content-length', p.size.toString())
} else if (p instanceof ReadableStream) {
} else { p = JSON.stringify(p)
h.set('content-length', p.length.toString())
if (!h.has('content-type')) { h.set('content-type', 'application/json; charset=utf-8;') }
if (((p as unknown) as { code: number }).code) { c = ((p as unknown) as { code: number }).code } }
return new Response(p, { headers: h, status: c, }) } catch (_err) { return __darkflare.onError('SOMETHING WENT WRONG', request) } }, }}