chore: add getBill endpoint

Author: felipemelozx

src/main/java/fun/trackmoney/transaction/controller/TransactionController.java

@@ -1,5 +1,6 @@
 package fun.trackmoney.transaction.controller;
 
+import fun.trackmoney.transaction.dto.BillResponseDTO;
 import fun.trackmoney.transaction.dto.CreateTransactionDTO;
 import fun.trackmoney.transaction.dto.TransactionResponseDTO;
 import fun.trackmoney.transaction.dto.TransactionUpdateDTO;
@@ -61,6 +62,12 @@ public ResponseEntity<ApiResponse<BigDecimal>> getExpense(@PathVariable Integer
         new ApiResponse<>(true,"Get expense", transactionService.getExpense(id),null));
   }
 
+  @GetMapping("/bill/{id}")
+  public ResponseEntity<ApiResponse<BillResponseDTO>> getBill(@PathVariable Integer id) {
+    return ResponseEntity.ok().body(
+        new ApiResponse<>(true,"Get expense", transactionService.getBill(id),null));
+  }
+
   @PutMapping("/{id}")
   public ResponseEntity<ApiResponse<TransactionResponseDTO>> updateTransaction(@PathVariable Integer id,
                                                                                @RequestBody TransactionUpdateDTO dto) {

src/main/java/fun/trackmoney/transaction/dto/BillResponseDTO.java

@@ -0,0 +1,8 @@
+package fun.trackmoney.transaction.dto;
+
+import java.math.BigDecimal;
+
+public record BillResponseDTO(BigDecimal bill,
+                              BigDecimal totalUpcoming,
+                              BigDecimal bueSoon) {
+}

src/main/java/fun/trackmoney/transaction/repository/TransactionRepository.java

@@ -9,5 +9,5 @@
 
 public interface TransactionRepository extends JpaRepository<TransactionEntity, Integer> {
   @Query("SELECT a FROM TransactionEntity a WHERE a.account.accountId = :accountId")
-  List<TransactionEntity> findAllByUserEmail(@Param("accountId") Integer accountId);
+  List<TransactionEntity> findAllByAccountId(@Param("accountId") Integer accountId);
 }

src/main/java/fun/trackmoney/transaction/service/TransactionService.java

@@ -6,6 +6,7 @@
 import fun.trackmoney.category.entity.CategoryEntity;
 import fun.trackmoney.category.service.CategoryService;
 import fun.trackmoney.enums.TransactionType;
+import fun.trackmoney.transaction.dto.BillResponseDTO;
 import fun.trackmoney.transaction.dto.CreateTransactionDTO;
 import fun.trackmoney.transaction.dto.TransactionResponseDTO;
 import fun.trackmoney.transaction.dto.TransactionUpdateDTO;
@@ -16,6 +17,7 @@
 import org.springframework.stereotype.Service;
 
 import java.math.BigDecimal;
+import java.time.LocalDate;
 import java.util.List;
 import java.util.Objects;
 
@@ -82,7 +84,7 @@ public void delete(Integer id) {
   }
 
   public BigDecimal getIncome(Integer accountId) {
-    return transactionRepository.findAllByUserEmail(accountId)
+    return transactionRepository.findAllByAccountId(accountId)
         .stream()
         .filter(t -> TransactionType.INCOME.equals(t.getTransactionType()))
         .map(TransactionEntity::getAmount)
@@ -91,11 +93,49 @@ public BigDecimal getIncome(Integer accountId) {
   }
 
   public BigDecimal getExpense(Integer accountId) {
-    return transactionRepository.findAllByUserEmail(accountId)
+    return transactionRepository.findAllByAccountId(accountId)
         .stream()
         .filter(t -> TransactionType.EXPENSE.equals(t.getTransactionType()))
         .map(TransactionEntity::getAmount)
         .filter(Objects::nonNull)
         .reduce(BigDecimal.ZERO, BigDecimal::add);
   }
+
+  public BillResponseDTO getBill(Integer id) {
+    var result = transactionRepository.findAllByAccountId(id);
+
+    BigDecimal totalBillsBeforeToday = result.stream()
+        .filter(transaction ->
+            transaction.getCategory().getName().equalsIgnoreCase("bill") &&
+                transaction.getTransactionDate().toLocalDateTime().toLocalDate().isBefore(LocalDate.now())
+        )
+        .map(TransactionEntity::getAmount)
+        .filter(Objects::nonNull)
+        .reduce(BigDecimal.ZERO, BigDecimal::add);
+
+
+    BigDecimal totalUpComing = result.stream()
+        .filter(transaction ->
+            transaction.getCategory().getName().equalsIgnoreCase("bill") &&
+                transaction.getTransactionDate().toLocalDateTime().toLocalDate().isAfter(LocalDate.now())
+        )
+        .map(TransactionEntity::getAmount)
+        .filter(Objects::nonNull)
+        .reduce(BigDecimal.ZERO, BigDecimal::add);
+
+    BigDecimal totalBueSoon = result.stream()
+        .filter(transaction -> {
+          LocalDate date = transaction.getTransactionDate().toLocalDateTime().toLocalDate();
+          return transaction.getCategory().getName().equalsIgnoreCase("bill")
+              && !date.isBefore(LocalDate.now())
+              && !date.isAfter(LocalDate.now().plusDays(7));
+        })
+
+        .map(TransactionEntity::getAmount)
+        .filter(Objects::nonNull)
+        .reduce(BigDecimal.ZERO, BigDecimal::add);
+
+
+    return new BillResponseDTO(totalBillsBeforeToday, totalUpComing, totalBueSoon);
+  }
 }

src/test/java/fun/trackmoney/transaction/service/TransactionServiceTest.java

@@ -151,7 +151,7 @@ void getIncome_shouldReturnTotalIncome() {
     tr2.setTransactionType(TransactionType.INCOME);
 
     var accountId = 1;
-    when(transactionRepository.findAllByUserEmail(accountId)).thenReturn(List.of(tr1, tr2));
+    when(transactionRepository.findAllByAccountId(accountId)).thenReturn(List.of(tr1, tr2));
 
     var result = transactionService.getIncome(accountId);
 
@@ -169,7 +169,7 @@ void getExpense_shouldReturnTotalExpense() {
     tr2.setTransactionType(TransactionType.EXPENSE);
 
     var accountId = 1;
-    when(transactionRepository.findAllByUserEmail(accountId)).thenReturn(List.of(tr1, tr2));
+    when(transactionRepository.findAllByAccountId(accountId)).thenReturn(List.of(tr1, tr2));
 
     var result = transactionService.getExpense(accountId);