@graphenedata/cli 0.0.18 → 0.0.19

package/README.md

@@ -74,7 +74,7 @@ GSQL is inspired by [Malloy](https://github.com/malloydata/malloy), from the cre
 - [Try the demo project](https://github.com/graphene-data/example-flights)
 - [Create a new Graphene project](/docs/setup.md)
 
-Graphene currently supports Snowflake, BigQuery, ClickHouse, and local data (via DuckDB) as data sources. It is easy for us to add more - just ask.
+Graphene currently supports Snowflake, BigQuery, ClickHouse, Postgres, and local data (via DuckDB) as data sources. It is easy for us to add more - just ask.
 
 Once your project is set up, simply start the dev server via `npm exec graphene serve` (or `pnpm graphene serve`, etc. based on your package manager) and then prompt your coding agent to do analytics work: answer a data question, build a dashboard, edit the model, etc.
 
@@ -84,6 +84,8 @@ Graphene itself is a CLI which can be installed via npm (or pnpm, yarn, etc.). T
 
 A Graphene project can either be a standalone repo or a directory within a larger codebase (such as dbt). It is comprised of _semantic models_ via .gsql files and _pages_ via .md files.
 
+<img alt="Architecture Diagram" src="./assets/marketecture.png"/>
+
 ### GSQL and Graphene markdown
 
 Semantic models are defined like so:

package/dist/cli/athena-WROJBSLV.js

@@ -0,0 +1,136 @@
+// connections/athena.ts
+import {
+  AthenaClient,
+  GetQueryExecutionCommand,
+  GetQueryResultsCommand,
+  GetTableMetadataCommand,
+  ListDatabasesCommand,
+  ListTableMetadataCommand,
+  StartQueryExecutionCommand
+} from "@aws-sdk/client-athena";
+var AthenaConnection = class {
+  client;
+  catalog;
+  database;
+  workGroup;
+  outputLocation;
+  constructor(opts = {}) {
+    if (!opts.region) throw new Error("Athena requires a region in config or AWS_REGION");
+    let clientConfig = { region: opts.region };
+    if (opts.accessKeyId && opts.secretAccessKey) clientConfig.credentials = { accessKeyId: opts.accessKeyId, secretAccessKey: opts.secretAccessKey, sessionToken: opts.sessionToken };
+    this.client = new AthenaClient(clientConfig);
+    this.catalog = opts.catalog || "AwsDataCatalog";
+    this.database = opts.database;
+    this.workGroup = opts.workGroup;
+    this.outputLocation = opts.outputLocation;
+  }
+  async runQuery(sql, params) {
+    let queryExecutionId = await this.startQuery(sql, params);
+    await this.waitForQuery(queryExecutionId);
+    return await this.fetchResults(queryExecutionId);
+  }
+  async listDatasets() {
+    let databases = [];
+    let nextToken;
+    while (true) {
+      let res = await this.client.send(new ListDatabasesCommand({ CatalogName: this.catalog, NextToken: nextToken }));
+      databases.push(...(res.DatabaseList || []).map((db) => String(db.Name).toLowerCase()));
+      if (!res.NextToken) return databases;
+      nextToken = res.NextToken;
+    }
+  }
+  async listTables(database = this.database) {
+    if (!database) throw new Error("Athena requires a database");
+    let resolvedDatabase = await this.resolveDatabaseName(database);
+    let tables = [];
+    let nextToken;
+    while (true) {
+      let res = await this.client.send(new ListTableMetadataCommand({ CatalogName: this.catalog, DatabaseName: resolvedDatabase, NextToken: nextToken }));
+      let validTypes = /* @__PURE__ */ new Set(["EXTERNAL_TABLE", "VIRTUAL_VIEW"]);
+      tables.push(...(res.TableMetadataList || []).filter((t) => validTypes.has(String(t.TableType))).map((t) => `${resolvedDatabase}.${String(t.Name).toLowerCase()}`));
+      if (!res.NextToken) return tables;
+      nextToken = res.NextToken;
+    }
+  }
+  async describeTable(target) {
+    let parts = target.split(".").filter(Boolean);
+    let table = parts.pop() || "";
+    let database = parts.join(".") || this.database;
+    if (!database) throw new Error("No Athena database specified and no default namespace configured");
+    let resolvedDatabase = await this.resolveDatabaseName(database);
+    let res = await this.client.send(new GetTableMetadataCommand({ CatalogName: this.catalog, DatabaseName: resolvedDatabase, TableName: table }));
+    return (res.TableMetadata?.Columns || []).map((col) => ({ name: String(col.Name).toLowerCase(), dataType: String(col.Type) }));
+  }
+  close() {
+    this.client.destroy();
+    return Promise.resolve();
+  }
+  async startQuery(sql, params) {
+    if (params && !Array.isArray(params)) throw new Error("Athena only supports positional query parameters");
+    let res = await this.client.send(
+      new StartQueryExecutionCommand({
+        QueryString: sql,
+        QueryExecutionContext: { Catalog: this.catalog, Database: this.database },
+        WorkGroup: this.workGroup,
+        ResultConfiguration: this.outputLocation ? { OutputLocation: this.outputLocation } : void 0,
+        ExecutionParameters: params?.map(renderAthenaParameter)
+      })
+    );
+    if (!res.QueryExecutionId) throw new Error("Athena did not return a query execution id");
+    return res.QueryExecutionId;
+  }
+  async waitForQuery(queryExecutionId) {
+    let started = Date.now();
+    let timeoutMs = 5 * 60 * 1e3;
+    while (Date.now() - started < timeoutMs) {
+      let res = await this.client.send(new GetQueryExecutionCommand({ QueryExecutionId: queryExecutionId }));
+      let status = res.QueryExecution?.Status;
+      if (status?.State == "SUCCEEDED") return;
+      if (status?.State == "FAILED" || status?.State == "CANCELLED") throw new Error(`Athena query ${status.State.toLowerCase()}: ${status.StateChangeReason || "unknown reason"}`);
+      await new Promise((resolve) => setTimeout(resolve, 1e3));
+    }
+    throw new Error(`Athena query timed out after ${Math.round(timeoutMs / 1e3)}s`);
+  }
+  async fetchResults(queryExecutionId) {
+    let rows = [];
+    let nextToken;
+    let firstPage = true;
+    while (true) {
+      let res = await this.client.send(new GetQueryResultsCommand({ QueryExecutionId: queryExecutionId, NextToken: nextToken, MaxResults: 1e3 }));
+      let columns = res.ResultSet?.ResultSetMetadata?.ColumnInfo || [];
+      let resultRows = firstPage ? (res.ResultSet?.Rows || []).slice(1) : res.ResultSet?.Rows || [];
+      for (let row of resultRows) {
+        let out = {};
+        columns.forEach((column, idx) => {
+          let value = row.Data?.[idx]?.VarCharValue;
+          out[String(column.Name).toLowerCase()] = parseAthenaValue(value, String(column.Type || ""));
+        });
+        rows.push(out);
+      }
+      firstPage = false;
+      if (!res.NextToken || rows.length >= 1e4) return { rows, totalRows: rows.length };
+      nextToken = res.NextToken;
+    }
+  }
+  async resolveDatabaseName(name) {
+    let databases = await this.listDatasets();
+    return databases.find((db) => db.toLowerCase() == name.toLowerCase()) || name;
+  }
+};
+function renderAthenaParameter(value) {
+  if (value === null || value === void 0) return "NULL";
+  if (typeof value == "number" || typeof value == "bigint") return String(value);
+  if (typeof value == "boolean") return value ? "true" : "false";
+  return `'${String(value).replace(/'/g, "''")}'`;
+}
+function parseAthenaValue(value, type) {
+  if (value === void 0) return null;
+  let normalizedType = type.toLowerCase();
+  if (normalizedType == "boolean") return value == "true";
+  if (/^(tinyint|smallint|integer|bigint|real|float|double|decimal)/.test(normalizedType)) return Number(value);
+  return value;
+}
+export {
+  AthenaConnection
+};
+//# sourceMappingURL=athena-WROJBSLV.js.map

package/dist/cli/athena-WROJBSLV.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../connections/athena.ts"],
+  "sourcesContent": ["import {\n  AthenaClient,\n  GetQueryExecutionCommand,\n  GetQueryResultsCommand,\n  GetTableMetadataCommand,\n  ListDatabasesCommand,\n  ListTableMetadataCommand,\n  StartQueryExecutionCommand,\n} from '@aws-sdk/client-athena'\n\nimport {type QueryConnection, type QueryParams, type QueryResult, type SchemaColumn} from './types.ts'\n\nexport interface AthenaOptions {\n  region?: string\n  catalog?: string\n  database?: string\n  workGroup?: string\n  outputLocation?: string\n  accessKeyId?: string\n  secretAccessKey?: string\n  sessionToken?: string\n}\n\n// Athena is serverless SQL over external data, usually S3 files registered in Glue.\n// Query execution is asynchronous: StartQueryExecution writes results to S3, then\n// GetQueryResults pages those materialized rows back through the Athena API.\nexport class AthenaConnection implements QueryConnection {\n  private client: AthenaClient\n  private catalog: string\n  private database?: string\n  private workGroup?: string\n  private outputLocation?: string\n\n  constructor(opts: AthenaOptions = {}) {\n    if (!opts.region) throw new Error('Athena requires a region in config or AWS_REGION')\n\n    let clientConfig: any = {region: opts.region}\n    if (opts.accessKeyId && opts.secretAccessKey) clientConfig.credentials = {accessKeyId: opts.accessKeyId, secretAccessKey: opts.secretAccessKey, sessionToken: opts.sessionToken}\n\n    this.client = new AthenaClient(clientConfig)\n    this.catalog = opts.catalog || 'AwsDataCatalog'\n    this.database = opts.database\n    this.workGroup = opts.workGroup\n    this.outputLocation = opts.outputLocation\n  }\n\n  async runQuery(sql: string, params?: QueryParams): Promise<QueryResult> {\n    let queryExecutionId = await this.startQuery(sql, params)\n    await this.waitForQuery(queryExecutionId)\n    return await this.fetchResults(queryExecutionId)\n  }\n\n  async listDatasets(): Promise<string[]> {\n    let databases: string[] = []\n    let nextToken: string | undefined\n    while (true) {\n      let res = await this.client.send(new ListDatabasesCommand({CatalogName: this.catalog, NextToken: nextToken}))\n      databases.push(...(res.DatabaseList || []).map(db => String(db.Name).toLowerCase()))\n      if (!res.NextToken) return databases\n      nextToken = res.NextToken\n    }\n  }\n\n  async listTables(database = this.database): Promise<string[]> {\n    if (!database) throw new Error('Athena requires a database')\n    let resolvedDatabase = await this.resolveDatabaseName(database)\n    let tables: string[] = []\n    let nextToken: string | undefined\n    while (true) {\n      let res = await this.client.send(new ListTableMetadataCommand({CatalogName: this.catalog, DatabaseName: resolvedDatabase, NextToken: nextToken}))\n      let validTypes = new Set(['EXTERNAL_TABLE', 'VIRTUAL_VIEW'])\n      tables.push(...(res.TableMetadataList || []).filter(t => validTypes.has(String(t.TableType))).map(t => `${resolvedDatabase}.${String(t.Name).toLowerCase()}`))\n      if (!res.NextToken) return tables\n      nextToken = res.NextToken\n    }\n  }\n\n  async describeTable(target: string): Promise<SchemaColumn[]> {\n    let parts = target.split('.').filter(Boolean)\n    let table = parts.pop() || ''\n    let database = parts.join('.') || this.database\n    if (!database) throw new Error('No Athena database specified and no default namespace configured')\n    let resolvedDatabase = await this.resolveDatabaseName(database)\n    let res = await this.client.send(new GetTableMetadataCommand({CatalogName: this.catalog, DatabaseName: resolvedDatabase, TableName: table}))\n    return (res.TableMetadata?.Columns || []).map(col => ({name: String(col.Name).toLowerCase(), dataType: String(col.Type)}))\n  }\n\n  close(): Promise<void> {\n    this.client.destroy()\n    return Promise.resolve()\n  }\n\n  private async startQuery(sql: string, params?: QueryParams) {\n    // Athena execution parameters are positional only and are rendered as SQL literals by the API.\n    // Named params would be misleading, because Athena cannot bind them by name.\n    if (params && !Array.isArray(params)) throw new Error('Athena only supports positional query parameters')\n    let res = await this.client.send(\n      new StartQueryExecutionCommand({\n        QueryString: sql,\n        QueryExecutionContext: {Catalog: this.catalog, Database: this.database},\n        WorkGroup: this.workGroup,\n        ResultConfiguration: this.outputLocation ? {OutputLocation: this.outputLocation} : undefined,\n        ExecutionParameters: params?.map(renderAthenaParameter),\n      }),\n    )\n    if (!res.QueryExecutionId) throw new Error('Athena did not return a query execution id')\n    return res.QueryExecutionId\n  }\n\n  private async waitForQuery(queryExecutionId: string) {\n    let started = Date.now()\n    let timeoutMs = 5 * 60 * 1000\n    while (Date.now() - started < timeoutMs) {\n      let res = await this.client.send(new GetQueryExecutionCommand({QueryExecutionId: queryExecutionId}))\n      let status = res.QueryExecution?.Status\n      if (status?.State == 'SUCCEEDED') return\n      if (status?.State == 'FAILED' || status?.State == 'CANCELLED') throw new Error(`Athena query ${status.State.toLowerCase()}: ${status.StateChangeReason || 'unknown reason'}`)\n      await new Promise(resolve => setTimeout(resolve, 1000))\n    }\n    throw new Error(`Athena query timed out after ${Math.round(timeoutMs / 1000)}s`)\n  }\n\n  private async fetchResults(queryExecutionId: string): Promise<QueryResult> {\n    let rows: Array<Record<string, unknown>> = []\n    let nextToken: string | undefined\n    let firstPage = true\n\n    while (true) {\n      let res = await this.client.send(new GetQueryResultsCommand({QueryExecutionId: queryExecutionId, NextToken: nextToken, MaxResults: 1000}))\n      let columns = res.ResultSet?.ResultSetMetadata?.ColumnInfo || []\n      let resultRows = firstPage ? (res.ResultSet?.Rows || []).slice(1) : res.ResultSet?.Rows || []\n\n      for (let row of resultRows) {\n        let out: Record<string, unknown> = {}\n        columns.forEach((column, idx) => {\n          let value = row.Data?.[idx]?.VarCharValue\n          out[String(column.Name).toLowerCase()] = parseAthenaValue(value, String(column.Type || ''))\n        })\n        rows.push(out)\n      }\n\n      firstPage = false\n      if (!res.NextToken || rows.length >= 10000) return {rows, totalRows: rows.length}\n      nextToken = res.NextToken\n    }\n  }\n\n  private async resolveDatabaseName(name: string): Promise<string> {\n    let databases = await this.listDatasets()\n    return databases.find(db => db.toLowerCase() == name.toLowerCase()) || name\n  }\n}\n\nfunction renderAthenaParameter(value: unknown) {\n  if (value === null || value === undefined) return 'NULL'\n  if (typeof value == 'number' || typeof value == 'bigint') return String(value)\n  if (typeof value == 'boolean') return value ? 'true' : 'false'\n  return `'${String(value).replace(/'/g, \"''\")}'`\n}\n\nfunction parseAthenaValue(value: string | undefined, type: string) {\n  if (value === undefined) return null\n  let normalizedType = type.toLowerCase()\n  if (normalizedType == 'boolean') return value == 'true'\n  if (/^(tinyint|smallint|integer|bigint|real|float|double|decimal)/.test(normalizedType)) return Number(value)\n  return value\n}\n"],
+  "mappings": ";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAkBA,IAAM,mBAAN,MAAkD;AAAA,EAC/C;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,OAAsB,CAAC,GAAG;AACpC,QAAI,CAAC,KAAK,OAAQ,OAAM,IAAI,MAAM,kDAAkD;AAEpF,QAAI,eAAoB,EAAC,QAAQ,KAAK,OAAM;AAC5C,QAAI,KAAK,eAAe,KAAK,gBAAiB,cAAa,cAAc,EAAC,aAAa,KAAK,aAAa,iBAAiB,KAAK,iBAAiB,cAAc,KAAK,aAAY;AAE/K,SAAK,SAAS,IAAI,aAAa,YAAY;AAC3C,SAAK,UAAU,KAAK,WAAW;AAC/B,SAAK,WAAW,KAAK;AACrB,SAAK,YAAY,KAAK;AACtB,SAAK,iBAAiB,KAAK;AAAA,EAC7B;AAAA,EAEA,MAAM,SAAS,KAAa,QAA4C;AACtE,QAAI,mBAAmB,MAAM,KAAK,WAAW,KAAK,MAAM;AACxD,UAAM,KAAK,aAAa,gBAAgB;AACxC,WAAO,MAAM,KAAK,aAAa,gBAAgB;AAAA,EACjD;AAAA,EAEA,MAAM,eAAkC;AACtC,QAAI,YAAsB,CAAC;AAC3B,QAAI;AACJ,WAAO,MAAM;AACX,UAAI,MAAM,MAAM,KAAK,OAAO,KAAK,IAAI,qBAAqB,EAAC,aAAa,KAAK,SAAS,WAAW,UAAS,CAAC,CAAC;AAC5G,gBAAU,KAAK,IAAI,IAAI,gBAAgB,CAAC,GAAG,IAAI,QAAM,OAAO,GAAG,IAAI,EAAE,YAAY,CAAC,CAAC;AACnF,UAAI,CAAC,IAAI,UAAW,QAAO;AAC3B,kBAAY,IAAI;AAAA,IAClB;AAAA,EACF;AAAA,EAEA,MAAM,WAAW,WAAW,KAAK,UAA6B;AAC5D,QAAI,CAAC,SAAU,OAAM,IAAI,MAAM,4BAA4B;AAC3D,QAAI,mBAAmB,MAAM,KAAK,oBAAoB,QAAQ;AAC9D,QAAI,SAAmB,CAAC;AACxB,QAAI;AACJ,WAAO,MAAM;AACX,UAAI,MAAM,MAAM,KAAK,OAAO,KAAK,IAAI,yBAAyB,EAAC,aAAa,KAAK,SAAS,cAAc,kBAAkB,WAAW,UAAS,CAAC,CAAC;AAChJ,UAAI,aAAa,oBAAI,IAAI,CAAC,kBAAkB,cAAc,CAAC;AAC3D,aAAO,KAAK,IAAI,IAAI,qBAAqB,CAAC,GAAG,OAAO,OAAK,WAAW,IAAI,OAAO,EAAE,SAAS,CAAC,CAAC,EAAE,IAAI,OAAK,GAAG,gBAAgB,IAAI,OAAO,EAAE,IAAI,EAAE,YAAY,CAAC,EAAE,CAAC;AAC7J,UAAI,CAAC,IAAI,UAAW,QAAO;AAC3B,kBAAY,IAAI;AAAA,IAClB;AAAA,EACF;AAAA,EAEA,MAAM,cAAc,QAAyC;AAC3D,QAAI,QAAQ,OAAO,MAAM,GAAG,EAAE,OAAO,OAAO;AAC5C,QAAI,QAAQ,MAAM,IAAI,KAAK;AAC3B,QAAI,WAAW,MAAM,KAAK,GAAG,KAAK,KAAK;AACvC,QAAI,CAAC,SAAU,OAAM,IAAI,MAAM,kEAAkE;AACjG,QAAI,mBAAmB,MAAM,KAAK,oBAAoB,QAAQ;AAC9D,QAAI,MAAM,MAAM,KAAK,OAAO,KAAK,IAAI,wBAAwB,EAAC,aAAa,KAAK,SAAS,cAAc,kBAAkB,WAAW,MAAK,CAAC,CAAC;AAC3I,YAAQ,IAAI,eAAe,WAAW,CAAC,GAAG,IAAI,UAAQ,EAAC,MAAM,OAAO,IAAI,IAAI,EAAE,YAAY,GAAG,UAAU,OAAO,IAAI,IAAI,EAAC,EAAE;AAAA,EAC3H;AAAA,EAEA,QAAuB;AACrB,SAAK,OAAO,QAAQ;AACpB,WAAO,QAAQ,QAAQ;AAAA,EACzB;AAAA,EAEA,MAAc,WAAW,KAAa,QAAsB;AAG1D,QAAI,UAAU,CAAC,MAAM,QAAQ,MAAM,EAAG,OAAM,IAAI,MAAM,kDAAkD;AACxG,QAAI,MAAM,MAAM,KAAK,OAAO;AAAA,MAC1B,IAAI,2BAA2B;AAAA,QAC7B,aAAa;AAAA,QACb,uBAAuB,EAAC,SAAS,KAAK,SAAS,UAAU,KAAK,SAAQ;AAAA,QACtE,WAAW,KAAK;AAAA,QAChB,qBAAqB,KAAK,iBAAiB,EAAC,gBAAgB,KAAK,eAAc,IAAI;AAAA,QACnF,qBAAqB,QAAQ,IAAI,qBAAqB;AAAA,MACxD,CAAC;AAAA,IACH;AACA,QAAI,CAAC,IAAI,iBAAkB,OAAM,IAAI,MAAM,4CAA4C;AACvF,WAAO,IAAI;AAAA,EACb;AAAA,EAEA,MAAc,aAAa,kBAA0B;AACnD,QAAI,UAAU,KAAK,IAAI;AACvB,QAAI,YAAY,IAAI,KAAK;AACzB,WAAO,KAAK,IAAI,IAAI,UAAU,WAAW;AACvC,UAAI,MAAM,MAAM,KAAK,OAAO,KAAK,IAAI,yBAAyB,EAAC,kBAAkB,iBAAgB,CAAC,CAAC;AACnG,UAAI,SAAS,IAAI,gBAAgB;AACjC,UAAI,QAAQ,SAAS,YAAa;AAClC,UAAI,QAAQ,SAAS,YAAY,QAAQ,SAAS,YAAa,OAAM,IAAI,MAAM,gBAAgB,OAAO,MAAM,YAAY,CAAC,KAAK,OAAO,qBAAqB,gBAAgB,EAAE;AAC5K,YAAM,IAAI,QAAQ,aAAW,WAAW,SAAS,GAAI,CAAC;AAAA,IACxD;AACA,UAAM,IAAI,MAAM,gCAAgC,KAAK,MAAM,YAAY,GAAI,CAAC,GAAG;AAAA,EACjF;AAAA,EAEA,MAAc,aAAa,kBAAgD;AACzE,QAAI,OAAuC,CAAC;AAC5C,QAAI;AACJ,QAAI,YAAY;AAEhB,WAAO,MAAM;AACX,UAAI,MAAM,MAAM,KAAK,OAAO,KAAK,IAAI,uBAAuB,EAAC,kBAAkB,kBAAkB,WAAW,WAAW,YAAY,IAAI,CAAC,CAAC;AACzI,UAAI,UAAU,IAAI,WAAW,mBAAmB,cAAc,CAAC;AAC/D,UAAI,aAAa,aAAa,IAAI,WAAW,QAAQ,CAAC,GAAG,MAAM,CAAC,IAAI,IAAI,WAAW,QAAQ,CAAC;AAE5F,eAAS,OAAO,YAAY;AAC1B,YAAI,MAA+B,CAAC;AACpC,gBAAQ,QAAQ,CAAC,QAAQ,QAAQ;AAC/B,cAAI,QAAQ,IAAI,OAAO,GAAG,GAAG;AAC7B,cAAI,OAAO,OAAO,IAAI,EAAE,YAAY,CAAC,IAAI,iBAAiB,OAAO,OAAO,OAAO,QAAQ,EAAE,CAAC;AAAA,QAC5F,CAAC;AACD,aAAK,KAAK,GAAG;AAAA,MACf;AAEA,kBAAY;AACZ,UAAI,CAAC,IAAI,aAAa,KAAK,UAAU,IAAO,QAAO,EAAC,MAAM,WAAW,KAAK,OAAM;AAChF,kBAAY,IAAI;AAAA,IAClB;AAAA,EACF;AAAA,EAEA,MAAc,oBAAoB,MAA+B;AAC/D,QAAI,YAAY,MAAM,KAAK,aAAa;AACxC,WAAO,UAAU,KAAK,QAAM,GAAG,YAAY,KAAK,KAAK,YAAY,CAAC,KAAK;AAAA,EACzE;AACF;AAEA,SAAS,sBAAsB,OAAgB;AAC7C,MAAI,UAAU,QAAQ,UAAU,OAAW,QAAO;AAClD,MAAI,OAAO,SAAS,YAAY,OAAO,SAAS,SAAU,QAAO,OAAO,KAAK;AAC7E,MAAI,OAAO,SAAS,UAAW,QAAO,QAAQ,SAAS;AACvD,SAAO,IAAI,OAAO,KAAK,EAAE,QAAQ,MAAM,IAAI,CAAC;AAC9C;AAEA,SAAS,iBAAiB,OAA2B,MAAc;AACjE,MAAI,UAAU,OAAW,QAAO;AAChC,MAAI,iBAAiB,KAAK,YAAY;AACtC,MAAI,kBAAkB,UAAW,QAAO,SAAS;AACjD,MAAI,+DAA+D,KAAK,cAAc,EAAG,QAAO,OAAO,KAAK;AAC5G,SAAO;AACT;",
+  "names": []
+}

package/dist/cli/{bigQuery-YIWXZPY6.js → bigQuery-3A7HPXZS.js}

@@ -1,6 +1,6 @@
 import {
   config
-} from "./chunk-SQVXTHE5.js";
+} from "./chunk-KW66YQ62.js";
 
 // connections/bigQuery.ts
 import { BigQuery, BigQueryDate, BigQueryTimestamp } from "@google-cloud/bigquery";
