@reldens/storage 0.36.0 → 0.37.0

package/README.md

@@ -104,7 +104,7 @@ Then, use the PrismaDataServer in your code:
 const { PrismaDataServer } = require('@reldens/storage');
 
 const server = new PrismaDataServer({
-    client: 'mysql2',
+    client: 'mysql',
     config: {
         user: 'reldens',
         password: 'reldens',

package/index.js

@@ -29,6 +29,11 @@ module.exports = {
         'mikro-orm': MikroOrmDataServer,
         'prisma': PrismaDataServer
     },
+    DriversClassMap: {
+        'ObjectionJsDataServer': 'objection-js',
+        'MikroOrmDataServer': 'mikro-orm',
+        'PrismaDataServer': 'prisma-server'
+    },
     // objection-js:
     ObjectionJsDataServer,
     ObjectionJsDriver,

package/lib/entities-generator.js

@@ -37,6 +37,11 @@ class EntitiesGenerator
             'mikro-orm': MikroOrmDataServer,
             'prisma': PrismaDataServer
         };
+        this.driversClassMap = {
+            'ObjectionJsDataServer': 'objection-js',
+            'MikroOrmDataServer': 'mikro-orm',
+            'PrismaDataServer': 'prisma-server'
+        };
         this.server = sc.get(props, 'server', false);
         this.connectionData = sc.get(props, 'connectionData', false);
         if(!this.server && !this.connectionData){
@@ -194,7 +199,7 @@ class EntitiesGenerator
 
     addTypeAttribute(props, column, type)
     {
-        if(this.isForeignKey(column)){
+        if(column.referencedTable){
             props.push('type: \'reference\'');
             props.push('reference: \''+this.getReferenceTable(column)+'\'');
             return;
@@ -247,8 +252,7 @@ class EntitiesGenerator
             return [];
         }
         let valuesString = match[1];
-        let values = valuesString.split('\',\'');
-        return values;
+        return valuesString.split('\',\'');
     }
 
     addRequiredAttribute(props, column)
@@ -258,14 +262,6 @@ class EntitiesGenerator
         }
     }
 
-    isForeignKey(column)
-    {
-        if(column.referencedTable){
-            return true;
-        }
-        return false;
-    }
-
     getReferenceTable(column)
     {
         if(column.referencedTable){
@@ -562,14 +558,24 @@ class EntitiesGenerator
         try {
             await this.server.connect();
         } catch (error) {
-            ErrorManager.error('Failed to connect to database: '+error.message);
+            Logger.critical('Failed to connect to database: '+error.message);
+            return false;
         }
         let tables = await this.server.fetchEntitiesFromDatabase();
         if(!tables){
-            ErrorManager.error('Failed to fetch tables from database.');
+            Logger.critical('EntitiesGenerator tables fetch failed.');
+            return false;
         }
         this.ensureFoldersExist();
-        let driverKey = sc.get(this.connectionData, 'driver', 'objection-js');
+        let driverKey = sc.get(
+            this.connectionData,
+            'driver',
+            (this.server ? sc.get(this.driversClassMap, this.server.constructor.name, false) : false)
+        );
+        if(!driverKey){
+            Logger.critical('Failed to fetch driver key.');
+            return false;
+        }
         for(let tableName in tables){
             let tableData = tables[tableName];
             await this.generateEntityFile(tableName, tableData);
@@ -583,6 +589,9 @@ class EntitiesGenerator
 
     createServer()
     {
+        if(this.server){
+            return this.server;
+        }
         let driverKey = sc.get(this.connectionData, 'driver', 'objection-js');
         let driverClassMapped = this.driverMap[driverKey];
         if(!driverClassMapped){
@@ -598,6 +607,7 @@ class EntitiesGenerator
                 port: sc.get(this.connectionData, 'port', 3306)
             }
         });
+        return this.server;
     }
 
 }

package/lib/mikro-orm/mikro-orm-data-server.js

@@ -102,7 +102,7 @@ class MikroOrmDataServer extends BaseDataServer
     async fetchEntitiesFromDatabase()
     {
         if(!this.initialized){
-            Logger.error('Connection was not initialized, please use the connect method first.');
+            Logger.critical('Connection was not initialized, please use the connect method first.');
             return false;
         }
         try {
@@ -120,10 +120,11 @@ class MikroOrmDataServer extends BaseDataServer
                 }
                 return tables;
             }
-            ErrorManager.error('Unsupported client type: '+this.client);
+            Logger.critical('Unsupported client type: '+this.client);
         } catch(error) {
-            ErrorManager.error('Failed to fetch tables from database: ' + error.message);
+            Logger.critical('MikroORM tables fetch failed. ' + error.message);
         }
+        return false;
     }
 
 }

package/lib/prisma/prisma-data-server.js

@@ -177,13 +177,14 @@ class PrismaDataServer extends BaseDataServer
     async fetchEntitiesFromDatabase()
     {
         if(!this.initialized){
-            Logger.error('Connection was not initialized, please use the connect method first.');
+            Logger.critical('Connection was not initialized, please use the connect method first.');
             return false;
         }
         try {
             return await MySQLTablesProvider.fetchTables(this);
         } catch(error) {
-            ErrorManager.error('Failed to fetch tables from database: '+error.message);
+            Logger.critical('Prisma Data Server tables fetch failed. '+error.message);
+            return false;
         }
     }
 

package/package.json

@@ -1,7 +1,7 @@
 {
     "name": "@reldens/storage",
     "scope": "@reldens",
-    "version": "0.36.0",
+    "version": "0.37.0",
     "description": "Reldens - Storage",
     "author": "Damian A. Pastorini",
     "license": "MIT",
@@ -25,6 +25,8 @@
         "mikro",
         "mikroorm",
         "mikro-orm",
+        "prisma",
+        "prisma-orm",
         "driver",
         "orm",
         "database",