feat: make plugins compatible with prisma client extensions (#39)

* feat: make plugins compatible with prisma client extensions

* chore: bump version, fix lint

* update

Author: ymc9

.vscode/tasks.json

@@ -18,7 +18,7 @@
         },
         {
             "label": "Build all - watch",
-            "command": "pnpm watch",
+            "command": "turbo watch build",
             "type": "shell",
             "group": {
                 "kind": "build"
@@ -28,6 +28,16 @@
                 "id": "server-process"
             }
         },
+        {
+            "label": "Lint all",
+            "command": "pnpm lint",
+            "type": "shell",
+            "icon": {
+                "color": "terminal.ansiYellow",
+                "id": "server-process"
+            },
+            "problemMatcher": []
+        },
         {
             "label": "Test all",
             "command": "pnpm test",

NEW-FEATURES.md

@@ -280,16 +280,20 @@ You can use a plugin to achieve the following goals:
 
 #### 1. ORM query interception
 
-ORM query interception allows you to intercept the high-level ORM API calls.
+ORM query interception allows you to intercept the high-level ORM API calls. The interceptor's configuration is compatible with Prisma's [query client extension](https://www.prisma.io/docs/orm/prisma-client/client-extensions/query).
 
 ```ts
 client.$use({
     id: 'cost-logger',
-    async onQuery({ model, operation, proceed, queryArgs }) {
-        const start = Date.now();
-        const result = await proceed(queryArgs);
-        console.log(`[cost] ${model} ${operation} took ${Date.now() - start}ms`);
-        return result;
+    onQuery: {
+        $allModels: {
+            $allOperations: async ({ model, operation, args, query }) => {
+                const start = Date.now();
+                const result = await query(args);
+                console.log(`[cost] ${model} ${operation} took ${Date.now() - start}ms`);
+                return result;
+            },
+        },
     },
 });
 ```

README.md

@@ -51,10 +51,11 @@
         - [x] Count
         - [x] Aggregate
         - [x] Group by
+    - [ ] Raw queries
     - [ ] Extensions
         - [x] Query builder API
         - [x] Computed fields
-        - [ ] Prisma client extension
+        - [x] Prisma client extension
     - [ ] Misc
         - [x] JSDoc for CRUD methods
         - [x] Cache validation schemas
@@ -66,7 +67,8 @@
         - [ ] Error system
         - [x] Custom table name
         - [x] Custom field name
-        - [ ] Implement changesets
+        - [ ] Strict undefined checks
+    - [ ] Benchmark
 - [ ] Polymorphism
 - [ ] Validation
 - [ ] Access Policy

TODO.md

@@ -1,6 +1,6 @@
 {
     "name": "zenstack-v3",
-    "version": "3.0.0-alpha.4",
+    "version": "3.0.0-alpha.5",
     "description": "ZenStack",
     "packageManager": "[email protected]",
     "scripts": {

package.json

@@ -3,7 +3,7 @@
     "publisher": "zenstack",
     "displayName": "ZenStack CLI",
     "description": "FullStack database toolkit with built-in access control and automatic API generation.",
-    "version": "3.0.0-alpha.4",
+    "version": "3.0.0-alpha.5",
     "type": "module",
     "author": {
         "name": "ZenStack Team"
@@ -22,20 +22,21 @@
     },
     "scripts": {
         "build": "tsup-node",
+        "watch": "tsup-node --watch",
         "lint": "eslint src --ext ts",
         "test": "vitest run",
         "pack": "pnpm pack"
     },
     "dependencies": {
         "@zenstackhq/language": "workspace:*",
         "@zenstackhq/sdk": "workspace:*",
+        "@zenstackhq/common-helpers": "workspace:*",
         "async-exit-hook": "^2.0.1",
         "colors": "1.4.0",
         "commander": "^8.3.0",
         "langium": "catalog:",
         "ora": "^5.4.1",
         "package-manager-detector": "^1.3.0",
-        "tiny-invariant": "^1.3.3",
         "ts-pattern": "catalog:"
     },
     "peerDependencies": {

packages/cli/package.json

@@ -1,10 +1,9 @@
+import { invariant } from '@zenstackhq/common-helpers';
 import { isPlugin, LiteralExpr, type Model } from '@zenstackhq/language/ast';
-import type { CliGenerator } from '@zenstackhq/runtime/client';
-import { PrismaSchemaGenerator, TsSchemaGenerator } from '@zenstackhq/sdk';
+import { PrismaSchemaGenerator, TsSchemaGenerator, type CliGenerator } from '@zenstackhq/sdk';
 import colors from 'colors';
 import fs from 'node:fs';
 import path from 'node:path';
-import invariant from 'tiny-invariant';
 import { getSchemaFile, loadSchemaDocument } from './action-utils';
 
 type Options = {

packages/cli/src/actions/generate.ts

@@ -0,0 +1,4 @@
+import config from '@zenstackhq/eslint-config/base.js';
+
+/** @type {import("eslint").Linter.Config} */
+export default config;

packages/common-helpers/eslint.config.js

@@ -0,0 +1,34 @@
+{
+    "name": "@zenstackhq/common-helpers",
+    "version": "3.0.0-alpha.5",
+    "description": "ZenStack Common Helpers",
+    "type": "module",
+    "scripts": {
+        "build": "tsup-node",
+        "watch": "tsup-node --watch",
+        "lint": "eslint src --ext ts",
+        "pack": "pnpm pack"
+    },
+    "keywords": [],
+    "author": "ZenStack Team",
+    "license": "MIT",
+    "files": [
+        "dist"
+    ],
+    "exports": {
+        ".": {
+            "import": {
+                "types": "./dist/index.d.ts",
+                "default": "./dist/index.js"
+            },
+            "require": {
+                "types": "./dist/index.d.cts",
+                "default": "./dist/index.cjs"
+            }
+        }
+    },
+    "devDependencies": {
+        "@zenstackhq/typescript-config": "workspace:*",
+        "@zenstackhq/eslint-config": "workspace:*"
+    }
+}

packages/common-helpers/package.json

@@ -0,0 +1,6 @@
+export * from './is-plain-object';
+export * from './lower-case-first';
+export * from './param-case';
+export * from './sleep';
+export * from './tiny-invariant';
+export * from './upper-case-first';