Fix migrateDb for postgres and sqlite

1 files changed, 18 insertions, 5 deletions

diff --git a/src/main/java/net/elytrium/limboauth/LimboAuth.java b/src/main/java/net/elytrium/limboauth/LimboAuth.java
index d2c442f..c98497f 100644
--- a/src/main/java/net/elytrium/limboauth/LimboAuth.java
+++ b/src/main/java/net/elytrium/limboauth/LimboAuth.java
@@ -422,16 +422,21 @@ public class LimboAuth {
     String findSql;
     String database = Settings.IMP.DATABASE.DATABASE;
     String tableName = tableInfo.getTableName();
-    switch (Settings.IMP.DATABASE.STORAGE_TYPE) {
-      case "h2": {
+    DatabaseLibrary databaseLibrary = DatabaseLibrary.valueOf(Settings.IMP.DATABASE.STORAGE_TYPE.toUpperCase(Locale.ROOT));
+    switch (databaseLibrary) {
+      case SQLITE: {
+        findSql = "SELECT name FROM PRAGMA_TABLE_INFO('" + tableName + "')";
+        break;
+      }
+      case H2: {
         findSql = "SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = '" + tableName + "';";
         break;
       }
-      case "postgresql": {
+      case POSTGRESQL: {
         findSql = "SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_CATALOG = '" + database + "' AND TABLE_NAME = '" + tableName + "';";
         break;
       }
-      case "mysql": {
+      case MYSQL: {
         findSql = "SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = '" + database + "' AND TABLE_NAME = '" + tableName + "';";
         break;
       }
@@ -447,7 +452,15 @@ public class LimboAuth {
 
       tables.forEach(table -> {
         try {
-          StringBuilder builder = new StringBuilder("ALTER TABLE " + tableName + " ADD ");
+          StringBuilder builder = new StringBuilder("ALTER TABLE ");
+          if (databaseLibrary == DatabaseLibrary.POSTGRESQL) {
+            builder.append('"');
+          }
+          builder.append(tableName);
+          if (databaseLibrary == DatabaseLibrary.POSTGRESQL) {
+            builder.append('"');
+          }
+          builder.append(" ADD ");
           String columnDefinition = table.getColumnDefinition();
           DatabaseType databaseType = dao.getConnectionSource().getDatabaseType();
           if (columnDefinition == null) {