@stacksjs/bun-router 0.0.16 → 0.0.18

package/src/router/server.ts

@@ -1,462 +0,0 @@
-import type { Server } from 'bun'
-import type { EnhancedRequest, HTTPMethod, ServerOptions } from '../types'
-import type { Router } from './router'
-import { RequestWithMacros } from '../request/macros'
-import { runWithRequest, setCurrentRequest } from '../request/context'
-
-/**
- * Server handling extension for Router class
- */
-export function registerServerHandling(RouterClass: typeof Router): void {
-  Object.defineProperties(RouterClass.prototype, {
-    /**
-     * Start the HTTP server
-     */
-    serve: {
-      async value(options: ServerOptions = {}): Promise<Server<any>> {
-        // Initialize automatic file-based routing (discovers views from src/views, views, etc.)
-        if (this._initFileRoutes) {
-          await this._initFileRoutes()
-        }
-
-        // Initialize automatic API route discovery (from routes/ directory)
-        if (this._initApiRoutes) {
-          await this._initApiRoutes()
-        }
-
-        // Invalidate route cache before starting server
-        this.invalidateCache()
-
-        // Create server options
-        const serverOptions: any = {
-          idleTimeout: 255, // Max allowed timeout (255 seconds)
-          ...options,
-          fetch: this.handleRequest.bind(this),
-        }
-
-        // Build Bun's static option from registered static Response routes.
-        // Static responses use Bun's native zero-allocation dispatch (~15% faster).
-        const staticRouteMap: Record<string, Response> = {}
-        if (this.staticResponses && this.staticResponses.size > 0) {
-          for (const [path, response] of this.staticResponses) {
-            staticRouteMap[path] = response
-          }
-        }
-        // Merge with user-provided static routes (user options take precedence)
-        if (options.static) {
-          Object.assign(staticRouteMap, options.static)
-        }
-        if (Object.keys(staticRouteMap).length > 0) {
-          serverOptions.static = staticRouteMap
-        }
-
-        // Forward development options for HMR and console streaming
-        if (options.development !== undefined) {
-          serverOptions.development = options.development
-        }
-
-        // Apply WebSocket configuration if provided
-        if (this.wsConfig) {
-          serverOptions.websocket = this.wsConfig
-        }
-
-        // Start the server
-        this.serverInstance = Bun.serve(serverOptions)
-
-        if (this.config.verbose) {
-          const port = this.serverInstance.port
-          const hostname = this.serverInstance.hostname
-          console.log(`\n🚀 Server running at http://${hostname}:${port}\n`)
-        }
-
-        return this.serverInstance
-      },
-      writable: true,
-      configurable: true,
-    },
-
-    /**
-     * Reload the HTTP server
-     */
-    reload: {
-      async value(): Promise<void> {
-        if (!this.serverInstance) {
-          throw new Error('Server not started, cannot reload')
-        }
-
-        // Invalidate route cache before reloading
-        this.invalidateCache()
-
-        // Save the current server port and hostname
-        const port = this.serverInstance.port
-        const hostname = this.serverInstance.hostname
-
-        // Close the current server
-        this.serverInstance.stop()
-
-        // Start a new server with the same configuration
-        this.serverInstance = Bun.serve({
-          port,
-          hostname,
-          fetch: this.handleRequest.bind(this),
-          websocket: this.wsConfig || undefined,
-        })
-
-        if (this.config.verbose) {
-          console.log(`🔄 Server reloaded at http://${hostname}:${port}`)
-        }
-      },
-      writable: true,
-      configurable: true,
-    },
-
-    /**
-     * Handle HTTP requests
-     */
-    handleRequest: {
-      async value(req: Request): Promise<Response> {
-        // The prototype is mutated via defineProperties; the static type doesn't see
-        // handleRequestImpl as a public method, so we cast through unknown.
-        const self = this as unknown as { handleRequestImpl: (r: Request) => Promise<Response> }
-        return runWithRequest(req as EnhancedRequest, () => self.handleRequestImpl(req))
-      },
-      writable: true,
-      configurable: true,
-    },
-
-    /**
-     * Internal: actual request handling, executed inside the request context scope.
-     */
-    handleRequestImpl: {
-      async value(req: Request): Promise<Response> {
-        try {
-          // Create URL for route matching
-          const url = new URL(req.url)
-
-          // Handle CORS preflight OPTIONS requests - but check for registered OPTIONS routes first
-          // This ensures explicitly registered OPTIONS routes work while still providing CORS support
-          if (req.method === 'OPTIONS') {
-            const hostname = url.hostname || req.headers.get('host')?.split(':')[0] || 'localhost'
-            const optionsMatch = this.matchRoute(url.pathname, 'OPTIONS', hostname)
-            if (!optionsMatch) {
-              // No explicit OPTIONS route - return generic CORS preflight response
-              return new Response(null, {
-                status: 204,
-                headers: {
-                  'Access-Control-Allow-Origin': '*',
-                  'Access-Control-Allow-Methods': 'GET, POST, PUT, DELETE, PATCH, OPTIONS',
-                  'Access-Control-Allow-Headers': 'Content-Type, Authorization, X-Requested-With, Accept, Origin',
-                  'Access-Control-Max-Age': '86400',
-                  'Access-Control-Allow-Credentials': 'true',
-                },
-              })
-            }
-            // Let the registered OPTIONS route handle it (fall through to normal route matching)
-          }
-
-          // Get domain from the host header
-          const hostname = url.hostname || req.headers.get('host')?.split(':')[0] || 'localhost'
-
-          // Find a matching route
-          const match = this.matchRoute(url.pathname, req.method as HTTPMethod, hostname)
-
-          // Enhance the request with params and other utilities
-          const enhancedReq = this.enhanceRequest(req, match?.params || {})
-          setCurrentRequest(enhancedReq)
-
-          if (match) {
-            // Add the matched route to the request
-            enhancedReq.route = match.route
-
-            // Collect all middleware to run
-            const middlewareStack = [...this.globalMiddleware]
-
-            // Add route-specific middleware
-            if (match.route.middleware && match.route.middleware.length > 0) {
-              middlewareStack.push(...match.route.middleware)
-            }
-
-            // Create a final middleware that executes the route handler
-            const routeHandlerMiddleware = async (req: EnhancedRequest, _next: any) => {
-              return await this.resolveHandler(match.route.handler, req)
-            }
-
-            // Add the route handler as the final middleware
-            middlewareStack.push(routeHandlerMiddleware)
-
-            // Run middleware stack with the route handler at the end
-            const response = await this.runMiddleware(enhancedReq, middlewareStack)
-
-            // Apply modified cookies to the response
-            if (response) {
-              return this.applyModifiedCookies(response, enhancedReq)
-            }
-
-            // This should not happen since we're always returning a response now
-            return new Response('No response from middleware chain', { status: 500 })
-          }
-
-          // No route found - check if the path exists with a different method (405 vs 404).
-          // The 404/405 responses below now (a) include path + method in the body so client
-          // debugging is one grep away, and (b) flow through globalMiddleware so cross-cutting
-          // concerns (X-Request-ID, Server-Timing, audit logging, custom CORS) can observe
-          // them. Previously these paths short-circuited entirely.
-          const allowedMethods = this.getAllowedMethods(url.pathname, hostname)
-          const corsHeaders = {
-            'Content-Type': 'application/json',
-            'Access-Control-Allow-Origin': '*',
-            'Access-Control-Allow-Methods': 'GET, POST, PUT, DELETE, PATCH, OPTIONS',
-            'Access-Control-Allow-Headers': 'Content-Type, Authorization, X-Requested-With, Accept, Origin',
-          }
-
-          if (allowedMethods.length > 0) {
-            const methodNotAllowedHandler = async (_req: EnhancedRequest, _next: any) => {
-              return new Response(JSON.stringify({
-                success: false,
-                message: 'Method Not Allowed',
-                path: url.pathname,
-                method: req.method,
-                allowed: allowedMethods,
-              }), {
-                status: 405,
-                headers: { ...corsHeaders, Allow: allowedMethods.join(', ') },
-              })
-            }
-            if (this.globalMiddleware.length > 0) {
-              const stack = [...this.globalMiddleware, methodNotAllowedHandler]
-              const response = await this.runMiddleware(enhancedReq, stack)
-              if (response) return this.applyModifiedCookies(response, enhancedReq)
-            }
-            return await methodNotAllowedHandler(enhancedReq, async () => null as any)
-          }
-
-          // No route found, try the fallback handler
-          if (this.fallbackHandler) {
-            const response = await this.resolveHandler(this.fallbackHandler, enhancedReq)
-            return this.applyModifiedCookies(response, enhancedReq)
-          }
-
-          // No fallback — emit a 404 enriched with path + method, also through
-          // globalMiddleware so user middleware sees it.
-          const notFoundHandler = async (_req: EnhancedRequest, _next: any) => {
-            return new Response(JSON.stringify({
-              success: false,
-              message: 'Not Found',
-              path: url.pathname,
-              method: req.method,
-            }), { status: 404, headers: corsHeaders })
-          }
-          if (this.globalMiddleware.length > 0) {
-            const stack = [...this.globalMiddleware, notFoundHandler]
-            const response = await this.runMiddleware(enhancedReq, stack)
-            if (response) return this.applyModifiedCookies(response, enhancedReq)
-          }
-          return await notFoundHandler(enhancedReq, async () => null as any)
-        }
-        catch (error) {
-          console.error('Error handling request:', error)
-
-          // Use custom error handler if available
-          if (this.errorHandler) {
-            return this.errorHandler(error as Error)
-          }
-
-          // Default error response with CORS headers
-          return new Response(JSON.stringify({
-            success: false,
-            message: 'Internal Server Error',
-            error: error instanceof Error ? error.message : String(error),
-          }), {
-            status: 500,
-            headers: {
-              'Content-Type': 'application/json',
-              'Access-Control-Allow-Origin': '*',
-              'Access-Control-Allow-Methods': 'GET, POST, PUT, DELETE, PATCH, OPTIONS',
-              'Access-Control-Allow-Headers': 'Content-Type, Authorization, X-Requested-With, Accept, Origin',
-            },
-          })
-        }
-      },
-      writable: true,
-      configurable: true,
-    },
-
-    /**
-     * Enhance a request with params and other utilities
-     */
-    enhanceRequest: {
-      value(req: Request, params: Record<string, string> = {}): EnhancedRequest {
-        // Lazy cookie parsing
-        let parsedCookies: Record<string, string> | null = null
-
-        const getCookies = () => {
-          if (parsedCookies === null) {
-            parsedCookies = {}
-            const cookieHeader = req.headers.get('cookie') || ''
-
-            cookieHeader.split(';').forEach((cookie) => {
-              const parts = cookie.trim().split('=')
-              if (parts.length >= 2) {
-                const name = parts[0].trim()
-                const value = parts.slice(1).join('=').trim()
-                parsedCookies![name] = decodeURIComponent(value)
-              }
-            })
-          }
-          return parsedCookies
-        }
-
-        // Create cookie utilities with lazy parsing
-        const cookies = {
-          get: (name: string) => getCookies()[name],
-          set: (name: string, value: string, options: any = {}) => {
-            const enhancedRequest = req as EnhancedRequest
-            if (!enhancedRequest._cookiesToSet) {
-              enhancedRequest._cookiesToSet = []
-            }
-            enhancedRequest._cookiesToSet.push({ name, value, options })
-          },
-          delete: (name: string, options: any = {}) => {
-            const enhancedRequest = req as EnhancedRequest
-            if (!enhancedRequest._cookiesToDelete) {
-              enhancedRequest._cookiesToDelete = []
-            }
-            enhancedRequest._cookiesToDelete.push({ name, options })
-          },
-          getAll: () => ({ ...getCookies() }),
-        }
-
-        // Create enhanced request
-        const enhancedReq = Object.assign(req, {
-          params,
-          cookies: getCookies(), // Set cookies as plain object for direct access
-          _cookiesToSet: [],
-          _cookiesToDelete: [],
-        }) as unknown as EnhancedRequest
-
-        // Attach registered request macros (input, has, bearerToken, etc.) so
-        // they're available in middleware and handlers without manual setup.
-        // Applied *before* the cookie utility so the cookies macro doesn't
-        // overwrite the get/set/delete object form consumers depend on.
-        RequestWithMacros.applyMacros(enhancedReq)
-
-        // Attach the helpers consumers reach for that aren't shaped as
-        // generic input macros — `getParam`, the function-form `cookie`,
-        // the parsed-query `get` shorthand. These mirror the methods the
-        // bare `Router.enhanceRequest` (router.ts) attaches; without them
-        // here, requests that flow through the registered server.ts
-        // override are missing exactly the helpers that frameworks
-        // layered on bun-router (Stacks, etc.) treat as guaranteed.
-        ;(enhancedReq as any).getParam = <T = string>(name: string, defaultValue?: T): T | undefined => {
-          const value = params?.[name] as T | undefined
-          return value !== undefined ? value : defaultValue
-        }
-        ;(enhancedReq as any).cookie = (name: string, defaultValue?: string): string | null => {
-          const value = getCookies()[name]
-          return value !== undefined ? value : (defaultValue ?? null)
-        }
-        // `get(key)` — read query params (and only query params; the full
-        // input merge with body etc. is provided by the `input` macro).
-        // Skip if a macro already registered it under the same name.
-        if (typeof (enhancedReq as any).get !== 'function') {
-          ;(enhancedReq as any).get = <T = unknown>(key: string, defaultValue?: T): T | undefined => {
-            const url = new URL(req.url)
-            const value = url.searchParams.get(key) as T | null
-            return value !== null && value !== undefined ? value : defaultValue
-          }
-        }
-
-        // Add cookie methods to the request (overrides any macro named `cookies`).
-        Object.assign(enhancedReq, { cookies: { ...getCookies(), ...cookies } })
-
-        return enhancedReq
-      },
-      writable: true,
-      configurable: true,
-    },
-
-    /**
-     * Apply modified cookies to a response
-     */
-    applyModifiedCookies: {
-      value(response: Response, req: EnhancedRequest): Response {
-        // Clone the response to modify headers
-        const newResponse = new Response(response.body, {
-          status: response.status,
-          statusText: response.statusText,
-          headers: response.headers,
-        })
-
-        // Apply cookies to set
-        if (req._cookiesToSet && req._cookiesToSet.length > 0) {
-          for (const { name, value, options } of req._cookiesToSet) {
-            const cookieString = this.serializeCookie(name, value, options)
-            newResponse.headers.append('Set-Cookie', cookieString)
-          }
-        }
-
-        // Apply cookies to delete
-        if (req._cookiesToDelete && req._cookiesToDelete.length > 0) {
-          for (const { name, options } of req._cookiesToDelete) {
-            const deletionOptions = {
-              ...options,
-              expires: new Date(0), // Set expiration to past date
-              maxAge: 0,
-            }
-            const cookieString = this.serializeCookie(name, '', deletionOptions)
-            newResponse.headers.append('Set-Cookie', cookieString)
-          }
-        }
-
-        return newResponse
-      },
-      writable: true,
-      configurable: true,
-    },
-
-    /**
-     * Serialize a cookie for the Set-Cookie header
-     */
-    serializeCookie: {
-      value(name: string, value: string, options: any = {}): string {
-        let cookie = `${encodeURIComponent(name)}=${encodeURIComponent(value)}`
-
-        if (options.maxAge !== undefined) {
-          cookie += `; Max-Age=${options.maxAge}`
-        }
-
-        if (options.expires && options.expires instanceof Date) {
-          cookie += `; Expires=${options.expires.toUTCString()}`
-        }
-
-        if (options.path) {
-          cookie += `; Path=${options.path}`
-        }
-        else {
-          cookie += '; Path=/'
-        }
-
-        if (options.domain) {
-          cookie += `; Domain=${options.domain}`
-        }
-
-        if (options.secure) {
-          cookie += '; Secure'
-        }
-
-        if (options.httpOnly) {
-          cookie += '; HttpOnly'
-        }
-
-        if (options.sameSite) {
-          const sameSite = options.sameSite.toLowerCase()
-          cookie += `; SameSite=${sameSite.charAt(0).toUpperCase() + sameSite.slice(1)}`
-        }
-
-        return cookie
-      },
-      writable: true,
-      configurable: true,
-    },
-  })
-}