@riktajs/core 0.5.0 → 0.7.0

package/dist/core/decorators/apply-decorators.d.ts

@@ -0,0 +1,8 @@
+import 'reflect-metadata';
+type ClassDecorator = <TFunction extends Function>(target: TFunction) => TFunction | void;
+type MethodDecorator = <T>(target: Object, propertyKey: string | symbol, descriptor: TypedPropertyDescriptor<T>) => TypedPropertyDescriptor<T> | void;
+type PropertyDecorator = (target: Object, propertyKey: string | symbol) => void;
+type AnyDecorator = ClassDecorator | MethodDecorator | PropertyDecorator;
+export declare function applyDecorators(...decorators: AnyDecorator[]): MethodDecorator & ClassDecorator & PropertyDecorator;
+export declare function SetMetadata<T = unknown>(key: string | symbol, value: T): MethodDecorator & ClassDecorator;
+export {};

package/dist/core/decorators/apply-decorators.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.applyDecorators = applyDecorators;
+exports.SetMetadata = SetMetadata;
+require("reflect-metadata");
+function applyDecorators(...decorators) {
+    return ((target, propertyKey, descriptor) => {
+        for (const decorator of decorators.reverse()) {
+            if (descriptor) {
+                const result = decorator(target, propertyKey, descriptor);
+                if (result) {
+                    descriptor = result;
+                }
+            }
+            else if (propertyKey !== undefined) {
+                decorator(target, propertyKey);
+            }
+            else {
+                const result = decorator(target);
+                if (result) {
+                    return result;
+                }
+            }
+        }
+        if (descriptor) {
+            return descriptor;
+        }
+    });
+}
+function SetMetadata(key, value) {
+    const decoratorFactory = (target, propertyKey, descriptor) => {
+        if (descriptor) {
+            Reflect.defineMetadata(key, value, target.constructor, propertyKey);
+        }
+        else {
+            Reflect.defineMetadata(key, value, target);
+        }
+    };
+    return decoratorFactory;
+}

package/dist/core/decorators/create-param-decorator.d.ts

@@ -0,0 +1,11 @@
+import 'reflect-metadata';
+import { ExecutionContext } from '../guards/execution-context';
+export declare const CUSTOM_PARAM_METADATA: unique symbol;
+export interface CustomParamMetadata<T = unknown> {
+    index: number;
+    factory: CustomParamFactory<T, unknown>;
+    data?: T;
+}
+export type CustomParamFactory<TData = unknown, TResult = unknown> = (data: TData, ctx: ExecutionContext) => TResult | Promise<TResult>;
+export declare function createParamDecorator<TData = unknown, TResult = unknown>(factory: CustomParamFactory<TData, TResult>): (data?: TData) => ParameterDecorator;
+export declare function getCustomParamMetadata<T = unknown>(target: Function, propertyKey: string | symbol): CustomParamMetadata<T>[];

package/dist/core/decorators/create-param-decorator.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CUSTOM_PARAM_METADATA = void 0;
+exports.createParamDecorator = createParamDecorator;
+exports.getCustomParamMetadata = getCustomParamMetadata;
+require("reflect-metadata");
+exports.CUSTOM_PARAM_METADATA = Symbol.for('rikta:custom:param:metadata');
+function createParamDecorator(factory) {
+    return (data) => {
+        return (target, propertyKey, parameterIndex) => {
+            if (propertyKey === undefined)
+                return;
+            const existingParams = Reflect.getMetadata(exports.CUSTOM_PARAM_METADATA, target.constructor, propertyKey) ?? [];
+            const metadata = {
+                index: parameterIndex,
+                factory: factory,
+                data,
+            };
+            existingParams.push(metadata);
+            Reflect.defineMetadata(exports.CUSTOM_PARAM_METADATA, existingParams, target.constructor, propertyKey);
+        };
+    };
+}
+function getCustomParamMetadata(target, propertyKey) {
+    return Reflect.getMetadata(exports.CUSTOM_PARAM_METADATA, target, propertyKey) ?? [];
+}

package/dist/core/decorators/index.d.ts

@@ -5,3 +5,5 @@ export * from './param.decorator';
 export * from './autowired.decorator';
 export * from './provider.decorator';
 export * from './config-property.decorator';
+export * from './create-param-decorator';
+export * from './apply-decorators';

package/dist/core/decorators/index.js

@@ -21,3 +21,5 @@ __exportStar(require("./param.decorator"), exports);
 __exportStar(require("./autowired.decorator"), exports);
 __exportStar(require("./provider.decorator"), exports);
 __exportStar(require("./config-property.decorator"), exports);
