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

package/dist/connectors/influxdb.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/influxdb/parameters.ts
+init_parameter_definition();
 var parameters = {
   url: new ParameterDefinition({
     slug: "url",
@@ -103,7 +115,7 @@ function createClient(params) {
       `influxdb: missing required parameter: ${parameters.database.slug}`
     );
   }
-  function authHeaders(extra) {
+  function authHeaders2(extra) {
     const headers = new Headers(extra);
     headers.set("Authorization", `Token ${token}`);
     if (!headers.has("Accept")) headers.set("Accept", "application/json");
@@ -139,7 +151,7 @@ function createClient(params) {
     async querySql(sql, options) {
       const res = await fetch(`${url}/api/v3/query_sql`, {
         method: "POST",
-        headers: authHeaders({ "Content-Type": "application/json" }),
+        headers: authHeaders2({ "Content-Type": "application/json" }),
         body: JSON.stringify({
           db: options?.database ?? database,
           q: sql
@@ -151,7 +163,7 @@ function createClient(params) {
     async queryInfluxql(influxql, options) {
       const res = await fetch(`${url}/api/v3/query_influxql`, {
         method: "POST",
-        headers: authHeaders({ "Content-Type": "application/json" }),
+        headers: authHeaders2({ "Content-Type": "application/json" }),
         body: JSON.stringify({
           db: options?.database ?? database,
           q: influxql
@@ -171,7 +183,7 @@ function createClient(params) {
         `${url}/api/v2/query?org=${encodeURIComponent(resolvedOrg)}`,
         {
           method: "POST",
-          headers: authHeaders({
+          headers: authHeaders2({
             "Content-Type": "application/vnd.flux",
             Accept: "application/csv"
           }),
@@ -188,7 +200,7 @@ function createClient(params) {
       const endpoint = resolvedOrg ? `${url}/api/v2/write?org=${encodeURIComponent(resolvedOrg)}&bucket=${encodeURIComponent(resolvedDatabase)}&precision=${precision}` : `${url}/api/v3/write_lp?db=${encodeURIComponent(resolvedDatabase)}&precision=${precision}`;
       const res = await fetch(endpoint, {
         method: "POST",
-        headers: authHeaders({ "Content-Type": "text/plain; charset=utf-8" }),
+        headers: authHeaders2({ "Content-Type": "text/plain; charset=utf-8" }),
         body: lineProtocol
       });
       await assertOk(res, "writeLineProtocol");
@@ -255,6 +267,28 @@ var ConnectorPlugin = class _ConnectorPlugin {
   tools;
   query;
   checkConnection;
+  /**
+   * SQPD-1212: Logic-based, rule-driven connection setup. Connectors that
+   * implement this expose a step-by-step exploration flow (database/schema/
+   * table/etc. discovery) that the dashboard backend drives via the
+   * `/connections/:connectionId/setup` endpoint. Implement by delegating to
+   * `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;
@@ -271,6 +305,8 @@ var ConnectorPlugin = class _ConnectorPlugin {
     this.tools = config.tools;
     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);
@@ -335,6 +371,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",
@@ -403,6 +484,244 @@ var influxdbOnboarding = new ConnectorOnboarding({
   }
 });
 
+// ../connectors/src/connectors/influxdb/utils.ts
+function authHeaders(token, extra) {
+  const headers = new Headers(extra);
+  headers.set("Authorization", `Token ${token}`);
+  if (!headers.has("Accept")) headers.set("Accept", "application/json");
+  return headers;
+}
+async function detectVariant(params) {
+  const url = (params[parameters.url.slug] ?? "").replace(/\/$/, "");
+  const token = params[parameters.token.slug];
+  const database = params[parameters.database.slug];
+  const res = await fetch(`${url}/api/v3/query_sql`, {
+    method: "POST",
+    headers: authHeaders(token, { "Content-Type": "application/json" }),
+    body: JSON.stringify({ db: database, q: "SELECT 1" })
+  });
+  if (res.ok) {
+    const contentType = res.headers.get("content-type") ?? "";
+    if (contentType.includes("application/json")) return "v3";
+  }
+  return "v2";
+}
+async function runSql(params, sql) {
+  const url = (params[parameters.url.slug] ?? "").replace(/\/$/, "");
+  const token = params[parameters.token.slug];
+  const database = params[parameters.database.slug];
+  const res = await fetch(`${url}/api/v3/query_sql`, {
+    method: "POST",
+    headers: authHeaders(token, { "Content-Type": "application/json" }),
+    body: JSON.stringify({ db: database, q: sql })
+  });
+  if (!res.ok) {
+    const body = await res.text().catch(() => res.statusText);
+    throw new Error(
+      `InfluxDB SQL query failed: HTTP ${res.status} \u2014 ${body.slice(0, 200)}`
+    );
+  }
+  return await res.json();
+}
+async function runFlux(params, flux) {
+  const url = (params[parameters.url.slug] ?? "").replace(/\/$/, "");
+  const token = params[parameters.token.slug];
+  const org = params[parameters.org.slug];
+  if (!org) {
+    throw new Error(
+      "InfluxDB 2 Flux queries require the 'org' connection parameter, which is not configured"
+    );
+  }
+  const res = await fetch(
+    `${url}/api/v2/query?org=${encodeURIComponent(org)}`,
+    {
+      method: "POST",
+      headers: authHeaders(token, {
+        "Content-Type": "application/vnd.flux",
+        Accept: "application/csv"
+      }),
+      body: flux
+    }
+  );
+  if (!res.ok) {
+    const body = await res.text().catch(() => res.statusText);
+    throw new Error(
+      `InfluxDB Flux query failed: HTTP ${res.status} \u2014 ${body.slice(0, 200)}`
+    );
+  }
+  return await res.text();
+}
+function parseAnnotatedCsv(csv) {
+  const lines = csv.split(/\r?\n/).filter((line) => line.trim().length > 0);
+  const rows = [];
+  let headers = null;
+  for (const line of lines) {
+    if (line.startsWith("#")) {
+      headers = null;
+      continue;
+    }
+    const fields = splitCsvLine(line);
+    if (headers === null) {
+      headers = fields;
+      continue;
+    }
+    const row = {};
+    headers.forEach((h, i) => {
+      row[h] = fields[i] ?? "";
+    });
+    rows.push(row);
+  }
+  return { columns: headers ?? [], rows };
+}
+function splitCsvLine(line) {
+  const out = [];
+  let cur = "";
+  let inQuotes = false;
+  for (let i = 0; i < line.length; i++) {
+    const c = line[i];
+    if (inQuotes) {
+      if (c === '"' && line[i + 1] === '"') {
+        cur += '"';
+        i += 1;
+      } else if (c === '"') {
+        inQuotes = false;
+      } else {
+        cur += c;
+      }
+    } else if (c === '"') {
+      inQuotes = true;
+    } else if (c === ",") {
+      out.push(cur);
+      cur = "";
+    } else {
+      cur += c;
+    }
+  }
+  out.push(cur);
+  return out;
+}
+
+// ../connectors/src/connectors/influxdb/setup-flow.ts
+var ALL_MEASUREMENTS = "__ALL_MEASUREMENTS__";
+var INFLUXDB_SETUP_MAX_MEASUREMENTS = 20;
+async function listMeasurements(params, variant) {
+  if (variant === "v3") {
+    const rows2 = await runSql(params, "SHOW TABLES");
+    return rows2.map((r) => String(r["table_name"] ?? r["name"] ?? "")).filter((name) => name).filter((name) => !name.includes("::"));
+  }
+  const bucket = params[parameters.database.slug];
+  if (!bucket) {
+    throw new Error(
+      "InfluxDB 2 setup: 'database' (bucket) parameter is required to list measurements"
+    );
+  }
+  const flux = [
+    `import "influxdata/influxdb/schema"`,
+    `schema.measurements(bucket: ${JSON.stringify(bucket)})`
+  ].join("\n");
+  const csv = await runFlux(params, flux);
+  const { rows } = parseAnnotatedCsv(csv);
+  return rows.map((r) => r["_value"] ?? "").filter((name) => name);
+}
+async function listMeasurementColumns(params, variant, measurement) {
+  if (variant === "v3") {
+    const safeName2 = measurement.replace(/"/g, '""');
+    const rows = await runSql(params, `SHOW COLUMNS FROM "${safeName2}"`);
+    return rows.map((r) => ({
+      name: String(r["column_name"] ?? r["name"] ?? ""),
+      kind: String(r["data_type"] ?? r["type"] ?? "")
+    }));
+  }
+  const bucket = params[parameters.database.slug];
+  const safeName = measurement.replace(/"/g, '\\"');
+  const fieldsFlux = [
+    `import "influxdata/influxdb/schema"`,
+    `schema.measurementFieldKeys(bucket: ${JSON.stringify(bucket)}, measurement: "${safeName}")`
+  ].join("\n");
+  const tagsFlux = [
+    `import "influxdata/influxdb/schema"`,
+    `schema.measurementTagKeys(bucket: ${JSON.stringify(bucket)}, measurement: "${safeName}")`
+  ].join("\n");
+  const [fieldsCsv, tagsCsv] = await Promise.all([
+    runFlux(params, fieldsFlux),
+    runFlux(params, tagsFlux)
+  ]);
+  const fields = parseAnnotatedCsv(fieldsCsv).rows.map((r) => ({
+    name: r["_value"] ?? "",
+    kind: "field"
+  }));
+  const tags = parseAnnotatedCsv(tagsCsv).rows.map((r) => ({
+    name: r["_value"] ?? "",
+    kind: "tag"
+  }));
+  return [...tags, ...fields].filter((c) => c.name);
+}
+var influxdbSetupFlow = {
+  initialState: () => ({}),
+  steps: [
+    {
+      slug: "measurements",
+      type: "multiSelect",
+      question: {
+        ja: "\u5BFE\u8C61 measurement \u3092\u9078\u3093\u3067\u304F\u3060\u3055\u3044\uFF08\u8907\u6570\u9078\u629E\u53EF\uFF09",
+        en: "Select target measurements (multi-select allowed)"
+      },
+      async fetchOptions(_state, rt) {
+        const variant = await detectVariant(rt.params);
+        const ms = await listMeasurements(rt.params, variant);
+        const options = ms.map((value) => ({ value }));
+        return [
+          {
+            value: ALL_MEASUREMENTS,
+            label: rt.language === "ja" ? "\u3059\u3079\u3066\u306E measurement" : "All measurements"
+          },
+          ...options
+        ];
+      },
+      applyAnswer: (state, answer) => ({ ...state, measurements: answer })
+    }
+  ],
+  async finalize(state, rt) {
+    if (!state.measurements) {
+      throw new Error("InfluxDB setup: incomplete state on finalize");
+    }
+    const variant = await detectVariant(rt.params);
+    const database = rt.params[parameters.database.slug];
+    const org = rt.params[parameters.org.slug];
+    const targetMeasurements = await resolveSetupSelection({
+      selected: state.measurements,
+      allSentinel: ALL_MEASUREMENTS,
+      fetchAll: () => listMeasurements(rt.params, variant),
+      limit: INFLUXDB_SETUP_MAX_MEASUREMENTS
+    });
+    const sections = [
+      "## InfluxDB",
+      "",
+      `### Variant: ${variant === "v3" ? "InfluxDB 3 (SQL)" : "InfluxDB 2 (Flux)"}`,
+      `### ${variant === "v3" ? "Database" : "Bucket"}: ${database}`
+    ];
+    if (variant === "v2" && org) {
+      sections.push(`### Organization: ${org}`);
+    }
+    sections.push("");
+    for (const measurement of targetMeasurements) {
+      const cols = await listMeasurementColumns(rt.params, variant, measurement);
+      sections.push(`#### Measurement: ${measurement}`, "");
+      if (cols.length === 0) {
+        sections.push("_(no fields or tags found)_", "");
+        continue;
+      }
+      sections.push("| Column | Kind |");
+      sections.push("|--------|------|");
+      for (const c of cols) {
+        sections.push(`| ${c.name} | ${c.kind || "-"} |`);
+      }
+      sections.push("");
+    }
+    return sections.join("\n");
+  }
+};
+
 // ../connectors/src/connectors/influxdb/tools/request.ts
 import { z } from "zod";
 var REQUEST_TIMEOUT_MS = 6e4;
@@ -669,6 +988,7 @@ export default async function handler(c: Context) {
 - measurement \u4E00\u89A7: \`SHOW TABLES\`\u3001\u5217\u4E00\u89A7: \`SHOW COLUMNS FROM <measurement>\``
   },
   tools,
+  setup: (params, ctx, config) => runSetupFlow(influxdbSetupFlow, params, ctx, config),
   async checkConnection(params) {
     const url = (params.url ?? "").replace(/\/$/, "");
     const token = params.token;
@@ -737,6 +1057,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;
@@ -745,6 +1066,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;
@@ -754,10 +1080,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",
@@ -783,10 +1116,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) {
@@ -794,6 +1126,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: {