@geekmidas/constructs 0.0.7 → 0.0.10

package/src/endpoints/Endpoint.ts

@@ -5,6 +5,7 @@ import set from 'lodash.set';
 import type { OpenAPIV3_1 } from 'openapi-types';
 
 import type { Service, ServiceRecord } from '@geekmidas/services';
+import type { Authorizer } from './Authorizer';
 import { ConstructType } from '../Construct';
 import { Function, type FunctionHandler } from '../functions';
 
@@ -93,6 +94,8 @@ export class Endpoint<
   public authorize: AuthorizeFn<TServices, TLogger, TSession> = () => true;
   /** Optional rate limiting configuration */
   public rateLimit?: RateLimitConfig;
+  /** Optional authorizer for this endpoint */
+  public authorizer?: Authorizer;
   /** The endpoint handler function */
   private endpointFn!: EndpointHandler<
     TInput,
@@ -174,12 +177,13 @@ export class Endpoint<
    * Creates a case-insensitive header lookup function from a headers object.
    *
    * @param headers - Object containing header key-value pairs
-   * @returns Function to retrieve header values by case-insensitive key
+   * @returns Function to retrieve header values by case-insensitive key, or all headers
    *
    * @example
    * ```typescript
    * const headerFn = Endpoint.createHeaders({ 'Content-Type': 'application/json' });
    * headerFn('content-type'); // Returns 'application/json'
+   * headerFn(); // Returns { 'content-type': 'application/json' }
    * ```
    */
   static createHeaders(headers: Record<string, string>): HeaderFn {
@@ -189,22 +193,30 @@ export class Endpoint<
       headerMap.set(key, v);
     }
 
-    return function get(key: string): string | undefined {
+    function get(): Record<string, string>;
+    function get(key: string): string | undefined;
+    function get(key?: string): string | undefined | Record<string, string> {
+      if (key === undefined) {
+        // Return all headers as plain object
+        return Object.fromEntries(headerMap.entries());
+      }
       return headerMap.get(key.toLowerCase());
-    };
+    }
+
+    return get;
   }
 
   /**
    * Parses cookie string and creates a cookie lookup function.
    *
    * @param cookieHeader - The Cookie header value
-   * @returns Function to retrieve cookie values by name
+   * @returns Function to retrieve cookie values by name, or all cookies
    *
    * @example
    * ```typescript
    * const cookieFn = Endpoint.createCookies('session=abc123; theme=dark');
    * cookieFn('session'); // Returns 'abc123'
-   * cookieFn('theme'); // Returns 'dark'
+   * cookieFn(); // Returns { session: 'abc123', theme: 'dark' }
    * ```
    */
   static createCookies(cookieHeader: string | undefined): CookieFn {
@@ -222,9 +234,17 @@ export class Endpoint<
       }
     }
 
-    return function get(name: string): string | undefined {
+    function get(): Record<string, string>;
+    function get(name: string): string | undefined;
+    function get(name?: string): string | undefined | Record<string, string> {
+      if (name === undefined) {
+        // Return all cookies as plain object
+        return Object.fromEntries(cookieMap.entries());
+      }
       return cookieMap.get(name);
-    };
+    }
+
+    return get;
   }
 
   /**
@@ -499,6 +519,7 @@ export class Endpoint<
    * @param options.getSession - Session extraction function
    * @param options.authorize - Authorization check function
    * @param options.status - Success HTTP status code (default: 200)
+   * @param options.authorizer - Optional authorizer configuration
    */
   constructor({
     fn,
@@ -511,12 +532,14 @@ export class Endpoint<
     output: outputSchema,
     services,
     timeout,
+    memorySize,
     getSession,
     authorize,
     rateLimit,
     status = SuccessStatus.OK,
     publisherService,
     events,
+    authorizer,
   }: EndpointOptions<
     TRoute,
     TMethod,
@@ -539,6 +562,7 @@ export class Endpoint<
       logger,
       publisherService,
       events,
+      memorySize,
     );
 
     this.route = route;
@@ -559,6 +583,10 @@ export class Endpoint<
     if (rateLimit) {
       this.rateLimit = rateLimit;
     }
+
+    if (authorizer) {
+      this.authorizer = authorizer;
+    }
   }
 }
 
@@ -625,6 +653,8 @@ export interface EndpointOptions<
   tags?: string[];
   /** Optional execution timeout in milliseconds */
   timeout: number | undefined;
+  /** Optional memory size in MB for serverless deployments */
+  memorySize: number | undefined;
   /** Input validation schemas */
   input: TInput | undefined;
   /** Output validation schema */