+__exportStar(require("./create-param-decorator"), exports);
+__exportStar(require("./apply-decorators"), exports);

package/dist/core/index.d.ts

@@ -9,6 +9,7 @@ export * from './application';
 export * from './decorators';
 export * from './exceptions';
 export * from './guards';
+export * from './middleware';
 export * from './config';
 export * from './metadata';
 export { z } from 'zod';

package/dist/core/index.js

@@ -26,6 +26,7 @@ __exportStar(require("./application"), exports);
 __exportStar(require("./decorators"), exports);
 __exportStar(require("./exceptions"), exports);
 __exportStar(require("./guards"), exports);
+__exportStar(require("./middleware"), exports);
 __exportStar(require("./config"), exports);
 __exportStar(require("./metadata"), exports);
 var zod_1 = require("zod");

package/dist/core/middleware/index.d.ts

@@ -0,0 +1,3 @@
+export { Middleware } from './middleware.decorator';
+export { RiktaMiddleware, NextFunction } from './rikta-middleware.interface';
+export { UseMiddleware, getMiddlewareMetadata, getMiddlewareMetadata as getMiddlewaresMetadata, MiddlewareClass } from './use-middleware.decorator';

package/dist/core/middleware/index.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getMiddlewaresMetadata = exports.getMiddlewareMetadata = exports.UseMiddleware = exports.Middleware = void 0;
+var middleware_decorator_1 = require("./middleware.decorator");
+Object.defineProperty(exports, "Middleware", { enumerable: true, get: function () { return middleware_decorator_1.Middleware; } });
+var use_middleware_decorator_1 = require("./use-middleware.decorator");
+Object.defineProperty(exports, "UseMiddleware", { enumerable: true, get: function () { return use_middleware_decorator_1.UseMiddleware; } });
+Object.defineProperty(exports, "getMiddlewareMetadata", { enumerable: true, get: function () { return use_middleware_decorator_1.getMiddlewareMetadata; } });
+Object.defineProperty(exports, "getMiddlewaresMetadata", { enumerable: true, get: function () { return use_middleware_decorator_1.getMiddlewareMetadata; } });

package/dist/core/middleware/middleware.decorator.d.ts

@@ -0,0 +1,2 @@
+import 'reflect-metadata';
+export declare function Middleware(): ClassDecorator;

package/dist/core/middleware/middleware.decorator.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Middleware = Middleware;
+require("reflect-metadata");
+function Middleware() {
+    return (target) => {
+        Reflect.defineMetadata('middleware', true, target);
+        Reflect.defineMetadata('injectable', true, target);
+    };
+}

package/dist/core/middleware/rikta-middleware.interface.d.ts

@@ -0,0 +1,5 @@
+import type { FastifyRequest, FastifyReply } from 'fastify';
+export type NextFunction = () => void | Promise<void>;
+export interface RiktaMiddleware {
+    use(req: FastifyRequest, res: FastifyReply, next: NextFunction): void | Promise<void>;
+}

package/dist/core/middleware/rikta-middleware.interface.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/core/middleware/use-middleware.decorator.d.ts

@@ -0,0 +1,6 @@
+import 'reflect-metadata';
+import type { Constructor } from '../types';
+import type { RiktaMiddleware } from './rikta-middleware.interface';
+export type MiddlewareClass = Constructor<RiktaMiddleware>;
+export declare function UseMiddleware(...middleware: MiddlewareClass[]): ClassDecorator & MethodDecorator;
+export declare function getMiddlewareMetadata(target: Constructor, propertyKey?: string | symbol): MiddlewareClass[];

package/dist/core/middleware/use-middleware.decorator.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UseMiddleware = UseMiddleware;
+exports.getMiddlewareMetadata = getMiddlewareMetadata;
+require("reflect-metadata");
+const constants_1 = require("../constants");
+function UseMiddleware(...middleware) {
+    return (target, propertyKey) => {
+        if (propertyKey !== undefined) {
+            const existingMiddleware = Reflect.getMetadata(constants_1.MIDDLEWARE_METADATA, target.constructor, propertyKey) ?? [];
+            Reflect.defineMetadata(constants_1.MIDDLEWARE_METADATA, [...existingMiddleware, ...middleware], target.constructor, propertyKey);
+        }
+        else {
+            const existingMiddleware = Reflect.getMetadata(constants_1.MIDDLEWARE_METADATA, target) ?? [];
+            Reflect.defineMetadata(constants_1.MIDDLEWARE_METADATA, [...existingMiddleware, ...middleware], target);
+        }
+    };
+}
+function getMiddlewareMetadata(target, propertyKey) {
+    const controllerMiddleware = Reflect.getMetadata(constants_1.MIDDLEWARE_METADATA, target) ?? [];
+    if (!propertyKey) {
+        return controllerMiddleware;
+    }
+    const methodMiddleware = Reflect.getMetadata(constants_1.MIDDLEWARE_METADATA, target, propertyKey) ?? [];
+    return [...controllerMiddleware, ...methodMiddleware];
+}

