feat(WHE-29): add v1 accounting schema migration and seed fixtures

2026-04-01 03:14:02 +00:00 · 2026-03-05 04:13:00 +04:00
parent 27205bc90b
commit b3ae1a51e4
10 changed files with 1841 additions and 2 deletions
--- a/packages/db/src/seed.ts
+++ b/packages/db/src/seed.ts
@@ -0,0 +1,226 @@
+import { and, eq } from 'drizzle-orm'
+import { drizzle } from 'drizzle-orm/postgres-js'
+import postgres from 'postgres'
+
+import {
+  billingCycles,
+  households,
+  members,
+  presenceOverrides,
+  processedBotMessages,
+  purchaseEntries,
+  rentRules,
+  settlementLines,
+  settlements,
+  utilityBills
+} from './schema'
+
+const databaseUrl = process.env.DATABASE_URL
+if (!databaseUrl) {
+  throw new Error('DATABASE_URL is required for db seed')
+}
+
+const queryClient = postgres(databaseUrl, {
+  prepare: false,
+  max: 2
+})
+
+const db = drizzle(queryClient)
+
+const FIXTURE_IDS = {
+  household: '11111111-1111-4111-8111-111111111111',
+  cycle: '22222222-2222-4222-8222-222222222222',
+  memberAlice: '33333333-3333-4333-8333-333333333331',
+  memberBob: '33333333-3333-4333-8333-333333333332',
+  memberCarol: '33333333-3333-4333-8333-333333333333',
+  settlement: '44444444-4444-4444-8444-444444444444'
+} as const
+
+async function seed(): Promise<void> {
+  await db
+    .insert(households)
+    .values({
+      id: FIXTURE_IDS.household,
+      name: 'Kojori Demo Household'
+    })
+    .onConflictDoNothing()
+
+  await db
+    .insert(members)
+    .values([
+      {
+        id: FIXTURE_IDS.memberAlice,
+        householdId: FIXTURE_IDS.household,
+        telegramUserId: '10001',
+        displayName: 'Alice',
+        isAdmin: 1
+      },
+      {
+        id: FIXTURE_IDS.memberBob,
+        householdId: FIXTURE_IDS.household,
+        telegramUserId: '10002',
+        displayName: 'Bob',
+        isAdmin: 0
+      },
+      {
+        id: FIXTURE_IDS.memberCarol,
+        householdId: FIXTURE_IDS.household,
+        telegramUserId: '10003',
+        displayName: 'Carol',
+        isAdmin: 0
+      }
+    ])
+    .onConflictDoNothing()
+
+  await db
+    .insert(billingCycles)
+    .values({
+      id: FIXTURE_IDS.cycle,
+      householdId: FIXTURE_IDS.household,
+      period: '2026-03',
+      currency: 'USD'
+    })
+    .onConflictDoNothing()
+
+  await db
+    .insert(rentRules)
+    .values({
+      householdId: FIXTURE_IDS.household,
+      amountMinor: 70000n,
+      currency: 'USD',
+      effectiveFromPeriod: '2026-03'
+    })
+    .onConflictDoNothing()
+
+  await db
+    .insert(utilityBills)
+    .values({
+      householdId: FIXTURE_IDS.household,
+      cycleId: FIXTURE_IDS.cycle,
+      billName: 'Electricity',
+      amountMinor: 12000n,
+      currency: 'USD',
+      source: 'manual',
+      createdByMemberId: FIXTURE_IDS.memberAlice
+    })
+    .onConflictDoNothing()
+
+  await db
+    .insert(presenceOverrides)
+    .values([
+      {
+        cycleId: FIXTURE_IDS.cycle,
+        memberId: FIXTURE_IDS.memberAlice,
+        utilityDays: 31,
+        reason: 'full month'
+      },
+      {
+        cycleId: FIXTURE_IDS.cycle,
+        memberId: FIXTURE_IDS.memberBob,
+        utilityDays: 31,
+        reason: 'full month'
+      },
+      {
+        cycleId: FIXTURE_IDS.cycle,
+        memberId: FIXTURE_IDS.memberCarol,
+        utilityDays: 20,
+        reason: 'partial month'
+      }
+    ])
+    .onConflictDoNothing()
+
+  await db
+    .insert(purchaseEntries)
+    .values({
+      householdId: FIXTURE_IDS.household,
+      cycleId: FIXTURE_IDS.cycle,
+      payerMemberId: FIXTURE_IDS.memberAlice,
+      amountMinor: 3000n,
+      currency: 'USD',
+      rawText: 'Bought toilet paper 30 gel',
+      normalizedText: 'bought toilet paper 30 gel',
+      parserMode: 'rules',
+      parserConfidence: 93,
+      telegramChatId: '-100householdchat',
+      telegramMessageId: '501',
+      telegramThreadId: 'general-buys'
+    })
+    .onConflictDoNothing()
+
+  await db
+    .insert(processedBotMessages)
+    .values({
+      householdId: FIXTURE_IDS.household,
+      source: 'telegram',
+      sourceMessageKey: 'chat:-100householdchat:message:501',
+      payloadHash: 'demo-hash'
+    })
+    .onConflictDoNothing()
+
+  await db
+    .insert(settlements)
+    .values({
+      id: FIXTURE_IDS.settlement,
+      householdId: FIXTURE_IDS.household,
+      cycleId: FIXTURE_IDS.cycle,
+      inputHash: 'demo-settlement-hash',
+      totalDueMinor: 82000n,
+      currency: 'USD'
+    })
+    .onConflictDoNothing()
+
+  await db
+    .insert(settlementLines)
+    .values([
+      {
+        settlementId: FIXTURE_IDS.settlement,
+        memberId: FIXTURE_IDS.memberAlice,
+        rentShareMinor: 23334n,
+        utilityShareMinor: 4000n,
+        purchaseOffsetMinor: -2000n,
+        netDueMinor: 25334n,
+        explanations: ['rent_share_minor=23334', 'utility_share_minor=4000']
+      },
+      {
+        settlementId: FIXTURE_IDS.settlement,
+        memberId: FIXTURE_IDS.memberBob,
+        rentShareMinor: 23333n,
+        utilityShareMinor: 4000n,
+        purchaseOffsetMinor: 1000n,
+        netDueMinor: 28333n,
+        explanations: ['rent_share_minor=23333', 'utility_share_minor=4000']
+      },
+      {
+        settlementId: FIXTURE_IDS.settlement,
+        memberId: FIXTURE_IDS.memberCarol,
+        rentShareMinor: 23333n,
+        utilityShareMinor: 4000n,
+        purchaseOffsetMinor: 1000n,
+        netDueMinor: 28333n,
+        explanations: ['rent_share_minor=23333', 'utility_share_minor=4000']
+      }
+    ])
+    .onConflictDoNothing()
+
+  const seededCycle = await db
+    .select({ period: billingCycles.period, currency: billingCycles.currency })
+    .from(billingCycles)
+    .where(
+      and(
+        eq(billingCycles.id, FIXTURE_IDS.cycle),
+        eq(billingCycles.householdId, FIXTURE_IDS.household)
+      )
+    )
+    .limit(1)
+
+  if (seededCycle.length === 0) {
+    throw new Error('Seed verification failed: billing cycle not found')
+  }
+}
+
+try {
+  await seed()
+  console.log('Seed completed')
+} finally {
+  await queryClient.end({ timeout: 5 })
+}