From 1551d399784a147f5d3d64ec912c63d6d2a956bc Mon Sep 17 00:00:00 2001
From: Max Koon <22125083+k2on@users.noreply.github.com>
Date: Wed, 15 Oct 2025 12:04:45 -0400
Subject: [PATCH] feat: update multiple balances

---
 api/src/zero.ts                | 36 ++++++++++++--
 app/index.tsx                  | 13 ++++-
 shared/src/db/schema/public.ts | 11 +++++
 shared/src/mutators.ts         |  1 +
 shared/src/queries.ts          |  5 ++
 shared/src/zero-schema.gen.ts  | 87 ++++++++++++++++++++++++++++++++++
 6 files changed, 147 insertions(+), 6 deletions(-)

diff --git a/api/src/zero.ts b/api/src/zero.ts
index d1eca5f..77e7078 100644
--- a/api/src/zero.ts
+++ b/api/src/zero.ts
@@ -23,8 +23,8 @@ import { getHono } from "./hono";
 import { Configuration, CountryCode, PlaidApi, PlaidEnvironments, Products } from "plaid";
 import { randomUUID } from "crypto";
 import { db } from "./db";
-import { plaidAccessTokens, plaidLink, transaction } from "@money/shared/db";
-import { eq } from "drizzle-orm";
+import { balance, plaidAccessTokens, plaidLink, transaction } from "@money/shared/db";
+import { asc, desc, eq, sql } from "drizzle-orm";
 
 
 const configuration = new Configuration({
@@ -118,11 +118,39 @@ const createMutators = (authData: AuthData | null) => {
             id: randomUUID(),
             user_id: authData.user.id,
             name: t.name,
-            amount: t.amount,
+            amount: t.amount.toString(),
           });
         }
 
-      }
+      },
+      async updateBalences() {
+        isLoggedIn(authData);
+        const accounts = await db.query.plaidAccessTokens.findMany({
+          where: eq(plaidAccessTokens.userId, authData.user.id),
+        });
+        if (accounts.length == 0) {
+          console.error("No accounts");
+          return;
+        }
+
+        for (const account of accounts) {
+          const { data } = await plaidClient.accountsBalanceGet({
+            access_token: account.token
+          });
+          await db.insert(balance).values(data.accounts.map(bal => ({
+            id: randomUUID(),
+            user_id: authData.user.id,
+            plaid_id: bal.account_id,
+            avaliable: bal.balances.available as any,
+            current: bal.balances.current as any,
+            name: bal.name,
+          }))).onConflictDoUpdate({
+            target: balance.plaid_id,
+            set: { current: sql.raw(`excluded.${balance.current.name}`), avaliable: sql.raw(`excluded.${balance.avaliable.name}`) }
+          })
+        }
+
+      },
     }
   } as const satisfies Mutators;
 }
diff --git a/app/index.tsx b/app/index.tsx
index 6c3aaab..819df8b 100644
--- a/app/index.tsx
+++ b/app/index.tsx
@@ -1,4 +1,3 @@
-import { SafeAreaView } from 'react-native-safe-area-context';
 import { authClient } from '@/lib/auth-client';
 import { Button, Linking, ScrollView, Text, View } from 'react-native';
 import { useQuery, useZero } from "@rocicorp/zero/react";