@@ -645,6 +675,8 @@ export interface EndpointOptions<
   publisherService?: Service<TEventPublisherServiceName, TEventPublisher>;
 
   events?: MappedEvent<TEventPublisher, OutSchema>[];
+  /** Optional authorizer configuration */
+  authorizer?: Authorizer;
 }
 
 /**
@@ -780,24 +812,45 @@ export type EndpointHeaders = Map<string, string>;
 
 /**
  * Function type for retrieving HTTP header values.
+ * Supports two calling patterns:
+ * - `header(key)` - Get a single header value (case-insensitive)
+ * - `header()` - Get all headers as a plain object
  *
- * @param key - The header name (case-insensitive)
- * @returns The header value or undefined if not found
+ * @example
+ * ```typescript
+ * // Get single header
+ * const contentType = header('content-type');
+ *
+ * // Get all headers
+ * const allHeaders = header();
+ * // { 'content-type': 'application/json', 'host': 'example.com', ... }
+ * ```
  */
-export type HeaderFn = SingleHeaderFn;
+export interface HeaderFn {
+  (): Record<string, string>;
+  (key: string): string | undefined;
+}
 
 /**
  * Function type for retrieving cookie values.
- *
- * @param name - The cookie name
- * @returns The cookie value or undefined if not found
+ * Supports two calling patterns:
+ * - `cookie(name)` - Get a single cookie value
+ * - `cookie()` - Get all cookies as a plain object
  *
  * @example
  * ```typescript
+ * // Get single cookie
  * const sessionId = cookie('session');
+ *
+ * // Get all cookies
+ * const allCookies = cookie();
+ * // { session: 'abc123', theme: 'dark', ... }
  * ```
  */
