@tetrascience-npm/tetrascience-react-ui 0.5.0-beta.20.1 → 0.5.0-beta.22.1

package/dist/lib/utils.cjs

@@ -0,0 +1,2 @@
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const r=require("clsx"),t=require("tailwind-merge");function n(...e){return t.twMerge(r.clsx(e))}exports.cn=n;
+//# sourceMappingURL=utils.cjs.map

package/dist/lib/utils.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.cjs","sources":["../../src/lib/utils.ts"],"sourcesContent":["import { clsx, type ClassValue } from \"clsx\"\nimport { twMerge } from \"tailwind-merge\"\n\nexport function cn(...inputs: ClassValue[]) {\n  return twMerge(clsx(inputs))\n}\n"],"names":["cn","inputs","twMerge","clsx"],"mappings":"oIAGO,SAASA,KAAMC,EAAsB,CAC1C,OAAOC,EAAAA,QAAQC,OAAKF,CAAM,CAAC,CAC7B"}

package/dist/lib/utils.js

@@ -0,0 +1,9 @@
+import { clsx as o } from "clsx";
+import { twMerge as t } from "tailwind-merge";
+function n(...r) {
+  return t(o(r));
+}
+export {
+  n as cn
+};
+//# sourceMappingURL=utils.js.map

package/dist/lib/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sources":["../../src/lib/utils.ts"],"sourcesContent":["import { clsx, type ClassValue } from \"clsx\"\nimport { twMerge } from \"tailwind-merge\"\n\nexport function cn(...inputs: ClassValue[]) {\n  return twMerge(clsx(inputs))\n}\n"],"names":["cn","inputs","twMerge","clsx"],"mappings":";;AAGO,SAASA,KAAMC,GAAsB;AAC1C,SAAOC,EAAQC,EAAKF,CAAM,CAAC;AAC7B;"}

package/dist/providers/athena.cjs

@@ -1,2 +1,2 @@
-"use strict";var g=Object.create;var p=Object.defineProperty;var f=Object.getOwnPropertyDescriptor;var S=Object.getOwnPropertyNames;var C=Object.getPrototypeOf,T=Object.prototype.hasOwnProperty;var k=(e,t,r,o)=>{if(t&&typeof t=="object"||typeof t=="function")for(let n of S(t))!T.call(e,n)&&n!==r&&p(e,n,{get:()=>t[n],enumerable:!(o=f(t,n))||o.enumerable});return e};var v=(e,t,r)=>(r=e!=null?g(C(e)):{},k(t||!e||!e.__esModule?p(r,"default",{value:e,enumerable:!0}):r,e));Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const i=require("../exceptions-DN25pCDi.cjs"),x=1e3;async function Q(){try{return await import("@aws-sdk/client-athena")}catch{throw new i.InvalidProviderConfigurationError("The '@aws-sdk/client-athena' package is required to use the Athena provider. Please install it: npm install @aws-sdk/client-athena")}}const _=262144;class m{client;sdk;workgroup;database;outputLocation;constructor(t,r,o,n,a){this.client=t,this.sdk=r,this.workgroup=o,this.database=n,this.outputLocation=a}async query(t,r={}){if(t.length>_)throw new Error("Query length exceeds the maximum allowed limit.");const o=new this.sdk.StartQueryExecutionCommand({QueryString:t,WorkGroup:this.workgroup,QueryExecutionContext:{Database:this.database},ResultConfiguration:this.outputLocation?{OutputLocation:this.outputLocation}:void 0}),a=(await this.client.send(o)).QueryExecutionId;if(!a)throw new i.QueryError("Failed to start query execution");return await this.waitForQueryCompletion(a,t),this.fetchAllResults(a)}async waitForQueryCompletion(t,r){const a=Date.now();for(;Date.now()-a<3e5;){const l=new this.sdk.GetQueryExecutionCommand({QueryExecutionId:t}),s=await this.client.send(l),u=s.QueryExecution?.Status?.State;if(u===this.sdk.QueryExecutionState.SUCCEEDED)return;if(u===this.sdk.QueryExecutionState.FAILED||u===this.sdk.QueryExecutionState.CANCELLED){const c=s.QueryExecution?.Status?.StateChangeReason??"Unknown error";if(c.includes("TABLE_NOT_FOUND")){const h=c.split(":").pop()?.trim()??"";throw new i.MissingTableError("Athena is unable to find the table. If the table is created by a tetraflow, make sure that the tetraflow has run successfully. "+h+".")}throw new i.QueryError(`Query failed: ${r}. Reason: ${c}`)}await new Promise(c=>setTimeout(c,1e3))}throw new i.QueryError(`Query timed out after ${3e5/x} seconds`)}async fetchAllResults(t){const r=[];let o,n=[],a=!0;do{const l=new this.sdk.GetQueryResultsCommand({QueryExecutionId:t,NextToken:o}),s=await this.client.send(l);a&&s.ResultSet?.ResultSetMetadata?.ColumnInfo&&(n=s.ResultSet.ResultSetMetadata.ColumnInfo.map((h,w)=>{const d=h.Name;return d&&d.trim()!==""?d:`column_${w}`}));const u=s.ResultSet?.Rows??[],c=a?u.slice(1):u;for(const h of c){const w={};h.Data?.forEach((d,E)=>{const y=n[E]??`column_${E}`;w[y]=d.VarCharValue??null}),r.push(w)}o=s.NextToken,a=!1}while(o);return r}async close(){this.client.destroy()}}async function A(){const e=await Q(),t=process.env.ORG_SLUG??"",r=t.replace(/-/g,"_"),o=process.env.ATHENA_S3_OUTPUT_LOCATION,n=process.env.AWS_REGION,a=`${r}__tss__default`,l=t,s=new e.AthenaClient({region:n});try{return await s.send(new e.GetWorkGroupCommand({WorkGroup:l})),new m(s,e,l,a)}catch{if(!o)throw new Error("ATHENA_S3_OUTPUT_LOCATION environment variable is required when using the 'primary' workgroup. Either set this variable or ensure the org-specific workgroup exists.");const u=`s3://${o}/${r}/`;return new m(s,e,"primary",a,u)}}exports.InvalidProviderConfigurationError=i.InvalidProviderConfigurationError;exports.MissingTableError=i.MissingTableError;exports.ProviderConnectionError=i.ProviderConnectionError;exports.ProviderError=i.ProviderError;exports.QueryError=i.QueryError;exports.AthenaProvider=m;exports.getTdpAthenaProvider=A;
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const e=require("../server/providers/AthenaProvider.cjs"),r=require("../server/providers/exceptions.cjs");exports.AthenaProvider=e.AthenaProvider;exports.getTdpAthenaProvider=e.getTdpAthenaProvider;exports.InvalidProviderConfigurationError=r.InvalidProviderConfigurationError;exports.MissingTableError=r.MissingTableError;exports.ProviderConnectionError=r.ProviderConnectionError;exports.ProviderError=r.ProviderError;exports.QueryError=r.QueryError;
 //# sourceMappingURL=athena.cjs.map

