@malloydata/malloy 0.0.433 → 0.0.435

package/dist/dialect/standardsql/standardsql.js

@@ -150,7 +150,7 @@ class StandardSQLDialect extends dialect_1.Dialect {
     // can array agg or any_value a struct...
     sqlAggregateTurtle(groupSet, fieldList, orderBy, limit, filterSQL) {
         const fields = fieldList
-            .map(f => `\n  ${f.sqlExpression} as ${f.sqlOutputName}`)
+            .map(f => `\n  ${f.sqlExpression} as ${this.sqlQuoteIdentifier(f.rawName)}`)
             .join(', ');
         const orderByClause = orderBy ? this.sqlTurtleOrderByClause(orderBy) : '';
         // BigQuery ARRAY_AGG takes LIMIT as its final sub-clause, after ORDER BY.
@@ -162,7 +162,7 @@ class StandardSQLDialect extends dialect_1.Dialect {
     }
     sqlAnyValueTurtle(groupSet, fieldList) {
         const fields = fieldList
-            .map(f => `${f.sqlExpression} as ${f.sqlOutputName}`)
+            .map(f => `${f.sqlExpression} as ${this.sqlQuoteIdentifier(f.rawName)}`)
             .join(', ');
         return `ANY_VALUE(CASE WHEN group_set=${groupSet} THEN STRUCT(${fields}) END)`;
     }
@@ -171,10 +171,10 @@ class StandardSQLDialect extends dialect_1.Dialect {
     }
     sqlCoaleseMeasuresInline(groupSet, fieldList) {
         const fields = fieldList
-            .map(f => `${f.sqlExpression} as ${f.sqlOutputName}`)
+            .map(f => `${f.sqlExpression} as ${this.sqlQuoteIdentifier(f.rawName)}`)
             .join(', ');
         const nullValues = fieldList
-            .map(f => `NULL as ${f.sqlOutputName}`)
+            .map(f => `NULL as ${this.sqlQuoteIdentifier(f.rawName)}`)
             .join(', ');
         return `COALESCE(ANY_VALUE(CASE WHEN group_set=${groupSet} THEN STRUCT(${fields}) END), STRUCT(${nullValues}))`;
     }
@@ -240,8 +240,11 @@ ${(0, utils_1.indent)(sql)}
     sqlCreateFunctionCombineLastStage(lastStageName) {
         return `SELECT ARRAY((SELECT AS STRUCT * FROM ${lastStageName}))\n`;
     }
-    sqlSelectAliasAsStruct(alias) {
-        return `(SELECT AS STRUCT ${alias}.*)`;
+    sqlSelectAliasAsStruct(alias, dialectFieldList) {
+        const fields = dialectFieldList
+            .map(d => `${alias}.${this.sqlQuoteIdentifier(d.rawName)}`)
+            .join(', ');
+        return `(SELECT AS STRUCT ${fields})`;
     }
     sqlNowExpr() {
         return 'CURRENT_TIMESTAMP()';
@@ -249,18 +252,18 @@ ${(0, utils_1.indent)(sql)}
     sqlTimeExtractExpr(qi, te) {
         const extractTo = extractMap[te.units] || te.units;
         const tz = malloy_types_1.TD.isAnyTimestamp(te.e.typeDef) && qtz(qi);
-        const tzAdd = tz ? ` AT TIME ZONE '${tz}'` : '';
+        const tzAdd = tz ? ` AT TIME ZONE ${this.sqlTimezoneLiteral(tz)}` : '';
         return `EXTRACT(${extractTo} FROM ${te.e.sql}${tzAdd})`;
     }
     sqlConvertToCivilTime(expr, timezone, _typeDef) {
         // BigQuery has no timestamptz type, so typeDef.timestamptz will never be true
         return {
-            sql: `DATETIME(${expr}, '${timezone}')`,
+            sql: `DATETIME(${expr}, ${this.sqlTimezoneLiteral(timezone)})`,
             typeDef: { type: 'timestamp' },
         };
     }
     sqlConvertFromCivilTime(expr, timezone, _destTypeDef) {
-        return `TIMESTAMP(${expr}, '${timezone}')`;
+        return `TIMESTAMP(${expr}, ${this.sqlTimezoneLiteral(timezone)})`;
     }
     sqlTruncate(expr, unit, typeDef, inCivilTime, timezone) {
         if (inCivilTime) {
@@ -272,7 +275,7 @@ ${(0, utils_1.indent)(sql)}
             return `DATE_TRUNC(${expr}, ${unit})`;
         }
         // TIMESTAMP truncation with optional timezone
-        const tzParam = timezone ? `, '${timezone}'` : '';
+        const tzParam = timezone ? `, ${this.sqlTimezoneLiteral(timezone)}` : '';
         return `TIMESTAMP_TRUNC(${expr}, ${unit}${tzParam})`;
     }
     sqlOffsetTime(expr, op, magnitude, unit, typeDef, inCivilTime, _timezone) {
@@ -299,10 +302,10 @@ ${(0, utils_1.indent)(sql)}
         const tz = qtz(qi);
         const src = cast.e.sql || '';
         if (op === 'timestamp::date' && tz) {
-            return `DATE(${src},'${tz}')`;
+            return `DATE(${src},${this.sqlTimezoneLiteral(tz)})`;
         }
         if (op === 'date::timestamp' && tz) {
-            return `TIMESTAMP(${src}, '${tz}')`;
+            return `TIMESTAMP(${src}, ${this.sqlTimezoneLiteral(tz)})`;
         }
         if (!malloy_types_1.TD.eq(srcTypeDef, dstTypeDef)) {
             const castFunc = cast.safe ? 'SAFE_CAST' : 'CAST';
@@ -320,7 +323,7 @@ ${(0, utils_1.indent)(sql)}
         let timestampArgs = `'${literal}'`;
         const tz = timezone || qtz(qi);
         if (tz && tz !== 'UTC') {
-            timestampArgs += `,'${tz}'`;
+            timestampArgs += `,${this.sqlTimezoneLiteral(tz)}`;
         }
         return `TIMESTAMP(${timestampArgs})`;
     }

package/dist/dialect/trino/trino.d.ts

@@ -1,4 +1,4 @@
-import type { Expr, Sampling, AtomicTypeDef, ATimestampTypeDef, TypecastExpr, RegexMatchExpr, MeasureTimeExpr, TimeExtractExpr, BasicAtomicTypeDef, RecordLiteralNode } from '../../model/malloy_types';
+import type { Expr, Sampling, AtomicTypeDef, ATimestampTypeDef, TypecastExpr, RegexMatchExpr, MeasureTimeExpr, TimeExtractExpr, BasicAtomicTypeDef, RecordLiteralNode, OrderBy } from '../../model/malloy_types';
 import type { DialectFunctionOverloadDef } from '../functions';
 import type { CompiledOrderBy, DialectFieldList, OrderByClauseType, QueryInfo } from '../dialect';
 import { PostgresBase } from '../pg_impl';
@@ -49,8 +49,8 @@ export declare class TrinoDialect extends PostgresBase {
     sqlUnnestPipelineHead(isSingleton: boolean, sourceSQLExpression: string): string;
     sqlCreateFunction(id: string, funcText: string): string;
     sqlCreateTableAsSelect(tableName: string, sql: string): string;
-    sqlCreateFunctionCombineLastStage(lastStageName: string, fieldList: DialectFieldList): string;
-    sqlSelectAliasAsStruct(alias: string, fieldList: any): string;
+    sqlCreateFunctionCombineLastStage(lastStageName: string, fieldList: DialectFieldList, orderBy: OrderBy[] | undefined): string;
+    sqlSelectAliasAsStruct(alias: string, fieldList: DialectFieldList): string;
     keywords: string[];
     sqlConvertToCivilTime(expr: string, timezone: string, typeDef: AtomicTypeDef): {
         sql: string;

package/dist/dialect/trino/trino.js

@@ -82,7 +82,7 @@ class TrinoDialect extends pg_impl_1.PostgresBase {
         this.orderByClause = 'output_name';
         this.nullMatchesFunctionSignature = false;
         this.supportsSelectReplace = false;
-        this.supportsComplexFilteredSources = false;
+        this.supportsComplexFilteredSources = true;
         this.supportsTempTables = false;
         this.supportsCountApprox = true;
         this.supportsHyperLogLog = true;
@@ -223,7 +223,8 @@ class TrinoDialect extends pg_impl_1.PostgresBase {
     }
     buildTypeExpression(fieldList) {
         return fieldList
-            .map(dlf => `${dlf.sqlOutputName} ${this.malloyTypeToSQLType(dlf.typeDef)}`)
+            .map(dlf => `${this.sqlQuoteIdentifier(dlf.rawName)} ` +
+            this.malloyTypeToSQLType(dlf.typeDef))
             .join(', \n');
     }
     // can array agg or any_value a struct...
@@ -325,15 +326,18 @@ ${(0, utils_1.indent)(sql)}
 );
 `;
     }
-    sqlCreateFunctionCombineLastStage(lastStageName, fieldList) {
+    sqlCreateFunctionCombineLastStage(lastStageName, fieldList, orderBy) {
         const fields = fieldList.map(f => f.sqlExpression).join(', ');
         const definitions = this.buildTypeExpression(fieldList);
-        return `SELECT ARRAY_AGG(CAST(ROW(${fields}) as ROW(${definitions}))) FROM ${lastStageName}\n`;
+        const o = this.sqlCombineLastStageOrderBy(orderBy, fieldList);
+        return `SELECT ARRAY_AGG(CAST(ROW(${fields}) as ROW(${definitions}))${o}) FROM ${lastStageName}\n`;
     }
     sqlSelectAliasAsStruct(alias, fieldList) {
-        const fields = fieldList.map(f => f.sqlExpression).join(', ');
+        const fields = fieldList
+            .map(f => `${alias}.${this.sqlQuoteIdentifier(f.rawName)}`)
+            .join(', ');
         const definitions = this.buildTypeExpression(fieldList);
-        return `CAST(ROW(${fields}) as ROW(${definitions})`;
+        return `CAST(ROW(${fields}) as ROW(${definitions}))`;
     }
     sqlConvertToCivilTime(expr, timezone, typeDef) {
         // Trino civil time = TIMESTAMPTZ with query timezone as stored timezone
@@ -341,13 +345,13 @@ ${(0, utils_1.indent)(sql)}
         if (typeDef.type === 'timestamp') {
             // TIMESTAMP (UTC wall clock) → TIMESTAMPTZ in query timezone
             return {
-                sql: `at_timezone(with_timezone(${expr}, 'UTC'), '${timezone}')`,
+                sql: `at_timezone(with_timezone(${expr}, 'UTC'), ${this.sqlTimezoneLiteral(timezone)})`,
                 typeDef: { type: 'timestamptz' },
             };
         }
         // TIMESTAMPTZ → TIMESTAMPTZ in query timezone (same instant, different stored tz)
         return {
-            sql: `at_timezone(${expr}, '${timezone}')`,
+            sql: `at_timezone(${expr}, ${this.sqlTimezoneLiteral(timezone)})`,
             typeDef: { type: 'timestamptz' },
         };
     }
@@ -392,27 +396,27 @@ ${(0, utils_1.indent)(sql)}
         if (tz && srcTypeDef && dstTypeDef) {
             // TIMESTAMP → DATE: interpret as UTC, convert to query timezone
             if (malloy_types_1.TD.isTimestamp(srcTypeDef) && malloy_types_1.TD.isDate(dstTypeDef)) {
-                return `CAST(at_timezone(with_timezone(${expr}, 'UTC'), '${tz}') AS DATE)`;
+                return `CAST(at_timezone(with_timezone(${expr}, 'UTC'), ${this.sqlTimezoneLiteral(tz)}) AS DATE)`;
             }
             // TIMESTAMPTZ → DATE: convert to query timezone
             if (malloy_types_1.TD.isTimestamptz(srcTypeDef) && malloy_types_1.TD.isDate(dstTypeDef)) {
-                return `CAST(at_timezone(${expr}, '${tz}') AS DATE)`;
+                return `CAST(at_timezone(${expr}, ${this.sqlTimezoneLiteral(tz)}) AS DATE)`;
             }
             // DATE → TIMESTAMP: interpret date in query timezone, return UTC wall clock
             if (malloy_types_1.TD.isDate(srcTypeDef) && malloy_types_1.TD.isTimestamp(dstTypeDef)) {
-                return `CAST(at_timezone(with_timezone(CAST(${expr} AS TIMESTAMP), '${tz}'), 'UTC') AS TIMESTAMP)`;
+                return `CAST(at_timezone(with_timezone(CAST(${expr} AS TIMESTAMP), ${this.sqlTimezoneLiteral(tz)}), 'UTC') AS TIMESTAMP)`;
             }
             // DATE → TIMESTAMPTZ: interpret date in query timezone
             if (malloy_types_1.TD.isDate(srcTypeDef) && malloy_types_1.TD.isTimestamptz(dstTypeDef)) {
-                return `with_timezone(CAST(${expr} AS TIMESTAMP), '${tz}')`;
+                return `with_timezone(CAST(${expr} AS TIMESTAMP), ${this.sqlTimezoneLiteral(tz)})`;
             }
             // TIMESTAMPTZ → TIMESTAMP: convert to query timezone wall clock
             if (malloy_types_1.TD.isTimestamptz(srcTypeDef) && malloy_types_1.TD.isTimestamp(dstTypeDef)) {
-                return `CAST(at_timezone(${expr}, '${tz}') AS TIMESTAMP)`;
+                return `CAST(at_timezone(${expr}, ${this.sqlTimezoneLiteral(tz)}) AS TIMESTAMP)`;
             }
             // TIMESTAMP → TIMESTAMPTZ: interpret TIMESTAMP as being in query timezone
             if (malloy_types_1.TD.isTimestamp(srcTypeDef) && malloy_types_1.TD.isTimestamptz(dstTypeDef)) {
-                return `with_timezone(${expr}, '${tz}')`;
+                return `with_timezone(${expr}, ${this.sqlTimezoneLiteral(tz)})`;
             }
         }
         // No special handling needed, or no query timezone
@@ -567,13 +571,13 @@ ${(0, utils_1.indent)(sql)}
         const tz = timezone || qtz(qi);
         if (tz) {
             // Interpret wall clock time in timezone, convert to UTC wall clock, cast to TIMESTAMP
-            return `CAST(at_timezone(with_timezone(TIMESTAMP '${literal}', '${tz}'), 'UTC') AS TIMESTAMP)`;
+            return `CAST(at_timezone(with_timezone(TIMESTAMP '${literal}', ${this.sqlTimezoneLiteral(tz)}), 'UTC') AS TIMESTAMP)`;
         }
         return `TIMESTAMP '${literal}'`;
     }
     sqlTimestamptzLiteral(_qi, literal, timezone) {
         // Use with_timezone to create a TIMESTAMP WITH TIME ZONE
-        return `with_timezone(TIMESTAMP '${literal}', '${timezone}')`;
+        return `with_timezone(TIMESTAMP '${literal}', ${this.sqlTimezoneLiteral(timezone)})`;
     }
     sqlTimeExtractExpr(qi, from) {
         const pgUnits = pg_impl_1.timeExtractMap[from.units] || from.units;
@@ -584,11 +588,11 @@ ${(0, utils_1.indent)(sql)}
                 // Convert both TIMESTAMP and TIMESTAMPTZ to query timezone for extraction
                 if (from.e.typeDef.type === 'timestamp') {
                     // TIMESTAMP: interpret as UTC, convert to query timezone
-                    extractFrom = `at_timezone(with_timezone(${extractFrom}, 'UTC'), '${tz}')`;
+                    extractFrom = `at_timezone(with_timezone(${extractFrom}, 'UTC'), ${this.sqlTimezoneLiteral(tz)})`;
                 }
                 else {
                     // TIMESTAMPTZ: convert to query timezone
-                    extractFrom = `at_timezone(${extractFrom}, '${tz}')`;
+                    extractFrom = `at_timezone(${extractFrom}, ${this.sqlTimezoneLiteral(tz)})`;
                 }
             }
         }
@@ -628,18 +632,18 @@ class PrestoDialect extends TrinoDialect {
     sqlTimestampLiteral(qi, literal, timezone) {
         const tz = timezone || qtz(qi);
         if (tz) {
-            return `CAST(TIMESTAMP '${literal} ${tz}' AT TIME ZONE 'UTC' AS TIMESTAMP)`;
+            return `CAST(TIMESTAMP '${literal} ${this.checkedTimezoneName(tz)}' AT TIME ZONE 'UTC' AS TIMESTAMP)`;
         }
         return `TIMESTAMP '${literal}'`;
     }
     sqlTimestamptzLiteral(_qi, literal, timezone) {
-        return `TIMESTAMP '${literal} ${timezone}'`;
+        return `TIMESTAMP '${literal} ${this.checkedTimezoneName(timezone)}'`;
     }
     sqlConvertToCivilTime(expr, timezone, _typeDef) {
         // Presto's AT TIME ZONE operator (not function) produces TIMESTAMPTZ
         // Reinterprets the instant in the target timezone
         return {
-            sql: `${expr} AT TIME ZONE '${timezone}'`,
+            sql: `${expr} AT TIME ZONE ${this.sqlTimezoneLiteral(timezone)}`,
             typeDef: { type: 'timestamptz' },
         };
     }
@@ -658,26 +662,26 @@ class PrestoDialect extends TrinoDialect {
         if (tz && srcTypeDef && dstTypeDef) {
             // TIMESTAMP → DATE: interpret as UTC, convert to query timezone
             if (malloy_types_1.TD.isTimestamp(srcTypeDef) && malloy_types_1.TD.isDate(dstTypeDef)) {
-                return `CAST((${expr} AT TIME ZONE 'UTC') AT TIME ZONE '${tz}' AS DATE)`;
+                return `CAST((${expr} AT TIME ZONE 'UTC') AT TIME ZONE ${this.sqlTimezoneLiteral(tz)} AS DATE)`;
             }
             // TIMESTAMPTZ → DATE: convert to query timezone
             if (malloy_types_1.TD.isTimestamptz(srcTypeDef) && malloy_types_1.TD.isDate(dstTypeDef)) {
-                return `CAST(${expr} AT TIME ZONE '${tz}' AS DATE)`;
+                return `CAST(${expr} AT TIME ZONE ${this.sqlTimezoneLiteral(tz)} AS DATE)`;
             }
             // DATE → TIMESTAMP: interpret date in query timezone, return UTC wall clock
             // Presto doesn't have a way to interpret TIMESTAMP in a non-UTC timezone,
             // so we build a TIMESTAMPTZ literal string and cast it
             if (malloy_types_1.TD.isDate(srcTypeDef) && malloy_types_1.TD.isTimestamp(dstTypeDef)) {
-                const tstzLiteral = `CAST(CAST(${expr} AS VARCHAR) || ' 00:00:00 ${tz}' AS TIMESTAMP WITH TIME ZONE)`;
+                const tstzLiteral = `CAST(CAST(${expr} AS VARCHAR) || ' 00:00:00 ${this.checkedTimezoneName(tz)}' AS TIMESTAMP WITH TIME ZONE)`;
                 return `CAST(${tstzLiteral} AS TIMESTAMP)`;
             }
             // DATE → TIMESTAMPTZ: interpret date in query timezone
             if (malloy_types_1.TD.isDate(srcTypeDef) && malloy_types_1.TD.isTimestamptz(dstTypeDef)) {
-                return `CAST(CAST(${expr} AS VARCHAR) || ' 00:00:00 ${tz}' AS TIMESTAMP WITH TIME ZONE)`;
+                return `CAST(CAST(${expr} AS VARCHAR) || ' 00:00:00 ${this.checkedTimezoneName(tz)}' AS TIMESTAMP WITH TIME ZONE)`;
             }
             // TIMESTAMPTZ → TIMESTAMP: convert to query timezone wall clock
             if (malloy_types_1.TD.isTimestamptz(srcTypeDef) && malloy_types_1.TD.isTimestamp(dstTypeDef)) {
-                return `CAST(${expr} AT TIME ZONE '${tz}' AS TIMESTAMP)`;
+                return `CAST(${expr} AT TIME ZONE ${this.sqlTimezoneLiteral(tz)} AS TIMESTAMP)`;
             }
             // TIMESTAMP → TIMESTAMPTZ: interpret TIMESTAMP as UTC
             if (malloy_types_1.TD.isTimestamp(srcTypeDef) && malloy_types_1.TD.isTimestamptz(dstTypeDef)) {
@@ -701,11 +705,11 @@ class PrestoDialect extends TrinoDialect {
                 if (from.e.typeDef.type === 'timestamp') {
                     // TIMESTAMP: interpret as UTC, convert to query timezone
                     // Presto uses AT TIME ZONE operator
-                    extractFrom = `(${extractFrom} AT TIME ZONE 'UTC') AT TIME ZONE '${tz}'`;
+                    extractFrom = `(${extractFrom} AT TIME ZONE 'UTC') AT TIME ZONE ${this.sqlTimezoneLiteral(tz)}`;
                 }
                 else {
                     // TIMESTAMPTZ: convert to query timezone
-                    extractFrom = `${extractFrom} AT TIME ZONE '${tz}'`;
+                    extractFrom = `${extractFrom} AT TIME ZONE ${this.sqlTimezoneLiteral(tz)}`;
                 }
             }
         }

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-export { DuckDBDialect, StandardSQLDialect, TrinoDialect, PostgresDialect, SnowflakeDialect, MySQLDialect, DatabricksDialect, registerDialect, arg, qtz, overload, minScalar, anyExprType, minAggregate, maxScalar, sql, makeParam, param, variadicParam, literal, spread, Dialect, decodeDottedTablePath, validateDottedTablePath, } from './dialect';
+export { DuckDBDialect, StandardSQLDialect, TrinoDialect, PostgresDialect, SnowflakeDialect, MySQLDialect, DatabricksDialect, SQLServerDialect, registerDialect, getDialect, arg, qtz, overload, minScalar, anyExprType, minAggregate, maxScalar, sql, makeParam, param, variadicParam, literal, spread, Dialect, decodeDottedTablePath, validateDottedTablePath, } from './dialect';
 export type { DialectFieldList, DialectFunctionOverloadDef, QueryInfo, MalloyStandardFunctionImplementations, DefinitionBlueprint, DefinitionBlueprintMap, OverloadedDefinitionBlueprint, DecodeDottedTablePathResult, DottedTablePathOptions, TablePathEscapeStyle, TablePathSegment, ValidateTablePathResult, } from './dialect';
 export type { QueryRecord, StructDef, TableSourceDef, SQLSourceDef, SourceDef, JoinFieldDef, NamedSourceDefs, MalloyQueryData, DateUnit, ExtractUnit, TimestampUnit, TemporalFieldType, QueryData, QueryValue, Expr, FilterCondition, Argument, Parameter, FieldDef, PipeSegment, QueryFieldDef, IndexFieldDef, TurtleDef, SearchValueMapResult, SearchIndexResult, ModelDef, Query, QueryResult, QueryResultDef, QueryRunStats, QueryScalar, NamedQueryDef, NamedModelObject, ExpressionType, FunctionDef, FunctionOverloadDef, FunctionParameterDef, ExpressionValueType, TypeDesc, FunctionParamTypeDesc, DocumentLocation, DocumentRange, DocumentPosition, Sampling, BasicAtomicTypeDef, BasicAtomicDef, AtomicTypeDef, AtomicFieldDef, ArrayDef, ArrayTypeDef, RecordTypeDef, RepeatedRecordTypeDef, RecordDef, RepeatedRecordDef, RecordLiteralNode, StringLiteralNode, ArrayLiteralNode, SourceComponentInfo, DateLiteralNode, TimestampLiteralNode, TimestamptzLiteralNode, TimeLiteralExpr, TypecastExpr, BuildID, BuildManifest, BuildManifestEntry, GivenValue, VirtualMap, } from './model';
 export { isSourceDef, isAtomic, isBasicAtomic, isCompoundArrayData, isJoined, isJoinedSource, isSamplingEnable, isSamplingPercent, isSamplingRows, isRepeatedRecord, isBasicArray, mkArrayDef, mkFieldDef, expressionIsAggregate, expressionIsAnalytic, expressionIsCalculation, expressionIsScalar, expressionIsUngroupedAggregate, indent, composeSQLExpr, isTimestampUnit, isDateUnit, constantExprToSQL, } from './model';

package/dist/index.js

@@ -37,9 +37,9 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Model = exports.MalloyTranslator = exports.malloyToQuery = exports.constantExprToSQL = exports.isDateUnit = exports.isTimestampUnit = exports.composeSQLExpr = exports.indent = exports.expressionIsUngroupedAggregate = exports.expressionIsScalar = exports.expressionIsCalculation = exports.expressionIsAnalytic = exports.expressionIsAggregate = exports.mkFieldDef = exports.mkArrayDef = exports.isBasicArray = exports.isRepeatedRecord = exports.isSamplingRows = exports.isSamplingPercent = exports.isSamplingEnable = exports.isJoinedSource = exports.isJoined = exports.isCompoundArrayData = exports.isBasicAtomic = exports.isAtomic = exports.isSourceDef = exports.validateDottedTablePath = exports.decodeDottedTablePath = exports.Dialect = exports.spread = exports.literal = exports.variadicParam = exports.param = exports.makeParam = exports.sql = exports.maxScalar = exports.minAggregate = exports.anyExprType = exports.minScalar = exports.overload = exports.qtz = exports.arg = exports.registerDialect = exports.DatabricksDialect = exports.MySQLDialect = exports.SnowflakeDialect = exports.PostgresDialect = exports.TrinoDialect = exports.StandardSQLDialect = exports.DuckDBDialect = void 0;
-exports.Reference = exports.EMPTY_BUILD_MANIFEST = exports.PersistSource = exports.annotationToTaglines = exports.annotationToTag = exports.tagFromAnnotations = exports.annotationsForRoute = exports.payloadOf = exports.routeOf = exports.RoutedNote = exports.Annotations = exports.sqlKey = exports.API = exports.sourceDefToSourceInfo = exports.modelDefToModelInfo = exports.toAsyncGenerator = exports.createConnectionsFromConfig = exports.getRegisteredConnectionTypes = exports.getConnectionTypeDisplayName = exports.getConnectionProperties = exports.registerConnectionType = exports.discoverConfig = exports.defaultConfigOverlays = exports.contextOverlay = exports.envOverlay = exports.MalloyConfig = exports.Manifest = exports.CacheManager = exports.InMemoryModelCache = exports.Explore = exports.DataWriter = exports.Parse = exports.JSONWriter = exports.CSVWriter = exports.QueryMaterializer = exports.Result = exports.PreparedResult = exports.TimestampTimeframe = exports.DateTimeframe = exports.SourceRelationship = exports.JoinRelationship = exports.MalloyError = exports.FixedConnectionMap = exports.InMemoryURLReader = exports.EmptyURLReader = exports.SingleConnectionRuntime = exports.ConnectionRuntime = exports.AtomicFieldType = exports.Runtime = exports.Malloy = void 0;
-exports.makeDigest = void 0;
+exports.malloyToQuery = exports.constantExprToSQL = exports.isDateUnit = exports.isTimestampUnit = exports.composeSQLExpr = exports.indent = exports.expressionIsUngroupedAggregate = exports.expressionIsScalar = exports.expressionIsCalculation = exports.expressionIsAnalytic = exports.expressionIsAggregate = exports.mkFieldDef = exports.mkArrayDef = exports.isBasicArray = exports.isRepeatedRecord = exports.isSamplingRows = exports.isSamplingPercent = exports.isSamplingEnable = exports.isJoinedSource = exports.isJoined = exports.isCompoundArrayData = exports.isBasicAtomic = exports.isAtomic = exports.isSourceDef = exports.validateDottedTablePath = exports.decodeDottedTablePath = exports.Dialect = exports.spread = exports.literal = exports.variadicParam = exports.param = exports.makeParam = exports.sql = exports.maxScalar = exports.minAggregate = exports.anyExprType = exports.minScalar = exports.overload = exports.qtz = exports.arg = exports.getDialect = exports.registerDialect = exports.SQLServerDialect = exports.DatabricksDialect = exports.MySQLDialect = exports.SnowflakeDialect = exports.PostgresDialect = exports.TrinoDialect = exports.StandardSQLDialect = exports.DuckDBDialect = void 0;
+exports.PersistSource = exports.annotationToTaglines = exports.annotationToTag = exports.tagFromAnnotations = exports.annotationsForRoute = exports.payloadOf = exports.routeOf = exports.RoutedNote = exports.Annotations = exports.sqlKey = exports.API = exports.sourceDefToSourceInfo = exports.modelDefToModelInfo = exports.toAsyncGenerator = exports.createConnectionsFromConfig = exports.getRegisteredConnectionTypes = exports.getConnectionTypeDisplayName = exports.getConnectionProperties = exports.registerConnectionType = exports.discoverConfig = exports.defaultConfigOverlays = exports.contextOverlay = exports.envOverlay = exports.MalloyConfig = exports.Manifest = exports.CacheManager = exports.InMemoryModelCache = exports.Explore = exports.DataWriter = exports.Parse = exports.JSONWriter = exports.CSVWriter = exports.QueryMaterializer = exports.Result = exports.PreparedResult = exports.TimestampTimeframe = exports.DateTimeframe = exports.SourceRelationship = exports.JoinRelationship = exports.MalloyError = exports.FixedConnectionMap = exports.InMemoryURLReader = exports.EmptyURLReader = exports.SingleConnectionRuntime = exports.ConnectionRuntime = exports.AtomicFieldType = exports.Runtime = exports.Malloy = exports.Model = exports.MalloyTranslator = void 0;
+exports.makeDigest = exports.Reference = exports.EMPTY_BUILD_MANIFEST = void 0;
 var dialect_1 = require("./dialect");
 Object.defineProperty(exports, "DuckDBDialect", { enumerable: true, get: function () { return dialect_1.DuckDBDialect; } });
 Object.defineProperty(exports, "StandardSQLDialect", { enumerable: true, get: function () { return dialect_1.StandardSQLDialect; } });
@@ -48,7 +48,9 @@ Object.defineProperty(exports, "PostgresDialect", { enumerable: true, get: funct
 Object.defineProperty(exports, "SnowflakeDialect", { enumerable: true, get: function () { return dialect_1.SnowflakeDialect; } });
 Object.defineProperty(exports, "MySQLDialect", { enumerable: true, get: function () { return dialect_1.MySQLDialect; } });
 Object.defineProperty(exports, "DatabricksDialect", { enumerable: true, get: function () { return dialect_1.DatabricksDialect; } });
+Object.defineProperty(exports, "SQLServerDialect", { enumerable: true, get: function () { return dialect_1.SQLServerDialect; } });
 Object.defineProperty(exports, "registerDialect", { enumerable: true, get: function () { return dialect_1.registerDialect; } });
+Object.defineProperty(exports, "getDialect", { enumerable: true, get: function () { return dialect_1.getDialect; } });
 Object.defineProperty(exports, "arg", { enumerable: true, get: function () { return dialect_1.arg; } });
 Object.defineProperty(exports, "qtz", { enumerable: true, get: function () { return dialect_1.qtz; } });
 Object.defineProperty(exports, "overload", { enumerable: true, get: function () { return dialect_1.overload; } });

package/dist/lang/ast/source-elements/named-source.js

@@ -52,7 +52,6 @@ class NamedSource extends source_1.Source {
         }
     }
     modelStruct() {
-        var _a;
         const modelEnt = this.modelEntry(this.ref);
         const entry = modelEnt === null || modelEnt === void 0 ? void 0 : modelEnt.entry;
         if (!entry) {
@@ -89,7 +88,6 @@ class NamedSource extends source_1.Source {
             return;
         }
         else {
-            (_a = this.document()) === null || _a === void 0 ? void 0 : _a.checkExperimentalDialect(this, entry.dialect);
             if ((0, malloy_types_1.isSourceDef)(entry)) {
                 // This struct is created as an unmodified reference to `entry`. Mark it
                 // with entry's own identity; the modification path (DynamicSpace) clears

package/dist/lang/ast/source-elements/refined-source.js

@@ -81,7 +81,7 @@ class RefinedSource extends source_1.Source {
                 filters.push(el);
             }
             else if (el instanceof timezone_statement_1.TimezoneStatement) {
-                newTimezone = el.tz;
+                newTimezone = el.timezoneName();
             }
             else {
                 errTo.logError('unexpected-source-property', `Unexpected source property: '${errTo.elementType}'`);

package/dist/lang/ast/source-elements/sql-source.js

@@ -19,12 +19,13 @@ class SQLSource extends source_1.Source {
         this.connectionNameInvalid = false;
     }
     sqlSourceRequest(doc) {
+        var _a;
         const partialModel = this.select.containsQueries
             ? doc.modelDef()
             : undefined;
         const [valid, phrases] = this.select.sqlPhrases(this.connectionName.refString);
         if (valid) {
-            return (0, sql_block_1.getSourceRequest)(phrases, this.connectionName.refString, partialModel);
+            return (0, sql_block_1.getSourceRequest)(phrases, this.connectionName.refString, partialModel, (_a = this.translator()) === null || _a === void 0 ? void 0 : _a.root.virtualMap);
         }
         return undefined;
     }
@@ -77,9 +78,6 @@ class SQLSource extends source_1.Source {
                 compileSQL: sql,
             };
         }
-        else if (lookup.status === 'present') {
-            doc.checkExperimentalDialect(this, lookup.value.dialect);
-        }
     }
     getSourceDef() {
         var _a;

package/dist/lang/ast/source-elements/table-source.js

@@ -12,7 +12,7 @@ const error_factory_1 = require("../error-factory");
 const dialect_1 = require("../../../dialect");
 class TableSource extends source_1.Source {
     getSourceDef() {
-        var _a, _b, _c;
+        var _a, _b;
         const info = this.getTableInfo();
         if (info === undefined) {
             return error_factory_1.ErrorFactory.structDef;
@@ -36,7 +36,6 @@ class TableSource extends source_1.Source {
         let msg = `Schema read failure for table '${tablePath}' for connection '${connectionName}'`;
         if (tableDefEntry) {
             if (tableDefEntry.status === 'present') {
-                (_c = this.document()) === null || _c === void 0 ? void 0 : _c.checkExperimentalDialect(this, tableDefEntry.value.dialect);
                 tableDefEntry.value.location = this.location;
                 tableDefEntry.value.fields.forEach(field => {
                     field.location = this.location;

package/dist/lang/ast/source-properties/timezone-statement.d.ts

@@ -1,12 +1,13 @@
+import type { ExprString } from '../expressions/expr-string';
 import type { QueryBuilder } from '../types/query-builder';
 import { MalloyElement } from '../types/malloy-element';
 import type { QueryPropertyInterface } from '../types/query-property-interface';
 export declare class TimezoneStatement extends MalloyElement implements QueryPropertyInterface {
-    readonly tz: string;
+    private readonly timezone;
     elementType: string;
     forceQueryClass: undefined;
     queryRefinementStage: undefined;
-    constructor(tz: string);
-    get isValid(): boolean;
+    constructor(timezone: ExprString);
+    timezoneName(): string | undefined;
     queryExecute(executeFor: QueryBuilder): void;
 }

package/dist/lang/ast/source-properties/timezone-statement.js

@@ -5,27 +5,29 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TimezoneStatement = void 0;
+const dialect_1 = require("../../../dialect/dialect");
 const malloy_element_1 = require("../types/malloy-element");
-const luxon_1 = require("luxon");
 class TimezoneStatement extends malloy_element_1.MalloyElement {
-    constructor(tz) {
-        super();
-        this.tz = tz;
+    constructor(timezone) {
+        super({ timezone });
+        this.timezone = timezone;
         this.elementType = 'timezone';
         this.forceQueryClass = undefined;
         this.queryRefinementStage = undefined;
     }
-    get isValid() {
-        try {
-            luxon_1.DateTime.fromISO('2020-02-19T00:00:00', { zone: this.tz });
-            return true;
-        }
-        catch {
-            return false;
+    timezoneName() {
+        const name = this.timezone.value;
+        if (!dialect_1.Dialect.isSafeTimezoneName(name)) {
+            this.logError('invalid-timezone', { timezone: name }, { at: this.timezone.location });
+            return undefined;
         }
+        return name;
     }
     queryExecute(executeFor) {
-        executeFor.resultFS.setTimezone(this.tz);
+        const timezone = this.timezoneName();
+        if (timezone !== undefined) {
+            executeFor.resultFS.setTimezone(timezone);
+        }
     }
 }
 exports.TimezoneStatement = TimezoneStatement;

package/dist/lang/ast/statements/import-statement.js

@@ -152,6 +152,10 @@ class ImportStatement extends malloy_element_1.ListOf {
                             }
                             const importMe = { ...sourceEntry };
                             importMe.as = dstName;
+                            const dialect = (0, source_def_utils_1.dialectOfEntry)(importMe, importedModel);
+                            if (dialect !== undefined) {
+                                trans.checkExperimentalDialect(dialect, this.location);
+                            }
                             doc.setEntry(dstName, { entry: importMe, exported: false });
                             // Every source the walk touched, routes included — not just the
                             // persistent ones. A `#@ -persist` wrapper that adds a join is

package/dist/lang/ast/types/malloy-element.d.ts

@@ -152,13 +152,6 @@ export declare class Document extends MalloyElement implements NameSpace {
     modelDef(): ModelDef;
     getEntry(str: string): ModelEntry | undefined;
     setEntry(str: string, ent: ModelEntry): void;
-    /**
-     * Return an error message if this dialect is the first reference to this particular
-     * dialect, and the dialect is marked as experimental, and we are not running tests.
-     * @param dialect The dialect name
-     * @returns The error message or undefined
-     */
-    checkExperimentalDialect(me: MalloyElement, dialect: string): void;
     private readonly dialectNameSpaces;
     getDialectNamespace(dialectName: string): NameSpace | undefined;
 }

package/dist/lang/ast/types/malloy-element.js

@@ -656,9 +656,6 @@ class Document extends MalloyElement {
         if (this.globalNameSpace.getEntry(str) !== undefined) {
             this.logError('name-conflict-with-global', `Cannot redefine '${str}', which is in global namespace`);
         }
-        if ((0, malloy_types_1.isSourceDef)(ent.entry)) {
-            this.checkExperimentalDialect(this, ent.entry.dialect);
-        }
         // Track if the model was modified after initialization
         if (this.didInitModel) {
             this.modelWasModified = true;
@@ -675,21 +672,6 @@ class Document extends MalloyElement {
             };
         }
     }
-    /**
-     * Return an error message if this dialect is the first reference to this particular
-     * dialect, and the dialect is marked as experimental, and we are not running tests.
-     * @param dialect The dialect name
-     * @returns The error message or undefined
-     */
-    checkExperimentalDialect(me, dialect) {
-        const t = this.translator();
-        if (t &&
-            t.firstReferenceToDialect(dialect) &&
-            (0, dialect_1.getDialect)(dialect).experimental &&
-            !t.experimentalDialectEnabled(dialect)) {
-            me.logError('experimental-dialect-not-enabled', { dialect });
-        }
-    }
     getDialectNamespace(dialectName) {
         if (this.dialectNameSpaces.has(dialectName)) {
             return this.dialectNameSpaces.get(dialectName);

package/dist/lang/composite-source-utils.js

@@ -407,9 +407,11 @@ function expandRefUsage(seed, fields, alwaysJoinNames = []) {
     // given usage into the accumulator. Used both in-loop (joins reached
     // via reference paths) and upfront (always-on segment-level joins).
     function activateJoin(joinDef, joinPath) {
+        if (!(0, malloy_types_1.isJoined)(joinDef))
+            return;
         const joinKey = (0, utils_2.pathToKey)('join', joinPath);
         const thisDep = getJoin(activeJoinGraph, joinKey, joinPath);
-        if (!(0, malloy_types_1.isJoined)(joinDef) || thisDep.checked)
+        if (thisDep.checked)
             return;
         thisDep.checked = true;
         const joinFieldUsage = getJoinFieldUsage(joinDef, joinPath);
@@ -462,8 +464,13 @@ function expandRefUsage(seed, fields, alwaysJoinNames = []) {
                 ungroupings.push(...joinedUngroupings(refPath, def.ungroupings));
             }
         }
-        // For paths through joins, additionaly track join relationships
-        for (let joinLen = 1; joinLen < reference.path.length; joinLen++) {
+        // Track the joins along the path.  A path which ends at a join and carries
+        // a unique key requirement is an aggregate computed over that join's rows,
+        // which should then include the join's on-clause fields in the field usage.
+        // A path which merely names a join-typed value does not.
+        const endsInTraversedJoin = reference.uniqueKeyRequirement !== undefined;
+        const joinDepth = reference.path.length - (endsInTraversedJoin ? 0 : 1);
+        for (let joinLen = 1; joinLen <= joinDepth; joinLen++) {
             const joinPath = reference.path.slice(0, joinLen);
             const joinDef = inNamespace(joinPath, fieldNameSpace);
             if (!joinDef)

package/dist/lang/malloy-to-ast.js

@@ -803,14 +803,8 @@ class MalloyToAST extends AbstractParseTreeVisitor_1.AbstractParseTreeVisitor {
         return this.visitTimezoneStatement(cx.timezoneStatement());
     }
     visitTimezoneStatement(cx) {
-        const timezone = this.getPlainStringFrom(cx);
-        const timezoneStatement = this.astAt(new ast.TimezoneStatement(timezone), cx.string());
-        if (!timezoneStatement.isValid) {
-            this.astError(timezoneStatement, 'invalid-timezone', {
-                timezone: timezoneStatement.tz,
-            });
-        }
-        return this.astAt(timezoneStatement, cx);
+        const timezone = this.astAt(new ast.ExprString(this.getPlainStringFrom(cx)), cx.string());
+        return this.astAt(new ast.TimezoneStatement(timezone), cx);
     }
     visitQueryProperties(pcx) {
         const qProps = this.only(pcx.queryStatement().map(qcx => this.astAt(this.visit(qcx), qcx)), x => ast.isQueryProperty(x) && x, 'query statement');