From 73eb5226bfc06ff744e0e1f0316ffe12a99af977 Mon Sep 17 00:00:00 2001
From: Thomas Gehrsitz <158038003+tgehrsitz-ite@users.noreply.github.com>
Date: Thu, 16 May 2024 14:42:53 +0200
Subject: [PATCH] tgz: Implemented ColumnSnapshotGenerator to handle numeric
 boolean default values in snapshots (#191)

* tgz: Implemented ColumnSnapshotGenerator to handle numeric boolean default values in snapshots

* ColumnSnapshotGeneratorHana, improved priority evaluation, restricting to HanaDatabase
---
 .../snapshot/ColumnSnapshotGeneratorHana.java | 47 +++++++++++++++++++
 .../liquibase.snapshot.SnapshotGenerator      |  1 +
 2 files changed, 48 insertions(+)
 create mode 100644 src/main/java/liquibase/ext/hana/snapshot/ColumnSnapshotGeneratorHana.java

diff --git a/src/main/java/liquibase/ext/hana/snapshot/ColumnSnapshotGeneratorHana.java b/src/main/java/liquibase/ext/hana/snapshot/ColumnSnapshotGeneratorHana.java
new file mode 100644
index 0000000..db8ecbd
--- /dev/null
+++ b/src/main/java/liquibase/ext/hana/snapshot/ColumnSnapshotGeneratorHana.java
@@ -0,0 +1,47 @@
+package liquibase.ext.hana.snapshot;
+
+import liquibase.database.Database;
+import liquibase.ext.hana.HanaDatabase;
+import liquibase.snapshot.CachedRow;
+import liquibase.snapshot.SnapshotGenerator;
+import liquibase.snapshot.jvm.ColumnSnapshotGenerator;
+import liquibase.statement.DatabaseFunction;
+import liquibase.structure.DatabaseObject;
+import liquibase.structure.core.Column;
+
+public class ColumnSnapshotGeneratorHana extends ColumnSnapshotGenerator {
+
+    @Override
+    public int getPriority(Class<? extends DatabaseObject> objectType, Database database) {
+        if (!(database instanceof HanaDatabase)) {
+            return PRIORITY_NONE;
+        }
+        int priority = super.getPriority(objectType, database);
+        if (priority > PRIORITY_NONE) {
+            priority += PRIORITY_DATABASE;
+        }
+        return priority;
+    }
+
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{ColumnSnapshotGenerator.class};
+    }
+
+    @Override
+    protected Object readDefaultValue(CachedRow columnMetadataResultSet, Column columnInfo, Database database) {
+        Object defaultValue = super.readDefaultValue(columnMetadataResultSet, columnInfo, database);
+        String typeName = columnMetadataResultSet.getString("TYPE_NAME").toUpperCase();
+        if ("BOOLEAN".equals(typeName)) {
+            if (defaultValue instanceof DatabaseFunction) {
+                String defaultAsString = ((DatabaseFunction) defaultValue).getValue();
+                if ("0".equals(defaultAsString)) {
+                    return "FALSE";
+                } else if ("1".equals(defaultAsString)) {
+                    return "TRUE";
+                }
+            }
+        }
+        return defaultValue;
+    }
+}
diff --git a/src/main/resources/META-INF/services/liquibase.snapshot.SnapshotGenerator b/src/main/resources/META-INF/services/liquibase.snapshot.SnapshotGenerator
index 42349ab..612dffb 100644
--- a/src/main/resources/META-INF/services/liquibase.snapshot.SnapshotGenerator
+++ b/src/main/resources/META-INF/services/liquibase.snapshot.SnapshotGenerator
@@ -1,2 +1,3 @@
 liquibase.ext.hana.snapshot.SequenceSnapshotGeneratorHana
 liquibase.ext.hana.snapshot.UniqueConstraintSnapshotGeneratorHana
+liquibase.ext.hana.snapshot.ColumnSnapshotGeneratorHana