package/dist/providers/athena.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"athena.cjs","sources":["../../src/server/providers/AthenaProvider.ts"],"sourcesContent":["/**\n * Athena Data Provider\n *\n * TypeScript equivalent of AthenaProvider from\n * ts-lib-ui-kit-streamlit/tetrascience/data_app_providers/provider.py\n *\n * @remarks\n * This provider requires the `@aws-sdk/client-athena` package to be installed.\n * It is an optional peer dependency - install it only if you need Athena support:\n * ```bash\n * npm install @aws-sdk/client-athena\n * # or\n * yarn add @aws-sdk/client-athena\n * ```\n */\n\nimport {\n  QueryError,\n  MissingTableError,\n  InvalidProviderConfigurationError,\n} from \"./exceptions\";\n\n/** Milliseconds per second */\nconst MILLISECONDS_PER_SECOND = 1000;\n\n// Type imports for @aws-sdk/client-athena (these don't require the package at runtime)\ntype AthenaClient = import(\"@aws-sdk/client-athena\").AthenaClient;\ntype AthenaSDK = typeof import(\"@aws-sdk/client-athena\");\n\n/**\n * Dynamically import @aws-sdk/client-athena\n * @throws {InvalidProviderConfigurationError} If @aws-sdk/client-athena is not installed\n */\nasync function getAthenaSDK(): Promise<AthenaSDK> {\n  try {\n    return await import(\"@aws-sdk/client-athena\");\n  } catch {\n    throw new InvalidProviderConfigurationError(\n      \"The '@aws-sdk/client-athena' package is required to use the Athena provider. \" +\n        \"Please install it: npm install @aws-sdk/client-athena\",\n    );\n  }\n}\n\n/** Maximum query length allowed by Athena */\nconst MAX_QUERY_LENGTH = 262144;\n\n/**\n * Athena data provider\n */\nexport class AthenaProvider {\n  private client: AthenaClient;\n  private sdk: AthenaSDK;\n  private workgroup: string;\n  private database: string;\n  private outputLocation?: string;\n\n  /**\n   * Initialize the Athena data provider\n   *\n   * @param client - AWS Athena client\n   * @param sdk - AWS Athena SDK module (for accessing command classes)\n   * @param workgroup - Athena workgroup to use\n   * @param database - Default database/schema\n   * @param outputLocation - Optional S3 output location\n   */\n  constructor(\n    client: AthenaClient,\n    sdk: AthenaSDK,\n    workgroup: string,\n    database: string,\n    outputLocation?: string,\n  ) {\n    this.client = client;\n    this.sdk = sdk;\n    this.workgroup = workgroup;\n    this.database = database;\n    this.outputLocation = outputLocation;\n  }\n\n  /**\n   * Query the Athena database\n   *\n   * @param sqlQuery - SQL query to execute\n   * @param _params - Parameters to pass to the query (currently not used - Athena doesn't support parameterized queries)\n   * @returns Promise resolving to array of row objects\n   *\n   * @remarks\n   * **Security Note:** AWS Athena does not support parameterized queries.\n   * Unlike traditional databases, there is no native way to use bind parameters\n   * with Athena. Callers are responsible for properly sanitizing any user input\n   * before constructing the SQL query string. This is a known limitation of the\n   * Athena service, not a design flaw in this implementation.\n   */\n  async query(\n    sqlQuery: string,\n    _params: Record<string, unknown> = {},\n  ): Promise<Array<Record<string, unknown>>> {\n    if (sqlQuery.length > MAX_QUERY_LENGTH) {\n      throw new Error(\"Query length exceeds the maximum allowed limit.\");\n    }\n\n    // Start query execution\n    // Note: Athena does not support parameterized queries. The sqlQuery is passed\n    // directly to Athena. Callers must sanitize user input before constructing queries.\n    const startCommand = new this.sdk.StartQueryExecutionCommand({\n      QueryString: sqlQuery,\n      WorkGroup: this.workgroup,\n      QueryExecutionContext: {\n        Database: this.database,\n      },\n      ResultConfiguration: this.outputLocation\n        ? { OutputLocation: this.outputLocation }\n        : undefined,\n    });\n\n    const startResponse = await this.client.send(startCommand);\n    const queryExecutionId = startResponse.QueryExecutionId;\n\n    if (!queryExecutionId) {\n      throw new QueryError(\"Failed to start query execution\");\n    }\n\n    // Wait for query to complete\n    await this.waitForQueryCompletion(queryExecutionId, sqlQuery);\n\n    // Fetch results\n    return this.fetchAllResults(queryExecutionId);\n  }\n\n  /**\n   * Wait for query to complete\n   */\n  private async waitForQueryCompletion(\n    queryExecutionId: string,\n    originalQuery: string,\n  ): Promise<void> {\n    const maxWaitTime = 300000; // 5 minutes\n    const pollInterval = 1000; // 1 second\n    const startTime = Date.now();\n\n    while (Date.now() - startTime < maxWaitTime) {\n      const command = new this.sdk.GetQueryExecutionCommand({\n        QueryExecutionId: queryExecutionId,\n      });\n      const response = await this.client.send(command);\n      const state = response.QueryExecution?.Status?.State;\n\n      if (state === this.sdk.QueryExecutionState.SUCCEEDED) {\n        return;\n      }\n\n      if (\n        state === this.sdk.QueryExecutionState.FAILED ||\n        state === this.sdk.QueryExecutionState.CANCELLED\n      ) {\n        const reason =\n          response.QueryExecution?.Status?.StateChangeReason ?? \"Unknown error\";\n\n        if (reason.includes(\"TABLE_NOT_FOUND\")) {\n          const errorTail = reason.split(\":\").pop()?.trim() ?? \"\";\n          throw new MissingTableError(\n            \"Athena is unable to find the table. If the table is created by a \" +\n              \"tetraflow, make sure that the tetraflow has run successfully. \" +\n              errorTail +\n              \".\",\n          );\n        }\n\n        throw new QueryError(\n          `Query failed: ${originalQuery}. Reason: ${reason}`,\n        );\n      }\n\n      await new Promise((resolve) => setTimeout(resolve, pollInterval));\n    }\n\n    throw new QueryError(\n      `Query timed out after ${maxWaitTime / MILLISECONDS_PER_SECOND} seconds`,\n    );\n  }\n\n  /**\n   * Fetch all results from a completed query\n   */\n  private async fetchAllResults(\n    queryExecutionId: string,\n  ): Promise<Array<Record<string, unknown>>> {\n    const results: Array<Record<string, unknown>> = [];\n    let nextToken: string | undefined;\n    let columnNames: string[] = [];\n    let isFirstPage = true;\n\n    do {\n      const command = new this.sdk.GetQueryResultsCommand({\n        QueryExecutionId: queryExecutionId,\n        NextToken: nextToken,\n      });\n\n      const response = await this.client.send(command);\n\n      if (isFirstPage && response.ResultSet?.ResultSetMetadata?.ColumnInfo) {\n        columnNames = response.ResultSet.ResultSetMetadata.ColumnInfo.map(\n          (col, idx) => {\n            // Handle empty or missing column names by using a fallback\n            const name = col.Name;\n            return name && name.trim() !== \"\" ? name : `column_${idx}`;\n          },\n        );\n      }\n\n      const rows = response.ResultSet?.Rows ?? [];\n      // Skip header row on first page\n      const dataRows = isFirstPage ? rows.slice(1) : rows;\n\n      for (const row of dataRows) {\n        const rowData: Record<string, unknown> = {};\n        row.Data?.forEach((cell, index) => {\n          // columnNames already has fallback values, so we can use them directly\n          const columnName = columnNames[index] ?? `column_${index}`;\n          rowData[columnName] = cell.VarCharValue ?? null;\n        });\n        results.push(rowData);\n      }\n\n      nextToken = response.NextToken;\n      isFirstPage = false;\n    } while (nextToken);\n\n    return results;\n  }\n\n  /**\n   * Close the Athena client (no-op for AWS SDK clients)\n   */\n  async close(): Promise<void> {\n    this.client.destroy();\n  }\n}\n\n/**\n * Get the TDP Athena provider\n *\n * Creates an Athena provider using TDP environment configuration\n *\n * @returns Promise resolving to Athena data provider\n * @throws {InvalidProviderConfigurationError} If @aws-sdk/client-athena is not installed\n * @throws {Error} If ATHENA_S3_OUTPUT_LOCATION is not set when using the 'primary' workgroup\n */\nexport async function getTdpAthenaProvider(): Promise<AthenaProvider> {\n  // Dynamically import @aws-sdk/client-athena\n  const athenaSDK = await getAthenaSDK();\n\n  const orgSlug = process.env.ORG_SLUG ?? \"\";\n  const orgSlugDbFriendly = orgSlug.replace(/-/g, \"_\");\n  const athenaQueryBucket = process.env.ATHENA_S3_OUTPUT_LOCATION;\n  const athenaRegion = process.env.AWS_REGION;\n  const athenaSchema = `${orgSlugDbFriendly}__tss__default`;\n  const athenaWorkgroup = orgSlug;\n\n  const client = new athenaSDK.AthenaClient({\n    region: athenaRegion,\n  });\n\n  // Check if the org-specific workgroup exists\n  try {\n    await client.send(\n      new athenaSDK.GetWorkGroupCommand({\n        WorkGroup: athenaWorkgroup,\n      }),\n    );\n\n    // Workgroup exists, use it\n    return new AthenaProvider(client, athenaSDK, athenaWorkgroup, athenaSchema);\n  } catch {\n    // Workgroup doesn't exist or access denied, use 'primary' workgroup\n    // The 'primary' workgroup requires an explicit output location\n    if (!athenaQueryBucket) {\n      throw new Error(\n        \"ATHENA_S3_OUTPUT_LOCATION environment variable is required when using the 'primary' workgroup. \" +\n          \"Either set this variable or ensure the org-specific workgroup exists.\",\n      );\n    }\n\n    const athenaOutputLocation = `s3://${athenaQueryBucket}/${orgSlugDbFriendly}/`;\n    return new AthenaProvider(\n      client,\n      athenaSDK,\n      \"primary\",\n      athenaSchema,\n      athenaOutputLocation,\n    );\n  }\n}\n\n"],"names":["MILLISECONDS_PER_SECOND","getAthenaSDK","InvalidProviderConfigurationError","MAX_QUERY_LENGTH","AthenaProvider","client","sdk","workgroup","database","outputLocation","sqlQuery","_params","startCommand","queryExecutionId","QueryError","originalQuery","startTime","command","response","state","reason","errorTail","MissingTableError","resolve","results","nextToken","columnNames","isFirstPage","col","idx","name","rows","dataRows","row","rowData","cell","index","columnName","getTdpAthenaProvider","athenaSDK","orgSlug","orgSlugDbFriendly","athenaQueryBucket","athenaRegion","athenaSchema","athenaWorkgroup","athenaOutputLocation"],"mappings":"ykBAuBMA,EAA0B,IAUhC,eAAeC,GAAmC,CAChD,GAAI,CACF,OAAO,KAAM,QAAO,wBAAwB,CAC9C,MAAQ,CACN,MAAM,IAAIC,EAAAA,kCACR,oIAAA,CAGJ,CACF,CAGA,MAAMC,EAAmB,OAKlB,MAAMC,CAAe,CAClB,OACA,IACA,UACA,SACA,eAWR,YACEC,EACAC,EACAC,EACAC,EACAC,EACA,CACA,KAAK,OAASJ,EACd,KAAK,IAAMC,EACX,KAAK,UAAYC,EACjB,KAAK,SAAWC,EAChB,KAAK,eAAiBC,CACxB,CAgBA,MAAM,MACJC,EACAC,EAAmC,GACM,CACzC,GAAID,EAAS,OAASP,EACpB,MAAM,IAAI,MAAM,iDAAiD,EAMnE,MAAMS,EAAe,IAAI,KAAK,IAAI,2BAA2B,CAC3D,YAAaF,EACb,UAAW,KAAK,UAChB,sBAAuB,CACrB,SAAU,KAAK,QAAA,EAEjB,oBAAqB,KAAK,eACtB,CAAE,eAAgB,KAAK,gBACvB,MAAA,CACL,EAGKG,GADgB,MAAM,KAAK,OAAO,KAAKD,CAAY,GAClB,iBAEvC,GAAI,CAACC,EACH,MAAM,IAAIC,EAAAA,WAAW,iCAAiC,EAIxD,aAAM,KAAK,uBAAuBD,EAAkBH,CAAQ,EAGrD,KAAK,gBAAgBG,CAAgB,CAC9C,CAKA,MAAc,uBACZA,EACAE,EACe,CAGf,MAAMC,EAAY,KAAK,IAAA,EAEvB,KAAO,KAAK,MAAQA,EAAY,KAAa,CAC3C,MAAMC,EAAU,IAAI,KAAK,IAAI,yBAAyB,CACpD,iBAAkBJ,CAAA,CACnB,EACKK,EAAW,MAAM,KAAK,OAAO,KAAKD,CAAO,EACzCE,EAAQD,EAAS,gBAAgB,QAAQ,MAE/C,GAAIC,IAAU,KAAK,IAAI,oBAAoB,UACzC,OAGF,GACEA,IAAU,KAAK,IAAI,oBAAoB,QACvCA,IAAU,KAAK,IAAI,oBAAoB,UACvC,CACA,MAAMC,EACJF,EAAS,gBAAgB,QAAQ,mBAAqB,gBAExD,GAAIE,EAAO,SAAS,iBAAiB,EAAG,CACtC,MAAMC,EAAYD,EAAO,MAAM,GAAG,EAAE,IAAA,GAAO,QAAU,GACrD,MAAM,IAAIE,EAAAA,kBACR,kIAEED,EACA,GAAA,CAEN,CAEA,MAAM,IAAIP,EAAAA,WACR,iBAAiBC,CAAa,aAAaK,CAAM,EAAA,CAErD,CAEA,MAAM,IAAI,QAASG,GAAY,WAAWA,EAAS,GAAY,CAAC,CAClE,CAEA,MAAM,IAAIT,EAAAA,WACR,yBAAyB,IAAcd,CAAuB,UAAA,CAElE,CAKA,MAAc,gBACZa,EACyC,CACzC,MAAMW,EAA0C,CAAA,EAChD,IAAIC,EACAC,EAAwB,CAAA,EACxBC,EAAc,GAElB,EAAG,CACD,MAAMV,EAAU,IAAI,KAAK,IAAI,uBAAuB,CAClD,iBAAkBJ,EAClB,UAAWY,CAAA,CACZ,EAEKP,EAAW,MAAM,KAAK,OAAO,KAAKD,CAAO,EAE3CU,GAAeT,EAAS,WAAW,mBAAmB,aACxDQ,EAAcR,EAAS,UAAU,kBAAkB,WAAW,IAC5D,CAACU,EAAKC,IAAQ,CAEZ,MAAMC,EAAOF,EAAI,KACjB,OAAOE,GAAQA,EAAK,KAAA,IAAW,GAAKA,EAAO,UAAUD,CAAG,EAC1D,CAAA,GAIJ,MAAME,EAAOb,EAAS,WAAW,MAAQ,CAAA,EAEnCc,EAAWL,EAAcI,EAAK,MAAM,CAAC,EAAIA,EAE/C,UAAWE,KAAOD,EAAU,CAC1B,MAAME,EAAmC,CAAA,EACzCD,EAAI,MAAM,QAAQ,CAACE,EAAMC,IAAU,CAEjC,MAAMC,EAAaX,EAAYU,CAAK,GAAK,UAAUA,CAAK,GACxDF,EAAQG,CAAU,EAAIF,EAAK,cAAgB,IAC7C,CAAC,EACDX,EAAQ,KAAKU,CAAO,CACtB,CAEAT,EAAYP,EAAS,UACrBS,EAAc,EAChB,OAASF,GAET,OAAOD,CACT,CAKA,MAAM,OAAuB,CAC3B,KAAK,OAAO,QAAA,CACd,CACF,CAWA,eAAsBc,GAAgD,CAEpE,MAAMC,EAAY,MAAMtC,EAAA,EAElBuC,EAAU,QAAQ,IAAI,UAAY,GAClCC,EAAoBD,EAAQ,QAAQ,KAAM,GAAG,EAC7CE,EAAoB,QAAQ,IAAI,0BAChCC,EAAe,QAAQ,IAAI,WAC3BC,EAAe,GAAGH,CAAiB,iBACnCI,EAAkBL,EAElBnC,EAAS,IAAIkC,EAAU,aAAa,CACxC,OAAQI,CAAA,CACT,EAGD,GAAI,CACF,aAAMtC,EAAO,KACX,IAAIkC,EAAU,oBAAoB,CAChC,UAAWM,CAAA,CACZ,CAAA,EAII,IAAIzC,EAAeC,EAAQkC,EAAWM,EAAiBD,CAAY,CAC5E,MAAQ,CAGN,GAAI,CAACF,EACH,MAAM,IAAI,MACR,sKAAA,EAKJ,MAAMI,EAAuB,QAAQJ,CAAiB,IAAID,CAAiB,IAC3E,OAAO,IAAIrC,EACTC,EACAkC,EACA,UACAK,EACAE,CAAA,CAEJ,CACF"}
+{"version":3,"file":"athena.cjs","sources":[],"sourcesContent":[],"names":[],"mappings":""}

package/dist/providers/athena.js

