Allow functions to accept and return sql native types

Author: christopherswenson

packages/malloy/src/dialect/duckdb/dialect_functions.ts

@@ -24,6 +24,18 @@ const dayname: DefinitionBlueprint = {
   impl: {function: 'DAYNAME'},
 };
 
+const date_part: DefinitionBlueprint = {
+  takes: {'part': 'string', 'interval': {sql_native: 'interval'}},
+  returns: 'number',
+  impl: {function: 'DATE_PART'},
+};
+
+const to_seconds: DefinitionBlueprint = {
+  takes: {'seconds': 'number'},
+  returns: {sql_native: 'interval'},
+  impl: {function: 'TO_SECONDS'},
+};
+
 const to_timestamp: DefinitionBlueprint = {
   takes: {'epoch_seconds': 'number'},
   returns: 'timestamp',
@@ -87,4 +99,6 @@ export const DUCKDB_DIALECT_FUNCTIONS: DefinitionBlueprintMap = {
   to_timestamp,
   string_agg,
   string_agg_distinct,
+  to_seconds,
+  date_part,
 };

packages/malloy/src/dialect/functions/util.ts

@@ -231,11 +231,17 @@ export type TypeDescElementBlueprintOrNamedGeneric =
 export interface RecordBlueprint {
   record: Record<string, TypeDescElementBlueprintOrNamedGeneric>;
 }
+
+export interface SQLNativeTypeBlueprint {
+  sql_native: string;
+}
+
 export type LeafPlusType = LeafExpressionType | 'any';
 export type TypeDescElementBlueprint =
   | LeafPlusType
   | ArrayBlueprint
-  | RecordBlueprint;
+  | RecordBlueprint
+  | SQLNativeTypeBlueprint;
 export type NamedGeneric = {generic: string};
 
 export type TypeDescBlueprint =
@@ -362,6 +368,8 @@ function expandTypeDescElementBlueprint(
       throw new Error('Cannot use generic');
     }
     return {type: 'generic', generic: blueprint.generic};
+  } else if ('sql_native' in blueprint) {
+    return {type: 'sql native', rawType: blueprint.sql_native};
   }
   throw new Error('Cannot figure out type');
 }
@@ -396,6 +404,8 @@ function expandReturnTypeBlueprint(
     return minAggregate(
       expandTypeDescElementBlueprint(blueprint.measure, false)
     );
+  } else if ('sql_native' in blueprint) {
+    return anyExprType({type: 'sql native', rawType: blueprint.sql_native});
   } else {
     return minAnalytic(
       expandTypeDescElementBlueprint(blueprint.calculation, false)
@@ -415,7 +425,8 @@ function isTypeDescBlueprint(
     'constant' in blueprint ||
     'dimension' in blueprint ||
     'measure' in blueprint ||
-    'calculation' in blueprint
+    'calculation' in blueprint ||
+    'sql_native' in blueprint
   );
 }
 
@@ -456,6 +467,8 @@ function expandParamTypeBlueprint(
     return anyExprType(expandTypeDescElementBlueprint(blueprint, false));
   } else if ('record' in blueprint) {
     return anyExprType(expandTypeDescElementBlueprint(blueprint, false));
+  } else if ('sql_native' in blueprint) {
+    return anyExprType({type: 'sql native', rawType: blueprint.sql_native});
   } else {
     return maxAnalytic(expandTypeDescElementBlueprint(blueprint.calculation));
   }

test/src/databases/all/functions.spec.ts

@@ -1322,6 +1322,9 @@ expressionModels.forEach((x, databaseName) => {
       it.when(isDuckdb)('list_extract', async () => {
         await funcTest('list_extract(list_extract([[5]], 1), 1)', 5);
       });
+      it.when(isDuckdb)('date_part,to_seconds', async () => {
+        await funcTest('date_part("seconds", to_seconds(5))', 5);
+      });
     });
 
     describe('trino', () => {