@tinycloud/sdk-services 2.3.0-beta.6 → 2.3.0-beta.7

package/dist/index.js

@@ -1670,7 +1670,7 @@ var SQLService = class extends BaseService {
       try {
         const response = await this.invokeSQL(
           dbName,
-          SQLAction.READ,
+          this.actionForSql(sql, SQLAction.READ),
           { action: "query", sql, params: params ?? [] },
           options?.signal
         );
@@ -1700,7 +1700,7 @@ var SQLService = class extends BaseService {
         }
         const response = await this.invokeSQL(
           dbName,
-          SQLAction.WRITE,
+          this.actionForSql(sql, SQLAction.WRITE),
           body,
           options?.signal
         );
@@ -1722,7 +1722,7 @@ var SQLService = class extends BaseService {
       try {
         const response = await this.invokeSQL(
           dbName,
-          SQLAction.WRITE,
+          this.actionForSqlBatch(statements),
           { action: "batch", statements },
           options?.signal
         );
@@ -1799,6 +1799,14 @@ var SQLService = class extends BaseService {
       signal: this.combineSignals(signal)
     });
   }
+  actionForSql(sql, fallback) {
+    return firstSqlToken(sql) === "pragma" ? SQLAction.ADMIN : fallback;
+  }
+  actionForSqlBatch(statements) {
+    return statements.some(
+      (statement) => this.actionForSql(statement.sql, SQLAction.WRITE) === SQLAction.ADMIN
+    ) ? SQLAction.ADMIN : SQLAction.WRITE;
+  }
   async handleErrorResponse(response, operation) {
     const errorText = await response.text();
     let errorBody = {};
@@ -1844,6 +1852,33 @@ var SQLService = class extends BaseService {
   }
 };
 SQLService.serviceName = "sql";
+function firstSqlToken(sql) {
+  let index = 0;
+  while (index < sql.length) {
+    while (index < sql.length && /\s/.test(sql[index])) {
+      index++;
+    }
+    if (sql.startsWith("--", index)) {
+      const newline = sql.indexOf("\n", index + 2);
+      if (newline === -1) {
+        return void 0;
+      }
+      index = newline + 1;
+      continue;
+    }
+    if (sql.startsWith("/*", index)) {
+      const end = sql.indexOf("*/", index + 2);
+      if (end === -1) {
+        return void 0;
+      }
+      index = end + 2;
+      continue;
+    }
+    break;
+  }
+  const match = /^[A-Za-z_]+/.exec(sql.slice(index));
+  return match?.[0].toLowerCase();
+}
 
 // src/duckdb/DuckDbDatabaseHandle.ts
 var DuckDbDatabaseHandle = class {