@@ -1,160 +1,12 @@
-import { Q as m, M as y, I as f } from "../exceptions-jCQ6h5C8.js";
-import { P as Q, a as R } from "../exceptions-jCQ6h5C8.js";
-const S = 1e3;
-async function g() {
-  try {
-    return await import("@aws-sdk/client-athena");
-  } catch {
-    throw new f(
-      "The '@aws-sdk/client-athena' package is required to use the Athena provider. Please install it: npm install @aws-sdk/client-athena"
-    );
-  }
-}
-const k = 262144;
-class p {
-  client;
-  sdk;
-  workgroup;
-  database;
-  outputLocation;
-  /**
-   * Initialize the Athena data provider
-   *
-   * @param client - AWS Athena client
-   * @param sdk - AWS Athena SDK module (for accessing command classes)
-   * @param workgroup - Athena workgroup to use
-   * @param database - Default database/schema
-   * @param outputLocation - Optional S3 output location
-   */
-  constructor(e, a, o, r, t) {
-    this.client = e, this.sdk = a, this.workgroup = o, this.database = r, this.outputLocation = t;
-  }
-  /**
-   * Query the Athena database
-   *
-   * @param sqlQuery - SQL query to execute
-   * @param _params - Parameters to pass to the query (currently not used - Athena doesn't support parameterized queries)
-   * @returns Promise resolving to array of row objects
-   *
-   * @remarks
-   * **Security Note:** AWS Athena does not support parameterized queries.
-   * Unlike traditional databases, there is no native way to use bind parameters
-   * with Athena. Callers are responsible for properly sanitizing any user input
-   * before constructing the SQL query string. This is a known limitation of the
-   * Athena service, not a design flaw in this implementation.
-   */
-  async query(e, a = {}) {
-    if (e.length > k)
-      throw new Error("Query length exceeds the maximum allowed limit.");
-    const o = new this.sdk.StartQueryExecutionCommand({
-      QueryString: e,
-      WorkGroup: this.workgroup,
-      QueryExecutionContext: {
-        Database: this.database
-      },
-      ResultConfiguration: this.outputLocation ? { OutputLocation: this.outputLocation } : void 0
-    }), t = (await this.client.send(o)).QueryExecutionId;
-    if (!t)
-      throw new m("Failed to start query execution");
-    return await this.waitForQueryCompletion(t, e), this.fetchAllResults(t);
-  }
-  /**
-   * Wait for query to complete
-   */
-  async waitForQueryCompletion(e, a) {
-    const t = Date.now();
-    for (; Date.now() - t < 3e5; ) {
-      const u = new this.sdk.GetQueryExecutionCommand({
-        QueryExecutionId: e
-      }), n = await this.client.send(u), s = n.QueryExecution?.Status?.State;
-      if (s === this.sdk.QueryExecutionState.SUCCEEDED)
-        return;
-      if (s === this.sdk.QueryExecutionState.FAILED || s === this.sdk.QueryExecutionState.CANCELLED) {
-        const i = n.QueryExecution?.Status?.StateChangeReason ?? "Unknown error";
-        if (i.includes("TABLE_NOT_FOUND")) {
-          const l = i.split(":").pop()?.trim() ?? "";
-          throw new y(
-            "Athena is unable to find the table. If the table is created by a tetraflow, make sure that the tetraflow has run successfully. " + l + "."
-          );
-        }
-        throw new m(
-          `Query failed: ${a}. Reason: ${i}`
-        );
-      }
-      await new Promise((i) => setTimeout(i, 1e3));
-    }
-    throw new m(
-      `Query timed out after ${3e5 / S} seconds`
-    );
-  }
-  /**
-   * Fetch all results from a completed query
-   */
-  async fetchAllResults(e) {
-    const a = [];
-    let o, r = [], t = !0;
-    do {
-      const u = new this.sdk.GetQueryResultsCommand({
-        QueryExecutionId: e,
-        NextToken: o
-      }), n = await this.client.send(u);
-      t && n.ResultSet?.ResultSetMetadata?.ColumnInfo && (r = n.ResultSet.ResultSetMetadata.ColumnInfo.map(
-        (l, d) => {
-          const h = l.Name;
-          return h && h.trim() !== "" ? h : `column_${d}`;
-        }
-      ));
-      const s = n.ResultSet?.Rows ?? [], i = t ? s.slice(1) : s;
-      for (const l of i) {
-        const d = {};
-        l.Data?.forEach((h, w) => {
-          const E = r[w] ?? `column_${w}`;
-          d[E] = h.VarCharValue ?? null;
-        }), a.push(d);
-      }
-      o = n.NextToken, t = !1;
-    } while (o);
-    return a;
-  }
-  /**
-   * Close the Athena client (no-op for AWS SDK clients)
-   */
-  async close() {
-    this.client.destroy();
-  }
-}
-async function C() {
-  const c = await g(), e = process.env.ORG_SLUG ?? "", a = e.replace(/-/g, "_"), o = process.env.ATHENA_S3_OUTPUT_LOCATION, r = process.env.AWS_REGION, t = `${a}__tss__default`, u = e, n = new c.AthenaClient({
-    region: r
-  });
-  try {
-    return await n.send(
-      new c.GetWorkGroupCommand({
-        WorkGroup: u
-      })
-    ), new p(n, c, u, t);
-  } catch {
-    if (!o)
-      throw new Error(
-        "ATHENA_S3_OUTPUT_LOCATION environment variable is required when using the 'primary' workgroup. Either set this variable or ensure the org-specific workgroup exists."
-      );
-    const s = `s3://${o}/${a}/`;
-    return new p(
-      n,
-      c,
-      "primary",
-      t,
-      s
-    );
-  }
-}
+import { AthenaProvider as e, getTdpAthenaProvider as i } from "../server/providers/AthenaProvider.js";
+import { InvalidProviderConfigurationError as d, MissingTableError as t, ProviderConnectionError as v, ProviderError as a, QueryError as E } from "../server/providers/exceptions.js";
 export {
-  p as AthenaProvider,
-  f as InvalidProviderConfigurationError,
-  y as MissingTableError,
-  Q as ProviderConnectionError,
-  R as ProviderError,
-  m as QueryError,
-  C as getTdpAthenaProvider
+  e as AthenaProvider,
+  d as InvalidProviderConfigurationError,
+  t as MissingTableError,
+  v as ProviderConnectionError,
+  a as ProviderError,
+  E as QueryError,
+  i as getTdpAthenaProvider
 };
 //# sourceMappingURL=athena.js.map

package/dist/providers/athena.js.map

