@suss/contract-prisma 0.11.0 → 0.13.0

package/dist/index.cjs

@@ -184,13 +184,21 @@ function relationField(field, modelNames) {
   if (!modelNames.has(field.fieldType)) {
     return null;
   }
+  const key = relationKeyOf(field);
   return {
     name: field.name,
     type: field.array === true ? `${field.fieldType}[]` : field.fieldType,
     nullable: field.optional === true,
-    derived: true
+    derived: true,
+    ...key === null ? {} : { relationKey: key }
   };
 }
+function relationKeyOf(field) {
+  const relation = (field.attributes ?? []).find(
+    (attr) => attr.name === "relation"
+  );
+  return relation === void 0 ? null : readKeyedArrayArg(relation, "fields");
+}
 function fieldToColumn(field, modelNames, enumNames) {
   if (field.array === true) {
     return null;
@@ -258,25 +266,39 @@ function blockAttributeToIndex(attr) {
   return { fields, unique: attr.name !== "index" };
 }
 function readArrayArg(attr) {
+  for (const arg of attr.args ?? []) {
+    const names = arrayNames(arg.value);
+    if (names !== null) {
+      return names;
+    }
+  }
+  return null;
+}
+function readKeyedArrayArg(attr, key) {
   for (const arg of attr.args ?? []) {
     const value = arg.value;
-    if (value === null || value === void 0) {
+    if (value?.type !== "keyValue" || value.key !== key) {
       continue;
     }
-    if (value.type === "array" && Array.isArray(value.args)) {
-      const out = [];
-      for (const item of value.args) {
-        if (typeof item === "string") {
-          out.push(item);
-        } else if (typeof item === "object" && item !== null && "name" in item && typeof item.name === "string") {
-          out.push(item.name);
-        }
-      }
-      return out;
-    }
+    return arrayNames(value.value);
   }
   return null;
 }
+function arrayNames(value) {
+  const array = value;
+  if (array?.type !== "array" || !Array.isArray(array.args)) {
+    return null;
+  }
+  const out = [];
+  for (const item of array.args) {
+    if (typeof item === "string") {
+      out.push(item);
+    } else if (typeof item === "object" && item !== null && "name" in item && typeof item.name === "string") {
+      out.push(item.name);
+    }
+  }
+  return out;
+}
 function readProviderString(ds) {
   for (const a of ds.assignments) {
     if (a.key !== "provider") {

package/dist/index.js

@@ -149,13 +149,21 @@ function relationField(field, modelNames) {
   if (!modelNames.has(field.fieldType)) {
     return null;
   }
+  const key = relationKeyOf(field);
   return {
     name: field.name,
     type: field.array === true ? `${field.fieldType}[]` : field.fieldType,
     nullable: field.optional === true,
-    derived: true
+    derived: true,
+    ...key === null ? {} : { relationKey: key }
   };
 }
+function relationKeyOf(field) {
+  const relation = (field.attributes ?? []).find(
+    (attr) => attr.name === "relation"
+  );
+  return relation === void 0 ? null : readKeyedArrayArg(relation, "fields");
+}
 function fieldToColumn(field, modelNames, enumNames) {
   if (field.array === true) {
     return null;
@@ -223,25 +231,39 @@ function blockAttributeToIndex(attr) {
   return { fields, unique: attr.name !== "index" };
 }
 function readArrayArg(attr) {
+  for (const arg of attr.args ?? []) {
+    const names = arrayNames(arg.value);
+    if (names !== null) {
+      return names;
+    }
+  }
+  return null;
+}
+function readKeyedArrayArg(attr, key) {
   for (const arg of attr.args ?? []) {
     const value = arg.value;
-    if (value === null || value === void 0) {
+    if (value?.type !== "keyValue" || value.key !== key) {
       continue;
     }
-    if (value.type === "array" && Array.isArray(value.args)) {
-      const out = [];
-      for (const item of value.args) {
-        if (typeof item === "string") {
-          out.push(item);
-        } else if (typeof item === "object" && item !== null && "name" in item && typeof item.name === "string") {
-          out.push(item.name);
-        }
-      }
-      return out;
-    }
+    return arrayNames(value.value);
   }
   return null;
 }
+function arrayNames(value) {
+  const array = value;
+  if (array?.type !== "array" || !Array.isArray(array.args)) {
+    return null;
+  }
+  const out = [];
+  for (const item of array.args) {
+    if (typeof item === "string") {
+      out.push(item);
+    } else if (typeof item === "object" && item !== null && "name" in item && typeof item.name === "string") {
+      out.push(item.name);
+    }
+  }
+  return out;
+}
 function readProviderString(ds) {
   for (const a of ds.assignments) {
     if (a.key !== "provider") {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@suss/contract-prisma",
-  "version": "0.11.0",
+  "version": "0.13.0",
   "type": "module",
   "description": "Generate suss behavioral summaries from Prisma schema declarations.",
   "keywords": [
@@ -47,7 +47,7 @@
   },
   "dependencies": {
     "@mrleebo/prisma-ast": "^0.15.0",
-    "@suss/behavioral-ir": "0.11.0"
+    "@suss/behavioral-ir": "0.13.0"
   },
   "devDependencies": {
     "tsup": "*",