From 605b2b86d83ca23456bf1d6da060742e45da85ff Mon Sep 17 00:00:00 2001
From: William Luke <williamluke4@gmail.com>
Date: Tue, 28 Nov 2023 18:42:31 +0300
Subject: [PATCH] fix: missing vpa mutation

---
 src/server/api/routers/me.ts   | 16 ++++++++++++++--
 src/server/api/routers/user.ts |  8 +-------
 2 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/src/server/api/routers/me.ts b/src/server/api/routers/me.ts
index d81c7ed..2c3425d 100644
--- a/src/server/api/routers/me.ts
+++ b/src/server/api/routers/me.ts
@@ -38,7 +38,7 @@ export const meRouter = createTRPCRouter({
 
   update: authenticatedProcedure
     .input(UserProfileFormSchema)
-    .mutation(async ({ ctx, input: { vpa: _vpa, ...pi } }) => {
+    .mutation(async ({ ctx, input: { vpa, ...pi } }) => {
       const address = ctx.session?.user?.account.blockchain_address;
       if (!address) throw new Error("No user found");
       const user = await ctx.kysely
@@ -54,7 +54,19 @@ export const meRouter = createTRPCRouter({
         .set(pi)
         .where("user_identifier", "=", user.userId)
         .execute();
-      
+      if (vpa && user.vpa) {
+        await ctx.kysely
+          .updateTable("vpa")
+          .set({ vpa })
+          .where("linked_account", "=", user.accountId)
+          .execute();
+      }
+      if (vpa && !user.vpa) {
+        await ctx.kysely
+          .insertInto("vpa")
+          .values({ vpa, linked_account: user.accountId })
+          .execute();
+      }
       return true;
     }),
   vouchers: authenticatedProcedure.query(async ({ ctx }) => {
diff --git a/src/server/api/routers/user.ts b/src/server/api/routers/user.ts
index 4f9f698..979acb1 100644
--- a/src/server/api/routers/user.ts
+++ b/src/server/api/routers/user.ts
@@ -69,13 +69,7 @@ export const userRouter = createTRPCRouter({
           .set(pi)
           .where("user_identifier", "=", user.userId)
           .execute();
-        if (vpa) {
-          await ctx.kysely
-            .updateTable("vpa")
-            .set({ vpa })
-            .where("linked_account", "=", user.accountId)
-            .execute();
-        }
+
         return true;
       }
     ),