-export type CookieFn = (name: string) => string | undefined;
+export interface CookieFn {
+  (): Record<string, string>;
+  (name: string): string | undefined;
+}
 
 /**
  * Cookie options matching standard Set-Cookie attributes

package/src/endpoints/EndpointBuilder.ts

@@ -3,6 +3,7 @@ import type { RateLimitConfig } from '@geekmidas/rate-limit';
 import type { Service } from '@geekmidas/services';
 import type { StandardSchemaV1 } from '@standard-schema/spec';
 import uniqBy from 'lodash.uniqby';
+import type { Authorizer } from './Authorizer';
 import { ConstructType } from '../Construct';
 import { BaseFunctionBuilder } from '../functions';
 import { Endpoint, type EndpointSchemas } from './Endpoint';
@@ -26,6 +27,7 @@ export class EndpointBuilder<
   TSession = unknown,
   TEventPublisher extends EventPublisher<any> | undefined = undefined,
   TEventPublisherServiceName extends string = string,
+  TAuthorizers extends readonly string[] = readonly string[],
 > extends BaseFunctionBuilder<
   TInput,
   OutSchema,
@@ -38,9 +40,12 @@ export class EndpointBuilder<
   protected _description?: string;
   protected _status?: SuccessStatus;
   protected _tags?: string[];
+  protected _memorySize?: number;
   _getSession: SessionFn<TServices, TLogger, TSession> = () => ({}) as TSession;
   _authorize: AuthorizeFn<TServices, TLogger, TSession> = () => true;
   _rateLimit?: RateLimitConfig;
+  _availableAuthorizers: Authorizer[] = [];
+  _authorizerName?: TAuthorizers[number];
 
   constructor(
     readonly route: TRoute,
@@ -78,6 +83,11 @@ export class EndpointBuilder<
     return this;
   }
 
+  memorySize(memorySize: number): this {
+    this._memorySize = memorySize;
+    return this;
+  }
+
   publisher<T extends EventPublisher<any>, TName extends string>(
     publisher: Service<TName, T>,
   ): EndpointBuilder<
@@ -89,7 +99,8 @@ export class EndpointBuilder<
     OutSchema,
     TSession,
     T,
-    TName
+    TName,
+    TAuthorizers
   > {
     this._publisher = publisher as unknown as Service<
       TEventPublisherServiceName,
@@ -105,7 +116,8 @@ export class EndpointBuilder<
       OutSchema,
       TSession,
       T,
-      TName
+      TName,
+      TAuthorizers
     >;
   }
 
@@ -119,7 +131,9 @@ export class EndpointBuilder<
     TLogger,
     OutSchema,
     TSession,
-    TEventPublisher
+    TEventPublisher,
+    TEventPublisherServiceName,
+    TAuthorizers
   > {
     this.schemas.body = schema as unknown as T;
     // @ts-ignore
@@ -136,7 +150,9 @@ export class EndpointBuilder<
     TLogger,
     OutSchema,
     TSession,
-    TEventPublisher
+    TEventPublisher,
+    TEventPublisherServiceName,
+    TAuthorizers
   > {
     this.schemas.query = schema as unknown as T;
     // @ts-ignore
@@ -153,7 +169,9 @@ export class EndpointBuilder<
     TLogger,
     OutSchema,
     TSession,
-    TEventPublisher
+    TEventPublisher,
+    TEventPublisherServiceName,
+    TAuthorizers
   > {
     return this.search(schema);
   }
@@ -168,7 +186,9 @@ export class EndpointBuilder<
     TLogger,
     OutSchema,
     TSession,
-    TEventPublisher
+    TEventPublisher,
+    TEventPublisherServiceName,
+    TAuthorizers
   > {
     this.schemas.params = schema as unknown as T;
     // @ts-ignore
@@ -180,6 +200,40 @@ export class EndpointBuilder<
     return this;
   }
 
+  authorizer(
+    name: TAuthorizers[number] | 'none',
+  ): EndpointBuilder<
+    TRoute,
+    TMethod,
+    TInput,
+    TServices,
+    TLogger,
+    OutSchema,
+    TSession,
+    TEventPublisher,
+    TEventPublisherServiceName,
+    TAuthorizers
+  > {
+    // Special case: 'none' explicitly marks endpoint as having no authorizer
+    if (name === 'none') {
+      this._authorizerName = undefined;
+      return this;
+    }
+
+    // Validate that the authorizer exists in available authorizers
+    const authorizerExists = this._availableAuthorizers.some(
+      (a) => a.name === name,
+    );
+    if (!authorizerExists && this._availableAuthorizers.length > 0) {
+      const available = this._availableAuthorizers.map((a) => a.name).join(', ');
+      throw new Error(
+        `Authorizer "${name as string}" not found in available authorizers: ${available}`,
+      );
+    }
+    this._authorizerName = name;
+    return this;
+  }
+
   services<T extends Service[]>(
     services: T,
   ): EndpointBuilder<
@@ -191,7 +245,8 @@ export class EndpointBuilder<
     OutSchema,
     TSession,
     TEventPublisher,
-    TEventPublisherServiceName
+    TEventPublisherServiceName,
+    TAuthorizers
   > {
     this._services = uniqBy(
       [...this._services, ...services],
@@ -207,7 +262,8 @@ export class EndpointBuilder<
       OutSchema,
       TSession,
       TEventPublisher,
-      TEventPublisherServiceName
+      TEventPublisherServiceName,
+      TAuthorizers
     >;
   }
 
@@ -222,7 +278,8 @@ export class EndpointBuilder<
     OutSchema,
     TSession,
     TEventPublisher,
-    TEventPublisherServiceName
+    TEventPublisherServiceName,
+    TAuthorizers
   > {
     this._logger = logger as unknown as TLogger;
 
@@ -235,7 +292,8 @@ export class EndpointBuilder<
       OutSchema,
       TSession,
       TEventPublisher,
-      TEventPublisherServiceName
+      TEventPublisherServiceName,
+      TAuthorizers
     >;
   }
 
@@ -250,7 +308,8 @@ export class EndpointBuilder<
     T,
     TSession,
     TEventPublisher,
-    TEventPublisherServiceName
+    TEventPublisherServiceName,
+    TAuthorizers
   > {
     this.outputSchema = schema as unknown as OutSchema;
 
@@ -263,7 +322,8 @@ export class EndpointBuilder<
       T,
       TSession,
       TEventPublisher,
-      TEventPublisherServiceName
+      TEventPublisherServiceName,
+      TAuthorizers
     >;
   }
 
@@ -286,6 +346,11 @@ export class EndpointBuilder<
     TSession,
     TEventPublisher
   > {
+    // Find authorizer metadata if name is set
+    const authorizer = this._authorizerName
+      ? this._availableAuthorizers.find((a) => a.name === this._authorizerName)
+      : undefined;
+
     return new Endpoint({
       fn,
       method: this.method,
@@ -297,12 +362,14 @@ export class EndpointBuilder<
       services: this._services,
       logger: this._logger,
       timeout: this._timeout,
+      memorySize: this._memorySize,
       authorize: this._authorize,
       status: this._status,
       getSession: this._getSession,
       rateLimit: this._rateLimit,
       publisherService: this._publisher,
       events: this._events,
+      authorizer,
     });
   }
 }