extract-mysql-schema 0.0.7 → 0.0.9

package/example.js

@@ -0,0 +1,10 @@
+module.exports = {
+  connection: {
+    engine: 'mysql',
+    host: '127.0.0.1',
+    user: 'root',
+    password: 'password',
+    database: 'db',
+    charset: 'utf8'
+  }
+};

package/index.js

@@ -19,6 +19,14 @@ const extractSchemas = async function (connection) {
     const schemaName = connection.database;
 
     let adapter = await getAdapter(connection);
+
+    let fkeys = await adapter.query(`
+    SELECT iif.*, iifc.FOR_COL_NAME, iifc.REF_COL_NAME
+    FROM INFORMATION_SCHEMA.INNODB_FOREIGN as iif
+    JOIN INFORMATION_SCHEMA.INNODB_FOREIGN_COLS as iifc on iifc.ID=iif.ID
+    `);
+    fkeys=fkeys[0];
+
     let columns = await adapter.query(`
     SELECT *
     FROM  INFORMATION_SCHEMA.COLUMNS
@@ -28,6 +36,20 @@ const extractSchemas = async function (connection) {
 
     columns = columns[0];
 
+    const foreign = {};
+    for(let i=0;i<fkeys.length;i++) {
+        const tableName = fkeys[i]['FOR_NAME'].substring(schemaName.length+1);
+        const keyName = fkeys[i]['ID'].substring(schemaName.length+1);
+        foreign[tableName+"_"+fkeys[i]['FOR_COL_NAME']] = {
+            "schemaName": schemaName,
+            "tableName": fkeys[i]['REF_NAME'].substring(schemaName.length+1),
+            "columnName": fkeys[i]['FOR_COL_NAME'],
+            "onUpdate": "CASCADE",
+            "onDelete": "RESTRICT",
+            "name": keyName
+          };
+    }
+
     let schema = {};
     let tables = [];
     for (let i = 0; i < columns.length; i++) {
@@ -63,9 +85,14 @@ const extractSchemas = async function (connection) {
             generated: columns[i]['EXTRA'].indexOf('DEFAULT_GENERATED') >= 0 ? (columns[i]['EXTRA'].indexOf('on update') > 0 ? "ALWAYS" : "BY DEFAULT") : "NEVER",
             isUpdatable: columns[i]['EXTRA'].indexOf('DEFAULT_GENERATED') < 0,
             type: columns[i]['DATA_TYPE'],
-            defaultValue: columns[i]['COLUMN_DEFAULT'] || ""
+            defaultValue: columns[i]['COLUMN_DEFAULT'] || "",
+            references:[]
         };
 
+        if(foreign[tableName+"_"+name]!==undefined) {
+            column.references.push(foreign[tableName+"_"+name]);
+        }
+
         column = Object.fromEntries(Object.entries(column).filter(([_, v]) => v != null));
         table.push(column);
     }
@@ -81,4 +108,4 @@ const extractSchemas = async function (connection) {
 
 module.exports = {
     extractSchemas
-};
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "extract-mysql-schema",
-  "version": "0.0.7",
+  "version": "0.0.9",
   "description": "",
   "main": "index.js",
   "repository": {

package/run.js

@@ -1,16 +1,50 @@
+const fs = require("fs");
 const { extractSchemas } = require('./index.js');
 
-async function run() {
-  const connection = {
-    host: '127.0.0.1',
-    database: 'db',
-    user: 'root',
-    password: 'password',
-  };
+async function main(options) {
+  const config = require(options.configFile);
+  const result = await extractSchemas(config.connection);
+  if(options.outputFile) {
+    fs.writeFileSync(options.outputFile, JSON.stringify(result,null,2) ,"utf8")
+  } else console.log(JSON.stringify(result,null,2));
+}
+
+let options = {
+  configFile:"",
+  outputFile:"",
+}
 
-  const result = await extractSchemas(connection);
+if (process.argv.length === 2) {
+console.error('Expected at least one argument!');
+process.exit(1);
+} else {
+let argv = process.argv;
 
-  console.log(JSON.stringify(result,null,2));
+for(let i=2;i<argv.length;i++) {
+  if(argv[i].substring(0,2)==="--") {
+      let name = argv[i].substring(2);
+      if(options[name]!==undefined) {
+          options[name] = argv[i+1];
+          i++;
+      } else {
+          console.error('Expected a known option');
+          process.exit(1);
+      }
+  }
+}
 }
 
-run();
+(async () => {
+try {
+    console.log('\x1b[32m%s\x1b[0m',"Running with options:\n",JSON.stringify(options,null,2));
+    main(options).then(()=>{
+      if(options.debug) console.log("DONE");
+      process.exit(0);
+    },()=>{
+      if(options.debug) console.log("ERROR");
+      process.exit(1);
+    })
+} catch (e) {
+    process.exit(1);
+}
+})();