package com.faf223.expensetrackerfaf.service; import com.faf223.expensetrackerfaf.model.*; import com.faf223.expensetrackerfaf.repository.CredentialRepository; import com.faf223.expensetrackerfaf.repository.ExpenseRepository; import com.faf223.expensetrackerfaf.repository.UserRepository; import com.faf223.expensetrackerfaf.util.exceptions.UserNotAuthenticatedException; import com.faf223.expensetrackerfaf.util.exceptions.UserNotFoundException; import lombok.RequiredArgsConstructor; import org.springframework.security.core.Authentication; import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.security.core.userdetails.UserDetails; import org.springframework.stereotype.Service; import java.time.LocalDate; import java.time.Month; import java.util.ArrayList; import java.util.List; import java.util.Optional; @Service @RequiredArgsConstructor public class ExpenseService implements ITransactionService { private final ExpenseRepository expenseRepository; private final CredentialRepository credentialRepository; private final UserRepository userRepository; public void createOrUpdate(IMoneyTransaction expense) { expenseRepository.save((Expense) expense); } public List getTransactionsByEmail(String email) { Optional credential = credentialRepository.findByEmail(email); if (credential.isPresent()) { return expenseRepository.findByUser(credential.get().getUser()); } return new ArrayList<>(); } @Override public List getTransactionsByDate(LocalDate date) { return expenseRepository.findByDate(date); } @Override public List getTransactionsByDate(LocalDate date, String email) { return getTransactionsByDate(date) .stream() .filter(transaction -> { Optional credential = credentialRepository.findByEmail(email); if(credential.isEmpty()) throw new UserNotFoundException("The user has not been found"); return credential.get().getUser().equals(transaction.getUser()); }) .toList(); } // TODO: store transaction month in a separate field in the DB and change this logic @Override public List getTransactionsByMonth(Month month) { LocalDate startOfMonth = LocalDate.of(LocalDate.now().getYear(), month, 1); LocalDate endOfMonth = startOfMonth.plusMonths(1).minusDays(1); return expenseRepository.findByDateBetween(startOfMonth, endOfMonth); } @Override public List getTransactionsByMonth(Month month, String email) { return getTransactionsByMonth(month) .stream() .filter(transaction -> { Optional credential = credentialRepository.findByEmail(email); if(credential.isEmpty()) throw new UserNotFoundException("The user has not been found"); return credential.get().getUser().equals(transaction.getUser()); }) .toList(); } public List getTransactions() { return expenseRepository.findAll(); } public Expense getTransactionById(long id) { return expenseRepository.findById(id).orElse(null); } public void deleteTransactionById(long id) { expenseRepository.deleteById(id); } @Override public boolean belongsToUser(IMoneyTransaction transaction) { Authentication authentication = SecurityContextHolder.getContext().getAuthentication(); if (authentication != null && authentication.getPrincipal() instanceof UserDetails userDetails) { if(authentication.getAuthorities().stream().noneMatch(authority -> authority.getAuthority().equals("ADMIN"))) { Optional credential = credentialRepository.findByEmail(userDetails.getUsername()); if(credential.isEmpty()) throw new UserNotFoundException("The user has not been found"); Optional user = userRepository.findById(credential.get().getUser().getUserUuid()); if(user.isEmpty()) throw new UserNotFoundException("The user has not been found"); return user.get().getExpenses().contains((Expense) transaction); } } throw new UserNotAuthenticatedException("You are not authenticated"); } }