@@ -1 +1 @@
-{"version":3,"file":"athena.js","sources":["../../src/server/providers/AthenaProvider.ts"],"sourcesContent":["/**\n * Athena Data Provider\n *\n * TypeScript equivalent of AthenaProvider from\n * ts-lib-ui-kit-streamlit/tetrascience/data_app_providers/provider.py\n *\n * @remarks\n * This provider requires the `@aws-sdk/client-athena` package to be installed.\n * It is an optional peer dependency - install it only if you need Athena support:\n * ```bash\n * npm install @aws-sdk/client-athena\n * # or\n * yarn add @aws-sdk/client-athena\n * ```\n */\n\nimport {\n  QueryError,\n  MissingTableError,\n  InvalidProviderConfigurationError,\n} from \"./exceptions\";\n\n/** Milliseconds per second */\nconst MILLISECONDS_PER_SECOND = 1000;\n\n// Type imports for @aws-sdk/client-athena (these don't require the package at runtime)\ntype AthenaClient = import(\"@aws-sdk/client-athena\").AthenaClient;\ntype AthenaSDK = typeof import(\"@aws-sdk/client-athena\");\n\n/**\n * Dynamically import @aws-sdk/client-athena\n * @throws {InvalidProviderConfigurationError} If @aws-sdk/client-athena is not installed\n */\nasync function getAthenaSDK(): Promise<AthenaSDK> {\n  try {\n    return await import(\"@aws-sdk/client-athena\");\n  } catch {\n    throw new InvalidProviderConfigurationError(\n      \"The '@aws-sdk/client-athena' package is required to use the Athena provider. \" +\n        \"Please install it: npm install @aws-sdk/client-athena\",\n    );\n  }\n}\n\n/** Maximum query length allowed by Athena */\nconst MAX_QUERY_LENGTH = 262144;\n\n/**\n * Athena data provider\n */\nexport class AthenaProvider {\n  private client: AthenaClient;\n  private sdk: AthenaSDK;\n  private workgroup: string;\n  private database: string;\n  private outputLocation?: string;\n\n  /**\n   * Initialize the Athena data provider\n   *\n   * @param client - AWS Athena client\n   * @param sdk - AWS Athena SDK module (for accessing command classes)\n   * @param workgroup - Athena workgroup to use\n   * @param database - Default database/schema\n   * @param outputLocation - Optional S3 output location\n   */\n  constructor(\n    client: AthenaClient,\n    sdk: AthenaSDK,\n    workgroup: string,\n    database: string,\n    outputLocation?: string,\n  ) {\n    this.client = client;\n    this.sdk = sdk;\n    this.workgroup = workgroup;\n    this.database = database;\n    this.outputLocation = outputLocation;\n  }\n\n  /**\n   * Query the Athena database\n   *\n   * @param sqlQuery - SQL query to execute\n   * @param _params - Parameters to pass to the query (currently not used - Athena doesn't support parameterized queries)\n   * @returns Promise resolving to array of row objects\n   *\n   * @remarks\n   * **Security Note:** AWS Athena does not support parameterized queries.\n   * Unlike traditional databases, there is no native way to use bind parameters\n   * with Athena. Callers are responsible for properly sanitizing any user input\n   * before constructing the SQL query string. This is a known limitation of the\n   * Athena service, not a design flaw in this implementation.\n   */\n  async query(\n    sqlQuery: string,\n    _params: Record<string, unknown> = {},\n  ): Promise<Array<Record<string, unknown>>> {\n    if (sqlQuery.length > MAX_QUERY_LENGTH) {\n      throw new Error(\"Query length exceeds the maximum allowed limit.\");\n    }\n\n    // Start query execution\n    // Note: Athena does not support parameterized queries. The sqlQuery is passed\n    // directly to Athena. Callers must sanitize user input before constructing queries.\n    const startCommand = new this.sdk.StartQueryExecutionCommand({\n      QueryString: sqlQuery,\n      WorkGroup: this.workgroup,\n      QueryExecutionContext: {\n        Database: this.database,\n      },\n      ResultConfiguration: this.outputLocation\n        ? { OutputLocation: this.outputLocation }\n        : undefined,\n    });\n\n    const startResponse = await this.client.send(startCommand);\n    const queryExecutionId = startResponse.QueryExecutionId;\n\n    if (!queryExecutionId) {\n      throw new QueryError(\"Failed to start query execution\");\n    }\n\n    // Wait for query to complete\n    await this.waitForQueryCompletion(queryExecutionId, sqlQuery);\n\n    // Fetch results\n    return this.fetchAllResults(queryExecutionId);\n  }\n\n  /**\n   * Wait for query to complete\n   */\n  private async waitForQueryCompletion(\n    queryExecutionId: string,\n    originalQuery: string,\n  ): Promise<void> {\n    const maxWaitTime = 300000; // 5 minutes\n    const pollInterval = 1000; // 1 second\n    const startTime = Date.now();\n\n    while (Date.now() - startTime < maxWaitTime) {\n      const command = new this.sdk.GetQueryExecutionCommand({\n        QueryExecutionId: queryExecutionId,\n      });\n      const response = await this.client.send(command);\n      const state = response.QueryExecution?.Status?.State;\n\n      if (state === this.sdk.QueryExecutionState.SUCCEEDED) {\n        return;\n      }\n\n      if (\n        state === this.sdk.QueryExecutionState.FAILED ||\n        state === this.sdk.QueryExecutionState.CANCELLED\n      ) {\n        const reason =\n          response.QueryExecution?.Status?.StateChangeReason ?? \"Unknown error\";\n\n        if (reason.includes(\"TABLE_NOT_FOUND\")) {\n          const errorTail = reason.split(\":\").pop()?.trim() ?? \"\";\n          throw new MissingTableError(\n            \"Athena is unable to find the table. If the table is created by a \" +\n              \"tetraflow, make sure that the tetraflow has run successfully. \" +\n              errorTail +\n              \".\",\n          );\n        }\n\n        throw new QueryError(\n          `Query failed: ${originalQuery}. Reason: ${reason}`,\n        );\n      }\n\n      await new Promise((resolve) => setTimeout(resolve, pollInterval));\n    }\n\n    throw new QueryError(\n      `Query timed out after ${maxWaitTime / MILLISECONDS_PER_SECOND} seconds`,\n    );\n  }\n\n  /**\n   * Fetch all results from a completed query\n   */\n  private async fetchAllResults(\n    queryExecutionId: string,\n  ): Promise<Array<Record<string, unknown>>> {\n    const results: Array<Record<string, unknown>> = [];\n    let nextToken: string | undefined;\n    let columnNames: string[] = [];\n    let isFirstPage = true;\n\n    do {\n      const command = new this.sdk.GetQueryResultsCommand({\n        QueryExecutionId: queryExecutionId,\n        NextToken: nextToken,\n      });\n\n      const response = await this.client.send(command);\n\n      if (isFirstPage && response.ResultSet?.ResultSetMetadata?.ColumnInfo) {\n        columnNames = response.ResultSet.ResultSetMetadata.ColumnInfo.map(\n          (col, idx) => {\n            // Handle empty or missing column names by using a fallback\n            const name = col.Name;\n            return name && name.trim() !== \"\" ? name : `column_${idx}`;\n          },\n        );\n      }\n\n      const rows = response.ResultSet?.Rows ?? [];\n      // Skip header row on first page\n      const dataRows = isFirstPage ? rows.slice(1) : rows;\n\n      for (const row of dataRows) {\n        const rowData: Record<string, unknown> = {};\n        row.Data?.forEach((cell, index) => {\n          // columnNames already has fallback values, so we can use them directly\n          const columnName = columnNames[index] ?? `column_${index}`;\n          rowData[columnName] = cell.VarCharValue ?? null;\n        });\n        results.push(rowData);\n      }\n\n      nextToken = response.NextToken;\n      isFirstPage = false;\n    } while (nextToken);\n\n    return results;\n  }\n\n  /**\n   * Close the Athena client (no-op for AWS SDK clients)\n   */\n  async close(): Promise<void> {\n    this.client.destroy();\n  }\n}\n\n/**\n * Get the TDP Athena provider\n *\n * Creates an Athena provider using TDP environment configuration\n *\n * @returns Promise resolving to Athena data provider\n * @throws {InvalidProviderConfigurationError} If @aws-sdk/client-athena is not installed\n * @throws {Error} If ATHENA_S3_OUTPUT_LOCATION is not set when using the 'primary' workgroup\n */\nexport async function getTdpAthenaProvider(): Promise<AthenaProvider> {\n  // Dynamically import @aws-sdk/client-athena\n  const athenaSDK = await getAthenaSDK();\n\n  const orgSlug = process.env.ORG_SLUG ?? \"\";\n  const orgSlugDbFriendly = orgSlug.replace(/-/g, \"_\");\n  const athenaQueryBucket = process.env.ATHENA_S3_OUTPUT_LOCATION;\n  const athenaRegion = process.env.AWS_REGION;\n  const athenaSchema = `${orgSlugDbFriendly}__tss__default`;\n  const athenaWorkgroup = orgSlug;\n\n  const client = new athenaSDK.AthenaClient({\n    region: athenaRegion,\n  });\n\n  // Check if the org-specific workgroup exists\n  try {\n    await client.send(\n      new athenaSDK.GetWorkGroupCommand({\n        WorkGroup: athenaWorkgroup,\n      }),\n    );\n\n    // Workgroup exists, use it\n    return new AthenaProvider(client, athenaSDK, athenaWorkgroup, athenaSchema);\n  } catch {\n    // Workgroup doesn't exist or access denied, use 'primary' workgroup\n    // The 'primary' workgroup requires an explicit output location\n    if (!athenaQueryBucket) {\n      throw new Error(\n        \"ATHENA_S3_OUTPUT_LOCATION environment variable is required when using the 'primary' workgroup. \" +\n          \"Either set this variable or ensure the org-specific workgroup exists.\",\n      );\n    }\n\n    const athenaOutputLocation = `s3://${athenaQueryBucket}/${orgSlugDbFriendly}/`;\n    return new AthenaProvider(\n      client,\n      athenaSDK,\n      \"primary\",\n      athenaSchema,\n      athenaOutputLocation,\n    );\n  }\n}\n\n"],"names":["MILLISECONDS_PER_SECOND","getAthenaSDK","InvalidProviderConfigurationError","MAX_QUERY_LENGTH","AthenaProvider","client","sdk","workgroup","database","outputLocation","sqlQuery","_params","startCommand","queryExecutionId","QueryError","originalQuery","startTime","command","response","state","reason","errorTail","MissingTableError","resolve","results","nextToken","columnNames","isFirstPage","col","idx","name","rows","dataRows","row","rowData","cell","index","columnName","getTdpAthenaProvider","athenaSDK","orgSlug","orgSlugDbFriendly","athenaQueryBucket","athenaRegion","athenaSchema","athenaWorkgroup","athenaOutputLocation"],"mappings":";;AAuBA,MAAMA,IAA0B;AAUhC,eAAeC,IAAmC;AAChD,MAAI;AACF,WAAO,MAAM,OAAO,wBAAwB;AAAA,EAC9C,QAAQ;AACN,UAAM,IAAIC;AAAA,MACR;AAAA,IAAA;AAAA,EAGJ;AACF;AAGA,MAAMC,IAAmB;AAKlB,MAAMC,EAAe;AAAA,EAClB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWR,YACEC,GACAC,GACAC,GACAC,GACAC,GACA;AACA,SAAK,SAASJ,GACd,KAAK,MAAMC,GACX,KAAK,YAAYC,GACjB,KAAK,WAAWC,GAChB,KAAK,iBAAiBC;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAM,MACJC,GACAC,IAAmC,IACM;AACzC,QAAID,EAAS,SAASP;AACpB,YAAM,IAAI,MAAM,iDAAiD;AAMnE,UAAMS,IAAe,IAAI,KAAK,IAAI,2BAA2B;AAAA,MAC3D,aAAaF;AAAA,MACb,WAAW,KAAK;AAAA,MAChB,uBAAuB;AAAA,QACrB,UAAU,KAAK;AAAA,MAAA;AAAA,MAEjB,qBAAqB,KAAK,iBACtB,EAAE,gBAAgB,KAAK,mBACvB;AAAA,IAAA,CACL,GAGKG,KADgB,MAAM,KAAK,OAAO,KAAKD,CAAY,GAClB;AAEvC,QAAI,CAACC;AACH,YAAM,IAAIC,EAAW,iCAAiC;AAIxD,iBAAM,KAAK,uBAAuBD,GAAkBH,CAAQ,GAGrD,KAAK,gBAAgBG,CAAgB;AAAA,EAC9C;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,uBACZA,GACAE,GACe;AAGf,UAAMC,IAAY,KAAK,IAAA;AAEvB,WAAO,KAAK,QAAQA,IAAY,OAAa;AAC3C,YAAMC,IAAU,IAAI,KAAK,IAAI,yBAAyB;AAAA,QACpD,kBAAkBJ;AAAA,MAAA,CACnB,GACKK,IAAW,MAAM,KAAK,OAAO,KAAKD,CAAO,GACzCE,IAAQD,EAAS,gBAAgB,QAAQ;AAE/C,UAAIC,MAAU,KAAK,IAAI,oBAAoB;AACzC;AAGF,UACEA,MAAU,KAAK,IAAI,oBAAoB,UACvCA,MAAU,KAAK,IAAI,oBAAoB,WACvC;AACA,cAAMC,IACJF,EAAS,gBAAgB,QAAQ,qBAAqB;AAExD,YAAIE,EAAO,SAAS,iBAAiB,GAAG;AACtC,gBAAMC,IAAYD,EAAO,MAAM,GAAG,EAAE,IAAA,GAAO,UAAU;AACrD,gBAAM,IAAIE;AAAA,YACR,oIAEED,IACA;AAAA,UAAA;AAAA,QAEN;AAEA,cAAM,IAAIP;AAAA,UACR,iBAAiBC,CAAa,aAAaK,CAAM;AAAA,QAAA;AAAA,MAErD;AAEA,YAAM,IAAI,QAAQ,CAACG,MAAY,WAAWA,GAAS,GAAY,CAAC;AAAA,IAClE;AAEA,UAAM,IAAIT;AAAA,MACR,yBAAyB,MAAcd,CAAuB;AAAA,IAAA;AAAA,EAElE;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,gBACZa,GACyC;AACzC,UAAMW,IAA0C,CAAA;AAChD,QAAIC,GACAC,IAAwB,CAAA,GACxBC,IAAc;AAElB,OAAG;AACD,YAAMV,IAAU,IAAI,KAAK,IAAI,uBAAuB;AAAA,QAClD,kBAAkBJ;AAAA,QAClB,WAAWY;AAAA,MAAA,CACZ,GAEKP,IAAW,MAAM,KAAK,OAAO,KAAKD,CAAO;AAE/C,MAAIU,KAAeT,EAAS,WAAW,mBAAmB,eACxDQ,IAAcR,EAAS,UAAU,kBAAkB,WAAW;AAAA,QAC5D,CAACU,GAAKC,MAAQ;AAEZ,gBAAMC,IAAOF,EAAI;AACjB,iBAAOE,KAAQA,EAAK,KAAA,MAAW,KAAKA,IAAO,UAAUD,CAAG;AAAA,QAC1D;AAAA,MAAA;AAIJ,YAAME,IAAOb,EAAS,WAAW,QAAQ,CAAA,GAEnCc,IAAWL,IAAcI,EAAK,MAAM,CAAC,IAAIA;AAE/C,iBAAWE,KAAOD,GAAU;AAC1B,cAAME,IAAmC,CAAA;AACzC,QAAAD,EAAI,MAAM,QAAQ,CAACE,GAAMC,MAAU;AAEjC,gBAAMC,IAAaX,EAAYU,CAAK,KAAK,UAAUA,CAAK;AACxD,UAAAF,EAAQG,CAAU,IAAIF,EAAK,gBAAgB;AAAA,QAC7C,CAAC,GACDX,EAAQ,KAAKU,CAAO;AAAA,MACtB;AAEA,MAAAT,IAAYP,EAAS,WACrBS,IAAc;AAAA,IAChB,SAASF;AAET,WAAOD;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAuB;AAC3B,SAAK,OAAO,QAAA;AAAA,EACd;AACF;AAWA,eAAsBc,IAAgD;AAEpE,QAAMC,IAAY,MAAMtC,EAAA,GAElBuC,IAAU,QAAQ,IAAI,YAAY,IAClCC,IAAoBD,EAAQ,QAAQ,MAAM,GAAG,GAC7CE,IAAoB,QAAQ,IAAI,2BAChCC,IAAe,QAAQ,IAAI,YAC3BC,IAAe,GAAGH,CAAiB,kBACnCI,IAAkBL,GAElBnC,IAAS,IAAIkC,EAAU,aAAa;AAAA,IACxC,QAAQI;AAAA,EAAA,CACT;AAGD,MAAI;AACF,iBAAMtC,EAAO;AAAA,MACX,IAAIkC,EAAU,oBAAoB;AAAA,QAChC,WAAWM;AAAA,MAAA,CACZ;AAAA,IAAA,GAII,IAAIzC,EAAeC,GAAQkC,GAAWM,GAAiBD,CAAY;AAAA,EAC5E,QAAQ;AAGN,QAAI,CAACF;AACH,YAAM,IAAI;AAAA,QACR;AAAA,MAAA;AAKJ,UAAMI,IAAuB,QAAQJ,CAAiB,IAAID,CAAiB;AAC3E,WAAO,IAAIrC;AAAA,MACTC;AAAA,MACAkC;AAAA,MACA;AAAA,MACAK;AAAA,MACAE;AAAA,IAAA;AAAA,EAEJ;AACF;"}
+{"version":3,"file":"athena.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;"}

package/dist/providers/databricks.cjs

@@ -1,2 +1,2 @@
-"use strict";var b=Object.create;var c=Object.defineProperty;var g=Object.getOwnPropertyDescriptor;var f=Object.getOwnPropertyNames;var w=Object.getPrototypeOf,y=Object.prototype.hasOwnProperty;var P=(t,e,r,s)=>{if(e&&typeof e=="object"||typeof e=="function")for(let i of f(e))!y.call(t,i)&&i!==r&&c(t,i,{get:()=>e[i],enumerable:!(s=g(e,i))||s.enumerable});return t};var S=(t,e,r)=>(r=t!=null?b(w(t)):{},P(e||!t||!t.__esModule?c(r,"default",{value:t,enumerable:!0}):r,t));Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const o=require("../exceptions-DN25pCDi.cjs");async function _(){try{return await import("@databricks/sql")}catch{throw new o.InvalidProviderConfigurationError("The '@databricks/sql' package is required to use the Databricks provider. Please install it: npm install @databricks/sql")}}class l{client;session;constructor(e,r){this.client=e,this.session=r}async query(e,r={}){const s=await this.session.executeStatement(e);try{return await s.fetchAll()}finally{await s.close()}}async close(){await this.session.close(),await this.client.close()}}function m(){return`${(process.env.ORG_SLUG??"").replace(/-/g,"_")}__tss__default`}async function E(t){const{DBSQLClient:e}=await _(),r=["server_hostname","http_path","client_id","client_secret","catalog"];for(const n of r)if(!t.fields[n])throw new o.InvalidProviderConfigurationError(`Missing field '${n}' in the provider '${t.name}' to connect to Databricks.`);const s=t.fields.server_hostname,i=t.fields.http_path,d=t.fields.client_id,u=t.fields.client_secret,h=t.fields.catalog,v=t.fields.schema??m(),a=new e;await a.connect({host:s,path:i,authType:"databricks-oauth",oauthClientId:d,oauthClientSecret:u});const p=await a.openSession({initialCatalog:h,initialSchema:v});return new l(a,p)}exports.InvalidProviderConfigurationError=o.InvalidProviderConfigurationError;exports.MissingTableError=o.MissingTableError;exports.ProviderConnectionError=o.ProviderConnectionError;exports.ProviderError=o.ProviderError;exports.QueryError=o.QueryError;exports.DatabricksProvider=l;exports.buildDatabricksProvider=E;
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const i=require("../server/providers/DatabricksProvider.cjs"),r=require("../server/providers/exceptions.cjs");exports.DatabricksProvider=i.DatabricksProvider;exports.buildDatabricksProvider=i.buildDatabricksProvider;exports.InvalidProviderConfigurationError=r.InvalidProviderConfigurationError;exports.MissingTableError=r.MissingTableError;exports.ProviderConnectionError=r.ProviderConnectionError;exports.ProviderError=r.ProviderError;exports.QueryError=r.QueryError;
 //# sourceMappingURL=databricks.cjs.map

