prisma-guard 1.30.0 → 1.31.0

package/dist/runtime/index.cjs

@@ -4141,12 +4141,39 @@ function validateAllowedKeys(value, allowed, method, kind) {
     return `Shape key "${key}" not valid for ${method}. Allowed: ${[...allowed].join(", ")}`;
   });
 }
+var RELATION_OPS_COVERING_FK = /* @__PURE__ */ new Set([
+  "connect",
+  "connectOrCreate",
+  "create"
+]);
+function collectRelationCoveredFks(modelFields, dataConfig) {
+  const covered = /* @__PURE__ */ new Set();
+  for (const [fieldName, value] of Object.entries(dataConfig)) {
+    const meta = modelFields[fieldName];
+    if (!meta || !meta.isRelation)
+      continue;
+    const fks = meta.relationFromFields;
+    if (!fks || fks.length === 0)
+      continue;
+    if (!isPlainObject(value))
+      continue;
+    const covers = Object.keys(value).some(
+      (op) => RELATION_OPS_COVERING_FK.has(op)
+    );
+    if (!covers)
+      continue;
+    for (const fk of fks)
+      covered.add(fk);
+  }
+  return covered;
+}
 function validateCreateCompleteness(modelName, dataConfig, typeMap, scopeFks, zodDefaults) {
   const modelFields = typeMap[modelName];
   if (!modelFields)
     return;
   const zodDefaultFields = zodDefaults[modelName];
   const zodDefaultSet = zodDefaultFields ? new Set(zodDefaultFields) : void 0;
+  const relationCoveredFks = collectRelationCoveredFks(modelFields, dataConfig);
   for (const [fieldName, meta] of Object.entries(modelFields)) {
     if (meta.isRelation)
       continue;
@@ -4160,10 +4187,12 @@ function validateCreateCompleteness(modelName, dataConfig, typeMap, scopeFks, zo
       continue;
     if (scopeFks.has(fieldName))
       continue;
+    if (relationCoveredFks.has(fieldName))
+      continue;
     if (zodDefaultSet && zodDefaultSet.has(fieldName))
       continue;
     throw new ShapeError(
-      `Required field "${fieldName}" on model "${modelName}" is missing from create data shape, has no default, and is not a scope FK`
+      `Required field "${fieldName}" on model "${modelName}" is missing from create data shape, has no default, is not a scope FK, and is not covered by a relation write in the shape`
     );
   }
 }