Auto-fix room_amenity schema for global amenities

src/main/kotlin/com/android/trisolarisserver/config/RoomAmenitySchemaFix.kt

@@ -0,0 +1,79 @@
+package com.android.trisolarisserver.config
+
+import org.slf4j.LoggerFactory
+import org.springframework.boot.ApplicationArguments
+import org.springframework.boot.ApplicationRunner
+import org.springframework.jdbc.core.JdbcTemplate
+import org.springframework.stereotype.Component
+
+@Component
+class RoomAmenitySchemaFix(
+    private val jdbcTemplate: JdbcTemplate
+) : ApplicationRunner {
+
+    private val logger = LoggerFactory.getLogger(RoomAmenitySchemaFix::class.java)
+
+    override fun run(args: ApplicationArguments) {
+        val version = jdbcTemplate.queryForObject("select version()", String::class.java) ?: return
+        if (!version.contains("PostgreSQL", ignoreCase = true)) {
+            return
+        }
+
+        val propertyIdExists = jdbcTemplate.queryForObject(
+            """
+            select count(*)
+            from information_schema.columns
+            where table_name = 'room_amenity'
+              and column_name = 'property_id'
+            """.trimIndent(),
+            Int::class.java
+        ) ?: 0
+
+        if (propertyIdExists > 0) {
+            val isNullable = jdbcTemplate.queryForObject(
+                """
+                select is_nullable
+                from information_schema.columns
+                where table_name = 'room_amenity'
+                  and column_name = 'property_id'
+                """.trimIndent(),
+                String::class.java
+            )
+            if (isNullable == "NO") {
+                logger.info("Altering room_amenity.property_id to be nullable")
+                jdbcTemplate.execute("alter table room_amenity alter column property_id drop not null")
+            }
+        }
+
+        val uniqueConstraints = jdbcTemplate.query(
+            """
+            select tc.constraint_name, array_agg(kcu.column_name order by kcu.column_name) as columns
+            from information_schema.table_constraints tc
+            join information_schema.key_column_usage kcu
+              on tc.constraint_name = kcu.constraint_name
+             and tc.table_name = kcu.table_name
+            where tc.table_name = 'room_amenity'
+              and tc.constraint_type = 'UNIQUE'
+            group by tc.constraint_name
+            """.trimIndent()
+        ) { rs, _ ->
+            rs.getString("constraint_name") to (rs.getArray("columns").array as Array<*>).map { it.toString() }
+        }
+
+        var hasNameUnique = false
+        for ((name, cols) in uniqueConstraints) {
+            if (cols.size == 1 && cols[0] == "name") {
+                hasNameUnique = true
+            }
+            if (cols.size == 2 && cols.contains("property_id") && cols.contains("name")) {
+                logger.info("Dropping old unique constraint {} on room_amenity(property_id, name)", name)
+                jdbcTemplate.execute("""alter table room_amenity drop constraint "$name"""")
+            }
+        }
+
+        if (!hasNameUnique) {
+            logger.info("Adding unique constraint on room_amenity(name)")
+            jdbcTemplate.execute("alter table room_amenity add constraint room_amenity_name_key unique (name)")
+        }
+    }
+}