package/dist/providers/databricks.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"databricks.cjs","sources":["../../src/server/providers/DatabricksProvider.ts"],"sourcesContent":["/**\n * Databricks Data Provider\n *\n * TypeScript equivalent of DatabricksProvider from\n * ts-lib-ui-kit-streamlit/tetrascience/data_app_providers/provider.py\n *\n * @remarks\n * This provider requires the `@databricks/sql` package to be installed.\n * It is an optional peer dependency - install it only if you need Databricks support:\n * ```bash\n * npm install @databricks/sql\n * # or\n * yarn add @databricks/sql\n * ```\n */\n\nimport { InvalidProviderConfigurationError } from \"./exceptions\";\n\nimport type { ProviderConfiguration } from \"./types\";\n\n// Type imports for @databricks/sql (these don't require the package at runtime)\ntype DBSQLClient = import(\"@databricks/sql\").DBSQLClient;\ntype IDBSQLSession = import(\"@databricks/sql/dist/contracts/IDBSQLSession\").default;\n\n/**\n * Dynamically import @databricks/sql\n * @throws {InvalidProviderConfigurationError} If @databricks/sql is not installed\n */\nasync function getDatabricksSQL(): Promise<typeof import(\"@databricks/sql\")> {\n  try {\n    return await import(\"@databricks/sql\");\n  } catch {\n    throw new InvalidProviderConfigurationError(\n      \"The '@databricks/sql' package is required to use the Databricks provider. \" +\n        \"Please install it: npm install @databricks/sql\",\n    );\n  }\n}\n\n/**\n * Databricks data provider\n */\nexport class DatabricksProvider {\n  private client: DBSQLClient;\n  private session: IDBSQLSession;\n\n  /**\n   * Initialize the Databricks data provider\n   *\n   * @param client - Databricks SQL client\n   * @param session - Databricks SQL session\n   */\n  constructor(client: DBSQLClient, session: IDBSQLSession) {\n    this.client = client;\n    this.session = session;\n  }\n\n  /**\n   * Query the Databricks database\n   *\n   * @param sqlQuery - SQL query to execute\n   * @param _params - Parameters to pass to the query (currently not used)\n   * @returns Promise resolving to array of row objects\n   */\n  async query(\n    sqlQuery: string,\n    _params: Record<string, unknown> = {},\n  ): Promise<Array<Record<string, unknown>>> {\n    const operation = await this.session.executeStatement(sqlQuery);\n    try {\n      const result = await operation.fetchAll();\n      return result as Array<Record<string, unknown>>;\n    } finally {\n      // Always close the operation to avoid leaking server-side resources\n      await operation.close();\n    }\n  }\n\n  /**\n   * Close the Databricks connection\n   */\n  async close(): Promise<void> {\n    await this.session.close();\n    await this.client.close();\n  }\n}\n\n/**\n * Get the default Athena schema from environment\n */\nfunction getDefaultSchema(): string {\n  const orgSlug = process.env.ORG_SLUG ?? \"\";\n  const orgSlugDbFriendly = orgSlug.replace(/-/g, \"_\");\n  return `${orgSlugDbFriendly}__tss__default`;\n}\n\n/**\n * Build a Databricks data provider from the configuration\n *\n * @param config - Provider configuration\n * @returns Promise resolving to Databricks data provider\n * @throws {InvalidProviderConfigurationError} If @databricks/sql is not installed or config is invalid\n */\nexport async function buildDatabricksProvider(\n  config: ProviderConfiguration,\n): Promise<DatabricksProvider> {\n  // Dynamically import @databricks/sql\n  const { DBSQLClient } = await getDatabricksSQL();\n\n  const requiredFields = [\n    \"server_hostname\",\n    \"http_path\",\n    \"client_id\",\n    \"client_secret\",\n    \"catalog\",\n  ] as const;\n\n  for (const field of requiredFields) {\n    if (!config.fields[field]) {\n      throw new InvalidProviderConfigurationError(\n        `Missing field '${field}' in the provider '${config.name}' to connect to Databricks.`,\n      );\n    }\n  }\n\n  const serverHostname = config.fields[\"server_hostname\"]!;\n  const httpPath = config.fields[\"http_path\"]!;\n  const clientId = config.fields[\"client_id\"]!;\n  const clientSecret = config.fields[\"client_secret\"]!;\n  const catalog = config.fields[\"catalog\"]!;\n  const schema = config.fields[\"schema\"] ?? getDefaultSchema();\n\n  const client = new DBSQLClient();\n\n  await client.connect({\n    host: serverHostname,\n    path: httpPath,\n    authType: \"databricks-oauth\",\n    oauthClientId: clientId,\n    oauthClientSecret: clientSecret,\n  });\n\n  const session = await client.openSession({\n    initialCatalog: catalog,\n    initialSchema: schema,\n  });\n\n  return new DatabricksProvider(client, session);\n}\n\n"],"names":["getDatabricksSQL","InvalidProviderConfigurationError","DatabricksProvider","client","session","sqlQuery","_params","operation","getDefaultSchema","buildDatabricksProvider","config","DBSQLClient","requiredFields","field","serverHostname","httpPath","clientId","clientSecret","catalog","schema"],"mappings":"ykBA4BA,eAAeA,GAA8D,CAC3E,GAAI,CACF,OAAO,KAAM,QAAO,iBAAiB,CACvC,MAAQ,CACN,MAAM,IAAIC,EAAAA,kCACR,0HAAA,CAGJ,CACF,CAKO,MAAMC,CAAmB,CACtB,OACA,QAQR,YAAYC,EAAqBC,EAAwB,CACvD,KAAK,OAASD,EACd,KAAK,QAAUC,CACjB,CASA,MAAM,MACJC,EACAC,EAAmC,GACM,CACzC,MAAMC,EAAY,MAAM,KAAK,QAAQ,iBAAiBF,CAAQ,EAC9D,GAAI,CAEF,OADe,MAAME,EAAU,SAAA,CAEjC,QAAA,CAEE,MAAMA,EAAU,MAAA,CAClB,CACF,CAKA,MAAM,OAAuB,CAC3B,MAAM,KAAK,QAAQ,MAAA,EACnB,MAAM,KAAK,OAAO,MAAA,CACpB,CACF,CAKA,SAASC,GAA2B,CAGlC,MAAO,IAFS,QAAQ,IAAI,UAAY,IACN,QAAQ,KAAM,GAAG,CACxB,gBAC7B,CASA,eAAsBC,EACpBC,EAC6B,CAE7B,KAAM,CAAE,YAAAC,GAAgB,MAAMX,EAAA,EAExBY,EAAiB,CACrB,kBACA,YACA,YACA,gBACA,SAAA,EAGF,UAAWC,KAASD,EAClB,GAAI,CAACF,EAAO,OAAOG,CAAK,EACtB,MAAM,IAAIZ,EAAAA,kCACR,kBAAkBY,CAAK,sBAAsBH,EAAO,IAAI,6BAAA,EAK9D,MAAMI,EAAiBJ,EAAO,OAAO,gBAC/BK,EAAWL,EAAO,OAAO,UACzBM,EAAWN,EAAO,OAAO,UACzBO,EAAeP,EAAO,OAAO,cAC7BQ,EAAUR,EAAO,OAAO,QACxBS,EAAST,EAAO,OAAO,QAAaF,EAAA,EAEpCL,EAAS,IAAIQ,EAEnB,MAAMR,EAAO,QAAQ,CACnB,KAAMW,EACN,KAAMC,EACN,SAAU,mBACV,cAAeC,EACf,kBAAmBC,CAAA,CACpB,EAED,MAAMb,EAAU,MAAMD,EAAO,YAAY,CACvC,eAAgBe,EAChB,cAAeC,CAAA,CAChB,EAED,OAAO,IAAIjB,EAAmBC,EAAQC,CAAO,CAC/C"}
+{"version":3,"file":"databricks.cjs","sources":[],"sourcesContent":[],"names":[],"mappings":""}

package/dist/providers/databricks.js

@@ -1,85 +1,12 @@
-import { I as o } from "../exceptions-jCQ6h5C8.js";
-import { M as g, P as v, a as y, Q as k } from "../exceptions-jCQ6h5C8.js";
-async function p() {
-  try {
-    return await import("@databricks/sql");
-  } catch {
-    throw new o(
-      "The '@databricks/sql' package is required to use the Databricks provider. Please install it: npm install @databricks/sql"
-    );
-  }
-}
-class f {
-  client;
-  session;
-  /**
-   * Initialize the Databricks data provider
-   *
-   * @param client - Databricks SQL client
-   * @param session - Databricks SQL session
-   */
-  constructor(e, s) {
-    this.client = e, this.session = s;
-  }
-  /**
-   * Query the Databricks database
-   *
-   * @param sqlQuery - SQL query to execute
-   * @param _params - Parameters to pass to the query (currently not used)
-   * @returns Promise resolving to array of row objects
-   */
-  async query(e, s = {}) {
-    const a = await this.session.executeStatement(e);
-    try {
-      return await a.fetchAll();
-    } finally {
-      await a.close();
-    }
-  }
-  /**
-   * Close the Databricks connection
-   */
-  async close() {
-    await this.session.close(), await this.client.close();
-  }
-}
-function m() {
-  return `${(process.env.ORG_SLUG ?? "").replace(/-/g, "_")}__tss__default`;
-}
-async function _(t) {
-  const { DBSQLClient: e } = await p(), s = [
-    "server_hostname",
-    "http_path",
-    "client_id",
-    "client_secret",
-    "catalog"
-  ];
-  for (const n of s)
-    if (!t.fields[n])
-      throw new o(
-        `Missing field '${n}' in the provider '${t.name}' to connect to Databricks.`
-      );
-  const a = t.fields.server_hostname, i = t.fields.http_path, c = t.fields.client_id, l = t.fields.client_secret, d = t.fields.catalog, h = t.fields.schema ?? m(), r = new e();
-  await r.connect({
-    host: a,
-    path: i,
-    authType: "databricks-oauth",
-    oauthClientId: c,
-    oauthClientSecret: l
-  });
-  const u = await r.openSession({
-    initialCatalog: d,
-    initialSchema: h
-  });
-  return new f(r, u);
-}
+import { DatabricksProvider as i, buildDatabricksProvider as e } from "../server/providers/DatabricksProvider.js";
+import { InvalidProviderConfigurationError as d, MissingTableError as n, ProviderConnectionError as t, ProviderError as v, QueryError as E } from "../server/providers/exceptions.js";
 export {
-  f as DatabricksProvider,
-  o as InvalidProviderConfigurationError,
-  g as MissingTableError,
-  v as ProviderConnectionError,
-  y as ProviderError,
-  k as QueryError,
-  _ as buildDatabricksProvider
+  i as DatabricksProvider,
+  d as InvalidProviderConfigurationError,
+  n as MissingTableError,
+  t as ProviderConnectionError,
+  v as ProviderError,
+  E as QueryError,
+  e as buildDatabricksProvider
 };
 //# sourceMappingURL=databricks.js.map

package/dist/providers/databricks.js.map

