@suss/contract-prisma 0.10.0 → 0.12.0

package/dist/index.cjs

@@ -118,6 +118,7 @@ function buildModelSummary(opts) {
   const columns = [];
   const indexes = [];
   const physicalTable = physicalTableOf(opts.model);
+  let hasRelation = false;
   for (const property of opts.model.properties) {
     if (property.type === "field") {
       const field = property;
@@ -125,6 +126,11 @@ function buildModelSummary(opts) {
       if (column !== null) {
         columns.push(column);
       }
+      const related = relationField(field, opts.modelNames);
+      if (related !== null) {
+        hasRelation = true;
+        columns.push(related);
+      }
     } else if (property.type === "attribute") {
       const attr = property;
       const index = blockAttributeToIndex(attr);
@@ -133,6 +139,14 @@ function buildModelSummary(opts) {
       }
     }
   }
+  if (hasRelation) {
+    columns.push({
+      name: "_count",
+      type: "PrismaCount",
+      nullable: false,
+      derived: true
+    });
+  }
   return {
     kind: "library",
     location: {
@@ -166,6 +180,25 @@ function buildModelSummary(opts) {
     }
   };
 }
+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,
+    ...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;
@@ -233,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

@@ -83,6 +83,7 @@ function buildModelSummary(opts) {
   const columns = [];
   const indexes = [];
   const physicalTable = physicalTableOf(opts.model);
+  let hasRelation = false;
   for (const property of opts.model.properties) {
     if (property.type === "field") {
       const field = property;
@@ -90,6 +91,11 @@ function buildModelSummary(opts) {
       if (column !== null) {
         columns.push(column);
       }
+      const related = relationField(field, opts.modelNames);
+      if (related !== null) {
+        hasRelation = true;
+        columns.push(related);
+      }
     } else if (property.type === "attribute") {
       const attr = property;
       const index = blockAttributeToIndex(attr);
@@ -98,6 +104,14 @@ function buildModelSummary(opts) {
       }
     }
   }
+  if (hasRelation) {
+    columns.push({
+      name: "_count",
+      type: "PrismaCount",
+      nullable: false,
+      derived: true
+    });
+  }
   return {
     kind: "library",
     location: {
@@ -131,6 +145,25 @@ function buildModelSummary(opts) {
     }
   };
 }
+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,
+    ...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;
@@ -198,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.10.0",
+  "version": "0.12.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.10.0"
+    "@suss/behavioral-ir": "0.12.0"
   },
   "devDependencies": {
     "tsup": "*",