Add Drizzle support

dbos-rules.test.ts

@@ -90,6 +90,10 @@ function makeSqlInjectionCode(code: string, sqlClient: string): string {
       query(query: string, ...values: any[]) {}
     }
 
+    class PgDatabase {
+      execute(query: string, ...values: any[]) {}
+    }
+
     function Transaction(target?: any, key?: any, descriptor?: any): any {
       return descriptor;
     }
@@ -244,7 +248,13 @@ const testSet: TestSet = [
       // Success test #9 (testing unsubstituted template literals)
       makeSqlInjectionSuccessTest(`
         ctxt.client.raw(\`foo\`);
-      `)
+      `),
+
+      // Success test #10 (testing Drizzle support)
+      makeSqlInjectionSuccessTest(`
+        ctxt.client.execute("foo");`,
+        "PgDatabase"
+      )
     ],
 
     [

dbos-rules.ts

@@ -57,7 +57,7 @@ const ormClientInfoForRawSqlQueries: Map<string, string[]> = new Map([
   ["PrismaClient", ["$queryRawUnsafe", "$executeRawUnsafe"]], // For Prisma
   ["EntityManager", ["query"]], // For TypeORM
   ["PoolClient", ["query"]], // This is supported in `dbos-transact` (see `user_database.ts`, but not sure what ORM this corresponds to)
-  // ["PgDatabase", []], // For Drizzle (TODO: add full support for this)
+  ["PgDatabase", []], // For Drizzle (Currently we don't detect raw SQL calls for Drizzle)
 ]);
 
 const assignmentTokenKinds = new Set([

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dbos-inc/eslint-plugin",
-  "version": "3.3.3",
+  "version": "3.3.4",
   "description": "eslint plugin for DBOS SDK",
   "license": "MIT",
   "repository": {