alter table

packages/tools/src/alterPlan.ts

@@ -248,10 +248,13 @@ export class AlterPlan {
 
       if (op.operationType == 'dropTable') {
         return [
-          ...(op.oldObject.dependencies || []).map(oldObject => ({
-            operationType: 'dropConstraint',
-            oldObject,
-          })),
+          ...(op.oldObject.dependencies || []).map(oldObject => {
+            const opRes: AlterOperation = {
+              operationType: 'dropConstraint',
+              oldObject,
+            };
+            return opRes;
+          }),
           op,
         ];
       }

packages/tools/src/nameTools.ts

@@ -1,4 +1,5 @@
-import { ColumnInfo, DatabaseInfo, DatabaseInfoObjects, TableInfo } from 'dbgate-types';
+import _ from 'lodash';
+import { ColumnInfo, ColumnReference, DatabaseInfo, DatabaseInfoObjects, TableInfo } from 'dbgate-types';
 
 export function fullNameFromString(name) {
   const m = name.match(/\[([^\]]+)\]\.\[([^\]]+)\]/);
@@ -68,3 +69,28 @@ export function makeUniqueColumnNames(res: ColumnInfo[]) {
     usedNames.add(res[i].columnName);
   }
 }
+
+function columnsConstraintName(prefix: string, table: TableInfo, columns: ColumnReference[]) {
+  return `${prefix}_${table.pureName}_${columns.map(x => x.columnName.replace(' ', '_')).join('_')}`;
+}
+
+export function fillConstraintNames(table: TableInfo) {
+  if (!table) return table;
+  const res = _.cloneDeep(table);
+  if (res.primaryKey && !res.primaryKey.constraintName) {
+    res.primaryKey.constraintName = `PK_${res.pureName}`;
+  }
+  for (const fk of res.foreignKeys) {
+    if (fk.constraintName) continue;
+    fk.constraintName = columnsConstraintName('FK', res, fk.columns);
+  }
+  for (const ix of res.indexes) {
+    if (ix.constraintName) continue;
+    ix.constraintName = columnsConstraintName('IX', res, ix.columns);
+  }
+  for (const uq of res.uniques) {
+    if (uq.constraintName) continue;
+    uq.constraintName = columnsConstraintName('UQ', res, uq.columns);
+  }
+  return res;
+}

packages/tools/src/schemaEditorTools.ts

@@ -1,6 +1,14 @@
 import uuidv1 from 'uuid/v1';
 import _omit from 'lodash/omit';
-import { ColumnInfo, ConstraintInfo, ForeignKeyInfo, PrimaryKeyInfo, TableInfo } from 'dbgate-types';
+import {
+  ColumnInfo,
+  ConstraintInfo,
+  ForeignKeyInfo,
+  IndexInfo,
+  PrimaryKeyInfo,
+  TableInfo,
+  UniqueInfo,
+} from 'dbgate-types';
 
 export interface EditorColumnInfo extends ColumnInfo {
   isPrimaryKey?: boolean;
@@ -8,7 +16,7 @@ export interface EditorColumnInfo extends ColumnInfo {
 
 export function fillEditorColumnInfo(column: ColumnInfo, table: TableInfo): EditorColumnInfo {
   return {
-    isPrimaryKey: !!(table?.primaryKey && table.primaryKey.columns.find(x => x.columnName == column.columnName)),
+    isPrimaryKey: !!table?.primaryKey?.columns?.find(x => x.columnName == column.columnName),
     dataType: column ? undefined : 'int',
     ...column,
   };
@@ -118,6 +126,26 @@ export function editorAddConstraint(table: TableInfo, constraint: ConstraintInfo
     ];
   }
 
+  if (constraint.constraintType == 'index') {
+    res.indexes = [
+      ...(res.indexes || []),
+      {
+        pairingId: uuidv1(),
+        ...constraint,
+      } as IndexInfo,
+    ];
+  }
+
+  if (constraint.constraintType == 'unique') {
+    res.uniques = [
+      ...(res.uniques || []),
+      {
+        pairingId: uuidv1(),
+        ...constraint,
+      } as UniqueInfo,
+    ];
+  }
+
   return res;
 }
 
@@ -139,6 +167,14 @@ export function editorModifyConstraint(table: TableInfo, constraint: ConstraintI
     );
   }
 
+  if (constraint.constraintType == 'index') {
+    res.indexes = table.indexes.map(fk => (fk.pairingId == constraint.pairingId ? { ...fk, ...constraint } : fk));
+  }
+
+  if (constraint.constraintType == 'unique') {
+    res.uniques = table.uniques.map(fk => (fk.pairingId == constraint.pairingId ? { ...fk, ...constraint } : fk));
+  }
+
   return res;
 }