WIP: Wallet migration

Author: ekzyis

prisma/migrations/20250502042253_wallet_v2/migration.sql

@@ -501,3 +501,104 @@ CREATE TRIGGER wallet_check_support
     BEFORE INSERT OR UPDATE ON "WalletRecvLNDGRPC"
     FOR EACH ROW
     EXECUTE FUNCTION wallet_check_support('RECEIVE');
+
+CREATE OR REPLACE FUNCTION wallet_v2_migration()
+RETURNS void AS
+$$
+DECLARE
+    row RECORD;
+BEGIN
+    -- the loops can create multiple wallets for the same user because for some reason, users already have multiple wallets of the same type
+    -- ( see `SELECT "userId", type, COUNT(1) FROM "Wallet" GROUP BY "userId", type ORDER BY count DESC` )
+    -- but that's no longer a problem because we can support multiple LNbits, NWC, lightning addresses etc. now
+    FOR row IN
+        SELECT w1.*, w2."userId", w2."priority", w2."enabled"
+        FROM "WalletLNbits" w1
+        JOIN "Wallet" w2 ON w1."walletId" = w2.id
+    LOOP
+        DECLARE
+            userWalletId INT;
+        BEGIN
+            INSERT INTO "UserWallet" ("userId", "walletId", "priority", "enabled")
+            SELECT row."userId", w.id, row."priority", row."enabled"
+            FROM "WalletV2" w
+            WHERE w.name = 'LNBITS'
+            RETURNING id INTO userWalletId;
+
+            IF row."adminKeyId" IS NOT NULL THEN
+                INSERT INTO "WalletSendLNbits" ("walletId", "url", "apiKeyId")
+                VALUES (userWalletId, row."url", row."adminKeyId");
+            END IF;
+
+            IF row."invoiceKey" IS NOT NULL THEN
+                INSERT INTO "WalletRecvLNbits" ("walletId", "url", "apiKey")
+                VALUES (userWalletId, row."url", row."invoiceKey");
+            END IF;
+        END;
+    END LOOP;
+
+    FOR row IN
+        SELECT w1.*, w2."userId", w2."userId", w2."priority", w2."enabled"
+        FROM "WalletPhoenixd" w1
+        JOIN "Wallet" w2 ON w1."walletId" = w2.id
+    LOOP
+        DECLARE
+            userWalletId INT;
+        BEGIN
+            INSERT INTO "UserWallet" ("userId", "walletId", "priority", "enabled")
+            SELECT row."userId", w.id, row."priority", row."enabled"
+            FROM "WalletV2" w
+            WHERE w.name = 'PHOENIXD'
+            RETURNING id INTO userWalletId;
+
+            IF row."primaryPasswordId" IS NOT NULL THEN
+                INSERT INTO "WalletSendPhoenixd" ("walletId", "url", "apiKeyId")
+                VALUES (userWalletId, row."url", row."primaryPasswordId");
+            END IF;
+
+            IF row."secondaryPassword" IS NOT NULL THEN
+                INSERT INTO "WalletRecvPhoenixd" ("walletId", "url", "apiKey")
+                VALUES (userWalletId, row."url", row."secondaryPassword");
+            END IF;
+        END;
+    END LOOP;
+
+    FOR row IN
+        SELECT w1.*, w2."userId", w2."userId", w2."priority", w2."enabled"
+        FROM "WalletBlink" w1
+        JOIN "Wallet" w2 ON w1."walletId" = w2.id
+    LOOP
+        DECLARE
+            userWalletId INT;
+        BEGIN
+            INSERT INTO "UserWallet" ("userId", "walletId", "priority", "enabled")
+            SELECT row."userId", w.id, row."priority", row."enabled"
+            FROM "WalletV2" w
+            WHERE w.name = 'BLINK'
+            RETURNING id INTO userWalletId;
+
+            IF row."apiKeyId" IS NOT NULL AND row."currencyId" IS NOT NULL THEN
+                INSERT INTO "WalletSendBlink" ("walletId", "apiKeyId", "currencyId")
+                VALUES (userWalletId, row."apiKeyId", row."currencyId");
+            END IF;
+
+            IF row."apiKeyRecv" IS NOT NULL AND row."currencyRecv" IS NOT NULL THEN
+                INSERT INTO "WalletRecvBlink" ("walletId", "currency", "apiKey")
+                VALUES (userWalletId, row."currencyRecv", row."apiKeyRecv");
+            END IF;
+        END;
+    END LOOP;
+
+    -- TODO: migrate remaining wallets
+    --   - WalletWebLN: 'ALBY', WalletSendWebLN
+    --   - WalletLND: 'LND', WalletRecvLND
+    --   - WalletCLN: 'CLN', WalletRecvCLN
+    --   - WalletLNC: 'LND', WalletSendLNC
+    --   - WalletLightningAddress: '<WALLET_BY_DOMAIN>', WalletRecvLightningAddress
+    --   - WalletNWC: '<WALLET_BY_RELAY>', WalletSendNWC, WalletRecvNWC
+END;
+$$ LANGUAGE plpgsql;
+
+SELECT wallet_v2_migration();
+
+DROP FUNCTION wallet_v2_migration();