@@ -11,6 +10,7 @@ export default function HomeScreen() {
   const z = useZero<Schema, Mutators>();
   const [plaidLink] = useQuery(queries.getPlaidLink(session)); 
   const [transactions] = useQuery(queries.allTransactions(session));
+  const [balances] = useQuery(queries.getBalances(session));
 
   const [idx, setIdx] = useState(0);
 
@@ -28,7 +28,6 @@ export default function HomeScreen() {
 
     window.addEventListener("keydown", handleKeyDown);
 
-    // Cleanup listener on unmount
     return () => {
       window.removeEventListener("keydown", handleKeyDown);
     };
@@ -39,6 +38,16 @@ export default function HomeScreen() {
       {plaidLink && <Button onPress={() => {
         z.mutate.link.updateTransactions();
       }} title="Update transactions" />}
+      {plaidLink && <Button onPress={() => {
+        z.mutate.link.updateBalences();
+      }} title="Update bal" />}
+
+      <View style={{ backgroundColor: 'red' }}>
+        {balances.map(bal => <View>
+          <Text style={{ fontFamily: 'mono', color: 'white' }}>{bal.name}: {bal.current} ({bal.avaliable})</Text>
+        </View>)}
+      </View>
+
       {transactions.map((t, i) => <View style={{ backgroundColor: i == idx ? 'black' : undefined }} key={t.id}>
         <Text style={{ fontFamily: 'mono', color: i == idx ? 'white' : undefined }}>{t.name} {t.amount}</Text>
       </View>)}
diff --git a/shared/src/db/schema/public.ts b/shared/src/db/schema/public.ts
index 545130d..4feb4a7 100644
--- a/shared/src/db/schema/public.ts
+++ b/shared/src/db/schema/public.ts
@@ -1,3 +1,4 @@
+import { definePermissions } from "@rocicorp/zero";
 import { pgTable, text, boolean, timestamp, uniqueIndex, decimal } from "drizzle-orm/pg-core";
 
 export const users = pgTable(
@@ -29,3 +30,13 @@ export const plaidLink = pgTable("plaidLink", {
   createdAt: timestamp("created_at").notNull().defaultNow(),
 });
 
+export const balance = pgTable("balance", {
+  id: text("id").primaryKey(),
+  user_id: text("userId").notNull(),
+  plaid_id: text("plaidId").notNull().unique(),
+  avaliable: decimal("avaliable").notNull(),
+  current: decimal("current").notNull(),
+  name: text("name").notNull(),
+  createdAt: timestamp("created_at").notNull().defaultNow(),
+  updatedAt: timestamp("updated_at").notNull().defaultNow(),
+})
diff --git a/shared/src/mutators.ts b/shared/src/mutators.ts
index 9f70849..f902909 100644
--- a/shared/src/mutators.ts
+++ b/shared/src/mutators.ts
@@ -28,6 +28,7 @@ export function createMutators(authData: AuthData | null) {
       async create() {},
       async get(tx: Tx, { link_token }: { link_token: string }) {},
       async updateTransactions() {},
+      async updateBalences() {},
     }
   } as const;
 }
diff --git a/shared/src/queries.ts b/shared/src/queries.ts
index 43a7609..8623545 100644
--- a/shared/src/queries.ts
+++ b/shared/src/queries.ts
@@ -25,4 +25,9 @@ export const queries = {
       .orderBy('createdAt', 'desc')
       .one();
   }),
+  getBalances: syncedQueryWithContext('getBalances', z.tuple([]), (authData: AuthData | null) => {
+    isLoggedIn(authData);
+    return builder.balance
+      .where('user_id', '=', authData.user.id);
+  })
 };
diff --git a/shared/src/zero-schema.gen.ts b/shared/src/zero-schema.gen.ts
index b08b311..de5b5a3 100644
--- a/shared/src/zero-schema.gen.ts
+++ b/shared/src/zero-schema.gen.ts
@@ -21,6 +21,88 @@ type ZeroSchema = DrizzleToZeroSchema<typeof drizzleSchema>;
  */
 export const schema = {
   tables: {
+    balance: {
+      name: "balance",
+      columns: {
+        id: {
+          type: "string",
+          optional: false,
+          customType: null as unknown as ZeroCustomType<
+            ZeroSchema,
+            "balance",
+            "id"
+          >,
+        },
+        user_id: {
+          type: "string",
+          optional: false,
+          customType: null as unknown as ZeroCustomType<
+            ZeroSchema,
+            "balance",
+            "user_id"
+          >,
+          serverName: "userId",
+        },
+        plaid_id: {
+          type: "string",
+          optional: false,
+          customType: null as unknown as ZeroCustomType<
+            ZeroSchema,
+            "balance",
+            "plaid_id"
+          >,
+          serverName: "plaidId",
+        },
+        avaliable: {
+          type: "number",
+          optional: false,
+          customType: null as unknown as ZeroCustomType<
+            ZeroSchema,
+            "balance",
+            "avaliable"
+          >,
+        },
+        current: {
+          type: "number",
+          optional: false,
+          customType: null as unknown as ZeroCustomType<
+            ZeroSchema,
+            "balance",
+            "current"
+          >,
+        },
+        name: {
+          type: "string",
+          optional: false,
+          customType: null as unknown as ZeroCustomType<
+            ZeroSchema,
+            "balance",
+            "name"
+          >,
+        },
+        createdAt: {
+          type: "number",
+          optional: true,
+          customType: null as unknown as ZeroCustomType<
+            ZeroSchema,
+            "balance",
+            "createdAt"
+          >,
+          serverName: "created_at",
+        },
+        updatedAt: {
+          type: "number",
+          optional: true,
+          customType: null as unknown as ZeroCustomType<
+            ZeroSchema,
+            "balance",
+            "updatedAt"
+          >,
+          serverName: "updated_at",
+        },
+      },
+      primaryKey: ["id"],
+    },
     plaidLink: {
       name: "plaidLink",
       columns: {
@@ -199,6 +281,11 @@ export const schema = {
  * This type is auto-generated from your Drizzle schema definition.
  */
 export type Schema = typeof schema;
+/**
+ * Represents a row from the "balance" table.
+ * This type is auto-generated from your Drizzle schema definition.
+ */
+export type Balance = Row<Schema["tables"]["balance"]>;
 /**
  * Represents a row from the "plaidLink" table.
  * This type is auto-generated from your Drizzle schema definition.