@squadbase/vite-server 0.1.12-dev.a9ac647 → 0.1.17-dev.3b633bb

package/dist/connectors/shopify-oauth.js

@@ -65,6 +65,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;
@@ -81,6 +103,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);
@@ -145,6 +169,71 @@ 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;
+  const pendingParameterUpdates = [];
+  for (const step of flow.steps) {
+    const ans = ctx.answers[answerIdx];
+    if (ans && ans.questionSlug === step.slug) {
+      state = step.applyAnswer(state, ans.answer);
+      if (step.toParameterUpdates) {
+        pendingParameterUpdates.push(...step.toParameterUpdates(state));
+      }
+      answerIdx += 1;
+      continue;
+    }
+    const resolvedAllowFreeText = step.allowFreeText !== void 0 ? step.allowFreeText : true;
+    if (step.type === "text") {
+      if (step.fetchOptions) {
+        const options2 = await step.fetchOptions(state, runtime);
+        if (options2.length === 0) {
+          continue;
+        }
+      }
+      return {
+        type: "nextQuestion",
+        questionSlug: step.slug,
+        question: step.question[ctx.language],
+        questionType: "text",
+        allowFreeText: resolvedAllowFreeText,
+        ...pendingParameterUpdates.length > 0 && {
+          parameterUpdates: pendingParameterUpdates
+        }
+      };
+    }
+    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,
+      allowFreeText: resolvedAllowFreeText,
+      ...pendingParameterUpdates.length > 0 && {
+        parameterUpdates: pendingParameterUpdates
+      }
+    };
+  }
+  const dataInvestigationResult = await flow.finalize(state, runtime);
+  return {
+    type: "fulfilled",
+    dataInvestigationResult,
+    ...pendingParameterUpdates.length > 0 && {
+      parameterUpdates: pendingParameterUpdates
+    }
+  };
+}
+
 // ../connectors/src/auth-types.ts
 var AUTH_TYPES = {
   OAUTH: "oauth",
@@ -307,11 +396,83 @@ var shopifyOauthOnboarding = new ConnectorOnboarding({
 // ../connectors/src/connectors/shopify-oauth/parameters.ts
 var parameters = {};
 
+// ../connectors/src/connectors/shopify-oauth/setup-flow.ts
+var SHOPIFY_API_VERSION = "2024-10";
+var SHOPIFY_OAUTH_SETUP_MAX_ENTITIES = 10;
+var ENTITY_LABELS = {
+  products: { en: "Products", ja: "Products (\u5546\u54C1)" },
+  orders: { en: "Orders", ja: "Orders (\u6CE8\u6587)" },
+  customers: { en: "Customers", ja: "Customers (\u9867\u5BA2)" },
+  collections: { en: "Collections", ja: "Collections (\u30B3\u30EC\u30AF\u30B7\u30E7\u30F3)" },
+  inventory: { en: "Inventory items", ja: "Inventory items (\u5728\u5EAB)" },
+  discounts: { en: "Discounts", ja: "Discounts (\u5272\u5F15)" }
+};
+var ENTITY_VALUES = [
+  "products",
+  "orders",
+  "customers",
+  "collections",
+  "inventory",
+  "discounts"
+];
+var COUNT_PATHS = {
+  products: `/admin/api/${SHOPIFY_API_VERSION}/products/count.json`,
+  orders: `/admin/api/${SHOPIFY_API_VERSION}/orders/count.json?status=any`,
+  customers: `/admin/api/${SHOPIFY_API_VERSION}/customers/count.json`,
+  collections: `/admin/api/${SHOPIFY_API_VERSION}/custom_collections/count.json`,
+  inventory: null,
+  discounts: `/admin/api/${SHOPIFY_API_VERSION}/price_rules/count.json`
+};
+var shopifyOauthSetupFlow = {
+  initialState: () => ({}),
+  steps: [
+    {
+      slug: "entities",
+      type: "multiSelect",
+      question: {
+        ja: "\u5BFE\u8C61\u306E\u30A8\u30F3\u30C6\u30A3\u30C6\u30A3\u3092\u9078\u3093\u3067\u304F\u3060\u3055\u3044\uFF08\u8907\u6570\u9078\u629E\u53EF\uFF09",
+        en: "Select target entities (multi-select allowed)"
+      },
+      async fetchOptions(_state, rt) {
+        return ENTITY_VALUES.map((value) => ({
+          value,
+          label: ENTITY_LABELS[value][rt.language]
+        }));
+      },
+      applyAnswer: (state, answer) => ({ ...state, entities: answer })
+    }
+  ],
+  async finalize(state, rt) {
+    if (!state.entities) {
+      throw new Error("Shopify setup: incomplete state on finalize");
+    }
+    const selected = state.entities.filter(
+      (e) => ENTITY_VALUES.includes(e)
+    ).slice(0, SHOPIFY_OAUTH_SETUP_MAX_ENTITIES);
+    const sections = ["## Shopify", ""];
+    for (const entity of selected) {
+      const path2 = COUNT_PATHS[entity];
+      let count = "available";
+      if (path2) {
+        const res = await rt.config.proxyFetch(path2, { method: "GET" });
+        if (res.ok) {
+          const data = await res.json();
+          if (typeof data.count === "number") count = String(data.count);
+        }
+      }
+      sections.push(`### ${ENTITY_LABELS[entity].en}`, "");
+      sections.push(`Count: ${count}`, "");
+    }
+    return sections.join("\n");
+  }
+};
+
 // ../connectors/src/connectors/shopify-oauth/index.ts
 var tools = { request: requestTool };
 var shopifyOauthConnector = new ConnectorPlugin({
   slug: "shopify",
   authType: AUTH_TYPES.OAUTH,
+  skipConnectionCheckOnCreate: true,
   name: "Shopify",
   description: "Connect to Shopify for e-commerce data including products, orders, and customers using OAuth.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/57KEjZCBskKgSxgKyU4Sm0/117d681a410f48dc36f97cdd9c0593c5/shopify-icon.svg",
@@ -424,6 +585,7 @@ const data = await res.json();
 \`\`\``
   },
   tools,
+  setup: (params, ctx, config) => runSetupFlow(shopifyOauthSetupFlow, params, ctx, config),
   async checkConnection(_params, config) {
     const { proxyFetch } = config;
     try {
@@ -473,6 +635,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;
@@ -481,6 +644,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;
@@ -490,10 +658,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",
@@ -519,10 +694,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) {
@@ -530,6 +704,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: {