feat: add methods to retrieve all tables and columns for MongoDB and PostgreSQL connectors

Author: NoOne7135

adminforth/dataConnectors/mongo.ts

@@ -46,7 +46,39 @@ class MongoConnector extends AdminForthBaseConnector implements IAdminForthDataS
         [AdminForthSortDirections.asc]: 1,
         [AdminForthSortDirections.desc]: -1,
     };
+    async getAllTables(): Promise<Array<string>>{
+        const db = this.client.db();
+    
+        const collections = await db.listCollections().toArray();
+    
+        return collections.map(col => col.name);
+    }
 
+    async getAllColumnsInTable(collectionName: string): Promise<Array<string>> {
+
+        const sampleDocs = await this.client.db().collection(collectionName).find({}).limit(100).toArray();
+        
+        const fieldSet = new Set<string>();
+        
+        function flattenObject(obj: any, prefix = '') {
+            Object.entries(obj).forEach(([key, value]) => {
+            const fullKey = prefix ? `${prefix}.${key}` : key;
+            if (value && typeof value === 'object' && !Array.isArray(value) && !(value instanceof Date)) {
+                flattenObject(value, fullKey);
+            } else {
+                fieldSet.add(fullKey);
+            }
+            });
+        }
+        
+        for (const doc of sampleDocs) {
+            flattenObject(doc);
+        }
+        
+        return Array.from(fieldSet);
+    }
+      
+    
     async discoverFields(resource) {
         return resource.columns.filter((col) => !col.virtual).reduce((acc, col) => {
             if (!col.type) {

adminforth/dataConnectors/postgres.ts

@@ -45,6 +45,24 @@ class PostgresConnector extends AdminForthBaseConnector implements IAdminForthDa
         [AdminForthSortDirections.desc]: 'DESC',
     };
 
+    async getAllTables(): Promise<Array<string>> {
+        const res = await this.client.query(`
+            SELECT table_name
+            FROM information_schema.tables
+            WHERE table_schema = 'public' AND table_type = 'BASE TABLE';
+        `);
+        return res.rows.map(row => row.table_name);
+    }
+    
+    async getAllColumnsInTable(tableName: string): Promise<Array<string>> {
+        const res = await this.client.query(`
+            SELECT column_name
+            FROM information_schema.columns
+            WHERE table_name = $1 AND table_schema = 'public';
+        `, [tableName]);
+        return res.rows.map(row => row.column_name);
+    }
+      
     async discoverFields(resource) {
 
         const tableName = resource.table;

adminforth/index.ts

@@ -391,6 +391,7 @@ class AdminForth implements IAdminForth {
           // console.log(`Connector ${dataSourceId} does not have getAllTables method`);
           results[dataSourceId] = [];
         }
+        await connector.close();
       })
     );
 
@@ -419,6 +420,7 @@ class AdminForth implements IAdminForth {
         } else {
           results[dataSourceId] = [];
         }
+        await connector.close();
       })
     );