@@ -72,4 +72,4 @@ var BigQueryConnection = class {
 export {
   BigQueryConnection
 };
-//# sourceMappingURL=bigQuery-YIWXZPY6.js.map
+//# sourceMappingURL=bigQuery-3A7HPXZS.js.map

package/dist/cli/{chunk-SQVXTHE5.js → chunk-KW66YQ62.js}

@@ -14,6 +14,8 @@ function normalizeConfig(input, defaultRoot = process.cwd()) {
   if (cfg.bigquery) dialect = "bigquery";
   else if (cfg.snowflake) dialect = "snowflake";
   else if (cfg.clickhouse) dialect = "clickhouse";
+  else if (cfg.postgres) dialect = "postgres";
+  else if (cfg.athena) dialect = "athena";
   else if (cfg.duckdb) dialect = "duckdb";
   let envFile = [".env"];
   if (Array.isArray(cfg.envFile)) envFile = cfg.envFile;
@@ -50,4 +52,4 @@ export {
   setGlobalConfig,
   loadConfig
 };
-//# sourceMappingURL=chunk-SQVXTHE5.js.map
+//# sourceMappingURL=chunk-KW66YQ62.js.map

package/dist/cli/chunk-KW66YQ62.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../lang/config.ts"],
+  "sourcesContent": ["import {existsSync} from 'node:fs'\nimport {readFile} from 'node:fs/promises'\nimport path from 'path'\n\nexport interface Config {\n  root: string\n  dialect: string\n  defaultNamespace?: string\n  ignoredFiles: string[]\n  telemetry?: boolean\n  updateNotifier?: boolean\n  port?: number\n  host?: string\n  envFile: string[] // array of paths where we can look for the env file\n\n  bigquery?: {\n    projectId?: string\n    keyPath?: string\n  }\n\n  snowflake?: {\n    account: string\n    username: string\n    privateKeyPath: string\n    schema?: string\n    database?: string\n  }\n\n  clickhouse?: {\n    url?: string\n    username?: string\n    database?: string\n    requestTimeout?: number\n  }\n\n  postgres?: {\n    connectionString?: string\n    host?: string\n    port?: number\n    database?: string\n    user?: string\n    username?: string\n    schema?: string\n    inMemory?: boolean\n    seedSql?: string\n    ssl?: boolean | {rejectUnauthorized?: boolean}\n    max?: number\n    idleTimeoutMillis?: number\n    connectionTimeoutMillis?: number\n    queryTimeout?: number\n    statementTimeout?: number\n  }\n\n  athena?: {\n    region?: string\n    catalog?: string\n    database?: string\n    workGroup?: string\n    outputLocation?: string\n  }\n\n  duckdb?: {\n    path?: string\n  }\n}\n\nexport type ConfigInput = Omit<Config, 'root' | 'dialect' | 'ignoredFiles' | 'envFile'> & {\n  root?: string\n  dialect?: Config['dialect']\n  ignoredFiles?: Config['ignoredFiles']\n  envFile?: string | string[]\n  namespace?: string\n}\n\nexport let config: Config = {dialect: 'duckdb', root: ''} as Config\n\nexport function setGlobalConfig(cfg: ConfigInput) {\n  Object.keys(config).forEach(key => delete config[key])\n  Object.assign(config, normalizeConfig(cfg))\n}\n\nexport function normalizeConfig(input: ConfigInput, defaultRoot = process.cwd()): Config {\n  let cfg = {...input}\n  if (cfg.namespace && !cfg.defaultNamespace) cfg.defaultNamespace = cfg.namespace\n\n  let dialect = cfg.dialect || 'duckdb'\n  if (cfg.bigquery) dialect = 'bigquery'\n  else if (cfg.snowflake) dialect = 'snowflake'\n  else if (cfg.clickhouse) dialect = 'clickhouse'\n  else if (cfg.postgres) dialect = 'postgres'\n  else if (cfg.athena) dialect = 'athena'\n  else if (cfg.duckdb) dialect = 'duckdb'\n  let envFile = ['.env']\n  if (Array.isArray(cfg.envFile)) envFile = cfg.envFile\n  else if (cfg.envFile) envFile = [cfg.envFile]\n\n  return {\n    ...cfg,\n    dialect,\n    root: path.resolve(cfg.root || defaultRoot),\n    port: cfg.port || Number(process.env.GRAPHENE_PORT) || 4000,\n    ignoredFiles: cfg.ignoredFiles || [],\n    envFile,\n  } as Config\n}\n\n// Read graphene config from the nearest parent package.json.\nexport async function loadConfig(dir: string, envLoader: (envFiles: string[]) => void): Promise<Config> {\n  // seek upwards from dir looking for package.json\n  let configDir = path.resolve(dir)\n  while (!existsSync(path.join(configDir, 'package.json'))) {\n    let parent = path.dirname(configDir)\n    if (parent == configDir) throw new Error(`No package.json found in ${path.resolve(dir)} or its parents`)\n    configDir = parent\n  }\n\n  let txt = await readFile(path.join(configDir, 'package.json'), 'utf8')\n  let graphene = JSON.parse(txt).graphene\n  if (!graphene || typeof graphene != 'object' || Array.isArray(graphene)) {\n    throw new Error(`No graphene config found in ${path.join(configDir, 'package.json')}`)\n  }\n\n  // config can provide 1 or more env files that Graphene should load. Default to just `.env`\n  let envFiles = Array.isArray(graphene.envFile) ? graphene.envFile : [graphene.envFile || '.env']\n  envLoader(envFiles.map(file => path.resolve(configDir, file)))\n\n  let cfg = normalizeConfig({...graphene, root: configDir}, configDir)\n  return cfg\n}\n"],
+  "mappings": ";AAAA,SAAQ,kBAAiB;AACzB,SAAQ,gBAAe;AACvB,OAAO,UAAU;AAwEV,IAAI,SAAiB,EAAC,SAAS,UAAU,MAAM,GAAE;AAEjD,SAAS,gBAAgB,KAAkB;AAChD,SAAO,KAAK,MAAM,EAAE,QAAQ,SAAO,OAAO,OAAO,GAAG,CAAC;AACrD,SAAO,OAAO,QAAQ,gBAAgB,GAAG,CAAC;AAC5C;AAEO,SAAS,gBAAgB,OAAoB,cAAc,QAAQ,IAAI,GAAW;AACvF,MAAI,MAAM,EAAC,GAAG,MAAK;AACnB,MAAI,IAAI,aAAa,CAAC,IAAI,iBAAkB,KAAI,mBAAmB,IAAI;AAEvE,MAAI,UAAU,IAAI,WAAW;AAC7B,MAAI,IAAI,SAAU,WAAU;AAAA,WACnB,IAAI,UAAW,WAAU;AAAA,WACzB,IAAI,WAAY,WAAU;AAAA,WAC1B,IAAI,SAAU,WAAU;AAAA,WACxB,IAAI,OAAQ,WAAU;AAAA,WACtB,IAAI,OAAQ,WAAU;AAC/B,MAAI,UAAU,CAAC,MAAM;AACrB,MAAI,MAAM,QAAQ,IAAI,OAAO,EAAG,WAAU,IAAI;AAAA,WACrC,IAAI,QAAS,WAAU,CAAC,IAAI,OAAO;AAE5C,SAAO;AAAA,IACL,GAAG;AAAA,IACH;AAAA,IACA,MAAM,KAAK,QAAQ,IAAI,QAAQ,WAAW;AAAA,IAC1C,MAAM,IAAI,QAAQ,OAAO,QAAQ,IAAI,aAAa,KAAK;AAAA,IACvD,cAAc,IAAI,gBAAgB,CAAC;AAAA,IACnC;AAAA,EACF;AACF;AAGA,eAAsB,WAAW,KAAa,WAA0D;AAEtG,MAAI,YAAY,KAAK,QAAQ,GAAG;AAChC,SAAO,CAAC,WAAW,KAAK,KAAK,WAAW,cAAc,CAAC,GAAG;AACxD,QAAI,SAAS,KAAK,QAAQ,SAAS;AACnC,QAAI,UAAU,UAAW,OAAM,IAAI,MAAM,4BAA4B,KAAK,QAAQ,GAAG,CAAC,iBAAiB;AACvG,gBAAY;AAAA,EACd;AAEA,MAAI,MAAM,MAAM,SAAS,KAAK,KAAK,WAAW,cAAc,GAAG,MAAM;AACrE,MAAI,WAAW,KAAK,MAAM,GAAG,EAAE;AAC/B,MAAI,CAAC,YAAY,OAAO,YAAY,YAAY,MAAM,QAAQ,QAAQ,GAAG;AACvE,UAAM,IAAI,MAAM,+BAA+B,KAAK,KAAK,WAAW,cAAc,CAAC,EAAE;AAAA,EACvF;AAGA,MAAI,WAAW,MAAM,QAAQ,SAAS,OAAO,IAAI,SAAS,UAAU,CAAC,SAAS,WAAW,MAAM;AAC/F,YAAU,SAAS,IAAI,UAAQ,KAAK,QAAQ,WAAW,IAAI,CAAC,CAAC;AAE7D,MAAI,MAAM,gBAAgB,EAAC,GAAG,UAAU,MAAM,UAAS,GAAG,SAAS;AACnE,SAAO;AACT;",
+  "names": []
+}