From 407128fd61a77bb2588f711d82974cad5ec8d35d Mon Sep 17 00:00:00 2001
From: callebtc <93376500+callebtc@users.noreply.github.com>
Date: Fri, 6 Jan 2023 11:13:25 +0100
Subject: [PATCH] fix migration for postgres

---
 lnbits/extensions/lnurlp/migrations.py | 133 +++++++++++++------------
 1 file changed, 69 insertions(+), 64 deletions(-)

diff --git a/lnbits/extensions/lnurlp/migrations.py b/lnbits/extensions/lnurlp/migrations.py
index 128f62ac..1ec85eb0 100644
--- a/lnbits/extensions/lnurlp/migrations.py
+++ b/lnbits/extensions/lnurlp/migrations.py
@@ -72,72 +72,77 @@ async def m005_webhook_headers_and_body(db):
 
 async def m006_redux(db):
     """
-    Add UUID ID's to links and migrates existing data
+    Migrate ID column type to string for UUIDs and migrate existing data
     """
-    await db.execute("ALTER TABLE lnurlp.pay_links RENAME TO pay_links_old")
-    await db.execute(
-        f"""
-        CREATE TABLE lnurlp.pay_links (
-            id TEXT PRIMARY KEY,
-            wallet TEXT NOT NULL,
-            description TEXT NOT NULL,
-            min {db.big_int} NOT NULL,
-            max {db.big_int},
-            currency TEXT,
-            fiat_base_multiplier INTEGER DEFAULT 1,
-            served_meta INTEGER NOT NULL,
-            served_pr INTEGER NOT NULL,
-            webhook_url TEXT,
-            success_text TEXT,
-            success_url TEXT,
-            comment_chars INTEGER DEFAULT 0,
-            webhook_headers TEXT,
-            webhook_body TEXT
-            );
-        """
-    )
-
-    for row in [
-        list(row) for row in await db.fetchall("SELECT * FROM lnurlp.pay_links_old")
-    ]:
+    # we can simply change the column type for postgres
+    if db.type != "SQLITE":
+        await db.execute("ALTER TABLE lnurlp.pay_links ALTER COLUMN id TYPE TEXT;")
+    else:
+        # but we have to do this for sqlite
+        await db.execute("ALTER TABLE lnurlp.pay_links RENAME TO pay_links_old")
         await db.execute(
+            f"""
+            CREATE TABLE lnurlp.pay_links (
+                id TEXT PRIMARY KEY,
+                wallet TEXT NOT NULL,
+                description TEXT NOT NULL,
+                min {db.big_int} NOT NULL,
+                max {db.big_int},
+                currency TEXT,
+                fiat_base_multiplier INTEGER DEFAULT 1,
+                served_meta INTEGER NOT NULL,
+                served_pr INTEGER NOT NULL,
+                webhook_url TEXT,
+                success_text TEXT,
+                success_url TEXT,
+                comment_chars INTEGER DEFAULT 0,
+                webhook_headers TEXT,
+                webhook_body TEXT
+                );
             """
-            INSERT INTO lnurlp.pay_links (
-                id,
-                wallet,
-                description,
-                min,
-                served_meta,
-                served_pr,
-                webhook_url,
-                success_text,
-                success_url,
-                currency,
-                comment_chars,
-                max,
-                fiat_base_multiplier,
-                webhook_headers,
-                webhook_body
-            )
-            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-            """,
-            (
-                row[0],
-                row[1],
-                row[2],
-                row[3],
-                row[4],
-                row[5],
-                row[6],
-                row[7],
-                row[8],
-                row[9],
-                row[10],
-                row[11],
-                row[12],
-                row[13],
-                row[14],
-            ),
         )
 
-    await db.execute("DROP TABLE lnurlp.pay_links_old")
+        for row in [
+            list(row) for row in await db.fetchall("SELECT * FROM lnurlp.pay_links_old")
+        ]:
+            await db.execute(
+                """
+                INSERT INTO lnurlp.pay_links (
+                    id,
+                    wallet,
+                    description,
+                    min,
+                    served_meta,
+                    served_pr,
+                    webhook_url,
+                    success_text,
+                    success_url,
+                    currency,
+                    comment_chars,
+                    max,
+                    fiat_base_multiplier,
+                    webhook_headers,
+                    webhook_body
+                )
+                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                """,
+                (
+                    row[0],
+                    row[1],
+                    row[2],
+                    row[3],
+                    row[4],
+                    row[5],
+                    row[6],
+                    row[7],
+                    row[8],
+                    row[9],
+                    row[10],
+                    row[11],
+                    row[12],
+                    row[13],
+                    row[14],
+                ),
+            )
+
+        await db.execute("DROP TABLE lnurlp.pay_links_old")