@@ -1 +1 @@
-{"version":3,"file":"databricks.js","sources":["../../src/server/providers/DatabricksProvider.ts"],"sourcesContent":["/**\n * Databricks Data Provider\n *\n * TypeScript equivalent of DatabricksProvider from\n * ts-lib-ui-kit-streamlit/tetrascience/data_app_providers/provider.py\n *\n * @remarks\n * This provider requires the `@databricks/sql` package to be installed.\n * It is an optional peer dependency - install it only if you need Databricks support:\n * ```bash\n * npm install @databricks/sql\n * # or\n * yarn add @databricks/sql\n * ```\n */\n\nimport { InvalidProviderConfigurationError } from \"./exceptions\";\n\nimport type { ProviderConfiguration } from \"./types\";\n\n// Type imports for @databricks/sql (these don't require the package at runtime)\ntype DBSQLClient = import(\"@databricks/sql\").DBSQLClient;\ntype IDBSQLSession = import(\"@databricks/sql/dist/contracts/IDBSQLSession\").default;\n\n/**\n * Dynamically import @databricks/sql\n * @throws {InvalidProviderConfigurationError} If @databricks/sql is not installed\n */\nasync function getDatabricksSQL(): Promise<typeof import(\"@databricks/sql\")> {\n  try {\n    return await import(\"@databricks/sql\");\n  } catch {\n    throw new InvalidProviderConfigurationError(\n      \"The '@databricks/sql' package is required to use the Databricks provider. \" +\n        \"Please install it: npm install @databricks/sql\",\n    );\n  }\n}\n\n/**\n * Databricks data provider\n */\nexport class DatabricksProvider {\n  private client: DBSQLClient;\n  private session: IDBSQLSession;\n\n  /**\n   * Initialize the Databricks data provider\n   *\n   * @param client - Databricks SQL client\n   * @param session - Databricks SQL session\n   */\n  constructor(client: DBSQLClient, session: IDBSQLSession) {\n    this.client = client;\n    this.session = session;\n  }\n\n  /**\n   * Query the Databricks database\n   *\n   * @param sqlQuery - SQL query to execute\n   * @param _params - Parameters to pass to the query (currently not used)\n   * @returns Promise resolving to array of row objects\n   */\n  async query(\n    sqlQuery: string,\n    _params: Record<string, unknown> = {},\n  ): Promise<Array<Record<string, unknown>>> {\n    const operation = await this.session.executeStatement(sqlQuery);\n    try {\n      const result = await operation.fetchAll();\n      return result as Array<Record<string, unknown>>;\n    } finally {\n      // Always close the operation to avoid leaking server-side resources\n      await operation.close();\n    }\n  }\n\n  /**\n   * Close the Databricks connection\n   */\n  async close(): Promise<void> {\n    await this.session.close();\n    await this.client.close();\n  }\n}\n\n/**\n * Get the default Athena schema from environment\n */\nfunction getDefaultSchema(): string {\n  const orgSlug = process.env.ORG_SLUG ?? \"\";\n  const orgSlugDbFriendly = orgSlug.replace(/-/g, \"_\");\n  return `${orgSlugDbFriendly}__tss__default`;\n}\n\n/**\n * Build a Databricks data provider from the configuration\n *\n * @param config - Provider configuration\n * @returns Promise resolving to Databricks data provider\n * @throws {InvalidProviderConfigurationError} If @databricks/sql is not installed or config is invalid\n */\nexport async function buildDatabricksProvider(\n  config: ProviderConfiguration,\n): Promise<DatabricksProvider> {\n  // Dynamically import @databricks/sql\n  const { DBSQLClient } = await getDatabricksSQL();\n\n  const requiredFields = [\n    \"server_hostname\",\n    \"http_path\",\n    \"client_id\",\n    \"client_secret\",\n    \"catalog\",\n  ] as const;\n\n  for (const field of requiredFields) {\n    if (!config.fields[field]) {\n      throw new InvalidProviderConfigurationError(\n        `Missing field '${field}' in the provider '${config.name}' to connect to Databricks.`,\n      );\n    }\n  }\n\n  const serverHostname = config.fields[\"server_hostname\"]!;\n  const httpPath = config.fields[\"http_path\"]!;\n  const clientId = config.fields[\"client_id\"]!;\n  const clientSecret = config.fields[\"client_secret\"]!;\n  const catalog = config.fields[\"catalog\"]!;\n  const schema = config.fields[\"schema\"] ?? getDefaultSchema();\n\n  const client = new DBSQLClient();\n\n  await client.connect({\n    host: serverHostname,\n    path: httpPath,\n    authType: \"databricks-oauth\",\n    oauthClientId: clientId,\n    oauthClientSecret: clientSecret,\n  });\n\n  const session = await client.openSession({\n    initialCatalog: catalog,\n    initialSchema: schema,\n  });\n\n  return new DatabricksProvider(client, session);\n}\n\n"],"names":["getDatabricksSQL","InvalidProviderConfigurationError","DatabricksProvider","client","session","sqlQuery","_params","operation","getDefaultSchema","buildDatabricksProvider","config","DBSQLClient","requiredFields","field","serverHostname","httpPath","clientId","clientSecret","catalog","schema"],"mappings":";;AA4BA,eAAeA,IAA8D;AAC3E,MAAI;AACF,WAAO,MAAM,OAAO,iBAAiB;AAAA,EACvC,QAAQ;AACN,UAAM,IAAIC;AAAA,MACR;AAAA,IAAA;AAAA,EAGJ;AACF;AAKO,MAAMC,EAAmB;AAAA,EACtB;AAAA,EACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQR,YAAYC,GAAqBC,GAAwB;AACvD,SAAK,SAASD,GACd,KAAK,UAAUC;AAAA,EACjB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,MACJC,GACAC,IAAmC,IACM;AACzC,UAAMC,IAAY,MAAM,KAAK,QAAQ,iBAAiBF,CAAQ;AAC9D,QAAI;AAEF,aADe,MAAME,EAAU,SAAA;AAAA,IAEjC,UAAA;AAEE,YAAMA,EAAU,MAAA;AAAA,IAClB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAuB;AAC3B,UAAM,KAAK,QAAQ,MAAA,GACnB,MAAM,KAAK,OAAO,MAAA;AAAA,EACpB;AACF;AAKA,SAASC,IAA2B;AAGlC,SAAO,IAFS,QAAQ,IAAI,YAAY,IACN,QAAQ,MAAM,GAAG,CACxB;AAC7B;AASA,eAAsBC,EACpBC,GAC6B;AAE7B,QAAM,EAAE,aAAAC,MAAgB,MAAMX,EAAA,GAExBY,IAAiB;AAAA,IACrB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA;AAGF,aAAWC,KAASD;AAClB,QAAI,CAACF,EAAO,OAAOG,CAAK;AACtB,YAAM,IAAIZ;AAAA,QACR,kBAAkBY,CAAK,sBAAsBH,EAAO,IAAI;AAAA,MAAA;AAK9D,QAAMI,IAAiBJ,EAAO,OAAO,iBAC/BK,IAAWL,EAAO,OAAO,WACzBM,IAAWN,EAAO,OAAO,WACzBO,IAAeP,EAAO,OAAO,eAC7BQ,IAAUR,EAAO,OAAO,SACxBS,IAAST,EAAO,OAAO,UAAaF,EAAA,GAEpCL,IAAS,IAAIQ,EAAA;AAEnB,QAAMR,EAAO,QAAQ;AAAA,IACnB,MAAMW;AAAA,IACN,MAAMC;AAAA,IACN,UAAU;AAAA,IACV,eAAeC;AAAA,IACf,mBAAmBC;AAAA,EAAA,CACpB;AAED,QAAMb,IAAU,MAAMD,EAAO,YAAY;AAAA,IACvC,gBAAgBe;AAAA,IAChB,eAAeC;AAAA,EAAA,CAChB;AAED,SAAO,IAAIjB,EAAmBC,GAAQC,CAAO;AAC/C;"}
+{"version":3,"file":"databricks.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;"}

package/dist/providers/snowflake.cjs

@@ -1,2 +1,2 @@
-"use strict";var f=Object.create;var u=Object.defineProperty;var v=Object.getOwnPropertyDescriptor;var E=Object.getOwnPropertyNames;var P=Object.getPrototypeOf,h=Object.prototype.hasOwnProperty;var k=(e,r,o,t)=>{if(r&&typeof r=="object"||typeof r=="function")for(let n of E(r))!h.call(e,n)&&n!==o&&u(e,n,{get:()=>r[n],enumerable:!(t=v(r,n))||t.enumerable});return e};var m=(e,r,o)=>(o=e!=null?f(P(e)):{},k(r||!e||!e.__esModule?u(o,"default",{value:e,enumerable:!0}):o,e));Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const i=require("../exceptions-DN25pCDi.cjs");async function S(){try{const e=await import("snowflake-sdk");return e.default||e}catch{throw new i.InvalidProviderConfigurationError("The 'snowflake-sdk' package is required to use the Snowflake provider. Please install it: npm install snowflake-sdk")}}class w{connection;constructor(r){this.connection=r}async query(r,o={}){const t=(Array.isArray(o),o);return new Promise((n,a)=>{this.connection.execute({sqlText:r,binds:t,complete:(c,d,s)=>{if(c){a(new i.QueryError(`Snowflake provider failed to query the database. Reason: ${c.message}`));return}n(s??[])}})})}async close(){return new Promise((r,o)=>{this.connection.destroy(t=>{if(t){o(t);return}r()})})}}async function y(e){const r=await S(),o=["user","password","account","warehouse","database","schema","role"];for(const n of o)if(!e.fields[n])throw new i.InvalidProviderConfigurationError(`Missing field '${n}' in the provider '${e.name}' to connect to Snowflake using password based authentication.`);const t={account:e.fields.account,username:e.fields.user,password:e.fields.password,warehouse:e.fields.warehouse,database:e.fields.database,schema:e.fields.schema,role:e.fields.role};return new Promise((n,a)=>{r.createConnection(t).connect((d,s)=>{if(d){a(new i.ProviderConnectionError(`Unable to connect to Snowflake. Reason: ${d.message}`));return}s.execute({sqlText:"ALTER SESSION SET TIMEZONE = 'UTC'",complete:l=>{l&&console.warn(`Warning: Failed to set timezone to UTC: ${l.message}`),n(new w(s))}})})})}exports.InvalidProviderConfigurationError=i.InvalidProviderConfigurationError;exports.MissingTableError=i.MissingTableError;exports.ProviderConnectionError=i.ProviderConnectionError;exports.ProviderError=i.ProviderError;exports.QueryError=i.QueryError;exports.SnowflakeProvider=w;exports.buildSnowflakeProvider=y;
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const o=require("../server/providers/SnowflakeProvider.cjs"),r=require("../server/providers/exceptions.cjs");exports.SnowflakeProvider=o.SnowflakeProvider;exports.buildSnowflakeProvider=o.buildSnowflakeProvider;exports.InvalidProviderConfigurationError=r.InvalidProviderConfigurationError;exports.MissingTableError=r.MissingTableError;exports.ProviderConnectionError=r.ProviderConnectionError;exports.ProviderError=r.ProviderError;exports.QueryError=r.QueryError;
 //# sourceMappingURL=snowflake.cjs.map

