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

package/lib/screens/rrsp_sun_life_screen.dart

@@ -1,197 +1,104 @@
 import 'package:flutter/material.dart';
 import 'package:intl/intl.dart';
-import 'package:personal_finance_frontend_core_services/services/transaction_service.dart';
-import 'package:personal_finance_frontend_core_ui/widgets/rrsp_contribution_form.dart';
 import 'package:provider/provider.dart';
 import 'package:personal_finance_frontend_core_services/providers/auth_provider.dart';
-import 'package:personal_finance_frontend_core_ui/utils/app_dialogs.dart'; // Added import for AppDialogs
+import 'package:personal_finance_frontend_core_ui/utils/app_dialogs.dart';
+import 'package:personal_finance_frontend_core_ui/widgets/rrsp_contribution_form.dart';
+import '../viewmodels/rrsp_sun_life_viewmodel.dart';
 
-class RrspSunLifeScreen extends StatefulWidget {
+class RrspSunLifeScreen extends StatelessWidget {
   const RrspSunLifeScreen({Key? key}) : super(key: key);
 
-  @override
-  _RrspSunLifeScreenState createState() => _RrspSunLifeScreenState();
-}
-
-class _RrspSunLifeScreenState extends State<RrspSunLifeScreen> {
-  final TransactionService _transactionService = TransactionService();
-  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;
-  String? _token;
-
-  @override
-  void initState() {
-    super.initState();
-    WidgetsBinding.instance.addPostFrameCallback((_) {
-      final authProvider = Provider.of<AuthProvider>(context, listen: false);
-      setState(() {
-        _token = authProvider.token;
-      });
-      _fetchData();
-    });
-  }
-
-  String currencySymbol = r'$'; // ou poderia vir de configuração, API, etc.
-
-  String _formatCurrency(double value) {
-    return '$currencySymbol${value.toStringAsFixed(2)}';
-  }
-
-  Future<void> _fetchData() async {
-    if (_token == null) return;
-    if (mounted) {
-      setState(() {
-        _isLoading = true;
-      });
-    }
-
-    try {
-      final contributions = await _transactionService.getRrspContributions(
-          investmentAccount: 'RRSP Sun Life', token: _token);
-      final balance =
-          await _transactionService.getAccountBalance('RRSP Sun Life', token: _token);
-      final totalPortfolioBookCost =
-          await _transactionService.getTotalPortfolioBookCost(token: _token);
-
-      final sinteticoSummaryData =
-          _calculateSinteticoSummary(contributions, totalPortfolioBookCost);
-
-      if (!mounted) return;
-      setState(() {
-        _contributions = contributions;
-        _sinteticoSummary = sinteticoSummaryData['summary'];
-        _rrspCashBalance = balance;
-        _rrspTotalValue = sinteticoSummaryData['total_value'];
-        _totalPortfolioBookCost = totalPortfolioBookCost;
-      });
-    } catch (e) {
-      if (!mounted) return;
-      ScaffoldMessenger.of(context).showSnackBar(
-        SnackBar(content: Text('Error fetching RRSP data: $e')),
-      );
-    } finally {
-      if (mounted) {
-        setState(() {
-          _isLoading = false;
-        });
-      }
-    }
-  }
-
-  Map<String, dynamic> _calculateSinteticoSummary(
-      List<Map<String, dynamic>> contributions, double totalPortfolioBookCost) {
-    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,
-    };
-
-    return {
-      'summary': [summaryData],
-      'total_value': marketValue
-    };
-  }
-
   @override
   Widget build(BuildContext context) {
-    return Scaffold(
-      appBar: AppBar(
-        title: const Text('RRSP Sun Life Portfolio'),
-        actions: [
-          Center(
-            child: Padding(
-              padding: const EdgeInsets.only(right: 16.0),
-              child: Column(
-                mainAxisAlignment: MainAxisAlignment.center,
-                crossAxisAlignment: CrossAxisAlignment.end,
-                children: [
-                  Text(
-                    'Total Portfolio: ${_formatCurrency(_rrspTotalValue)}',
-                    style: const TextStyle(
-                        fontSize: 16, fontWeight: FontWeight.bold),
-                  ),
-                  Text(
-                    'Account Balance: ${_formatCurrency(_rrspCashBalance)}',
-                    style: const TextStyle(fontSize: 12),
-                  ),
-                ],
-              ),
-            ),
-          ),
-          IconButton(
-            icon: const Icon(Icons.refresh),
-            onPressed: _fetchData,
-            tooltip: 'Refresh Data',
-          ),
-        ],
-      ),
-      body: _isLoading
-          ? const Center(child: CircularProgressIndicator())
-          : RefreshIndicator(
-              onRefresh: _fetchData,
-              child: SingleChildScrollView(
-                padding: const EdgeInsets.all(16.0),
-                child: Column(
-                  crossAxisAlignment: CrossAxisAlignment.stretch,
-                  children: [
-                    Text('Summary',
-                        style: Theme.of(context).textTheme.titleLarge),
-                    const SizedBox(height: 8),
-                    _buildSinteticoTable(),
-                    const SizedBox(height: 24),
-                    RrspContributionForm(
-                      investmentAccount: 'RRSP Sun Life',
-                      onContributionLogged: () => _fetchData(),
-                      token: _token,
-                    ),
-                    const SizedBox(height: 24),
-                    Text(
-                      'Contribution History',
-                      style: Theme.of(context).textTheme.titleLarge,
+    final authProvider = Provider.of<AuthProvider>(context, listen: false);
+
+    return ChangeNotifierProvider(
+      create: (_) => RrspSunLifeViewModel(token: authProvider.token),
+      child: Consumer<RrspSunLifeViewModel>(
+        builder: (context, viewModel, child) {
+          return Scaffold(
+            appBar: AppBar(
+              title: const Text('RRSP Sun Life Portfolio'),
+              actions: [
+                Center(
+                  child: Padding(
+                    padding: const EdgeInsets.only(right: 16.0),
+                    child: Column(
+                      mainAxisAlignment: MainAxisAlignment.center,
+                      crossAxisAlignment: CrossAxisAlignment.end,
+                      children: [
+                        Text(
+                          'Total Portfolio: ${_formatCurrency(viewModel.rrspTotalValue)}',
+                          style: const TextStyle(
+                              fontSize: 16, fontWeight: FontWeight.bold),
+                        ),
+                        Text(
+                          'Account Balance: ${_formatCurrency(viewModel.rrspCashBalance)}',
+                          style: const TextStyle(fontSize: 12),
+                        ),
+                      ],
                     ),
-                    const SizedBox(height: 8),
-                    _buildContributionTable(),
-                  ],
+                  ),
                 ),
-              ),
+                IconButton(
+                  icon: const Icon(Icons.refresh),
+                  onPressed: viewModel.fetchData,
+                  tooltip: 'Refresh Data',
+                ),
+              ],
             ),
+            body: viewModel.isLoading
+                ? const Center(child: CircularProgressIndicator())
+                : RefreshIndicator(
+                    onRefresh: viewModel.fetchData,
+                    child: SingleChildScrollView(
+                      padding: const EdgeInsets.all(16.0),
+                      child: Column(
+                        crossAxisAlignment: CrossAxisAlignment.stretch,
+                        children: [
+                          Text('Summary',
+                              style: Theme.of(context).textTheme.titleLarge),
+                          const SizedBox(height: 8),
+                          _buildSinteticoTable(context, viewModel),
+                          const SizedBox(height: 24),
+                          RrspContributionForm(
+                            investmentAccount: 'RRSP Sun Life',
+                            onContributionLogged: viewModel.fetchData,
+                            token: viewModel.token,
+                          ),
+                          const SizedBox(height: 24),
+                          Text(
+                            'Contribution History',
+                            style: Theme.of(context).textTheme.titleLarge,
+                          ),
+                          const SizedBox(height: 8),
+                          _buildContributionTable(context, viewModel),
+                        ],
+                      ),
+                    ),
+                  ),
+          );
+        },
+      ),
     );
   }
 
-  Widget _buildSinteticoTable() {
-    if (_sinteticoSummary.isEmpty) {
+  String _formatCurrency(double value) {
+    String currencySymbol = r'$';
+    return '$currencySymbol${value.toStringAsFixed(2)}';
+  }
+
+  Widget _buildSinteticoTable(
+      BuildContext context, RrspSunLifeViewModel viewModel) {
+    if (viewModel.sinteticoSummary.isEmpty) {
       return const SizedBox.shrink();
     }
 
-    final summary = _sinteticoSummary.first;
+    final summary = viewModel.sinteticoSummary.first;
     final double unrealizedPL = summary['unrealized_pl'];
     final plColor = (unrealizedPL >= 0) ? Colors.green : Colors.red;
 
-
     return DataTable(
       columns: const [
         DataColumn(label: Text('User Contr.')),
@@ -205,24 +112,33 @@ class _RrspSunLifeScreenState extends State<RrspSunLifeScreen> {
       rows: [
         DataRow(
           cells: [
-            DataCell(Text(_formatCurrency(summary['user_contribution'] as double))),
-            DataCell(Text(_formatCurrency(summary['company_contribution'] as double))),
             DataCell(
-                Text(_formatCurrency(summary['total_contributed'] as double))),
-            DataCell(Text(_formatCurrency(unrealizedPL), style: TextStyle(color: plColor))),
-            DataCell(Text(
-                '${(summary['percent_return'] as double).toStringAsFixed(2)}%', style: TextStyle(color: plColor))),
-            DataCell(Text(_formatCurrency(summary['market_value'] as double))),
+                Text(_formatCurrency(summary['user_contribution'] as double))),
             DataCell(Text(
-                '${(summary['portfolio_allocation_percent'] as double).toStringAsFixed(2)}%')),
+                _formatCurrency(summary['company_contribution'] as double))),
+            DataCell(
+                Text(_formatCurrency(summary['total_contributed'] as double))),
+            DataCell(Text(_formatCurrency(unrealizedPL),
+                style: TextStyle(color: plColor))),
+            DataCell(
+              Text('${(summary['percent_return'] as double).toStringAsFixed(2)}%',
+                  style: TextStyle(color: plColor)),
+            ),
+            DataCell(
+                Text(_formatCurrency(summary['market_value'] as double))),
+            DataCell(
+              Text(
+                  '${(summary['portfolio_allocation_percent'] as double).toStringAsFixed(2)}%'),
+            ),
           ],
         ),
       ],
     );
   }
 
-  Widget _buildContributionTable() {
-    if (_contributions.isEmpty) {
+  Widget _buildContributionTable(
+      BuildContext context, RrspSunLifeViewModel viewModel) {
+    if (viewModel.contributions.isEmpty) {
       return const Center(
         child: Padding(
           padding: EdgeInsets.all(16.0),
@@ -243,7 +159,7 @@ class _RrspSunLifeScreenState extends State<RrspSunLifeScreen> {
         DataColumn(label: Text('Portfolio %')),
         DataColumn(label: Text('Actions')),
       ],
-      rows: _contributions.map((c) {
+      rows: viewModel.contributions.map((c) {
         final contributionId = c['id'] as int;
         final rrspAmount = double.parse(c['rrsp_amount'].toString());
         final dpspAmount = double.parse(c['dpsp_amount'].toString());
@@ -251,13 +167,14 @@ class _RrspSunLifeScreenState extends State<RrspSunLifeScreen> {
 
         final returnAmount =
             double.parse(c['return_amount']?.toString() ?? '0.0');
-        final percentReturn =
-            totalContributed > 0 ? (returnAmount / totalContributed) * 100 : 0.0;
+        final percentReturn = totalContributed > 0
+            ? (returnAmount / totalContributed) * 100
+            : 0.0;
 
         final cumulativeValue = totalContributed + returnAmount;
 
-        final portfolioAllocation = _totalPortfolioBookCost > 0
-            ? (cumulativeValue / _totalPortfolioBookCost) * 100
+        final portfolioAllocation = viewModel.totalPortfolioBookCost > 0
+            ? (cumulativeValue / viewModel.totalPortfolioBookCost) * 100
             : 0.0;
 
         String dateStr;
@@ -283,7 +200,8 @@ class _RrspSunLifeScreenState extends State<RrspSunLifeScreen> {
             DataCell(
               IconButton(
                 icon: const Icon(Icons.delete, color: Colors.red),
-                onPressed: () => _confirmAndDeleteContribution(contributionId),
+                onPressed: () => _confirmAndDeleteContribution(
+                    context, viewModel, contributionId),
                 tooltip: 'Delete Contribution',
               ),
             ),
@@ -293,34 +211,22 @@ class _RrspSunLifeScreenState extends State<RrspSunLifeScreen> {
     );
   }
 
-  Future<void> _confirmAndDeleteContribution(int contributionId) async {
+  Future<void> _confirmAndDeleteContribution(BuildContext context,
+      RrspSunLifeViewModel viewModel, int contributionId) async {
     final bool? confirm = await AppDialogs.showDeleteConfirmationDialog(
       context,
-      'this contribution', // A generic string for the item being deleted
+      'this contribution',
     );
 
     if (confirm == true) {
-      if (_token == null) {
+      final success = await viewModel.deleteContribution(contributionId);
+      if (ScaffoldMessenger.of(context).mounted) {
         ScaffoldMessenger.of(context).showSnackBar(
-          const SnackBar(content: Text('Authentication token not available.')),
-        );
-        return;
-      }
-      try {
-        final success = await _transactionService.deleteRrspContribution(contributionId, token: _token);
-        if (success) {
-          ScaffoldMessenger.of(context).showSnackBar(
-            const SnackBar(content: Text('Contribution deleted successfully.')),
-          );
-          _fetchData(); // Refresh data after deletion
-        } else {
-          ScaffoldMessenger.of(context).showSnackBar(
-            const SnackBar(content: Text('Failed to delete contribution.')),
-          );
-        }
-      } catch (e) {
-        ScaffoldMessenger.of(context).showSnackBar(
-          SnackBar(content: Text('Error deleting contribution: $e')),
+          SnackBar(
+            content: Text(success
+                ? 'Contribution deleted successfully.'
+                : 'Failed to delete contribution.'),
+          ),
         );
       }
     }

package/lib/viewmodels/crypto_account_viewmodel.dart

@@ -0,0 +1,180 @@
+import 'package:flutter/material.dart';
+import 'package:personal_finance_frontend_core_services/services/crypto_service.dart';
+import 'package:personal_finance_frontend_core_services/services/transaction_service.dart';
+
+class CryptoAccountViewModel extends ChangeNotifier {
+  final TransactionService _transactionService = TransactionService();
+  final CryptoService _cryptoService = CryptoService();
+
+  final String accountName;
+  String? _token;
+
+  List<Map<String, dynamic>> _trades = [];
+  List<Map<String, dynamic>> _assets = [];
+  List<Map<String, dynamic>> _portfolioSummary = [];
+  Map<String, double> _livePrices = {};
+  double _cashBalance = 0.0;
+  double _accountTotalValue = 0.0;
+  bool _isLoading = true;
+  bool _isFetchingPrices = false;
+
+  // Getters
+  List<Map<String, dynamic>> get trades => _trades;
+  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 token => _token;
+
+  CryptoAccountViewModel({required this.accountName, 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 futures = <Future>[
+        _transactionService.getTrades(investmentAccount: accountName, token: _token),
+        _transactionService.getAccountBalance(accountName, token: _token),
+        _transactionService.getAssets(investmentAccount: accountName, token: _token),
+        _transactionService.getTotalPortfolioBookCost(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>>;
+      final totalPortfolioBookCost = results[3] as double;
+
+      _calculateAndApplyPortfolioSummary(totalPortfolioBookCost);
+      
+      await _fetchLivePrices();
+
+    } catch (e) {
+      debugPrint('Error fetching data: $e');
+    } finally {
+      _isLoading = false;
+      notifyListeners();
+    }
+  }
+
+  Future<void> _fetchLivePrices() async {
+    if (_portfolioSummary.isEmpty) return;
+    _isFetchingPrices = true;
+    notifyListeners();
+
+    final idsToFetch = _portfolioSummary
+        .map((p) => p['id_crypto'] as String? ?? p['symbol'] as String)
+        .where((s) => s.isNotEmpty)
+        .toList();
+
+    if (idsToFetch.isEmpty) {
+      _isFetchingPrices = false;
+      notifyListeners();
+      return;
+    }
+
+    final livePrices = await _cryptoService.getLiveCryptoPrices(idsToFetch);
+    _livePrices = livePrices;
+    
+    _recalculateMarketValue();
+
+    _isFetchingPrices = false;
+    notifyListeners();
+  }
+
+  void _calculateAndApplyPortfolioSummary(double totalPortfolioBookCost) {
+    Map<String, dynamic> summary = {};
+    double accountPortfolioBookCost = 0;
+
+    for (var trade in _trades) {
+      final asset = trade['asset'];
+      if (asset == null || asset['symbol'] == null) continue;
+      String symbol = asset['symbol'];
+      String idCrypto = asset['id_crypto'] ?? 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,
+          'id_crypto': idCrypto,
+          'name': asset['name'] ?? symbol,
+          'shares': 0.0,
+          'total_cost': 0.0,
+        };
+      }
+
+      if (tradeType == 'buy') {
+        summary[symbol]['shares'] += shares;
+        summary[symbol]['total_cost'] += shares * price;
+      } else if (tradeType == 'sell') {
+        double originalShares = summary[symbol]['shares'];
+        if (originalShares > 0) {
+          double avgPrice = summary[symbol]['total_cost'] / originalShares;
+          summary[symbol]['total_cost'] -= shares * avgPrice;
+        }
+        summary[symbol]['shares'] -= shares;
+      }
+    }
+
+    summary.removeWhere((key, value) => value['shares'] < 0.01);
+    accountPortfolioBookCost = summary.values.fold(0.0, (sum, item) => sum + item['total_cost']);
+
+    List<Map<String, dynamic>> result = [];
+    summary.forEach((symbol, data) {
+      double shares = data['shares'];
+      double totalCost = data['total_cost'];
+      data['avg_price'] = (shares > 0) ? totalCost / shares : 0.0;
+      data['account_allocation'] = (accountPortfolioBookCost > 0) ? (totalCost / accountPortfolioBookCost) * 100 : 0.0;
+      data['stocks_allocation'] = (totalPortfolioBookCost > 0) ? (totalCost / totalPortfolioBookCost) * 100 : 0.0;
+      result.add(data);
+    });
+
+    _portfolioSummary = result;
+    _recalculateMarketValue();
+  }
+
+  void _recalculateMarketValue() {
+      double newTotalValue = _cashBalance;
+      for (final position in _portfolioSummary) {
+        final shares = double.parse(position['shares'].toString());
+        final idForLookup = (position['id_crypto'] as String? ?? position['symbol'] as String).toLowerCase();
+        final livePrice = _livePrices[idForLookup];
+        if (livePrice != null) {
+          newTotalValue += shares * livePrice;
+        } else {
+          newTotalValue += (position['total_cost'] as num?)?.toDouble() ?? 0.0;
+        }
+      }
+      _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();
+      }
+      return success;
+    } catch (e) {
+      debugPrint("Error deleting trade: $e");
+      return false;
+    }
+  }
+}