bxo 0.0.5-dev.21 → 0.0.5-dev.22

package/README.md

@@ -20,7 +20,7 @@ BXO is a fast, lightweight, and fully type-safe web framework built specifically
 ### Installation
 
 ```bash
-bun add zod
+bun add bxo
 ```
 
 ### Basic Usage

package/index.ts

@@ -3,6 +3,18 @@ import { z } from 'zod';
 // Type utilities for extracting types from Zod schemas
 type InferZodType<T> = T extends z.ZodType<infer U> ? U : never;
 
+// Response configuration types
+type ResponseSchema = z.ZodSchema<any>;
+type StatusResponseSchema = Record<number, ResponseSchema>;
+type ResponseConfig = ResponseSchema | StatusResponseSchema;
+
+// Type utility to extract response type from response config
+type InferResponseType<T> = T extends ResponseSchema 
+  ? InferZodType<T>
+  : T extends StatusResponseSchema 
+    ? { [K in keyof T]: InferZodType<T[K]> }[keyof T]
+    : never;
+
 // Cookie interface
 interface Cookie {
   name: string;
@@ -35,7 +47,7 @@ interface RouteConfig {
   body?: z.ZodSchema<any>;
   headers?: z.ZodSchema<any>;
   cookies?: z.ZodSchema<any>;
-  response?: z.ZodSchema<any>;
+  response?: ResponseConfig;
   detail?: RouteDetail;
 }
 
@@ -53,11 +65,12 @@ export type Context<TConfig extends RouteConfig = {}> = {
     headers?: Record<string, string>;
     cookies?: Cookie[];
   };
+  status: (code: number, data?: any) => any;
   [key: string]: any;
 };
 
-// Handler function type
-type Handler<TConfig extends RouteConfig = {}, EC = {}> = (ctx: Context<TConfig> & EC) => Promise<any> | any;
+// Handler function type with proper response typing
+type Handler<TConfig extends RouteConfig = {}, EC = {}> = (ctx: Context<TConfig> & EC) => Promise<InferResponseType<TConfig['response']> | any> | InferResponseType<TConfig['response']> | any;
 
 // Route definition
 interface Route {
@@ -433,6 +446,38 @@ export default class BXO {
     return schema.parse(data);
   }
 
+  // Validate response against response config (supports both simple and status-based schemas)
+  private validateResponse(responseConfig: ResponseConfig | undefined, data: any, status: number = 200): any {
+    if (!responseConfig) return data;
+    
+    // If it's a simple schema (not status-based)
+    if ('parse' in responseConfig && typeof responseConfig.parse === 'function') {
+      return responseConfig.parse(data);
+    }
+    
+    // If it's a status-based schema
+    if (typeof responseConfig === 'object' && !('parse' in responseConfig)) {
+      const statusSchema = responseConfig[status];
+      if (statusSchema) {
+        return statusSchema.parse(data);
+      }
+      
+      // If no specific status schema found, try to find a fallback
+      // Common fallback statuses: 200, 201, 400, 500
+      const fallbackStatuses = [200, 201, 400, 500];
+      for (const fallbackStatus of fallbackStatuses) {
+        if (responseConfig[fallbackStatus]) {
+          return responseConfig[fallbackStatus].parse(data);
+        }
+      }
+      
+      // If no schema found for the status, return data as-is
+      return data;
+    }
+    
+    return data;
+  }
+
   // Main request handler
   private async handleRequest(request: Request, server?: any): Promise<Response | undefined> {
     const url = new URL(request.url);
@@ -514,7 +559,11 @@ export default class BXO {
         cookies: validatedCookies,
         path: pathname,
         request,
-        set: {}
+        set: {},
+        status: (code: number, data?: any) => {
+          ctx.set.status = code;
+          return data;
+        }
       };
     } catch (validationError) {
       // Validation failed - return error response
@@ -575,7 +624,8 @@ export default class BXO {
       if (route.config?.response && !(response instanceof Response)) {
         try {
           console.log('response', response);
-          response = this.validateData(route.config.response, response);
+          const status = ctx.set.status || 200;
+          response = this.validateResponse(route.config.response, response, status);
         } catch (validationError) {
           // Response validation failed
           

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "bxo",
   "module": "index.ts",
-  "version": "0.0.5-dev.21",
+  "version": "0.0.5-dev.22",
   "description": "A simple and lightweight web framework for Bun",
   "type": "module",
   "devDependencies": {