@514labs/moose-lib 0.6.446 → 0.6.447

package/dist/compilerPlugin.mjs

@@ -984,6 +984,50 @@ var generateAssertFunction = (ctx, type, typeName) => {
     config: { equals: false, guard: false }
   });
 };
+var getPropertyName = (prop) => prop.key.constants?.[0]?.values?.[0]?.value;
+var patchMetadataForInsert = (metadata, computedColumns, defaultColumns) => {
+  for (const obj of metadata.objects) {
+    const keep = [];
+    for (const prop of obj.type.properties) {
+      const name = getPropertyName(prop);
+      if (name !== void 0 && computedColumns.has(name)) continue;
+      if (name !== void 0 && defaultColumns.has(name)) {
+        prop.value.optional = true;
+        prop.value.required = false;
+      }
+      keep.push(prop);
+    }
+    obj.type.properties.length = 0;
+    obj.type.properties.push(...keep);
+  }
+};
+var withInsertableMetadata = (columns, fn) => {
+  const original = MetadataFactory.analyze;
+  MetadataFactory.analyze = (props) => {
+    const result = original(props);
+    if (result.success) {
+      patchMetadataForInsert(result.data, columns.computed, columns.defaults);
+    }
+    return result;
+  };
+  try {
+    return fn();
+  } finally {
+    MetadataFactory.analyze = original;
+  }
+};
+var generateInsertValidateFunction = (ctx, type, columns, typeName) => withInsertableMetadata(
+  columns,
+  () => generateValidateFunction(ctx, type, typeName)
+);
+var generateInsertIsFunction = (ctx, type, columns, typeName) => withInsertableMetadata(
+  columns,
+  () => generateIsFunction(ctx, type, typeName)
+);
+var generateInsertAssertFunction = (ctx, type, columns, typeName) => withInsertableMetadata(
+  columns,
+  () => generateAssertFunction(ctx, type, typeName)
+);
 var generateHttpAssertQueryFunction = (ctx, type, typeName) => {
   const typiaCtx = toTypiaContext(ctx);
   return HttpAssertQueryProgrammer.write({
@@ -1187,18 +1231,16 @@ var transformNewMooseResource = (node, checker, ctx) => {
   }
   const typiaCtx = ctx.typiaContext;
   let internalArguments;
+  let columns;
   if (typeName === "DeadLetterQueue") {
     internalArguments = [generateAssertFunction(typiaCtx, typeAtLocation)];
   } else {
+    columns = toColumns(typeAtLocation, checker, {
+      allowIndexSignatures
+    });
     internalArguments = [
       generateJsonSchemas(typiaCtx, typeAtLocation),
-      parseAsAny(
-        JSON.stringify(
-          toColumns(typeAtLocation, checker, {
-            allowIndexSignatures
-          })
-        )
-      )
+      parseAsAny(JSON.stringify(columns))
     ];
   }
   const resourceName = checker.getSymbolAtLocation(node.expression).name;
@@ -1230,6 +1272,48 @@ var transformNewMooseResource = (node, checker, ctx) => {
       true
     );
     updatedArgs = [...updatedArgs, validatorsObject];
+    if (resourceName === "OlapTable" && columns) {
+      const insertColumnSets = {
+        computed: new Set(
+          columns.filter((c) => c.alias != null || c.materialized != null).map((c) => c.name)
+        ),
+        defaults: new Set(
+          columns.filter((c) => c.default != null).map((c) => c.name)
+        )
+      };
+      const insertValidatorsObject = factory.createObjectLiteralExpression(
+        [
+          factory.createPropertyAssignment(
+            factory.createIdentifier("validate"),
+            wrapValidateFunction(
+              generateInsertValidateFunction(
+                typiaCtx,
+                typeAtLocation,
+                insertColumnSets
+              )
+            )
+          ),
+          factory.createPropertyAssignment(
+            factory.createIdentifier("assert"),
+            generateInsertAssertFunction(
+              typiaCtx,
+              typeAtLocation,
+              insertColumnSets
+            )
+          ),
+          factory.createPropertyAssignment(
+            factory.createIdentifier("is"),
+            generateInsertIsFunction(
+              typiaCtx,
+              typeAtLocation,
+              insertColumnSets
+            )
+          )
+        ],
+        true
+      );
+      updatedArgs = [...updatedArgs, insertValidatorsObject];
+    }
     if (resourceName === "IngestPipeline") {
       updatedArgs = [
         ...updatedArgs,