BREAKING CHANGE: relation support

Author: adeyahya

prisma/typebox/Post.ts

@@ -1,7 +1,22 @@
-import { Type, Static } from '@sinclair/typebox';
+import { Type, Static } from "@sinclair/typebox";
+import { Role } from "./Role";
 
 export const Post = Type.Object({
-  id: Type.Number(),
+  id: Type.Optional(Type.Number()),
+  user: Type.Optional(
+    Type.Object({
+      id: Type.Optional(Type.Number()),
+      createdAt: Type.Optional(Type.String()),
+      email: Type.String(),
+      weight: Type.Optional(Type.Number()),
+      is18: Type.Optional(Type.Boolean()),
+      name: Type.Optional(Type.String()),
+      successorId: Type.Optional(Type.Number()),
+      role: Type.Optional(Role),
+      keywords: Type.Array(Type.String()),
+      biography: Type.String(),
+    })
+  ),
   userId: Type.Optional(Type.Number()),
 });
 

prisma/typebox/Role.ts

@@ -1,8 +1,8 @@
-import { Type, Static } from '@sinclair/typebox';
+import { Type, Static } from "@sinclair/typebox";
 
 export const RoleConst = {
-  USER: Type.Literal('USER'),
-  ADMIN: Type.Literal('ADMIN'),
+  USER: Type.Literal("USER"),
+  ADMIN: Type.Literal("ADMIN"),
 };
 
 export const Role = Type.KeyOf(Type.Object(RoleConst));

prisma/typebox/User.ts

@@ -1,13 +1,21 @@
-import { Type, Static } from '@sinclair/typebox';
+import { Type, Static } from "@sinclair/typebox";
+import { Role } from "./Role";
 
 export const User = Type.Object({
-  id: Type.Number(),
-  createdAt: Type.String(),
+  id: Type.Optional(Type.Number()),
+  createdAt: Type.Optional(Type.String()),
   email: Type.String(),
   weight: Type.Optional(Type.Number()),
   is18: Type.Optional(Type.Boolean()),
   name: Type.Optional(Type.String()),
   successorId: Type.Optional(Type.Number()),
+  role: Type.Optional(Role),
+  posts: Type.Array(
+    Type.Object({
+      id: Type.Optional(Type.Number()),
+      userId: Type.Optional(Type.Number()),
+    })
+  ),
   keywords: Type.Array(Type.String()),
   biography: Type.String(),
 });

src/generator/transformDMMF.ts

@@ -2,6 +2,7 @@ import type { DMMF } from '@prisma/generator-helper';
 
 const transformField = (field: DMMF.Field) => {
   const tokens = [field.name + ':'];
+  const deps = new Set();
 
   if (['Int', 'Float'].includes(field.type)) {
     tokens.push('Type.Number()');
@@ -10,28 +11,32 @@ const transformField = (field: DMMF.Field) => {
   } else if (field.type === 'Boolean') {
     tokens.push('Type.Boolean()');
   } else {
-    return '';
+    tokens.push(`::${field.type}::`);
+    deps.add(field.type);
   }
 
   if (field.isList) {
     tokens.splice(1, 0, 'Type.Array(');
     tokens.splice(tokens.length, 0, ')');
   }
 
-  if (!field.isRequired) {
+  if (!field.isRequired || field.hasDefaultValue) {
     tokens.splice(1, 0, 'Type.Optional(');
     tokens.splice(tokens.length, 0, ')');
   }
 
-  return tokens.join(' ').concat('\n');
+  return { str: tokens.join(' ').concat('\n'), deps };
 };
 
 const transformFields = (fields: DMMF.Field[]) => {
   let dependencies = new Set();
   const _fields: string[] = [];
 
   fields.map(transformField).forEach((field) => {
-    _fields.push(field);
+    _fields.push(field.str);
+    [...field.deps].forEach((d) => {
+      dependencies.add(d);
+    });
   });
 
   return {
@@ -40,16 +45,23 @@ const transformFields = (fields: DMMF.Field[]) => {
   };
 };
 
-const transformModel = (model: DMMF.Model) => {
+const transformModel = (model: DMMF.Model, dependant: string | null = null) => {
   const fields = transformFields(model.fields);
-
-  return [
-    `import {Type, Static} from '@sinclair/typebox'\n`,
-    `\nexport const ${model.name} = Type.Object({\n\t`,
+  let raw = [
+    `${dependant ? '' : `export const ${model.name} = `}Type.Object({\n\t`,
     fields.rawString,
-    '})\n',
-    `\nexport type ${model.name}Type = Static<typeof ${model.name}>`,
-  ].join('');
+    '})',
+  ].join('\n');
+
+  if (typeof dependant === 'string') {
+    const re = new RegExp(`.+::${dependant}.+\n`);
+    raw = raw.replace(re, '');
+  }
+
+  return {
+    raw,
+    deps: fields.dependencies,
+  };
 };
 
 export const transformEnum = (enm: DMMF.DatamodelEnum) => {
@@ -58,7 +70,6 @@ export const transformEnum = (enm: DMMF.DatamodelEnum) => {
     .join('');
 
   return [
-    "import {Type, Static} from '@sinclair/typebox'\n",
     `export const ${enm.name}Const = {`,
     values,
     '}\n',
@@ -69,18 +80,46 @@ export const transformEnum = (enm: DMMF.DatamodelEnum) => {
 
 export function transformDMMF(dmmf: DMMF.Document) {
   const { models, enums } = dmmf.datamodel;
+  const importStatements = new Set([
+    'import {Type, Static} from "@sinclair/typebox"',
+  ]);
 
   return [
     ...models.map((model) => {
+      let { raw, deps } = transformModel(model);
+
+      [...deps].forEach((d) => {
+        const depsModel = models.find((m) => m.name === d) as DMMF.Model;
+        if (depsModel) {
+          const replacer = transformModel(depsModel, model.name);
+          const re = new RegExp(`::${d}::`, 'gm');
+          raw = raw.replace(re, replacer.raw);
+        }
+      });
+
+      enums.forEach((enm) => {
+        const re = new RegExp(`::${enm.name}::`, 'gm');
+        if (raw.match(re)) {
+          raw = raw.replace(re, enm.name);
+          importStatements.add(`import { ${enm.name} } from './${enm.name}'`);
+        }
+      });
+
       return {
         name: model.name,
-        rawString: transformModel(model),
+        rawString: [
+          [...importStatements].join('\n'),
+          raw,
+          `export type ${model.name}Type = Static<typeof ${model.name}>`,
+        ].join('\n\n'),
       };
     }),
     ...enums.map((enm) => {
       return {
         name: enm.name,
-        rawString: transformEnum(enm),
+        rawString:
+          'import {Type, Static} from "@sinclair/typebox"\n\n' +
+          transformEnum(enm),
       };
     }),
   ];