@squadbase/connectors 0.0.12 → 0.0.13

package/dist/index.js

@@ -285,6 +285,19 @@ var parameters3 = {
 
 // src/connectors/snowflake/tools/execute-query.ts
 import { z } from "zod";
+
+// src/connectors/snowflake/utils.ts
+import crypto from "crypto";
+function decryptPrivateKey(pem, passphrase) {
+  if (!passphrase) return pem;
+  return crypto.createPrivateKey({
+    key: pem,
+    format: "pem",
+    passphrase
+  }).export({ type: "pkcs8", format: "pem" });
+}
+
+// src/connectors/snowflake/tools/execute-query.ts
 var MAX_ROWS = 500;
 var QUERY_TIMEOUT_MS = 6e4;
 var inputSchema = z.object({
@@ -334,18 +347,18 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
       const role = parameters3.role.getValue(connection);
       const warehouse = parameters3.warehouse.getValue(connection);
       const privateKeyBase64 = parameters3.privateKeyBase64.getValue(connection);
-      const privateKey = Buffer.from(privateKeyBase64, "base64").toString(
+      const privateKeyPem = Buffer.from(privateKeyBase64, "base64").toString(
         "utf-8"
       );
       const privateKeyPass = parameters3.privateKeyPassphrase.tryGetValue(connection);
+      const privateKey = decryptPrivateKey(privateKeyPem, privateKeyPass ?? void 0);
       const conn = snowflake.createConnection({
         account,
         username: user,
         role,
         warehouse,
         authenticator: "SNOWFLAKE_JWT",
-        privateKey,
-        ...privateKeyPass ? { privateKeyPass } : {}
+        privateKey
       });
       await new Promise((resolve, reject) => {
         conn.connect((err) => {
@@ -418,19 +431,19 @@ var snowflakeConnector = new ConnectorPlugin({
     try {
       const snowflake = (await import("snowflake-sdk")).default;
       snowflake.configure({ logLevel: "ERROR" });
-      const privateKey = Buffer.from(
+      const privateKeyPem = Buffer.from(
         params[parameters3.privateKeyBase64.slug],
         "base64"
       ).toString("utf-8");
       const privateKeyPass = params[parameters3.privateKeyPassphrase.slug] || void 0;
+      const privateKey = decryptPrivateKey(privateKeyPem, privateKeyPass);
       const conn = snowflake.createConnection({
         account: params[parameters3.account.slug],
         username: params[parameters3.user.slug],
         role: params[parameters3.role.slug],
         warehouse: params[parameters3.warehouse.slug],
         authenticator: "SNOWFLAKE_JWT",
-        privateKey,
-        ...privateKeyPass ? { privateKeyPass } : {}
+        privateKey
       });
       await new Promise((resolve, reject) => {
         conn.connect((err) => {
@@ -459,19 +472,19 @@ var snowflakeConnector = new ConnectorPlugin({
     const resolvedSql = replaceLiteralParams(sql, namedParams);
     const snowflake = (await import("snowflake-sdk")).default;
     snowflake.configure({ logLevel: "ERROR" });
-    const privateKey = Buffer.from(
+    const privateKeyPem = Buffer.from(
       params[parameters3.privateKeyBase64.slug],
       "base64"
     ).toString("utf-8");
     const privateKeyPass = params[parameters3.privateKeyPassphrase.slug] || void 0;
+    const privateKey = decryptPrivateKey(privateKeyPem, privateKeyPass);
     const conn = snowflake.createConnection({
       account: params[parameters3.account.slug],
       username: params[parameters3.user.slug],
       role: params[parameters3.role.slug],
       warehouse: params[parameters3.warehouse.slug],
       authenticator: "SNOWFLAKE_JWT",
-      privateKey,
-      ...privateKeyPass ? { privateKeyPass } : {}
+      privateKey
     });
     await new Promise((resolve, reject) => {
       conn.connect((err) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@squadbase/connectors",
-  "version": "0.0.12",
+  "version": "0.0.13",
   "description": "Squadbase Connectors",
   "type": "module",
   "main": "dist/index.js",
@@ -27,11 +27,9 @@
   },
   "scripts": {
     "build": "tsup",
-    "add-connector": "tsx scripts/add-connector.ts",
     "sync:dev1": "API_BASE_URL=https://dev1-api.squadbase.dev/v0 dotenv tsx scripts/sync-connectors.ts",
     "sync:dev2": "API_BASE_URL=https://dev2-api.squadbase.dev/v0 dotenv tsx scripts/sync-connectors.ts",
     "sync:prod": "API_BASE_URL=https://api.squadbase.dev/v0 dotenv tsx scripts/sync-connectors.ts",
-    "prepare-test-datasource": "tsx scripts/prepare-test-datasource.ts",
     "test": "vitest"
   },
   "peerDependencies": {