package/dist/providers/snowflake.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"snowflake.cjs","sources":["../../src/server/providers/SnowflakeProvider.ts"],"sourcesContent":["/**\n * Snowflake Data Provider\n *\n * TypeScript equivalent of SnowflakeProvider from\n * ts-lib-ui-kit-streamlit/tetrascience/data_app_providers/provider.py\n *\n * @remarks\n * This provider requires the `snowflake-sdk` package to be installed.\n * It is an optional peer dependency - install it only if you need Snowflake support:\n * ```bash\n * npm install snowflake-sdk\n * # or\n * yarn add snowflake-sdk\n * ```\n */\n\nimport {\n  QueryError,\n  ProviderConnectionError,\n  InvalidProviderConfigurationError,\n} from \"./exceptions\";\n\nimport type { ProviderConfiguration } from \"./types\";\n\n// Type imports for snowflake-sdk (these don't require the package at runtime)\ntype SnowflakeSDK = typeof import(\"snowflake-sdk\");\ntype SnowflakeConnection = import(\"snowflake-sdk\").Connection;\ntype SnowflakeBinds = import(\"snowflake-sdk\").Binds;\ntype SnowflakeError = import(\"snowflake-sdk\").SnowflakeError;\ntype SnowflakeConnectionOptions = import(\"snowflake-sdk\").ConnectionOptions;\ntype SnowflakeRowStatement = import(\"snowflake-sdk\").RowStatement;\ntype SnowflakeFileAndStageBindStatement =\n  import(\"snowflake-sdk\").FileAndStageBindStatement;\n\n/**\n * Dynamically import snowflake-sdk\n * @throws {InvalidProviderConfigurationError} If snowflake-sdk is not installed\n */\nasync function getSnowflakeSDK(): Promise<SnowflakeSDK> {\n  try {\n    const snowflake = await import(\"snowflake-sdk\");\n    return snowflake.default || snowflake;\n  } catch {\n    throw new InvalidProviderConfigurationError(\n      \"The 'snowflake-sdk' package is required to use the Snowflake provider. \" +\n        \"Please install it: npm install snowflake-sdk\",\n    );\n  }\n}\n\n/**\n * Snowflake data provider\n */\nexport class SnowflakeProvider {\n  private connection: SnowflakeConnection;\n\n  /**\n   * Initialize the Snowflake data provider\n   *\n   * @param connection - Snowflake connection\n   */\n  constructor(connection: SnowflakeConnection) {\n    this.connection = connection;\n  }\n\n  /**\n   * Query the Snowflake database\n   *\n   * @param sqlText - SQL query to execute\n   * @param params - Parameters to pass to the query. For positional binds, use an array.\n   *                 For named binds, use an object with keys matching the bind variable names.\n   * @returns Promise resolving to array of row objects\n   */\n  async query(\n    sqlText: string,\n    params: Record<string, unknown> | unknown[] = {},\n  ): Promise<Array<Record<string, unknown>>> {\n    // Snowflake SDK supports both positional binds (array) and named binds (object)\n    // Pass params directly to preserve named bind semantics\n    const binds = Array.isArray(params)\n      ? (params as unknown as SnowflakeBinds)\n      : (params as unknown as SnowflakeBinds);\n\n    return new Promise((resolve, reject) => {\n      this.connection.execute({\n        sqlText,\n        binds,\n        complete: (\n          err: SnowflakeError | undefined,\n          _stmt: SnowflakeRowStatement | SnowflakeFileAndStageBindStatement,\n          rows?: Array<Record<string, unknown>>,\n        ) => {\n          if (err) {\n            reject(\n              new QueryError(\n                `Snowflake provider failed to query the database. Reason: ${err.message}`,\n              ),\n            );\n            return;\n          }\n          resolve(rows ?? []);\n        },\n      });\n    });\n  }\n\n  /**\n   * Close the Snowflake connection\n   */\n  async close(): Promise<void> {\n    return new Promise((resolve, reject) => {\n      this.connection.destroy((err: SnowflakeError | undefined) => {\n        if (err) {\n          reject(err);\n          return;\n        }\n        resolve();\n      });\n    });\n  }\n}\n\n/**\n * Build a Snowflake data provider from the configuration\n *\n * @param config - Provider configuration\n * @returns Promise resolving to Snowflake data provider\n * @throws {InvalidProviderConfigurationError} If snowflake-sdk is not installed or config is invalid\n */\nexport async function buildSnowflakeProvider(\n  config: ProviderConfiguration,\n): Promise<SnowflakeProvider> {\n  // Dynamically import snowflake-sdk\n  const snowflake = await getSnowflakeSDK();\n\n  const requiredFields = [\n    \"user\",\n    \"password\",\n    \"account\",\n    \"warehouse\",\n    \"database\",\n    \"schema\",\n    \"role\",\n  ] as const;\n\n  for (const field of requiredFields) {\n    if (!config.fields[field]) {\n      throw new InvalidProviderConfigurationError(\n        `Missing field '${field}' in the provider '${config.name}' to connect to Snowflake ` +\n          \"using password based authentication.\",\n      );\n    }\n  }\n\n  const connectionOptions: SnowflakeConnectionOptions = {\n    account: config.fields[\"account\"]!,\n    username: config.fields[\"user\"]!,\n    password: config.fields[\"password\"]!,\n    warehouse: config.fields[\"warehouse\"]!,\n    database: config.fields[\"database\"]!,\n    schema: config.fields[\"schema\"]!,\n    role: config.fields[\"role\"]!,\n  };\n\n  return new Promise((resolve, reject) => {\n    const connection = snowflake.createConnection(connectionOptions);\n\n    connection.connect(\n      (err: SnowflakeError | undefined, conn: SnowflakeConnection) => {\n        if (err) {\n          reject(\n            new ProviderConnectionError(\n              `Unable to connect to Snowflake. Reason: ${err.message}`,\n            ),\n          );\n          return;\n        }\n\n        // Set timezone to UTC after connection.\n        // The default timezone is America/Los_Angeles. Timestamps from delta tables will be\n        // returned with the timezone `timezone`. NOTE, this is not a timezone conversion.\n        // The timezone is replaced without translating the clock time. `timezone` is set UTC\n        // to match the timezone of delta table timestamps.\n        conn.execute({\n          sqlText: \"ALTER SESSION SET TIMEZONE = 'UTC'\",\n          complete: (tzErr: SnowflakeError | undefined) => {\n            if (tzErr) {\n              // Log warning but don't fail - timezone setting is not critical\n              console.warn(\n                `Warning: Failed to set timezone to UTC: ${tzErr.message}`,\n              );\n            }\n            resolve(new SnowflakeProvider(conn));\n          },\n        });\n      },\n    );\n  });\n}\n\n"],"names":["getSnowflakeSDK","snowflake","InvalidProviderConfigurationError","SnowflakeProvider","connection","sqlText","params","binds","resolve","reject","err","_stmt","rows","QueryError","buildSnowflakeProvider","config","requiredFields","field","connectionOptions","conn","ProviderConnectionError","tzErr"],"mappings":"ykBAsCA,eAAeA,GAAyC,CACtD,GAAI,CACF,MAAMC,EAAY,KAAM,QAAO,eAAe,EAC9C,OAAOA,EAAU,SAAWA,CAC9B,MAAQ,CACN,MAAM,IAAIC,EAAAA,kCACR,qHAAA,CAGJ,CACF,CAKO,MAAMC,CAAkB,CACrB,WAOR,YAAYC,EAAiC,CAC3C,KAAK,WAAaA,CACpB,CAUA,MAAM,MACJC,EACAC,EAA8C,GACL,CAGzC,MAAMC,GAAQ,MAAM,QAAQD,CAAM,EAC7BA,GAGL,OAAO,IAAI,QAAQ,CAACE,EAASC,IAAW,CACtC,KAAK,WAAW,QAAQ,CACtB,QAAAJ,EACA,MAAAE,EACA,SAAU,CACRG,EACAC,EACAC,IACG,CACH,GAAIF,EAAK,CACPD,EACE,IAAII,EAAAA,WACF,4DAA4DH,EAAI,OAAO,EAAA,CACzE,EAEF,MACF,CACAF,EAAQI,GAAQ,EAAE,CACpB,CAAA,CACD,CACH,CAAC,CACH,CAKA,MAAM,OAAuB,CAC3B,OAAO,IAAI,QAAQ,CAACJ,EAASC,IAAW,CACtC,KAAK,WAAW,QAASC,GAAoC,CAC3D,GAAIA,EAAK,CACPD,EAAOC,CAAG,EACV,MACF,CACAF,EAAA,CACF,CAAC,CACH,CAAC,CACH,CACF,CASA,eAAsBM,EACpBC,EAC4B,CAE5B,MAAMd,EAAY,MAAMD,EAAA,EAElBgB,EAAiB,CACrB,OACA,WACA,UACA,YACA,WACA,SACA,MAAA,EAGF,UAAWC,KAASD,EAClB,GAAI,CAACD,EAAO,OAAOE,CAAK,EACtB,MAAM,IAAIf,EAAAA,kCACR,kBAAkBe,CAAK,sBAAsBF,EAAO,IAAI,gEAAA,EAM9D,MAAMG,EAAgD,CACpD,QAASH,EAAO,OAAO,QACvB,SAAUA,EAAO,OAAO,KACxB,SAAUA,EAAO,OAAO,SACxB,UAAWA,EAAO,OAAO,UACzB,SAAUA,EAAO,OAAO,SACxB,OAAQA,EAAO,OAAO,OACtB,KAAMA,EAAO,OAAO,IAAM,EAG5B,OAAO,IAAI,QAAQ,CAACP,EAASC,IAAW,CACnBR,EAAU,iBAAiBiB,CAAiB,EAEpD,QACT,CAACR,EAAiCS,IAA8B,CAC9D,GAAIT,EAAK,CACPD,EACE,IAAIW,EAAAA,wBACF,2CAA2CV,EAAI,OAAO,EAAA,CACxD,EAEF,MACF,CAOAS,EAAK,QAAQ,CACX,QAAS,qCACT,SAAWE,GAAsC,CAC3CA,GAEF,QAAQ,KACN,2CAA2CA,EAAM,OAAO,EAAA,EAG5Db,EAAQ,IAAIL,EAAkBgB,CAAI,CAAC,CACrC,CAAA,CACD,CACH,CAAA,CAEJ,CAAC,CACH"}
+{"version":3,"file":"snowflake.cjs","sources":[],"sourcesContent":[],"names":[],"mappings":""}

package/dist/providers/snowflake.js

@@ -1,122 +1,12 @@
-import { Q as u, I as d, P as w } from "../exceptions-jCQ6h5C8.js";
-import { M as y, a as P } from "../exceptions-jCQ6h5C8.js";
-async function f() {
-  try {
-    const e = await import("snowflake-sdk");
-    return e.default || e;
-  } catch {
-    throw new d(
-      "The 'snowflake-sdk' package is required to use the Snowflake provider. Please install it: npm install snowflake-sdk"
-    );
-  }
-}
-class m {
-  connection;
-  /**
-   * Initialize the Snowflake data provider
-   *
-   * @param connection - Snowflake connection
-   */
-  constructor(o) {
-    this.connection = o;
-  }
-  /**
-   * Query the Snowflake database
-   *
-   * @param sqlText - SQL query to execute
-   * @param params - Parameters to pass to the query. For positional binds, use an array.
-   *                 For named binds, use an object with keys matching the bind variable names.
-   * @returns Promise resolving to array of row objects
-   */
-  async query(o, n = {}) {
-    const s = (Array.isArray(n), n);
-    return new Promise((t, a) => {
-      this.connection.execute({
-        sqlText: o,
-        binds: s,
-        complete: (i, c, r) => {
-          if (i) {
-            a(
-              new u(
-                `Snowflake provider failed to query the database. Reason: ${i.message}`
-              )
-            );
-            return;
-          }
-          t(r ?? []);
-        }
-      });
-    });
-  }
-  /**
-   * Close the Snowflake connection
-   */
-  async close() {
-    return new Promise((o, n) => {
-      this.connection.destroy((s) => {
-        if (s) {
-          n(s);
-          return;
-        }
-        o();
-      });
-    });
-  }
-}
-async function k(e) {
-  const o = await f(), n = [
-    "user",
-    "password",
-    "account",
-    "warehouse",
-    "database",
-    "schema",
-    "role"
-  ];
-  for (const t of n)
-    if (!e.fields[t])
-      throw new d(
-        `Missing field '${t}' in the provider '${e.name}' to connect to Snowflake using password based authentication.`
-      );
-  const s = {
-    account: e.fields.account,
-    username: e.fields.user,
-    password: e.fields.password,
-    warehouse: e.fields.warehouse,
-    database: e.fields.database,
-    schema: e.fields.schema,
-    role: e.fields.role
-  };
-  return new Promise((t, a) => {
-    o.createConnection(s).connect(
-      (c, r) => {
-        if (c) {
-          a(
-            new w(
-              `Unable to connect to Snowflake. Reason: ${c.message}`
-            )
-          );
-          return;
-        }
-        r.execute({
-          sqlText: "ALTER SESSION SET TIMEZONE = 'UTC'",
-          complete: (l) => {
-            l && console.warn(
-              `Warning: Failed to set timezone to UTC: ${l.message}`
-            ), t(new m(r));
-          }
-        });
-      }
-    );
-  });
-}
+import { SnowflakeProvider as e, buildSnowflakeProvider as i } from "../server/providers/SnowflakeProvider.js";
+import { InvalidProviderConfigurationError as d, MissingTableError as v, ProviderConnectionError as a, ProviderError as f, QueryError as l } from "../server/providers/exceptions.js";
 export {
   d as InvalidProviderConfigurationError,
-  y as MissingTableError,
-  w as ProviderConnectionError,
-  P as ProviderError,
-  u as QueryError,
-  m as SnowflakeProvider,
-  k as buildSnowflakeProvider
+  v as MissingTableError,
+  a as ProviderConnectionError,
+  f as ProviderError,
+  l as QueryError,
+  e as SnowflakeProvider,
+  i as buildSnowflakeProvider
 };
 //# sourceMappingURL=snowflake.js.map

package/dist/providers/snowflake.js.map

