@squadbase/vite-server 0.1.12-dev.93b8799 → 0.1.17-dev.24af54e

package/dist/connectors/mongodb.js

@@ -1,48 +1,60 @@
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __esm = (fn, res) => function __init() {
+  return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
+};
+
 // ../connectors/src/parameter-definition.ts
-var ParameterDefinition = class {
-  slug;
-  name;
-  description;
-  envVarBaseKey;
-  type;
-  secret;
-  required;
-  constructor(config) {
-    this.slug = config.slug;
-    this.name = config.name;
-    this.description = config.description;
-    this.envVarBaseKey = config.envVarBaseKey;
-    this.type = config.type;
-    this.secret = config.secret;
-    this.required = config.required;
-  }
-  /**
-   * Get the parameter value from a ConnectorConnectionObject.
-   */
-  getValue(connection2) {
-    const param = connection2.parameters.find(
-      (p) => p.parameterSlug === this.slug
-    );
-    if (!param || param.value == null) {
-      throw new Error(
-        `Parameter "${this.slug}" not found or has no value in connection "${connection2.id}"`
-      );
-    }
-    return param.value;
-  }
-  /**
-   * Try to get the parameter value. Returns undefined if not found (for optional params).
-   */
-  tryGetValue(connection2) {
-    const param = connection2.parameters.find(
-      (p) => p.parameterSlug === this.slug
-    );
-    if (!param || param.value == null) return void 0;
-    return param.value;
+var ParameterDefinition;
+var init_parameter_definition = __esm({
+  "../connectors/src/parameter-definition.ts"() {
+    "use strict";
+    ParameterDefinition = class {
+      slug;
+      name;
+      description;
+      envVarBaseKey;
+      type;
+      secret;
+      required;
+      constructor(config) {
+        this.slug = config.slug;
+        this.name = config.name;
+        this.description = config.description;
+        this.envVarBaseKey = config.envVarBaseKey;
+        this.type = config.type;
+        this.secret = config.secret;
+        this.required = config.required;
+      }
+      /**
+       * Get the parameter value from a ConnectorConnectionObject.
+       */
+      getValue(connection2) {
+        const param = connection2.parameters.find(
+          (p) => p.parameterSlug === this.slug
+        );
+        if (!param || param.value == null) {
+          throw new Error(
+            `Parameter "${this.slug}" not found or has no value in connection "${connection2.id}"`
+          );
+        }
+        return param.value;
+      }
+      /**
+       * Try to get the parameter value. Returns undefined if not found (for optional params).
+       */
+      tryGetValue(connection2) {
+        const param = connection2.parameters.find(
+          (p) => p.parameterSlug === this.slug
+        );
+        if (!param || param.value == null) return void 0;
+        return param.value;
+      }
+    };
   }
-};
+});
 
 // ../connectors/src/connectors/mongodb/parameters.ts
+init_parameter_definition();
 var parameters = {
   connectionUrl: new ParameterDefinition({
     slug: "connection-url",
@@ -134,7 +146,7 @@ function createClient(params) {
       return db.collection(collection).countDocuments(filter);
     });
   }
-  async function listCollections() {
+  async function listCollections2() {
     return withDb(async (db) => {
       const list = await db.listCollections().toArray();
       return list.map((c) => ({ name: c.name, type: c.type ?? "collection" }));
@@ -145,7 +157,7 @@ function createClient(params) {
     find,
     findOne,
     countDocuments,
-    listCollections
+    listCollections: listCollections2
   };
 }
 
@@ -216,6 +228,20 @@ var ConnectorPlugin = class _ConnectorPlugin {
    * `runSetupFlow` from `setup-flow.ts`.
    */
   setup;
+  /**
+   * Opt-out of the default "verify before save" behavior on connection
+   * creation. The backend invokes `checkConnection` synchronously while
+   * creating the connection and aborts (no row inserted) if it fails — this
+   * flag disables that for connectors where the check cannot succeed pre-save:
+   *
+   *   - `squadbase-db` populates `connection-url` only after Neon provisioning
+   *   - OAuth connectors require an OAuth-aware proxyFetch keyed by the
+   *     connectionId, which doesn't exist until the row is saved
+   *
+   * Exceptions are the explicit position; new credential-input connectors get
+   * the default verify-on-create behavior without opt-in.
+   */
+  skipConnectionCheckOnCreate;
   constructor(config) {
     this.slug = config.slug;
     this.authType = config.authType;
@@ -233,6 +259,7 @@ var ConnectorPlugin = class _ConnectorPlugin {
     this.query = config.query;
     this.checkConnection = config.checkConnection;
     this.setup = config.setup;
+    this.skipConnectionCheckOnCreate = config.skipConnectionCheckOnCreate;
   }
   get connectorKey() {
     return _ConnectorPlugin.deriveKey(this.slug, this.authType);
@@ -297,6 +324,51 @@ var ConnectorPlugin = class _ConnectorPlugin {
   }
 };
 
+// ../connectors/src/setup-flow.ts
+async function runSetupFlow(flow, params, ctx, config) {
+  const runtime = {
+    params,
+    language: ctx.language,
+    config
+  };
+  let state = flow.initialState();
+  let answerIdx = 0;
+  for (const step of flow.steps) {
+    const ans = ctx.answers[answerIdx];
+    if (ans && ans.questionSlug === step.slug) {
+      state = step.applyAnswer(state, ans.answer);
+      answerIdx += 1;
+      continue;
+    }
+    if (step.type === "text") {
+      return {
+        type: "nextQuestion",
+        questionSlug: step.slug,
+        question: step.question[ctx.language],
+        questionType: "text"
+      };
+    }
+    const options = step.fetchOptions ? await step.fetchOptions(state, runtime) : [];
+    if (options.length === 0) {
+      continue;
+    }
+    return {
+      type: "nextQuestion",
+      questionSlug: step.slug,
+      question: step.question[ctx.language],
+      questionType: step.type,
+      options
+    };
+  }
+  const dataInvestigationResult = await flow.finalize(state, runtime);
+  return { type: "fulfilled", dataInvestigationResult };
+}
+async function resolveSetupSelection(params) {
+  const { selected, allSentinel, fetchAll, limit } = params;
+  const resolved = selected.includes(allSentinel) ? await fetchAll() : selected.filter((v) => v !== allSentinel);
+  return resolved.slice(0, limit);
+}
+
 // ../connectors/src/auth-types.ts
 var AUTH_TYPES = {
   OAUTH: "oauth",
@@ -321,10 +393,145 @@ var mongodbOnboarding = new ConnectorOnboarding({
   }
 });
 
+// ../connectors/src/connectors/mongodb/utils.ts
+var CONNECT_TIMEOUT_MS2 = 1e4;
+async function withClient(params, op) {
+  const { MongoClient } = await import("mongodb");
+  const connectionUrl = params[parameters.connectionUrl.slug];
+  const client = new MongoClient(connectionUrl, {
+    connectTimeoutMS: CONNECT_TIMEOUT_MS2,
+    serverSelectionTimeoutMS: CONNECT_TIMEOUT_MS2
+  });
+  try {
+    await client.connect();
+    return await op(client);
+  } finally {
+    await client.close().catch(() => {
+    });
+  }
+}
+var MONGO_SYSTEM_DATABASES = /* @__PURE__ */ new Set(["admin", "local", "config"]);
+async function listDatabases(params) {
+  return withClient(params, async (client) => {
+    const res = await client.db("admin").admin().listDatabases();
+    return (res.databases ?? []).map((d) => d.name).filter(
+      (name) => typeof name === "string" && name !== "" && !MONGO_SYSTEM_DATABASES.has(name)
+    );
+  });
+}
+async function listCollections(params, database) {
+  return withClient(params, async (client) => {
+    const collections = await client.db(database).listCollections({}, { nameOnly: true }).toArray();
+    return collections.map((c) => c.name).filter((name) => name);
+  });
+}
+async function sampleCollectionShape(params, database, collection) {
+  return withClient(params, async (client) => {
+    const doc = await client.db(database).collection(collection).findOne({});
+    if (!doc) return [];
+    return Object.entries(doc).map(([field, value]) => ({
+      field,
+      type: mongoFieldType(value)
+    }));
+  });
+}
+function mongoFieldType(value) {
+  if (value === null) return "null";
+  if (Array.isArray(value)) return "array";
+  if (value instanceof Date) return "date";
+  if (typeof value === "object") {
+    const ctorName = value.constructor?.name;
+    if (ctorName === "ObjectId") return "ObjectId";
+    if (ctorName && ctorName !== "Object") return ctorName;
+    return "object";
+  }
+  return typeof value;
+}
+
+// ../connectors/src/connectors/mongodb/setup-flow.ts
+var ALL_COLLECTIONS = "__ALL_COLLECTIONS__";
+var MONGODB_SETUP_MAX_COLLECTIONS = 20;
+var mongodbSetupFlow = {
+  initialState: () => ({}),
+  steps: [
+    {
+      slug: "database",
+      type: "select",
+      question: {
+        ja: "\u30BB\u30C3\u30C8\u30A2\u30C3\u30D7\u306B\u4F7F\u3046\u30C7\u30FC\u30BF\u30D9\u30FC\u30B9\u3092\u9078\u3093\u3067\u304F\u3060\u3055\u3044",
+        en: "Select the database to use for setup"
+      },
+      async fetchOptions(_state, rt) {
+        const dbs = await listDatabases(rt.params);
+        return dbs.map((value) => ({ value }));
+      },
+      applyAnswer: (state, answer) => ({ ...state, database: answer[0] })
+    },
+    {
+      slug: "collections",
+      type: "multiSelect",
+      question: {
+        ja: "\u5BFE\u8C61\u30B3\u30EC\u30AF\u30B7\u30E7\u30F3\u3092\u9078\u3093\u3067\u304F\u3060\u3055\u3044\uFF08\u8907\u6570\u9078\u629E\u53EF\uFF09",
+        en: "Select target collections (multi-select allowed)"
+      },
+      async fetchOptions(state, rt) {
+        if (!state.database) return [];
+        const cols = await listCollections(rt.params, state.database);
+        const options = cols.map((value) => ({ value }));
+        return [
+          {
+            value: ALL_COLLECTIONS,
+            label: rt.language === "ja" ? "\u3059\u3079\u3066\u306E\u30B3\u30EC\u30AF\u30B7\u30E7\u30F3" : "All collections"
+          },
+          ...options
+        ];
+      },
+      applyAnswer: (state, answer) => ({ ...state, collections: answer })
+    }
+  ],
+  async finalize(state, rt) {
+    if (!state.database || !state.collections) {
+      throw new Error("MongoDB setup: incomplete state on finalize");
+    }
+    const database = state.database;
+    const targetCollections = await resolveSetupSelection({
+      selected: state.collections,
+      allSentinel: ALL_COLLECTIONS,
+      fetchAll: () => listCollections(rt.params, database),
+      limit: MONGODB_SETUP_MAX_COLLECTIONS
+    });
+    const sections = [
+      "## MongoDB",
+      "",
+      `### Database: ${database}`,
+      ""
+    ];
+    for (const collection of targetCollections) {
+      const fields = await sampleCollectionShape(
+        rt.params,
+        database,
+        collection
+      );
+      sections.push(`#### Collection: ${collection}`, "");
+      if (fields.length === 0) {
+        sections.push("_(empty collection \u2014 no sample document available)_", "");
+        continue;
+      }
+      sections.push("| Field | Sampled Type |");
+      sections.push("|-------|--------------|");
+      for (const f of fields) {
+        sections.push(`| ${f.field} | ${f.type} |`);
+      }
+      sections.push("");
+    }
+    return sections.join("\n");
+  }
+};
+
 // ../connectors/src/connectors/mongodb/tools/find-documents.ts
 import { z } from "zod";
 var MAX_DOCUMENTS = 500;
-var CONNECT_TIMEOUT_MS2 = 1e4;
+var CONNECT_TIMEOUT_MS3 = 1e4;
 var QUERY_TIMEOUT_MS2 = 6e4;
 var inputSchema = z.object({
   toolUseIntent: z.string().optional().describe(
@@ -382,8 +589,8 @@ For listing collections, use the aggregate tool with an empty pipeline on the sp
       connectionUrl = parameters.connectionUrl.getValue(connection2);
       const database = parameters.database.getValue(connection2);
       const client = new MongoClient(connectionUrl, {
-        connectTimeoutMS: CONNECT_TIMEOUT_MS2,
-        serverSelectionTimeoutMS: CONNECT_TIMEOUT_MS2,
+        connectTimeoutMS: CONNECT_TIMEOUT_MS3,
+        serverSelectionTimeoutMS: CONNECT_TIMEOUT_MS3,
         socketTimeoutMS: QUERY_TIMEOUT_MS2
       });
       try {
@@ -429,7 +636,7 @@ For listing collections, use the aggregate tool with an empty pipeline on the sp
 // ../connectors/src/connectors/mongodb/tools/aggregate.ts
 import { z as z2 } from "zod";
 var MAX_DOCUMENTS2 = 500;
-var CONNECT_TIMEOUT_MS3 = 1e4;
+var CONNECT_TIMEOUT_MS4 = 1e4;
 var QUERY_TIMEOUT_MS3 = 6e4;
 var inputSchema2 = z2.object({
   toolUseIntent: z2.string().optional().describe(
@@ -479,8 +686,8 @@ Common stages: $match (filter), $group (aggregate), $sort (order), $project (res
       connectionUrl = parameters.connectionUrl.getValue(connection2);
       const database = parameters.database.getValue(connection2);
       const client = new MongoClient(connectionUrl, {
-        connectTimeoutMS: CONNECT_TIMEOUT_MS3,
-        serverSelectionTimeoutMS: CONNECT_TIMEOUT_MS3,
+        connectTimeoutMS: CONNECT_TIMEOUT_MS4,
+        serverSelectionTimeoutMS: CONNECT_TIMEOUT_MS4,
         socketTimeoutMS: QUERY_TIMEOUT_MS3
       });
       try {
@@ -518,7 +725,7 @@ Common stages: $match (filter), $group (aggregate), $sort (order), $project (res
 
 // ../connectors/src/connectors/mongodb/tools/list-collections.ts
 import { z as z3 } from "zod";
-var CONNECT_TIMEOUT_MS4 = 1e4;
+var CONNECT_TIMEOUT_MS5 = 1e4;
 var inputSchema3 = z3.object({
   toolUseIntent: z3.string().optional().describe(
     "Brief description of what you intend to accomplish with this tool call"
@@ -562,8 +769,8 @@ After listing collections, use the find tool to sample documents and understand
       connectionUrl = parameters.connectionUrl.getValue(connection2);
       const database = parameters.database.getValue(connection2);
       const client = new MongoClient(connectionUrl, {
-        connectTimeoutMS: CONNECT_TIMEOUT_MS4,
-        serverSelectionTimeoutMS: CONNECT_TIMEOUT_MS4
+        connectTimeoutMS: CONNECT_TIMEOUT_MS5,
+        serverSelectionTimeoutMS: CONNECT_TIMEOUT_MS5
       });
       try {
         await client.connect();
@@ -718,6 +925,7 @@ export default async function handler(_c: Context) {
 - find \u30AF\u30A8\u30EA\u306B\u306F\u5FC5\u305A limit \u3092\u3001\u96C6\u7D04\u30D1\u30A4\u30D7\u30E9\u30A4\u30F3\u306B\u306F $limit \u3092\u542B\u3081\u3066\u304F\u3060\u3055\u3044`
   },
   tools,
+  setup: (params, ctx, config) => runSetupFlow(mongodbSetupFlow, params, ctx, config),
   async checkConnection(params, _config) {
     try {
       const { MongoClient } = await import("mongodb");
@@ -765,6 +973,7 @@ function resolveEnvVarOptional(entry, key) {
 import { getContext } from "hono/context-storage";
 import { getCookie } from "hono/cookie";
 var APP_SESSION_COOKIE_NAME = "__Host-squadbase-session";
+var TABLEAU_SESSION_SENTINEL_URL = "squadbase://tableau-session/";
 function normalizeHeaders(input) {
   const out = {};
   if (!input) return out;
@@ -773,6 +982,11 @@ function normalizeHeaders(input) {
   });
   return out;
 }
+function extractInputUrl(input) {
+  if (typeof input === "string") return input;
+  if (input instanceof URL) return input.href;
+  return input.url;
+}
 function createSandboxProxyFetch(connectionId) {
   return async (input, init) => {
     const token = process.env.INTERNAL_SQUADBASE_OAUTH_MACHINE_CREDENTIAL;
@@ -782,10 +996,17 @@ function createSandboxProxyFetch(connectionId) {
         "Connection proxy is not configured. Please check your deployment settings."
       );
     }
-    const originalUrl = typeof input === "string" ? input : input instanceof URL ? input.href : input.url;
+    const originalUrl = extractInputUrl(input);
+    const baseDomain = process.env["SQUADBASE_PREVIEW_BASE_DOMAIN"] ?? "preview.app.squadbase.dev";
+    if (originalUrl === TABLEAU_SESSION_SENTINEL_URL) {
+      const sessionUrl = `https://${sandboxId}.${baseDomain}/_sqcore/connections/${connectionId}/tableau-session`;
+      return fetch(sessionUrl, {
+        method: "POST",
+        headers: { Authorization: `Bearer ${token}` }
+      });
+    }
     const originalMethod = init?.method ?? "GET";
     const originalBody = init?.body ? JSON.parse(init.body) : void 0;
-    const baseDomain = process.env["SQUADBASE_PREVIEW_BASE_DOMAIN"] ?? "preview.app.squadbase.dev";
     const proxyUrl = `https://${sandboxId}.${baseDomain}/_sqcore/connections/${connectionId}/request`;
     return fetch(proxyUrl, {
       method: "POST",
@@ -811,10 +1032,9 @@ function createDeployedAppProxyFetch(connectionId) {
   }
   const baseDomain = process.env["SQUADBASE_APP_BASE_DOMAIN"] ?? "squadbase.app";
   const proxyUrl = `https://${projectId}.${baseDomain}/_sqcore/connections/${connectionId}/request`;
+  const sessionUrl = `https://${projectId}.${baseDomain}/_sqcore/connections/${connectionId}/tableau-session`;
   return async (input, init) => {
-    const originalUrl = typeof input === "string" ? input : input instanceof URL ? input.href : input.url;
-    const originalMethod = init?.method ?? "GET";
-    const originalBody = init?.body ? JSON.parse(init.body) : void 0;
+    const originalUrl = extractInputUrl(input);
     const c = getContext();
     const appSession = getCookie(c, APP_SESSION_COOKIE_NAME);
     if (!appSession) {
@@ -822,6 +1042,14 @@ function createDeployedAppProxyFetch(connectionId) {
         "No authentication method available for connection proxy."
       );
     }
+    if (originalUrl === TABLEAU_SESSION_SENTINEL_URL) {
+      return fetch(sessionUrl, {
+        method: "POST",
+        headers: { Authorization: `Bearer ${appSession}` }
+      });
+    }
+    const originalMethod = init?.method ?? "GET";
+    const originalBody = init?.body ? JSON.parse(init.body) : void 0;
     return fetch(proxyUrl, {
       method: "POST",
       headers: {