feat: including relationships in migrations

src/generators/src/infra/database/migrations/migrations.js

@@ -2,7 +2,7 @@ const { snakeCase, camelCase } = require('lodash')
 const fs = require("fs")
 const glob = require('glob')
 const path = require('path')
-const { requireHerbarium } = require('../../../../utils')
+const { requireHerbarium, requireHerbs } = require('../../../../utils')
 
 module.exports =
   async ({ template: { generate }, filesystem }, command) =>
@@ -12,6 +12,7 @@ module.exports =
       process.stdout.write(`Generating Migration\n`)
 
       const herbarium = requireHerbarium(command, filesystem.cwd())
+      const herbs = requireHerbs(filesystem.cwd())
       const entities = herbarium.entities.all
 
       const cwd = filesystem.cwd().replace(new RegExp('\\\\', 'g'), '/')
@@ -26,15 +27,11 @@ module.exports =
         }
 
         function type2Str(Type) {
-          const nativeTypes = [Boolean, Number, String, Array, Object, Date, Function]
-
-          if (nativeTypes.includes(Type)) {
-            const _type = new Type()
-            if (_type instanceof String) return 'string'
-            if (_type instanceof Number) return 'integer'
-            if (_type instanceof Boolean) return 'boolean'
-            if (_type instanceof Date) return 'timestamp'
-          }
+          const _type = new Type()
+          if (_type instanceof String) return 'string'
+          if (_type instanceof Number) return 'integer'
+          if (_type instanceof Boolean) return 'boolean'
+          if (_type instanceof Date) return 'timestamp'
         }
 
         function getDBType(appDir) {
@@ -47,11 +44,35 @@ module.exports =
           const columns = []
           Object.keys(schema).forEach((prop) => {
             const { name, type, options } = schema[prop]
-            columns.push(`table.${type2Str(type)}('${snakeCase(name)}')${options.isId ? '.primary()' : ''}`)
+
+            const nativeTypes = [Boolean, Number, String, Date]
+            if (!nativeTypes.includes(type)) return
+
+            const typeString = type2Str(type)
+            columns.push(`table.${typeString}('${snakeCase(name)}')${options.isId ? '.primary()' : ''}`)
           })
           return columns
         }
 
+        function isArrayWithType(value) {
+          return Array.isArray(value) && value.length === 1
+        }
+
+        function identifyRef(schema) {
+          const refs = []
+          Object.values(schema).forEach(({ type, name }) => {
+            if (herbs.entity.isEntity(type)) {
+              const typeSchema = type.prototype.meta.schema
+              const idRef = Object.values(typeSchema).find(column => column.options.isId)?.name
+              refs.push({ id: idRef, columnName: `${camelCase(name)}Id`, table: `${camelCase(type.name)}s`, relationship: 'One-to-One' })
+            }
+
+            if (isArrayWithType(type) && herbs.entity.isEntity(type[0]))
+              refs.push({ table: `${camelCase(type[0].name)}s`, relationship: 'One-to-Many' })
+          })
+          return refs
+        }
+
         const db = getDBType(filesystem.cwd())
         const migrationName = new Date()
           .toISOString()
@@ -60,11 +81,14 @@ module.exports =
         const migrationFullPath = path.normalize(`${migrationsPath}/${migrationName}_${camelCase(name)}s.js`)
 
         const columns = createColumns(schema)
+        const ref = identifyRef(schema)
+        const idColumn = Object.values(schema).find(column => column.options.isId)?.name
+
 
         await generate({
           template: `infra/data/database/${db.toLowerCase()}/migration.ejs`,
           target: migrationFullPath,
-          props: { table: `${camelCase(name)}s`, columns: columns }
+          props: { table: `${camelCase(name)}s`, externalColumnName: `${camelCase(name)}Id`, columns, ref, idColumn }
         })
         process.stdout.write(`  New: ${migrationFullPath}\n`)
 

src/generators/utils.js

@@ -39,12 +39,17 @@ module.exports = {
     const herbariumPath = `${appPath}/src/domain/herbarium.js`
     let herbarium
     if (command === "update" || command === "spec") {
-      herbarium = require(herbariumPath)
+      herbarium = require(herbariumPath).herbarium
     }
     else
       herbarium = require('@herbsjs/herbarium').herbarium
     herbarium.requireAll()
     return herbarium
   },
+  requireHerbs: (appPath) => {
+    const herbariumPath = `${appPath}/src/domain/herbarium.js`
+    const herbs = require(herbariumPath).herbs
+    return herbs
+  },
   usingMongo: (base) => fs.existsSync(path.normalize(`${base}/src/infra/config/mongo.js`))
 }

src/templates/domain/herbarium.ejs

@@ -1,2 +1,3 @@
 const { herbarium } = require('@herbsjs/herbarium')
-module.exports = herbarium
+const herbs = require('@herbsjs/herbs')
+module.exports = { herbarium, herbs }

src/templates/infra/data/database/postgres/migration.ejs

@@ -1,17 +1,44 @@
-
 exports.up = async function (knex) {
-    knex.schema.hasTable('<%- props.table %>')
-        .then(function (exists) {
-            if (exists) return
-            return knex.schema
-                .createTable('<%- props.table %>', function (table) {<% for(colum of props.columns) { %>
-                    <%- colum %><% } %>
-                    table.timestamps()
-                })
-        })
+    return Promise.all([
+        knex.schema.hasTable('<%- props.table %>')
+            .then(function (exists) {
+                if (exists) return
+                return knex.schema
+                    .createTable('<%- props.table %>', function (table) {
+                        <% for(colum of props.columns) {%> 
+                        <%- colum %>  <%}%>
+                        <%if (props.ref) { 
+                            props.ref.forEach(function(link){ 
+                                if (link.relationship === 'One-to-One') { %>
+                        // table.integer('<%-link.columnName %>').unsigned().nullable()
+                        // table.foreign('<%-link.columnName %>').references('<%- link.id %>').inTable('<%- link.table %>')
+                                <% }                       
+                            })                      
+                        } %>
+                        table.timestamps()
+                    })
+            }),
+        <%if (props.ref) { 
+            props.ref.forEach(function(link){ 
+                if (link.relationship === 'One-to-Many') { %>
+            // knex.schema.table('<%- link.table %>', function (table) {
+            //     table.integer('<%-props.externalColumnName %>').unsigned().index().references('<%- props.idColumn %>').inTable('<%- props.table %>')
+            // })
+                <% } 
+            })
+        } %>
+    ])
 }
 
 exports.down = function (knex) {
-    return knex.schema
-    .dropTableIfExists('<%- props.table %>')
+    return Promise.all([
+        knex.schema
+            .dropTableIfExists('<%- props.table %>'),
+        <%if (props.ref) { 
+            props.ref.forEach(function(link){ 
+                if (link.relationship === 'One-to-Many') { %>
+        // knex.schema.table('<%- link.table %>', function (table) {
+        //     table.dropColumn('<%-props.externalColumnName %>')})
+        <% }})} %>
+    ])
 }