feat(server): add ApiHandlerService to handle request with NestJS

Author: ppodds

packages/server/src/nestjs/api-handler.service.ts

@@ -0,0 +1,52 @@
+import { DbClientContract } from '@zenstackhq/runtime';
+import { HttpException, Inject, Injectable, Scope,  } from "@nestjs/common";
+import { HttpAdapterHost, REQUEST,  } from "@nestjs/core";
+import { loadAssets } from "../shared";
+import { RPCApiHandler } from '../api/rpc';
+import { ENHANCED_PRISMA } from "./zenstack.constants";
+import { ApiHandlerOptions } from './interfaces';
+
+/**
+ * The ZenStack API handler service for NestJS. The service is used to handle API requests
+ * and forward them to the ZenStack API handler. It is platform agnostic and can be used
+ * with any HTTP adapter.
+ */
+@Injectable({scope: Scope.REQUEST})
+export class ApiHandlerService {
+    constructor(private readonly httpAdapterHost: HttpAdapterHost, @Inject(ENHANCED_PRISMA) private readonly prisma: DbClientContract, @Inject(REQUEST) private readonly request: unknown) {}
+
+    async handleRequest(options?: ApiHandlerOptions): Promise<unknown> {
+        const { modelMeta, zodSchemas } = loadAssets(options || {});
+        const requestHandler = options?.handler || RPCApiHandler();
+        const hostname = this.httpAdapterHost.httpAdapter.getRequestHostname(this.request);
+        const requestUrl = this.httpAdapterHost.httpAdapter.getRequestUrl(this.request);
+        // prefix with http:// to make a valid url accepted by URL constructor
+        const url = new URL(`http://${hostname}${requestUrl}`);
+        const method = this.httpAdapterHost.httpAdapter.getRequestMethod(this.request);
+        const path = options?.baseUrl && url.pathname.startsWith(options.baseUrl) ? url.pathname.slice(options.baseUrl.length) : url.pathname;
+        const searchParams = url.searchParams;
+        const query = Object.fromEntries(searchParams);
+        const requestBody = (this.request as {body: unknown}).body;
+
+        const response =  await requestHandler({
+            method,
+            path,
+            query,
+            requestBody,
+            prisma: this.prisma,
+            modelMeta,
+            zodSchemas,
+            logger: options?.logger,
+        });
+        console.log(response)
+        // handle handler error
+        // if reponse code >= 400 throw nestjs HttpException
+        // the error response will be generated by nestjs
+        // caller can use try/catch to deal with this manually also
+        if (response.status >= 400) {
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            throw new HttpException(response.body as Record<string, any>, response.status)
+        }
+        return response.body
+    }
+}

packages/server/src/nestjs/index.ts

@@ -1 +1,3 @@
 export * from './zenstack.module';
+export * from './api-handler.service';
+export * from './zenstack.constants';

packages/server/src/nestjs/interfaces/api-handler-options.interface.ts

@@ -0,0 +1,5 @@
+import { AdapterBaseOptions } from "../../types";
+
+export interface ApiHandlerOptions extends AdapterBaseOptions {
+    baseUrl?: string;
+}

packages/server/src/nestjs/interfaces/index.ts

@@ -0,0 +1,2 @@
+export * from './zenstack-module-options.interface'
+export * from './api-handler-options.interface'

packages/server/src/nestjs/interfaces/zenstack-module-options.interface.ts

@@ -0,0 +1,41 @@
+import { FactoryProvider, ModuleMetadata, Provider } from "@nestjs/common";
+
+/**
+ * ZenStack module options.
+ */
+export interface ZenStackModuleOptions {
+    /**
+     * A callback for getting an enhanced `PrismaClient`.
+     */
+    getEnhancedPrisma: (model?: string | symbol ) => unknown;
+}
+
+/**
+ * ZenStack module async registration options.
+ */
+export interface ZenStackModuleAsyncOptions extends Pick<ModuleMetadata, 'imports'> {
+    /**
+     * Whether the module is global-scoped.
+     */
+    global?: boolean;
+
+    /**
+     * The token to export the enhanced Prisma service. Default is {@link ENHANCED_PRISMA}.
+     */
+    exportToken?: string;
+
+    /**
+     * The factory function to create the enhancement options.
+     */
+    useFactory: (...args: unknown[]) => Promise<ZenStackModuleOptions> | ZenStackModuleOptions;
+
+    /**
+     * The dependencies to inject into the factory function.
+     */
+    inject?: FactoryProvider['inject'];
+
+    /**
+     * Extra providers to facilitate dependency injection.
+     */
+    extraProviders?: Provider[];
+}

packages/server/src/nestjs/zenstack.constants.ts

@@ -0,0 +1,4 @@
+/**
+ * The default token used to export the enhanced Prisma service.
+ */
+export const ENHANCED_PRISMA = 'ENHANCED_PRISMA';

packages/server/src/nestjs/zenstack.module.ts

@@ -1,49 +1,6 @@
-import { Module, type DynamicModule, type FactoryProvider, type ModuleMetadata, type Provider } from '@nestjs/common';
-
-/**
- * The default token used to export the enhanced Prisma service.
- */
-export const ENHANCED_PRISMA = 'ENHANCED_PRISMA';
-
-/**
- * ZenStack module options.
- */
-export interface ZenStackModuleOptions {
-    /**
-     * A callback for getting an enhanced `PrismaClient`.
-     */
-    getEnhancedPrisma: (model?: string | symbol ) => unknown;
-}
-
-/**
- * ZenStack module async registration options.
- */
-export interface ZenStackModuleAsyncOptions extends Pick<ModuleMetadata, 'imports'> {
-    /**
-     * Whether the module is global-scoped.
-     */
-    global?: boolean;
-
-    /**
-     * The token to export the enhanced Prisma service. Default is {@link ENHANCED_PRISMA}.
-     */
-    exportToken?: string;
-
-    /**
-     * The factory function to create the enhancement options.
-     */
-    useFactory: (...args: unknown[]) => Promise<ZenStackModuleOptions> | ZenStackModuleOptions;
-
-    /**
-     * The dependencies to inject into the factory function.
-     */
-    inject?: FactoryProvider['inject'];
-
-    /**
-     * Extra providers to facilitate dependency injection.
-     */
-    extraProviders?: Provider[];
-}
+import { Module, type DynamicModule } from '@nestjs/common';
+import { ENHANCED_PRISMA } from './zenstack.constants';
+import { ZenStackModuleAsyncOptions } from './interfaces';
 
 /**
  * The ZenStack module for NestJS. The module exports an enhanced Prisma service,

packages/server/tests/adapter/nestjs.test.ts

@@ -1,7 +1,6 @@
 import { Test } from '@nestjs/testing';
 import { loadSchema } from '@zenstackhq/testtools';
-import { ZenStackModule } from '../../src/nestjs';
-import { ENHANCED_PRISMA } from '../../src/nestjs/zenstack.module';
+import { ZenStackModule, ENHANCED_PRISMA } from '../../src/nestjs';
 
 describe('NestJS adapter tests', () => {
     const schema = `

packages/server/tsconfig.json

@@ -4,7 +4,8 @@
         "target": "ES2020",
         "lib": ["ESNext", "DOM"],
         "outDir": "dist",
-        "strictPropertyInitialization": false
+        "strictPropertyInitialization": false,
+        "emitDecoratorMetadata": true
     },
     "include": ["src/**/*.ts"]
 }