load views, procedures, functions

packages/engines/mssql/MsSqlAnalyser.js

@@ -208,25 +208,64 @@ class MsSqlAnalyser extends DatabaseAnalayser {
     return res;
   }
   async _runAnalysis() {
-    const tables = await this.driver.query(this.pool, this.createQuery('tables', ['tables']));
-    const columns = await this.driver.query(this.pool, this.createQuery('columns', ['tables']));
-    const pkColumns = await this.driver.query(this.pool, this.createQuery('primaryKeys', ['tables']));
-    const fkColumns = await this.driver.query(this.pool, this.createQuery('foreignKeys', ['tables']));
+    const tablesRows = await this.driver.query(this.pool, this.createQuery('tables', ['tables']));
+    const columnsRows = await this.driver.query(this.pool, this.createQuery('columns', ['tables']));
+    const pkColumnsRows = await this.driver.query(this.pool, this.createQuery('primaryKeys', ['tables']));
+    const fkColumnsRows = await this.driver.query(this.pool, this.createQuery('foreignKeys', ['tables']));
+
+    const sqlCodeRows = await this.driver.query(
+      this.pool,
+      this.createQuery('loadSqlCode', ['views', 'procedures', 'functions', 'triggers'])
+    );
+    const getCreateSql = (row) =>
+      sqlCodeRows.rows
+        .filter((x) => x.pureName == row.pureName && x.schemaName == row.schemaName)
+        .map((x) => x.codeText)
+        .join('');
+    const viewsRows = await this.driver.query(this.pool, this.createQuery('views', ['views']));
+    const programmableRows = await this.driver.query(
+      this.pool,
+      this.createQuery('programmables', ['procedures', 'functions'])
+    );
+
+    const tables = tablesRows.rows.map((row) => ({
+      ...row,
+      columns: columnsRows.rows
+        .filter((col) => col.objectId == row.objectId)
+        .map(({ isNullable, isIdentity, ...col }) => ({
+          ...col,
+          notNull: !isNullable,
+          autoIncrement: !!isIdentity,
+          commonType: detectType(col),
+        })),
+      primaryKey: extractPrimaryKeys(row, pkColumnsRows.rows),
+      foreignKeys: extractForeignKeys(row, fkColumnsRows.rows),
+    }));
+
+    const views = viewsRows.rows.map((row) => ({
+      ...row,
+      createSql: getCreateSql(row),
+    }));
+
+    const procedures = programmableRows.rows
+      .filter((x) => x.sqlObjectType.trim() == 'P')
+      .map((row) => ({
+        ...row,
+        createSql: getCreateSql(row),
+      }));
+
+    const functions = programmableRows.rows
+      .filter((x) => ['FN', 'IF', 'TF'].includes(x.sqlObjectType.trim()))
+      .map((row) => ({
+        ...row,
+        createSql: getCreateSql(row),
+      }));
 
     return this.mergeAnalyseResult({
-      tables: tables.rows.map((table) => ({
-        ...table,
-        columns: columns.rows
-          .filter((col) => col.objectId == table.objectId)
-          .map(({ isNullable, isIdentity, ...col }) => ({
-            ...col,
-            notNull: !isNullable,
-            autoIncrement: !!isIdentity,
-            commonType: detectType(col),
-          })),
-        primaryKey: extractPrimaryKeys(table, pkColumns.rows),
-        foreignKeys: extractForeignKeys(table, fkColumns.rows),
-      })),
+      tables,
+      views,
+      procedures,
+      functions,
     });
   }
 

packages/engines/mssql/sql/index.js

@@ -3,6 +3,9 @@ const foreignKeys = require('./foreignKeys');
 const primaryKeys = require('./primaryKeys');
 const tables = require('./tables');
 const modifications = require('./modifications');
+const loadSqlCode = require('./loadSqlCode');
+const views = require('./views');
+const programmables = require('./programmables');
 
 module.exports = {
   columns,
@@ -10,4 +13,7 @@ module.exports = {
   foreignKeys,
   primaryKeys,
   modifications,
+  loadSqlCode,
+  views,
+  programmables,
 };

packages/engines/mssql/sql/loadSqlCode.js

@@ -0,0 +1,8 @@
+module.exports = `
+select s.name as pureName, u.name as schemaName, c.text AS codeText
+    from sys.objects s
+    inner join sys.syscomments c on s.object_id = c.id
+    inner join sys.schemas u on u.schema_id = s.schema_id
+where (s.object_id =[OBJECT_ID_CONDITION])
+order by u.name, s.name, c.colid
+`;

packages/engines/mssql/sql/programmables.js

@@ -0,0 +1,6 @@
+module.exports = `
+select o.name as pureName, s.name as schemaName, o.object_id as objectId, o.create_date as createDate, o.modify_date as modifyDate, o.type as sqlObjectType
+from sys.objects o 
+inner join sys.schemas s on o.schema_id = s.schema_id
+where o.type in ('P', 'IF', 'FN', 'TF') and o.object_id =[OBJECT_ID_CONDITION]
+`;

packages/engines/mssql/sql/views.js

@@ -0,0 +1,10 @@
+module.exports = `
+SELECT 
+	o.name as pureName,
+	u.name as schemaName,
+	o.object_id as objectId,
+	o.create_date as createDate,
+	o.modify_date as modifyDate
+FROM sys.objects o INNER JOIN sys.schemas u ON u.schema_id=o.schema_id 
+WHERE type in ('V') and o.object_id =[OBJECT_ID_CONDITION]
+`;