view models

apps/website/lib/view-models/WalletViewModel.ts

@@ -1,19 +1,37 @@
-import { WalletDto, WalletTransactionDto } from '../dtos';
+import { WalletDto } from '../types/generated/WalletDto';
 import { WalletTransactionViewModel } from './WalletTransactionViewModel';
 
 export class WalletViewModel implements WalletDto {
-  driverId: string;
+  id: string;
+  leagueId: string;
   balance: number;
+  totalRevenue: number;
+  totalPlatformFees: number;
+  totalWithdrawn: number;
+  createdAt: string;
   currency: string;
-  transactions: WalletTransactionViewModel[];
 
-  constructor(dto: WalletDto & { transactions: WalletTransactionDto[] }) {
-    this.driverId = dto.driverId;
+  constructor(dto: WalletDto & { transactions?: any[] }) {
+    this.id = dto.id;
+    this.leagueId = dto.leagueId;
     this.balance = dto.balance;
+    this.totalRevenue = dto.totalRevenue;
+    this.totalPlatformFees = dto.totalPlatformFees;
+    this.totalWithdrawn = dto.totalWithdrawn;
+    this.createdAt = dto.createdAt;
     this.currency = dto.currency;
-    this.transactions = dto.transactions.map(t => new WalletTransactionViewModel(t));
+    
+    // Map transactions if provided
+    if (dto.transactions) {
+      this.transactions = dto.transactions.map(t => new WalletTransactionViewModel(t));
+    }
   }
 
+  // Note: The generated DTO doesn't have driverId or transactions
+  // These will need to be added when the OpenAPI spec is updated
+  driverId: string = '';
+  transactions: WalletTransactionViewModel[] = [];
+
   /** UI-specific: Formatted balance */
   get formattedBalance(): string {
     return `${this.currency} ${this.balance.toFixed(2)}`;