@marcos_feitoza/personal-finance-frontend-feature-investments 1.1.0 → 1.1.2

package/lib/viewmodels/investment_account_viewmodel.dart

@@ -0,0 +1,221 @@
+import 'package:flutter/material.dart';
+import 'package:personal_finance_frontend_core_services/services/stock_service.dart';
+import 'package:personal_finance_frontend_core_services/services/transaction_service.dart';
+
+class InvestmentAccountViewModel extends ChangeNotifier {
+  final TransactionService _transactionService = TransactionService();
+  final StockService _stockService = StockService();
+
+  final String accountName;
+  final bool showDividends;
+  String? _token;
+
+  List<Map<String, dynamic>> _trades = [];
+  List<Map<String, dynamic>> _dividends = [];
+  List<Map<String, dynamic>> _assets = [];
+  List<Map<String, dynamic>> _portfolioSummary = [];
+  Map<String, double> _livePrices = {};
+  double _cashBalance = 0.0;
+  double _accountTotalValue = 0.0;
+  double _totalPortfolioBookCost = 0.0;
+  bool _isLoading = true;
+  bool _isFetchingPrices = false;
+  String? _selectedSymbolForFilter;
+
+  // Getters
+  List<Map<String, dynamic>> get trades => _trades;
+  List<Map<String, dynamic>> get dividends => _dividends;
+  List<Map<String, dynamic>> get assets => _assets;
+  List<Map<String, dynamic>> get portfolioSummary => _portfolioSummary;
+  Map<String, double> get livePrices => _livePrices;
+  double get cashBalance => _cashBalance;
+  double get accountTotalValue => _accountTotalValue;
+  bool get isLoading => _isLoading;
+  bool get isFetchingPrices => _isFetchingPrices;
+  String? get selectedSymbolForFilter => _selectedSymbolForFilter;
+  String? get token => _token;
+
+  InvestmentAccountViewModel({required this.accountName, required this.showDividends, required String? token}) {
+    _token = token;
+    fetchData();
+  }
+
+  void setTokenAndFetch(String? token) {
+    _token = token;
+    fetchData();
+  }
+  
+  void setSymbolFilter(String? symbol) {
+    _selectedSymbolForFilter = symbol;
+    notifyListeners();
+  }
+
+  Future<void> fetchData() async {
+    if (_token == null) return;
+    _isLoading = true;
+    notifyListeners();
+
+    try {
+      final futures = <Future>[
+        _transactionService.getTrades(investmentAccount: accountName, token: _token),
+        _transactionService.getAccountBalance(accountName, token: _token),
+        _transactionService.getAssets(investmentAccount: accountName, token: _token),
+        _transactionService.getTotalPortfolioBookCost(token: _token),
+      ];
+      if (showDividends) {
+        futures.add(_transactionService.getDividends(investmentAccount: accountName, token: _token));
+      }
+      final results = await Future.wait(futures);
+
+      _trades = results[0] as List<Map<String, dynamic>>;
+      _cashBalance = results[1] as double;
+      _assets = results[2] as List<Map<String, dynamic>>;
+      _totalPortfolioBookCost = results[3] as double;
+      _dividends = showDividends ? results[4] as List<Map<String, dynamic>> : <Map<String, dynamic>>[];
+
+      _calculateAndApplyPortfolioSummary();
+      
+      await _fetchLivePrices();
+
+    } catch (e) {
+      // Handle error appropriately
+      debugPrint('Error fetching data: $e');
+    } finally {
+      _isLoading = false;
+      notifyListeners();
+    }
+  }
+
+  Future<void> _fetchLivePrices() async {
+    if (_portfolioSummary.isEmpty) return;
+    _isFetchingPrices = true;
+    notifyListeners();
+
+    final symbolsToFetch = _portfolioSummary
+        .map((p) => p['symbol'] as String)
+        .where((s) => s.isNotEmpty)
+        .toList();
+
+    if (symbolsToFetch.isEmpty) {
+      _isFetchingPrices = false;
+      notifyListeners();
+      return;
+    }
+
+    final livePrices = await _stockService.getLivePrices(symbolsToFetch);
+    _livePrices = livePrices;
+    
+    _recalculateMarketValues();
+
+    _isFetchingPrices = false;
+    notifyListeners();
+  }
+
+  void _calculateAndApplyPortfolioSummary() {
+    Map<String, dynamic> summary = {};
+    double accountTotalBookCost = 0;
+
+    Map<String, double> dividendSummary = {};
+    if (showDividends) {
+      for (var dividend in _dividends) {
+        final asset = dividend['asset'];
+        if (asset != null && asset['symbol'] != null) {
+          String symbol = asset['symbol'];
+          double amount = double.parse(dividend['amount'].toString());
+          dividendSummary.update(symbol, (value) => value + amount, ifAbsent: () => amount);
+        }
+      }
+    }
+
+    for (var trade in _trades) {
+      final asset = trade['asset'];
+      if (asset == null || asset['symbol'] == null) continue;
+      String symbol = asset['symbol'];
+
+      double shares = double.parse(trade['shares'].toString());
+      double price = double.parse(trade['price'].toString());
+      String tradeType = trade['trade_type'];
+
+      if (!summary.containsKey(symbol)) {
+        summary[symbol] = {
+          'symbol': symbol,
+          'name': asset['name'] ?? symbol,
+          'industry': asset['industry'] ?? 'N/A',
+          'shares': 0.0,
+          'book_cost': 0.0,
+          'total_dividends': dividendSummary[symbol] ?? 0.0,
+        };
+      }
+      
+      if (tradeType == 'buy') {
+        summary[symbol]['shares'] += shares;
+        summary[symbol]['book_cost'] += shares * price;
+      } else if (tradeType == 'sell') {
+        double originalShares = summary[symbol]['shares'];
+        if (originalShares > 0) {
+          double avgPrice = summary[symbol]['book_cost'] / originalShares;
+          summary[symbol]['book_cost'] -= shares * avgPrice;
+        }
+        summary[symbol]['shares'] -= shares;
+      }
+    }
+
+    summary.removeWhere((key, value) => value['shares'] < 0.01);
+
+    accountTotalBookCost = summary.values.fold(0.0, (sum, item) => sum + item['book_cost']);
+
+    List<Map<String, dynamic>> result = [];
+    summary.forEach((symbol, data) {
+      double shares = data['shares'];
+      double bookCost = data['book_cost'];
+
+      data['avg_price'] = (shares > 0) ? bookCost / shares : 0.0;
+      data['account_allocation_percent'] = (accountTotalBookCost > 0) ? (bookCost / accountTotalBookCost) * 100 : 0.0;
+      data['portfolio_allocation_percent'] = (_totalPortfolioBookCost > 0) ? (bookCost / _totalPortfolioBookCost) * 100 : 0.0;
+      
+      result.add(data);
+    });
+
+    _portfolioSummary = result;
+    _recalculateMarketValues(); // Initial calculation with book cost as fallback
+  }
+
+  void _recalculateMarketValues() {
+    double newTotalValue = _cashBalance;
+    for (final position in _portfolioSummary) {
+      final symbol = position['symbol'];
+      final shares = position['shares'] as double;
+      final bookCost = position['book_cost'] as double;
+      final totalDividends = position['total_dividends'] as double;
+
+      final livePrice = _livePrices[symbol];
+      double marketValue = livePrice != null ? shares * livePrice : bookCost;
+      double unrealizedPL = marketValue - bookCost;
+      double totalReturn = unrealizedPL + totalDividends;
+      
+      position['market_value'] = marketValue;
+      position['unrealized_pl'] = unrealizedPL;
+      position['total_return'] = totalReturn;
+      position['percent_unrealized_pl'] = (bookCost > 0) ? (unrealizedPL / bookCost) * 100 : 0.0;
+      position['percent_total_return'] = (bookCost > 0) ? (totalReturn / bookCost) * 100 : 0.0;
+      
+      newTotalValue += marketValue;
+    }
+    _accountTotalValue = newTotalValue;
+  }
+  
+  Future<bool> deleteTrade(int tradeId) async {
+    if (_token == null) return false;
+    
+    try {
+      final success = await _transactionService.deleteTrade(tradeId, token: _token);
+      if (success) {
+        fetchData(); // Refresh data
+      }
+      return success;
+    } catch (e) {
+      debugPrint("Error deleting trade: $e");
+      return false;
+    }
+  }
+}

