Last active
January 16, 2024 19:25
-
-
Save manuelbieh/ae3b028286db10770c81 to your computer and use it in GitHub Desktop.
Automatically generates migration files from your sequelize models
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import * as models from "models"; | |
import Sequelize from "sequelize"; | |
import fs from "fs"; | |
delete models.default; | |
const sequelize = new Sequelize( | |
'', | |
'', | |
'', { | |
host: '', | |
dialect: '', | |
} | |
); | |
for(let model in models) { | |
let attributes = models[model].attributes; | |
for(let column in attributes) { | |
delete attributes[column].Model; | |
delete attributes[column].fieldName; | |
delete attributes[column].field; | |
for(let property in attributes[column]) { | |
if(property.startsWith('_')) { | |
delete attributes[column][property]; | |
} | |
} | |
if(typeof attributes[column]['type'] !== 'undefined') { | |
if(typeof attributes[column]['type']['options'] !== 'undefined' && typeof attributes[column]['type']['options'].toString === 'function') { | |
attributes[column]['type']['options'] = attributes[column]['type']['options'].toString(sequelize); | |
} | |
if(typeof attributes[column]['type'].toString === 'function') { | |
attributes[column]['type'] = attributes[column]['type'].toString(sequelize); | |
} | |
} | |
} | |
let schema = JSON.stringify(attributes, null, 4); | |
let tableName = models[model].tableName; | |
let indexes = ['\n']; | |
if(models[model].options.indexes.length) { | |
models[model].options.indexes.forEach((obj) => { | |
indexes.push(' .then(() => {'); | |
indexes.push(' return queryInterface.addIndex('); | |
indexes.push(` '${tableName}',`); | |
indexes.push(` ['${obj.fields.join("','")}']`); | |
let opts = {}; | |
if(obj.name) { | |
opts.indexName = obj.name; | |
} | |
if(obj.unique === true) { | |
opts.indicesType = 'UNIQUE'; | |
} | |
if(obj.method === true) { | |
opts.indexType = obj.method; | |
} | |
if(Object.keys(opts).length) { | |
indexes.push(` , ${JSON.stringify(opts)}`) | |
} | |
indexes.push(' )'); | |
indexes.push(' })'); | |
}); | |
} | |
schema = schema.split('\n').map((line) => ' ' + line).join('\n'); | |
let template = `'use strict'; | |
module.exports = { | |
up: function(queryInterface, Sequelize) { | |
return queryInterface.sequelize.query('SET FOREIGN_KEY_CHECKS = 0') | |
.then(() => { | |
return queryInterface.createTable('${tableName}', | |
${schema}) | |
})${indexes.join('\n')} | |
.then(() => { | |
return queryInterface.sequelize.query('SET FOREIGN_KEY_CHECKS = 1'); | |
}); | |
}, | |
down: function(queryInterface, Sequelize) { | |
return queryInterface.sequelize.query('SET FOREIGN_KEY_CHECKS = 0') | |
.then(() => { | |
return queryInterface.dropTable('${tableName}'); | |
}) | |
.then(() => { | |
return queryInterface.sequelize.query('SET FOREIGN_KEY_CHECKS = 1'); | |
}); | |
} | |
};` | |
let d = new Date(); | |
let filename = [d.getFullYear(), d.getMonth()+1, d.getDate(), d.getHours(), d.getMinutes(), d.getSeconds()] | |
.map((num) => num <= 60 && (num + 100).toString().substring(1) || num) | |
.join('') + `-${models[model].tableName}`; | |
fs.writeFileSync(`./${filename}.js`, template); | |
}; |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I needed also to order the files to create tables in correct order (dependencies of related database tables). I used TopologicalSort order algorithm for correct order and add ordering number to exported file.
This does the trick:
Full source code: