Merge pull request #5 from zenstackhq/dev

dev to main

Author: jiashengguo

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zenstackhq/studio-mcp-remote",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "Local stdio MCP server bridging client calls and ZenStack local database proxy",
   "type": "module",
   "main": "./dist/index.js",
@@ -36,4 +36,4 @@
   "engines": {
     "node": ">=18.0.0"
   }
-}
+}

src/proxy.ts

@@ -1,7 +1,16 @@
 import SuperJSON from 'superjson'
 import type { McpOperation } from './types.js'
 
-const READ_OPERATIONS = new Set(['findMany', 'findFirst', 'findUnique', 'count'])
+const READ_OPERATIONS = new Set([
+  'findMany',
+  'findUnique',
+  'findFirst',
+  'count',
+  'aggregate',
+  'groupBy',
+])
+const DELETE_OPERATIONS = new Set(['delete', 'deleteMany'])
+const UPDATE_OPERATIONS = new Set(['update', 'updateMany', 'updateManyAndReturn'])
 
 export async function executeProxyOperation(
   proxyUrl: string,
@@ -10,7 +19,8 @@ export async function executeProxyOperation(
 ): Promise<unknown> {
   const { json, meta } = SuperJSON.serialize(operation.args ?? {})
   const isRead = READ_OPERATIONS.has(operation.operation)
-  const isDelete = operation.operation === 'deleteMany'
+  const isDelete = DELETE_OPERATIONS.has(operation.operation)
+  const isUpdate = UPDATE_OPERATIONS.has(operation.operation)
 
   const query = `q=${encodeURIComponent(JSON.stringify(json))}${
     meta && Object.keys(meta).length
@@ -30,13 +40,7 @@ export async function executeProxyOperation(
   const authHeader = authorization.startsWith('Bearer ') ? authorization : `Bearer ${authorization}`
 
   const response = await fetch(url, {
-    method: isRead
-      ? 'GET'
-      : isDelete
-        ? 'DELETE'
-        : operation.operation === 'update'
-          ? 'PUT'
-          : 'POST',
+    method: isRead ? 'GET' : isDelete ? 'DELETE' : isUpdate ? 'PUT' : 'POST',
     headers: {
       Authorization: authHeader,
       'Content-Type': 'application/json',