package/lib/viewmodels/rrsp_sun_life_viewmodel.dart

@@ -0,0 +1,103 @@
+import 'package:flutter/material.dart';
+import 'package:personal_finance_frontend_core_services/services/transaction_service.dart';
+
+class RrspSunLifeViewModel extends ChangeNotifier {
+  final TransactionService _transactionService = TransactionService();
+  String? _token;
+
+  List<Map<String, dynamic>> _contributions = [];
+  List<Map<String, dynamic>> _sinteticoSummary = [];
+  double _rrspTotalValue = 0.0;
+  double _rrspCashBalance = 0.0;
+  double _totalPortfolioBookCost = 0.0;
+  bool _isLoading = true;
+
+  // Getters
+  List<Map<String, dynamic>> get contributions => _contributions;
+  List<Map<String, dynamic>> get sinteticoSummary => _sinteticoSummary;
+  double get rrspTotalValue => _rrspTotalValue;
+  double get rrspCashBalance => _rrspCashBalance;
+  double get totalPortfolioBookCost => _totalPortfolioBookCost;
+  bool get isLoading => _isLoading;
+  String? get token => _token;
+
+  RrspSunLifeViewModel({required String? token}) {
+    _token = token;
+    fetchData();
+  }
+
+  void setTokenAndFetch(String? token) {
+    _token = token;
+    fetchData();
+  }
+
+  Future<void> fetchData() async {
+    if (_token == null) return;
+    _isLoading = true;
+    notifyListeners();
+
+    try {
+      final contributionsFuture = _transactionService.getRrspContributions(
+          investmentAccount: 'RRSP Sun Life', token: _token);
+      final balanceFuture = _transactionService.getAccountBalance('RRSP Sun Life', token: _token);
+      final totalPortfolioBookCostFuture = _transactionService.getTotalPortfolioBookCost(token: _token);
+
+      final results = await Future.wait([contributionsFuture, balanceFuture, totalPortfolioBookCostFuture]);
+      
+      _contributions = results[0] as List<Map<String, dynamic>>;
+      _rrspCashBalance = results[1] as double;
+      _totalPortfolioBookCost = results[2] as double;
+
+      _calculateAndApplySinteticoSummary();
+
+    } catch (e) {
+      debugPrint('Error fetching RRSP data: $e');
+    } finally {
+      _isLoading = false;
+      notifyListeners();
+    }
+  }
+
+  void _calculateAndApplySinteticoSummary() {
+    double totalUserContribution = 0;
+    double totalCompanyContribution = 0;
+    double totalUnrealizedPL = 0;
+
+    for (var c in _contributions) {
+      totalUserContribution += double.parse(c['rrsp_amount'].toString());
+      totalCompanyContribution += double.parse(c['dpsp_amount'].toString());
+      totalUnrealizedPL += double.parse(c['return_amount']?.toString() ?? '0.0');
+    }
+
+    final totalContributed = totalUserContribution + totalCompanyContribution;
+    final marketValue = totalContributed + totalUnrealizedPL;
+
+    final summaryData = {
+      'user_contribution': totalUserContribution,
+      'company_contribution': totalCompanyContribution,
+      'total_contributed': totalContributed,
+      'unrealized_pl': totalUnrealizedPL,
+      'percent_return': (totalContributed > 0) ? (totalUnrealizedPL / totalContributed) * 100 : 0.0,
+      'market_value': marketValue,
+      'portfolio_allocation_percent': (_totalPortfolioBookCost > 0) ? (marketValue / _totalPortfolioBookCost) * 100 : 0.0,
+    };
+
+    _sinteticoSummary = [summaryData];
+    _rrspTotalValue = marketValue;
+  }
+  
+  Future<bool> deleteContribution(int contributionId) async {
+    if (_token == null) return false;
+    
+    try {
+      final success = await _transactionService.deleteRrspContribution(contributionId, token: _token);
+      if (success) {
+        fetchData();
+      }
+      return success;
+    } catch (e) {
+      debugPrint("Error deleting contribution: $e");
+      return false;
+    }
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@marcos_feitoza/personal-finance-frontend-feature-investments",
-    "version": "1.1.0",
+    "version": "1.1.2",
     "publishConfig": {
         "access": "public"
     },