pk, fk analyse, show in structure tab

api/src/engines/mssql/MsSqlAnalyser.js

@@ -1,4 +1,5 @@
 const fs = require('fs-extra');
+const fp = require('lodash/fp');
 const path = require('path');
 const _ = require('lodash');
 
@@ -9,6 +10,28 @@ async function loadQuery(name) {
   return await fs.readFile(path.join(__dirname, name), 'utf-8');
 }
 
+const byTableFilter = table => x => x.pureName == table.pureName && x.schemaName == x.schemaName;
+
+function extractPrimaryKeys(table, pkColumns) {
+  const filtered = pkColumns.filter(byTableFilter(table));
+  if (filtered.length == 0) return undefined;
+  return {
+    constraintName: filtered[0].constraintName,
+    constraintType: 'primaryKey',
+    columns: filtered.map(fp.pick('columnName')),
+  };
+}
+
+function extractForeignKeys(table, fkColumns) {
+  const grouped = _.groupBy(fkColumns.filter(byTableFilter(table)), 'constraintName');
+  return _.keys(grouped).map(constraintName => ({
+    constraintName,
+    constraintType: 'foreignKey',
+    ..._.pick(fkColumns[0], ['refSchemaName', 'refTableName', 'updateAction', 'deleteAction']),
+    columns: grouped[constraintName].map(fp.pick(['columnName', 'refColumnName'])),
+  }));
+}
+
 class MsSqlAnalyser extends DatabaseAnalayser {
   constructor(pool, driver) {
     super(pool, driver);
@@ -29,6 +52,8 @@ class MsSqlAnalyser extends DatabaseAnalayser {
   async runAnalysis() {
     const tables = await this.driver.query(this.pool, await this.createQuery('tables.sql'));
     const columns = await this.driver.query(this.pool, await this.createQuery('columns.sql'));
+    const pkColumns = await this.driver.query(this.pool, await this.createQuery('primary_keys.sql'));
+    const fkColumns = await this.driver.query(this.pool, await this.createQuery('foreign_keys.sql'));
 
     this.result.tables = tables.rows.map(table => ({
       ...table,
@@ -39,6 +64,8 @@ class MsSqlAnalyser extends DatabaseAnalayser {
           notNull: !isNullable,
           autoIncrement: !!isIdentity,
         })),
+      primaryKey: extractPrimaryKeys(table, pkColumns.rows),
+      foreignKeys: extractForeignKeys(table, fkColumns.rows),
     }));
   }
 }

api/src/engines/mssql/foreign_keys.sql

@@ -0,0 +1,38 @@
+SELECT
+    schemaName = FK.TABLE_SCHEMA,
+    pureName = FK.TABLE_NAME,
+    columnName = CU.COLUMN_NAME,
+
+    refSchemaName = ISNULL(IXS.name, PK.TABLE_SCHEMA),
+    refTableName = ISNULL(IXT.name, PK.TABLE_NAME),
+    refColumnName = IXCC.name,
+
+    constraintName = C.CONSTRAINT_NAME,
+    updateAction = rc.UPDATE_RULE,
+    deleteAction = rc.DELETE_RULE,
+
+    objectId = o.object_id 
+
+FROM INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS C
+INNER JOIN INFORMATION_SCHEMA.TABLE_CONSTRAINTS FK ON C.CONSTRAINT_NAME = FK.CONSTRAINT_NAME
+
+LEFT JOIN INFORMATION_SCHEMA.TABLE_CONSTRAINTS PK ON C.UNIQUE_CONSTRAINT_NAME = PK.CONSTRAINT_NAME
+LEFT JOIN INFORMATION_SCHEMA.KEY_COLUMN_USAGE CU ON C.CONSTRAINT_NAME = CU.CONSTRAINT_NAME
+--LEFT JOIN (
+--SELECT i1.TABLE_NAME, i2.COLUMN_NAME
+--FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS i1
+--INNER JOIN INFORMATION_SCHEMA.KEY_COLUMN_USAGE i2 ON i1.CONSTRAINT_NAME = i2.CONSTRAINT_NAME
+--WHERE i1.CONSTRAINT_TYPE = 'PRIMARY KEY'
+--) PT ON PT.TABLE_NAME = PK.TABLE_NAME
+INNER JOIN INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS rc ON FK.CONSTRAINT_NAME = rc.CONSTRAINT_NAME
+
+LEFT JOIN sys.indexes IX ON IX.name = C.UNIQUE_CONSTRAINT_NAME
+LEFT JOIN sys.objects IXT ON IXT.object_id = IX.object_id
+LEFT JOIN sys.index_columns IXC ON IX.index_id = IXC.index_id and IX.object_id = IXC.object_id
+LEFT JOIN sys.columns IXCC ON IXCC.column_id = IXC.column_id AND IXCC.object_id = IXC.object_id
+LEFT JOIN sys.schemas IXS ON IXT.schema_id = IXS.schema_id
+
+inner join sys.objects o on FK.TABLE_NAME = o.name
+inner join sys.schemas s on o.schema_id = s.schema_id and FK.TABLE_SCHEMA = s.name
+
+where o.object_id =[OBJECT_ID_CONDITION]

api/src/engines/mssql/primary_keys.sql

@@ -0,0 +1,12 @@
+select o.object_id, pureName = t.Table_Name, schemaName = t.Table_Schema, columnName = c.Column_Name, constraintName=t.constraint_name from 
+    INFORMATION_SCHEMA.TABLE_CONSTRAINTS t,
+    sys.objects o,
+    sys.schemas s,
+    INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE c
+where
+    c.Constraint_Name = t.Constraint_Name
+    and t.table_name = o.name 
+    and o.schema_id = s.schema_id and t.Table_Schema = s.name
+    and c.Table_Name = t.Table_Name
+    and Constraint_Type = 'PRIMARY KEY'
+	and o.object_id =[OBJECT_ID_CONDITION]