From 12b1cdb52a9d2723c319083b45a4e9a21896b3b4 Mon Sep 17 00:00:00 2001
From: Sebastien Castiel <sebastien@castiel.me>
Date: Wed, 6 Dec 2023 17:02:37 -0500
Subject: Balances

---
 src/lib/balances.ts | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 src/lib/balances.ts

(limited to 'src/lib/balances.ts')

diff --git a/src/lib/balances.ts b/src/lib/balances.ts
new file mode 100644
index 0000000..fa00640
--- /dev/null
+++ b/src/lib/balances.ts
@@ -0,0 +1,42 @@
+import { getGroupExpenses } from '@/lib/api'
+import { Participant } from '@prisma/client'
+
+export type Balances = Record<
+  Participant['id'],
+  { paid: number; paidFor: number; total: number }
+>
+
+export function getBalances(
+  expenses: NonNullable<Awaited<ReturnType<typeof getGroupExpenses>>>,
+): Balances {
+  const balances: Balances = {}
+
+  for (const expense of expenses) {
+    const paidBy = expense.paidById
+    const paidFors = expense.paidFor.map((p) => p.participantId)
+
+    if (!balances[paidBy]) balances[paidBy] = { paid: 0, paidFor: 0, total: 0 }
+    balances[paidBy].paid += expense.amount
+    balances[paidBy].total += expense.amount
+    paidFors.forEach((paidFor, index) => {
+      if (!balances[paidFor])
+        balances[paidFor] = { paid: 0, paidFor: 0, total: 0 }
+
+      const dividedAmount = divide(
+        expense.amount,
+        paidFors.length,
+        index === paidFors.length - 1,
+      )
+      balances[paidFor].paidFor += dividedAmount
+      balances[paidFor].total -= dividedAmount
+    })
+  }
+
+  return balances
+}
+
+function divide(total: number, count: number, isLast: boolean): number {
+  if (!isLast) return Math.floor((total * 100) / count) / 100
+
+  return total - divide(total, count, false) * (count - 1)
+}
-- 
cgit v1.3