package/dist/core/router/router.d.ts

@@ -7,13 +7,16 @@ export declare class Router {
     private readonly container;
     private readonly globalPrefix;
     private readonly guardCache;
+    private readonly middlewareCache;
     constructor(server: FastifyInstance, container: Container, globalPrefix?: string);
     registerController(controllerClass: Constructor, silent?: boolean): void;
     private registerRoute;
     private compileParamResolvers;
     private createParamExtractor;
-    private resolveParamsOptimized;
+    private resolveAllParams;
     private resolveGuardInstances;
     private executeGuardsOptimized;
+    private resolveMiddlewareInstances;
+    private executeMiddlewareChain;
     private buildPath;
 }

package/dist/core/router/router.js

@@ -3,15 +3,18 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.Router = void 0;
 require("reflect-metadata");
 const constants_1 = require("../constants");
+const create_param_decorator_1 = require("../decorators/create-param-decorator");
 const validation_exception_1 = require("../exceptions/validation.exception");
 const exceptions_1 = require("../exceptions/exceptions");
 const execution_context_1 = require("../guards/execution-context");
 const use_guards_decorator_1 = require("../guards/use-guards.decorator");
+const use_middleware_decorator_1 = require("../middleware/use-middleware.decorator");
 class Router {
     server;
     container;
     globalPrefix;
     guardCache = new Map();
+    middlewareCache = new Map();
     constructor(server, container, globalPrefix = '') {
         this.server = server;
         this.container = container;
@@ -36,49 +39,46 @@ class Router {
             throw new Error(`Handler ${String(route.handlerName)} not found on ${controllerClass.name}`);
         }
         const paramsMeta = Reflect.getMetadata(constants_1.PARAM_METADATA, controllerClass, route.handlerName) ?? [];
+        const customParamsMeta = (0, create_param_decorator_1.getCustomParamMetadata)(controllerClass, route.handlerName);
         const statusCode = Reflect.getMetadata(constants_1.HTTP_CODE_METADATA, controllerClass, route.handlerName);
         const guards = (0, use_guards_decorator_1.getGuardsMetadata)(controllerClass, route.handlerName);
+        const middleware = (0, use_middleware_decorator_1.getMiddlewareMetadata)(controllerClass, route.handlerName);
         const compiledResolvers = this.compileParamResolvers(paramsMeta);
-        const hasParams = compiledResolvers.length > 0;
-        const maxParamIndex = hasParams ? Math.max(...compiledResolvers.map(r => r.index)) : -1;
+        const hasBuiltinParams = compiledResolvers.length > 0;
+        const hasCustomParams = customParamsMeta.length > 0;
+        const hasParams = hasBuiltinParams || hasCustomParams;
+        const allParamIndexes = [
+            ...compiledResolvers.map(r => r.index),
+            ...customParamsMeta.map(r => r.index)
+        ];
+        const maxParamIndex = hasParams ? Math.max(...allParamIndexes) : -1;
         const guardInstances = this.resolveGuardInstances(guards);
         const hasGuards = guardInstances.length > 0;
-        let routeHandler;
-        if (!hasGuards && !hasParams && !statusCode) {
-            routeHandler = async (_request, _reply) => {
-                return handler.call(controllerInstance);
-            };
-        }
-        else if (!hasGuards && !hasParams) {
-            routeHandler = async (_request, reply) => {
-                const result = await handler.call(controllerInstance);
-                if (statusCode)
-                    reply.status(statusCode);
-                return result;
-            };
-        }
-        else if (!hasGuards) {
-            routeHandler = async (request, reply) => {
-                const args = this.resolveParamsOptimized(compiledResolvers, maxParamIndex, request, reply);
-                const result = await handler.apply(controllerInstance, args);
-                if (statusCode)
-                    reply.status(statusCode);
-                return result;
-            };
-        }
-        else {
-            const createContext = (req, rep) => new execution_context_1.ExecutionContextImpl(req, rep, controllerClass, route.handlerName);
-            routeHandler = async (request, reply) => {
-                await this.executeGuardsOptimized(guardInstances, createContext(request, reply));
-                const args = hasParams
-                    ? this.resolveParamsOptimized(compiledResolvers, maxParamIndex, request, reply)
-                    : [];
-                const result = await handler.apply(controllerInstance, args);
-                if (statusCode)
-                    reply.status(statusCode);
-                return result;
-            };
-        }
+        const middlewareInstances = this.resolveMiddlewareInstances(middleware);
+        const hasMiddleware = middlewareInstances.length > 0;
+        const needsContext = hasGuards || hasCustomParams;
+        const createContext = needsContext
+            ? (req, rep) => new execution_context_1.ExecutionContextImpl(req, rep, controllerClass, route.handlerName)
+            : null;
+        const routeHandler = async (request, reply) => {
+            const executionContext = createContext ? createContext(request, reply) : null;
+            if (hasGuards && executionContext) {
+                await this.executeGuardsOptimized(guardInstances, executionContext);
+            }
+            if (hasMiddleware) {
+                await this.executeMiddlewareChain(middlewareInstances, request, reply);
+            }
+            let args;
+            if (hasParams) {
+                args = await this.resolveAllParams(compiledResolvers, customParamsMeta, maxParamIndex, request, reply, executionContext);
+            }
+            const result = args
+                ? await handler.apply(controllerInstance, args)
+                : await handler.call(controllerInstance);
+            if (statusCode)
+                reply.status(statusCode);
+            return result;
+        };
         const method = route.method.toLowerCase();
         this.server[method](fullPath, routeHandler);
         if (!silent)
@@ -127,8 +127,8 @@ class Router {
                 return () => undefined;
         }
     }
-    resolveParamsOptimized(resolvers, maxIndex, request, reply) {
-        const context = {
+    async resolveAllParams(compiledResolvers, customParams, maxIndex, request, reply, executionContext) {
+        const routeContext = {
             request,
             reply,
             params: request.params,
@@ -136,9 +136,9 @@ class Router {
             body: request.body,
         };
         const args = new Array(maxIndex + 1);
-        for (let i = 0; i < resolvers.length; i++) {
-            const resolver = resolvers[i];
-            let value = resolver.extract(context);
+        for (let i = 0; i < compiledResolvers.length; i++) {
+            const resolver = compiledResolvers[i];
+            let value = resolver.extract(routeContext);
             if (resolver.zodSchema) {
                 const result = resolver.zodSchema.safeParse(value);
                 if (!result.success) {
@@ -148,6 +148,13 @@ class Router {
             }
             args[resolver.index] = value;
         }
+        if (customParams.length > 0) {
+            const ctx = executionContext ?? new execution_context_1.ExecutionContextImpl(request, reply, {}, '');
+            for (const customParam of customParams) {
+                const value = await customParam.factory(customParam.data, ctx);
+                args[customParam.index] = value;
+            }
+        }
         return args;
     }
     resolveGuardInstances(guards) {
@@ -179,6 +186,37 @@ class Router {
             }
         }
     }
+    resolveMiddlewareInstances(middleware) {
+        return middleware.map(MiddlewareClass => {
+            let instance = this.middlewareCache.get(MiddlewareClass);
+            if (instance)
+                return instance;
+            try {
+                instance = this.container.resolve(MiddlewareClass);
+            }
+            catch (error) {
+                throw new Error(`Failed to resolve middleware ${MiddlewareClass.name}. ` +
+                    `Make sure it is decorated with @Injectable(). ` +
+                    `Original error: ${error instanceof Error ? error.message : String(error)}`);
+            }
+            if (typeof instance.use !== 'function') {
+                throw new Error(`${MiddlewareClass.name} does not implement RiktaMiddleware interface. ` +
+                    `The middleware must have a use(req, res, next) method.`);
+            }
+            this.middlewareCache.set(MiddlewareClass, instance);
+            return instance;
+        });
+    }
+    async executeMiddlewareChain(middlewareInstances, request, reply) {
+        let index = 0;
+        const next = async () => {
+            if (index < middlewareInstances.length) {
+                const middleware = middlewareInstances[index++];
+                await middleware.use(request, reply, next);
+            }
+        };
+        await next();
+    }
     buildPath(controllerPrefix, routePath) {
         const parts = [this.globalPrefix, controllerPrefix, routePath]
             .filter(Boolean)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@riktajs/core",
-  "version": "0.5.0",
+  "version": "0.7.0",
   "description": "A fast and modern TypeScript backend framework with zero-config autowiring, powered by Fastify",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",