@@ -1 +1 @@
-{"version":3,"file":"snowflake.js","sources":["../../src/server/providers/SnowflakeProvider.ts"],"sourcesContent":["/**\n * Snowflake Data Provider\n *\n * TypeScript equivalent of SnowflakeProvider from\n * ts-lib-ui-kit-streamlit/tetrascience/data_app_providers/provider.py\n *\n * @remarks\n * This provider requires the `snowflake-sdk` package to be installed.\n * It is an optional peer dependency - install it only if you need Snowflake support:\n * ```bash\n * npm install snowflake-sdk\n * # or\n * yarn add snowflake-sdk\n * ```\n */\n\nimport {\n  QueryError,\n  ProviderConnectionError,\n  InvalidProviderConfigurationError,\n} from \"./exceptions\";\n\nimport type { ProviderConfiguration } from \"./types\";\n\n// Type imports for snowflake-sdk (these don't require the package at runtime)\ntype SnowflakeSDK = typeof import(\"snowflake-sdk\");\ntype SnowflakeConnection = import(\"snowflake-sdk\").Connection;\ntype SnowflakeBinds = import(\"snowflake-sdk\").Binds;\ntype SnowflakeError = import(\"snowflake-sdk\").SnowflakeError;\ntype SnowflakeConnectionOptions = import(\"snowflake-sdk\").ConnectionOptions;\ntype SnowflakeRowStatement = import(\"snowflake-sdk\").RowStatement;\ntype SnowflakeFileAndStageBindStatement =\n  import(\"snowflake-sdk\").FileAndStageBindStatement;\n\n/**\n * Dynamically import snowflake-sdk\n * @throws {InvalidProviderConfigurationError} If snowflake-sdk is not installed\n */\nasync function getSnowflakeSDK(): Promise<SnowflakeSDK> {\n  try {\n    const snowflake = await import(\"snowflake-sdk\");\n    return snowflake.default || snowflake;\n  } catch {\n    throw new InvalidProviderConfigurationError(\n      \"The 'snowflake-sdk' package is required to use the Snowflake provider. \" +\n        \"Please install it: npm install snowflake-sdk\",\n    );\n  }\n}\n\n/**\n * Snowflake data provider\n */\nexport class SnowflakeProvider {\n  private connection: SnowflakeConnection;\n\n  /**\n   * Initialize the Snowflake data provider\n   *\n   * @param connection - Snowflake connection\n   */\n  constructor(connection: SnowflakeConnection) {\n    this.connection = connection;\n  }\n\n  /**\n   * Query the Snowflake database\n   *\n   * @param sqlText - SQL query to execute\n   * @param params - Parameters to pass to the query. For positional binds, use an array.\n   *                 For named binds, use an object with keys matching the bind variable names.\n   * @returns Promise resolving to array of row objects\n   */\n  async query(\n    sqlText: string,\n    params: Record<string, unknown> | unknown[] = {},\n  ): Promise<Array<Record<string, unknown>>> {\n    // Snowflake SDK supports both positional binds (array) and named binds (object)\n    // Pass params directly to preserve named bind semantics\n    const binds = Array.isArray(params)\n      ? (params as unknown as SnowflakeBinds)\n      : (params as unknown as SnowflakeBinds);\n\n    return new Promise((resolve, reject) => {\n      this.connection.execute({\n        sqlText,\n        binds,\n        complete: (\n          err: SnowflakeError | undefined,\n          _stmt: SnowflakeRowStatement | SnowflakeFileAndStageBindStatement,\n          rows?: Array<Record<string, unknown>>,\n        ) => {\n          if (err) {\n            reject(\n              new QueryError(\n                `Snowflake provider failed to query the database. Reason: ${err.message}`,\n              ),\n            );\n            return;\n          }\n          resolve(rows ?? []);\n        },\n      });\n    });\n  }\n\n  /**\n   * Close the Snowflake connection\n   */\n  async close(): Promise<void> {\n    return new Promise((resolve, reject) => {\n      this.connection.destroy((err: SnowflakeError | undefined) => {\n        if (err) {\n          reject(err);\n          return;\n        }\n        resolve();\n      });\n    });\n  }\n}\n\n/**\n * Build a Snowflake data provider from the configuration\n *\n * @param config - Provider configuration\n * @returns Promise resolving to Snowflake data provider\n * @throws {InvalidProviderConfigurationError} If snowflake-sdk is not installed or config is invalid\n */\nexport async function buildSnowflakeProvider(\n  config: ProviderConfiguration,\n): Promise<SnowflakeProvider> {\n  // Dynamically import snowflake-sdk\n  const snowflake = await getSnowflakeSDK();\n\n  const requiredFields = [\n    \"user\",\n    \"password\",\n    \"account\",\n    \"warehouse\",\n    \"database\",\n    \"schema\",\n    \"role\",\n  ] as const;\n\n  for (const field of requiredFields) {\n    if (!config.fields[field]) {\n      throw new InvalidProviderConfigurationError(\n        `Missing field '${field}' in the provider '${config.name}' to connect to Snowflake ` +\n          \"using password based authentication.\",\n      );\n    }\n  }\n\n  const connectionOptions: SnowflakeConnectionOptions = {\n    account: config.fields[\"account\"]!,\n    username: config.fields[\"user\"]!,\n    password: config.fields[\"password\"]!,\n    warehouse: config.fields[\"warehouse\"]!,\n    database: config.fields[\"database\"]!,\n    schema: config.fields[\"schema\"]!,\n    role: config.fields[\"role\"]!,\n  };\n\n  return new Promise((resolve, reject) => {\n    const connection = snowflake.createConnection(connectionOptions);\n\n    connection.connect(\n      (err: SnowflakeError | undefined, conn: SnowflakeConnection) => {\n        if (err) {\n          reject(\n            new ProviderConnectionError(\n              `Unable to connect to Snowflake. Reason: ${err.message}`,\n            ),\n          );\n          return;\n        }\n\n        // Set timezone to UTC after connection.\n        // The default timezone is America/Los_Angeles. Timestamps from delta tables will be\n        // returned with the timezone `timezone`. NOTE, this is not a timezone conversion.\n        // The timezone is replaced without translating the clock time. `timezone` is set UTC\n        // to match the timezone of delta table timestamps.\n        conn.execute({\n          sqlText: \"ALTER SESSION SET TIMEZONE = 'UTC'\",\n          complete: (tzErr: SnowflakeError | undefined) => {\n            if (tzErr) {\n              // Log warning but don't fail - timezone setting is not critical\n              console.warn(\n                `Warning: Failed to set timezone to UTC: ${tzErr.message}`,\n              );\n            }\n            resolve(new SnowflakeProvider(conn));\n          },\n        });\n      },\n    );\n  });\n}\n\n"],"names":["getSnowflakeSDK","snowflake","InvalidProviderConfigurationError","SnowflakeProvider","connection","sqlText","params","binds","resolve","reject","err","_stmt","rows","QueryError","buildSnowflakeProvider","config","requiredFields","field","connectionOptions","conn","ProviderConnectionError","tzErr"],"mappings":";;AAsCA,eAAeA,IAAyC;AACtD,MAAI;AACF,UAAMC,IAAY,MAAM,OAAO,eAAe;AAC9C,WAAOA,EAAU,WAAWA;AAAA,EAC9B,QAAQ;AACN,UAAM,IAAIC;AAAA,MACR;AAAA,IAAA;AAAA,EAGJ;AACF;AAKO,MAAMC,EAAkB;AAAA,EACrB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOR,YAAYC,GAAiC;AAC3C,SAAK,aAAaA;AAAA,EACpB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,MACJC,GACAC,IAA8C,IACL;AAGzC,UAAMC,KAAQ,MAAM,QAAQD,CAAM,GAC7BA;AAGL,WAAO,IAAI,QAAQ,CAACE,GAASC,MAAW;AACtC,WAAK,WAAW,QAAQ;AAAA,QACtB,SAAAJ;AAAA,QACA,OAAAE;AAAA,QACA,UAAU,CACRG,GACAC,GACAC,MACG;AACH,cAAIF,GAAK;AACP,YAAAD;AAAA,cACE,IAAII;AAAA,gBACF,4DAA4DH,EAAI,OAAO;AAAA,cAAA;AAAA,YACzE;AAEF;AAAA,UACF;AACA,UAAAF,EAAQI,KAAQ,EAAE;AAAA,QACpB;AAAA,MAAA,CACD;AAAA,IACH,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAuB;AAC3B,WAAO,IAAI,QAAQ,CAACJ,GAASC,MAAW;AACtC,WAAK,WAAW,QAAQ,CAACC,MAAoC;AAC3D,YAAIA,GAAK;AACP,UAAAD,EAAOC,CAAG;AACV;AAAA,QACF;AACA,QAAAF,EAAA;AAAA,MACF,CAAC;AAAA,IACH,CAAC;AAAA,EACH;AACF;AASA,eAAsBM,EACpBC,GAC4B;AAE5B,QAAMd,IAAY,MAAMD,EAAA,GAElBgB,IAAiB;AAAA,IACrB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA;AAGF,aAAWC,KAASD;AAClB,QAAI,CAACD,EAAO,OAAOE,CAAK;AACtB,YAAM,IAAIf;AAAA,QACR,kBAAkBe,CAAK,sBAAsBF,EAAO,IAAI;AAAA,MAAA;AAM9D,QAAMG,IAAgD;AAAA,IACpD,SAASH,EAAO,OAAO;AAAA,IACvB,UAAUA,EAAO,OAAO;AAAA,IACxB,UAAUA,EAAO,OAAO;AAAA,IACxB,WAAWA,EAAO,OAAO;AAAA,IACzB,UAAUA,EAAO,OAAO;AAAA,IACxB,QAAQA,EAAO,OAAO;AAAA,IACtB,MAAMA,EAAO,OAAO;AAAA,EAAM;AAG5B,SAAO,IAAI,QAAQ,CAACP,GAASC,MAAW;AAGtC,IAFmBR,EAAU,iBAAiBiB,CAAiB,EAEpD;AAAA,MACT,CAACR,GAAiCS,MAA8B;AAC9D,YAAIT,GAAK;AACP,UAAAD;AAAA,YACE,IAAIW;AAAA,cACF,2CAA2CV,EAAI,OAAO;AAAA,YAAA;AAAA,UACxD;AAEF;AAAA,QACF;AAOA,QAAAS,EAAK,QAAQ;AAAA,UACX,SAAS;AAAA,UACT,UAAU,CAACE,MAAsC;AAC/C,YAAIA,KAEF,QAAQ;AAAA,cACN,2CAA2CA,EAAM,OAAO;AAAA,YAAA,GAG5Db,EAAQ,IAAIL,EAAkBgB,CAAI,CAAC;AAAA,UACrC;AAAA,QAAA,CACD;AAAA,MACH;AAAA,IAAA;AAAA,EAEJ,CAAC;AACH;"}
+{"version":3,"file":"snowflake.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;"}

package/dist/server/auth/JwtTokenManager.cjs

@@ -0,0 +1,2 @@
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const a=require("@tetrascience-npm/ts-connectors-sdk"),c=60,s=1e3,l=5*c*s;class i{baseUrlOverride;connectorId;orgSlug;tokenCache;tokenRefreshThresholdMs;tdpClient;constructor(e={}){this.baseUrlOverride=e.baseUrl,this.connectorId=process.env.CONNECTOR_ID,this.orgSlug=process.env.ORG_SLUG,this.tokenCache=new Map,this.tokenRefreshThresholdMs=e.tokenRefreshThresholdMs||l,this.tdpClient=null}getBaseUrl(){const e=this.baseUrlOverride||process.env.TDP_ENDPOINT;if(!e)throw new Error("TDP base URL not configured. Set TDP_ENDPOINT environment variable or pass baseUrl in config.");return e}decodeJwtPayload(e){try{const t=e.split(".");if(t.length!==3)return console.warn("Invalid JWT token format"),null;const n=t[1].replace(/-/g,"+").replace(/_/g,"/"),o=n.padEnd(n.length+(4-n.length%4)%4,"=");return JSON.parse(Buffer.from(o,"base64").toString("utf-8"))}catch(t){return console.warn("Error decoding JWT token:",t),null}}isPayloadExpiringSoon(e){if(!e.exp)return console.warn("JWT token has no expiration claim"),!0;const t=e.exp*s,r=Date.now()+this.tokenRefreshThresholdMs;return t<=r}getValidUserJwt(e){const t=this.tokenCache.get(e);return t&&!this.isPayloadExpiringSoon(t.payload)?t.token:(t&&this.tokenCache.delete(e),null)}async getTdpClient(){if(this.tdpClient!==null)return this.tdpClient;if(!this.connectorId||!this.orgSlug)throw new Error("Missing required configuration: CONNECTOR_ID or ORG_SLUG");const e=this.getBaseUrl();try{const t=new a.TDPClient({tdpEndpoint:e,connectorId:this.connectorId,orgSlug:this.orgSlug,artifactType:"data-app"});return await t.init(),this.tdpClient=t,this.tdpClient}catch(t){throw console.error("Failed to initialize TDP client:",t),t}}async getJwtFromTokenRefInternal(e){const t=await this.getTdpClient();try{const r=await t.getValues([e]);if(r&&r.length>0&&r[0]?.jwt){const n=r[0].jwt,o=this.decodeJwtPayload(n);return o&&this.tokenCache.set(e,{token:n,payload:o}),n}console.error(`No JWT found for key '${e}' in connector store`)}catch(r){console.error("Error retrieving JWT token:",r)}return null}async getJwtFromTokenRef(e){if(!e||!this.orgSlug)return console.warn("Missing required parameters for JWT token retrieval"),null;const t=this.getValidUserJwt(e);return t||this.getJwtFromTokenRefInternal(e)}async getUserToken(e){const t=e["ts-auth-token"]||process.env.TS_AUTH_TOKEN;if(t)return t;const r=e["ts-token-ref"];if(r&&this.connectorId){const n=await this.getJwtFromTokenRef(r);if(n)return n;console.warn("Failed to resolve ts-token-ref to JWT token")}else r&&console.error("Connector ID not configured");return console.warn("No valid authentication token found"),null}async getTokenFromExpressRequest(e){return this.getUserToken(e.cookies||{})}clearCache(){this.tokenCache.clear()}}const h=new i;exports.JwtTokenManager=i;exports.jwtManager=h;
+//# sourceMappingURL=JwtTokenManager.cjs.map