From 47c00d7eb0bd9d5875bf15bc4f3a097a7a9dafda Mon Sep 17 00:00:00 2001
From: "SPRINX0\\prochazka" <jan.prochazka@sprinx.com>
Date: Fri, 8 Nov 2024 15:36:07 +0100
Subject: [PATCH] alter plan utility functions

---
 packages/tools/src/alterPlan.ts       |  2 +-
 packages/tools/src/computeDiffRows.ts | 64 ++++++++++++++++++++++++++-
 2 files changed, 63 insertions(+), 3 deletions(-)

diff --git a/packages/tools/src/alterPlan.ts b/packages/tools/src/alterPlan.ts
index 9d7aafdb9..0351df84a 100644
--- a/packages/tools/src/alterPlan.ts
+++ b/packages/tools/src/alterPlan.ts
@@ -110,7 +110,7 @@ interface AlterOperation_SetTableOption {
   optionValue: string;
 }
 
-type AlterOperation =
+export type AlterOperation =
   | AlterOperation_CreateColumn
   | AlterOperation_ChangeColumn
   | AlterOperation_DropColumn
diff --git a/packages/tools/src/computeDiffRows.ts b/packages/tools/src/computeDiffRows.ts
index a1fa836ac..edde4cfa0 100644
--- a/packages/tools/src/computeDiffRows.ts
+++ b/packages/tools/src/computeDiffRows.ts
@@ -1,6 +1,14 @@
-import { DbDiffOptions, testEqualColumns, testEqualTables, testEqualSqlObjects } from './diffTools';
-import type { DatabaseInfo, EngineDriver, SqlObjectInfo, TableInfo } from 'dbgate-types';
+import {
+  DbDiffOptions,
+  testEqualColumns,
+  testEqualTables,
+  testEqualSqlObjects,
+  createAlterDatabasePlan,
+} from './diffTools';
+import type { DatabaseInfo, EngineDriver, NamedObjectInfo, SqlObjectInfo, TableInfo } from 'dbgate-types';
 import _ from 'lodash';
+import { extendDatabaseInfo } from './structureTools';
+import { AlterOperation, runAlterOperation } from './alterPlan';
 
 export function computeDiffRowsCore(sourceList, targetList, testEqual) {
   const res = [];
@@ -124,6 +132,58 @@ export function computeTableDiffColumns(
   }));
 }
 
+export interface DiffOperationItemDisplay {
+  operationType: string;
+  name: string;
+  sqlScript: string;
+  identifier?: string;
+}
+
+function getOperationDisplay(operation: AlterOperation, driver: EngineDriver, index: number): DiffOperationItemDisplay {
+  const op = operation as any;
+  const name =
+    op?.newName ??
+    op?.newObject?.columnName ??
+    op?.newObject?.constraintName ??
+    op?.newObject?.pureName ??
+    op?.oldObject?.columnName ??
+    op?.oldObject?.constraintName ??
+    op?.oldObject?.pureName ??
+    op?.table?.pureName ??
+    op?.object?.columnName ??
+    op?.object?.constraintName ??
+    op?.object?.pureName;
+
+  const dmp = driver.createDumper();
+  runAlterOperation(operation, dmp);
+
+  return {
+    operationType: operation.operationType,
+    name,
+    sqlScript: dmp.s,
+    identifier: `${name}-${index}-${operation.operationType}`,
+  };
+}
+
+export function computeObjectDiffOperations(
+  sourceObject: { objectTypeField: string },
+  targetObject: { objectTypeField: string },
+  sourceDb: DatabaseInfo,
+  targetDb: DatabaseInfo,
+  opts: DbDiffOptions,
+  driver: EngineDriver
+): DiffOperationItemDisplay[] {
+  if (!driver) return [];
+  const srcdb = sourceObject
+    ? extendDatabaseInfo({ [sourceObject.objectTypeField]: [sourceObject] } as unknown as DatabaseInfo)
+    : extendDatabaseInfo({} as unknown as DatabaseInfo);
+  const dstdb = targetObject
+    ? extendDatabaseInfo({ [targetObject.objectTypeField]: [targetObject] } as unknown as DatabaseInfo)
+    : extendDatabaseInfo({} as unknown as DatabaseInfo);
+  const plan = createAlterDatabasePlan(dstdb, srcdb, opts, targetDb, sourceDb, driver);
+  return plan.operations.map((item, index) => getOperationDisplay(item, driver, index));
+}
+
 export function getCreateObjectScript(obj: TableInfo | SqlObjectInfo, driver: EngineDriver) {
   if (!obj || !driver) return '';
   if (obj.objectTypeField == 'tables') {