@tetrascience-npm/tetrascience-react-ui 0.4.0-beta.9.1 → 0.4.0

package/dist/providers/athena.cjs

@@ -0,0 +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;
+//# sourceMappingURL=athena.cjs.map

package/dist/providers/athena.cjs.map

@@ -0,0 +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"}

package/dist/providers/athena.d.ts

@@ -1,79 +1,9 @@
-import * as _aws_sdk_client_athena from '@aws-sdk/client-athena';
-export { I as InvalidProviderConfigurationError, M as MissingTableError, P as ProviderConfiguration, a as ProviderConnectionError, b as ProviderError, Q as QueryError, c as QueryResult } from './types-Ck4uFaGp.js';
+export * from '../server/providers/entries/athena'
+export {}
 
-/**
- * Athena Data Provider
- *
- * TypeScript equivalent of AthenaProvider from
- * ts-lib-ui-kit-streamlit/tetrascience/data_app_providers/provider.py
- *
- * @remarks
- * This provider requires the `@aws-sdk/client-athena` package to be installed.
- * It is an optional peer dependency - install it only if you need Athena support:
- * ```bash
- * npm install @aws-sdk/client-athena
- * # or
- * yarn add @aws-sdk/client-athena
- * ```
- */
-type AthenaClient = _aws_sdk_client_athena.AthenaClient;
-type AthenaSDK = typeof _aws_sdk_client_athena;
-/**
- * Athena data provider
- */
-declare class AthenaProvider {
-    private client;
-    private sdk;
-    private workgroup;
-    private database;
-    private 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(client: AthenaClient, sdk: AthenaSDK, workgroup: string, database: string, outputLocation?: string);
-    /**
-     * 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.
-     */
-    query(sqlQuery: string, _params?: Record<string, unknown>): Promise<Array<Record<string, unknown>>>;
-    /**
-     * Wait for query to complete
-     */
-    private waitForQueryCompletion;
-    /**
-     * Fetch all results from a completed query
-     */
-    private fetchAllResults;
-    /**
-     * Close the Athena client (no-op for AWS SDK clients)
-     */
-    close(): Promise<void>;
-}
-/**
- * Get the TDP Athena provider
- *
- * Creates an Athena provider using TDP environment configuration
- *
- * @returns Promise resolving to Athena data provider
- * @throws {InvalidProviderConfigurationError} If @aws-sdk/client-athena is not installed
- * @throws {Error} If ATHENA_S3_OUTPUT_LOCATION is not set when using the 'primary' workgroup
- */
-declare function getTdpAthenaProvider(): Promise<AthenaProvider>;
 
-export { AthenaProvider, getTdpAthenaProvider };
+declare module "styled-components" {
+    interface DefaultTheme {
+        position: ToastPosition;
+    }
+}

package/dist/providers/athena.js

@@ -0,0 +1,160 @@
+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
+    );
+  }
+}
+export {
+  p as AthenaProvider,
+  f as InvalidProviderConfigurationError,
+  y as MissingTableError,
+  Q as ProviderConnectionError,
+  R as ProviderError,
+  m as QueryError,
+  C as getTdpAthenaProvider
+};
+//# sourceMappingURL=athena.js.map

package/dist/providers/athena.js.map

@@ -0,0 +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;"}

package/dist/providers/databricks.cjs

@@ -0,0 +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;
+//# sourceMappingURL=databricks.cjs.map

package/dist/providers/databricks.cjs.map

@@ -0,0 +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"}

package/dist/providers/databricks.d.ts

@@ -1,43 +1,9 @@
-import * as _databricks_sql_dist_contracts_IDBSQLSession from '@databricks/sql/dist/contracts/IDBSQLSession';
-import * as _databricks_sql from '@databricks/sql';
-import { P as ProviderConfiguration } from './types-Ck4uFaGp.js';
-export { I as InvalidProviderConfigurationError, M as MissingTableError, a as ProviderConnectionError, b as ProviderError, Q as QueryError, c as QueryResult } from './types-Ck4uFaGp.js';
+export * from '../server/providers/entries/databricks'
+export {}
 
