import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'expense_model.dart';
import 'expense_api.dart';

class ExpenseApplyState {
  final ExpenseModel expense;
  final bool isSubmitting;

  const ExpenseApplyState({required this.expense, this.isSubmitting = false});

  ExpenseApplyState copyWith({ExpenseModel? expense, bool? isSubmitting}) {
    return ExpenseApplyState(
      expense: expense ?? this.expense,
      isSubmitting: isSubmitting ?? this.isSubmitting,
    );
  }
}

class ExpenseApplyController extends StateNotifier<ExpenseApplyState> {
  final ExpenseApi _api;

  ExpenseApplyController(this._api, {ExpenseModel? initial})
    : super(
        ExpenseApplyState(
          expense:
              initial ??
              ExpenseModel(
                id: '',
                reportNo: '',
                applicantId: '',
                applicantName: '',
                deptId: '',
                deptName: '',
                expenseType: '',
                totalAmount: 0.0,
                purpose: '',
                createTime: DateTime.now(),
                updateTime: DateTime.now(),
              ),
        ),
      );

  void updateCurrencyCode(String code) {
    state = state.copyWith(expense: state.expense.copyWith(currencyCode: code));
  }

  void updateApprovedAmount(double amount) {
    state = state.copyWith(expense: state.expense.copyWith(approvedAmount: amount));
  }

  void updateDetailApprovedAmount(int index, double amount) {
    final details = [...state.expense.details];
    details[index] = details[index].copyWith(approvedAmount: amount);
    state = state.copyWith(expense: state.expense.copyWith(details: details));
  }

  void updateDetailCustomerVendor(int index, String name) {
    final details = [...state.expense.details];
    details[index] = details[index].copyWith(customerVendorName: name);
    state = state.copyWith(expense: state.expense.copyWith(details: details));
  }

  void updateDetailProjectCode(int index, String code) {
    final details = [...state.expense.details];
    details[index] = details[index].copyWith(projectCode: code);
    state = state.copyWith(expense: state.expense.copyWith(details: details));
  }

  void updateDetailSubjectCode(int index, String code) {
    final details = [...state.expense.details];
    details[index] = details[index].copyWith(subjectCode: code);
    state = state.copyWith(expense: state.expense.copyWith(details: details));
  }

  void updateDetailProjectCategory(int index, String category) {
    final details = [...state.expense.details];
    details[index] = details[index].copyWith(projectCategory: category);
    state = state.copyWith(expense: state.expense.copyWith(details: details));
  }

  void updateDetailOffsetAmount(int index, double amount) {
    final details = [...state.expense.details];
    details[index] = details[index].copyWith(offsetAmount: amount);
    state = state.copyWith(expense: state.expense.copyWith(details: details));
  }

  void updatePurpose(String purpose) {
    state = state.copyWith(expense: state.expense.copyWith(purpose: purpose));
  }

  void addDetail(ExpenseDetailModel detail) {
    final details = [...state.expense.details, detail];
    state = state.copyWith(expense: state.expense.copyWith(details: details));
  }

  void removeDetail(int index) {
    final details = [...state.expense.details]..removeAt(index);
    state = state.copyWith(expense: state.expense.copyWith(details: details));
  }

  void recalculateAmount() {
    var totalAmount = 0.0;
    var approvedAmount = 0.0;
    for (final d in state.expense.details) {
      totalAmount += d.totalAmount;
      approvedAmount += d.approvedAmount;
    }
    state = state.copyWith(
      expense: state.expense.copyWith(
        totalAmount: totalAmount,
        approvedAmount: approvedAmount,
      ),
    );
  }

  Future<bool> submit() async {
    state = state.copyWith(isSubmitting: true);
    try {
      await _api.submit(state.expense.copyWith(status: 'pending'));
      return true;
    } catch (_) {
      return false;
    } finally {
      state = state.copyWith(isSubmitting: false);
    }
  }

  Future<bool> saveDraft() async {
    state = state.copyWith(isSubmitting: true);
    try {
      await _api.saveDraft(state.expense);
      return true;
    } catch (_) {
      return false;
    } finally {
      state = state.copyWith(isSubmitting: false);
    }
  }
}

final expenseApplyProvider = StateNotifierProvider.autoDispose
    .family<ExpenseApplyController, ExpenseApplyState, String?>((ref, editId) {
      final api = ref.watch(expenseApiProvider);
      return ExpenseApplyController(api);
    });