spiceflow 1.0.2 → 1.0.4

package/src/spiceflow.ts

@@ -89,7 +89,10 @@ export type InternalRoute = {
 	path: string
 	handler: InlineHandler<any, any, any>
 	hooks: LocalHook<any, any, any, any, any, any, any>
-	validate?: ValidateFunction
+	validateBody?: ValidateFunction
+	validateQuery?: ValidateFunction
+	validateParams?: ValidateFunction
+
 	prefix: string
 
 	// store: Record<any, any>
@@ -160,15 +163,9 @@ export class Spiceflow<
 		// }
 
 		let bodySchema: TypeSchema = hooks?.body
-		let validate: ValidateFunction | undefined
-
-		if (isZodSchema(bodySchema)) {
-			let jsonSchema = zodToJsonSchema(bodySchema, {})
-			validate = ajv.compile(jsonSchema)
-		} else if (bodySchema) {
-			// console.log(bodySchema)
-			validate = ajv.compile(bodySchema)
-		}
+		let validateBody = getValidateFunction(bodySchema)
+		let validateQuery = getValidateFunction(hooks?.query)
+		let validateParams = getValidateFunction(hooks?.params)
 
 		const store = router.router.register(path)
 		let route: InternalRoute = {
@@ -180,7 +177,9 @@ export class Spiceflow<
 			// prefix,
 			handler: handler!,
 			hooks,
-			validate,
+			validateBody,
+			validateParams,
+			validateQuery,
 		}
 		router.routes.push(route)
 		store[method] = route
@@ -216,6 +215,7 @@ export class Spiceflow<
 
 				const { onErrorHandlers, onRequestHandlers } = router
 				const params = route['params'] || {}
+
 				return {
 					...data,
 					router,
@@ -355,7 +355,7 @@ export class Spiceflow<
 							? ResolvePath<Path>
 							: Schema['params']
 						query: Schema['query']
-						headers: Schema['headers']
+
 						response: ComposeSpiceflowResponse<
 							Schema['response'],
 							Handle
@@ -422,7 +422,7 @@ export class Spiceflow<
 							? ResolvePath<Path>
 							: Schema['params']
 						query: Schema['query']
-						headers: Schema['headers']
+
 						response: ComposeSpiceflowResponse<
 							Schema['response'],
 							Handle
@@ -487,7 +487,7 @@ export class Spiceflow<
 							? ResolvePath<Path>
 							: Schema['params']
 						query: Schema['query']
-						headers: Schema['headers']
+
 						response: ComposeSpiceflowResponse<
 							Schema['response'],
 							Handle
@@ -553,7 +553,7 @@ export class Spiceflow<
 							? ResolvePath<Path>
 							: Schema['params']
 						query: Schema['query']
-						headers: Schema['headers']
+
 						response: ComposeSpiceflowResponse<
 							Schema['response'],
 							Handle
@@ -619,7 +619,7 @@ export class Spiceflow<
 							? ResolvePath<Path>
 							: Schema['params']
 						query: Schema['query']
-						headers: Schema['headers']
+
 						response: ComposeSpiceflowResponse<
 							Schema['response'],
 							Handle
@@ -685,7 +685,7 @@ export class Spiceflow<
 							? ResolvePath<Path>
 							: Schema['params']
 						query: Schema['query']
-						headers: Schema['headers']
+
 						response: ComposeSpiceflowResponse<
 							Schema['response'],
 							Handle
@@ -751,7 +751,7 @@ export class Spiceflow<
 							? ResolvePath<Path>
 							: Schema['params']
 						query: Schema['query']
-						headers: Schema['headers']
+
 						response: ComposeSpiceflowResponse<
 							Schema['response'],
 							Handle
@@ -819,7 +819,7 @@ export class Spiceflow<
 							? ResolvePath<Path>
 							: Schema['params']
 						query: Schema['query']
-						headers: Schema['headers']
+
 						response: ComposeSpiceflowResponse<
 							Schema['response'],
 							Handle
@@ -976,6 +976,7 @@ export class Spiceflow<
 	 */
 	async handle(request: Request, platform?: P): Promise<Response> {
 		platform ??= {} as P
+
 		let u = new URL(request.url)
 		let path = u.pathname + u.search
 		const defaultContext = {
@@ -995,7 +996,7 @@ export class Spiceflow<
 			onErrorHandlers = this.getRouteAndParents(route.router)
 				.reverse()
 				.flatMap((x) => x.onErrorHandlers)
-			const { params, store: defaultStore } = route
+			let { params, store: defaultStore } = route
 			const onReqHandlers = this.getRouteAndParents(route.router)
 				.reverse()
 				.flatMap((x) => x.onRequestHandlers)
@@ -1004,25 +1005,17 @@ export class Spiceflow<
 			// TODO add content type
 
 			let content = route?.hooks?.content
-			let body = await getRequestBody({ request, content })
+			// let body = await getRequestBody({ request, content })
 
-			if (route.validate) {
-				// TODO move compile to the router
+			if (route.validateBody) {
+				// TODO don't clone the request
+				let typedRequest = new TypedRequest(request)
+				typedRequest.validateBody = route.validateBody
+				request = typedRequest
+			}
 
-				const valid = route.validate(body)
-				if (!valid) {
-					const error = ajv.errorsText(route.validate.errors, {
-						separator: '\n',
-					})
+			let query = parseQuery.parse((u.search || '').slice(1))
 
-					return new Response(error, {
-						status: 400,
-						headers: {
-							'content-type': 'text/plain',
-						},
-					})
-				}
-			}
 			if (onReqHandlers.length > 0) {
 				for (const handler of onReqHandlers) {
 					const res = await handler({
@@ -1030,6 +1023,7 @@ export class Spiceflow<
 						response,
 						store,
 						path,
+						query,
 					} satisfies Context<any, any, any>)
 					if (res) {
 						return await turnHandlerResultIntoResponse(res)
@@ -1037,6 +1031,9 @@ export class Spiceflow<
 				}
 			}
 
+			query = runValidation(query, route.validateQuery)
+			params = runValidation(params, route.validateParams)
+
 			// console.log(route)
 
 			const res = route.handler({
@@ -1045,7 +1042,8 @@ export class Spiceflow<
 				response,
 				params: params as any,
 				store,
-				body,
+				query,
+				// body,
 				path,
 
 				// platform
@@ -1060,14 +1058,16 @@ export class Spiceflow<
 
 			return await turnHandlerResultIntoResponse(res)
 		} catch (err: any) {
+			if (err instanceof Response) return err
 			let res = await this.runErrorHandlers({
 				onErrorHandlers,
 				error: err,
 				request,
 			})
 			if (res) return res
+			let status = err?.status ?? 500
 			return new Response(err?.message || 'Internal Server Error', {
-				status: 500,
+				status,
 			})
 		}
 	}
@@ -1201,59 +1201,59 @@ export class Spiceflow<
 	}
 }
 
-async function getRequestBody({
-	request,
-	content,
-}: {
-	content
-	request: Request
-}) {
-	let body: string | Record<string, any> | undefined
-	if (request.method === 'GET' || request.method === 'HEAD') {
-		return
-	}
+// async function getRequestBody({
+// 	request,
+// 	content,
+// }: {
+// 	content
+// 	request: Request
+// }) {
+// 	let body: string | Record<string, any> | undefined
+// 	if (request.method === 'GET' || request.method === 'HEAD') {
+// 		return
+// 	}
 
-	const contentType =
-		content || request.headers.get('content-type')?.split(';')?.[0]
+// 	const contentType =
+// 		content || request.headers.get('content-type')?.split(';')?.[0]
 
-	if (!contentType) {
-		return
-	}
+// 	if (!contentType) {
+// 		return
+// 	}
 
-	switch (contentType) {
-		case 'application/json':
-			body = (await request.json()) as any
-			break
+// 	switch (contentType) {
+// 		case 'application/json':
+// 			body = (await request.json()) as any
+// 			break
 
-		case 'text/plain':
-			body = await request.text()
-			break
+// 		case 'text/plain':
+// 			body = await request.text()
+// 			break
 
-		case 'application/x-www-form-urlencoded':
-			body = parseQuery.parse(await request.text()) as any
-			break
+// 		case 'application/x-www-form-urlencoded':
+// 			body = parseQuery.parse(await request.text()) as any
+// 			break
 
-		case 'application/octet-stream':
-			body = await request.arrayBuffer()
-			break
+// 		case 'application/octet-stream':
+// 			body = await request.arrayBuffer()
+// 			break
 
-		case 'multipart/form-data':
-			body = {}
+// 		case 'multipart/form-data':
+// 			body = {}
 
-			const form = await request.formData()
-			for (const key of form.keys()) {
-				if (body[key]) continue
+// 			const form = await request.formData()
+// 			for (const key of form.keys()) {
+// 				if (body[key]) continue
 
-				const value = form.getAll(key)
-				if (value.length === 1) body[key] = value[0]
-				else body[key] = value
-			}
+// 				const value = form.getAll(key)
+// 				if (value.length === 1) body[key] = value[0]
+// 				else body[key] = value
+// 			}
 
-			break
-	}
+// 			break
+// 	}
 
-	return body
-}
+// 	return body
+// }
 
 const METHODS = [
 	'ALL',
@@ -1288,6 +1288,15 @@ function bfsFind<T>(
 	}
 	return
 }
+export class TypedRequest<T = any> extends Request {
+	validateBody?: ValidateFunction
+
+	async json(): Promise<T> {
+		const body = (await super.json()) as Promise<T>
+		return runValidation(body, this.validateBody)
+	}
+}
+
 export function bfs(tree: RouterTree) {
 	const queue = [tree]
 	let nodes: RouterTree[] = []
@@ -1346,3 +1355,26 @@ export function isZodSchema(value: unknown): value is ZodType {
 			'nullable' in value)
 	)
 }
+
+function getValidateFunction(schema: TypeSchema) {
+	if (isZodSchema(schema)) {
+		let jsonSchema = zodToJsonSchema(schema, {})
+		return ajv.compile(jsonSchema)
+	}
+
+	if (schema) {
+		return ajv.compile(schema)
+	}
+}
+
+function runValidation(value: any, validate?: ValidateFunction) {
+	if (!validate) return value
+	const valid = validate(value)
+	if (!valid) {
+		const error = ajv.errorsText(validate.errors, {
+			separator: '\n',
+		})
+		throw new ValidationError(error)
+	}
+	return value
+}

package/src/zod.test.ts

@@ -11,13 +11,14 @@ test('body is parsed as json', async () => {
 
 		.post(
 			'/post',
-			(c) => {
-				name = c.body.name
+			async (c) => {
+				const body = await c.request.json()
+				name = body.name
 				// @ts-expect-error
-				c.body.nonExistingField
+				body.nonExistingField
 				return {
 					name,
-					nameEcho: c.body.name,
+					nameEcho: body.name,
 					// add: 3,
 				}
 			},
@@ -33,13 +34,14 @@ test('body is parsed as json', async () => {
 		)
 		.post(
 			'/post2',
-			(c) => {
-				name = c.body.name
+			async (c) => {
+				const body = await c.request.json()
+				name = body.name
 				// @ts-expect-error
-				c.body.nonExistingField
+				body.nonExistingField
 				return {
 					name,
-					nameEcho: c.body.name,
+					nameEcho: body.name,
 				}
 			},
 			{