-type DBSQLClient = _databricks_sql.DBSQLClient;
-type IDBSQLSession = _databricks_sql_dist_contracts_IDBSQLSession.default;
-/**
- * Databricks data provider
- */
-declare class DatabricksProvider {
-    private client;
-    private session;
-    /**
-     * Initialize the Databricks data provider
-     *
-     * @param client - Databricks SQL client
-     * @param session - Databricks SQL session
-     */
-    constructor(client: DBSQLClient, session: IDBSQLSession);
-    /**
-     * 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
-     */
-    query(sqlQuery: string, _params?: Record<string, unknown>): Promise<Array<Record<string, unknown>>>;
-    /**
-     * Close the Databricks connection
-     */
-    close(): Promise<void>;
-}
-/**
- * Build a Databricks data provider from the configuration
- *
- * @param config - Provider configuration
- * @returns Promise resolving to Databricks data provider
- * @throws {InvalidProviderConfigurationError} If @databricks/sql is not installed or config is invalid
- */
-declare function buildDatabricksProvider(config: ProviderConfiguration): Promise<DatabricksProvider>;
 
-export { DatabricksProvider, ProviderConfiguration, buildDatabricksProvider };
+declare module "styled-components" {
+    interface DefaultTheme {
+        position: ToastPosition;
+    }
+}

package/dist/providers/databricks.js

@@ -0,0 +1,85 @@
+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);
+}
+export {
+  f as DatabricksProvider,
+  o as InvalidProviderConfigurationError,
+  g as MissingTableError,
+  v as ProviderConnectionError,
+  y as ProviderError,
+  k as QueryError,
+  _ as buildDatabricksProvider
+};
+//# sourceMappingURL=databricks.js.map

package/dist/providers/databricks.js.map

@@ -0,0 +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;"}

package/dist/providers/snowflake.cjs

@@ -0,0 +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;
+//# sourceMappingURL=snowflake.cjs.map

package/dist/providers/snowflake.cjs.map

@@ -0,0 +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"}

package/dist/providers/snowflake.d.ts

@@ -1,40 +1,9 @@
-import * as snowflake_sdk from 'snowflake-sdk';
-import { P as ProviderConfiguration } from './types-Ck4uFaGp.js';
-export { I as InvalidProviderConfigurationError, M as MissingTableError, a as ProviderConnectionError, b as ProviderError, Q as QueryError, c as QueryResult } from './types-Ck4uFaGp.js';
+export * from '../server/providers/entries/snowflake'
+export {}
 
-type SnowflakeConnection = snowflake_sdk.Connection;
-/**
- * Snowflake data provider
- */
-declare class SnowflakeProvider {
-    private connection;
-    /**
-     * Initialize the Snowflake data provider
-     *
-     * @param connection - Snowflake connection
-     */
-    constructor(connection: SnowflakeConnection);
-    /**
-     * 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
-     */
-    query(sqlText: string, params?: Record<string, unknown> | unknown[]): Promise<Array<Record<string, unknown>>>;
-    /**
-     * Close the Snowflake connection
-     */
-    close(): Promise<void>;
-}
-/**
- * Build a Snowflake data provider from the configuration
- *
- * @param config - Provider configuration
- * @returns Promise resolving to Snowflake data provider
- * @throws {InvalidProviderConfigurationError} If snowflake-sdk is not installed or config is invalid
- */
-declare function buildSnowflakeProvider(config: ProviderConfiguration): Promise<SnowflakeProvider>;
 
-export { ProviderConfiguration, SnowflakeProvider, buildSnowflakeProvider };
+declare module "styled-components" {
+    interface DefaultTheme {
+        position: ToastPosition;
+    }
+}