@cargo-ai/cdk 1.0.11 → 1.0.12

package/build/src/resources/model.js

@@ -34,6 +34,9 @@ export function defineModel(slug, spec) {
             unification: spec.unification === undefined
                 ? undefined
                 : compileUnification(spec.unification),
+            additionalColumns: spec.additionalColumns === undefined
+                ? undefined
+                : spec.additionalColumns.map(compileAdditionalColumn),
         },
     });
     // A small Proxy mirroring `connector.actions`: `model.columns.<columnSlug>`
@@ -49,7 +52,51 @@ export function defineModel(slug, spec) {
     });
     return { slug, uuid: token(id, "uuid"), resource: "model", columns };
 }
-// Lower a `UnificationSpec` to the wire shape `StorageTypes.ModelUnification`.
+// Lower an `ModelAdditionalColumnSpec` to the wire `ModelAdditionalColumn` shape: the
+// metric's relationship and the lookup's join model carry handles/refs in code
+// (so the dependency shows up in the graph), which become uuid tokens resolved
+// at apply time.
+function compileAdditionalColumn(column) {
+    const base = {
+        slug: column.slug,
+        type: column.type,
+        label: column.label === undefined ? toTitle(column.slug) : column.label,
+        description: column.description,
+        properties: column.properties,
+    };
+    switch (column.kind) {
+        case "custom":
+            return { ...base, kind: "custom" };
+        case "computed":
+            return {
+                ...base,
+                kind: "computed",
+                expression: column.expression,
+                columnsUsed: column.columnsUsed,
+            };
+        case "metric":
+            return {
+                ...base,
+                kind: "metric",
+                relationshipUuid: refUuid(column.relationship),
+                aggregation: column.aggregation,
+                filter: column.filter,
+            };
+        case "lookup":
+            return {
+                ...base,
+                kind: "lookup",
+                join: {
+                    toModelUuid: refUuid(column.join.model),
+                    fromColumnSlug: column.join.fromColumnSlug,
+                    toColumnSlug: column.join.toColumnSlug,
+                },
+                extractColumnSlug: column.extractColumnSlug,
+                filter: column.filter,
+            };
+    }
+}
+// Lower a `ModelUnificationSpec` to the wire shape `StorageTypes.ModelUnification`.
 // The only reshaping is the parent link: a `model` parent carries a handle/ref
 // in code (so the dependency shows up in the graph), which becomes a
 // `parentModelUuid` token resolved at apply time.