feat: Add ability to link files to transactions

BudgetBuddyDE · Feb 4, 2024 · 581599d · 581599d
1 parent 7864d43
commit 581599d
Show file tree

Hide file tree

Showing 5 changed files with 42 additions and 5 deletions.
diff --git a/src/main/java/de/budgetbuddy/backend/log/Logger.java b/src/main/java/de/budgetbuddy/backend/log/Logger.java
@@ -2,8 +2,6 @@
 
 import org.springframework.stereotype.Component;
 
-import java.util.Date;
-
 @Component
 public class Logger {
     private static Logger instance;

diff --git a/src/main/java/de/budgetbuddy/backend/transaction/Transaction.java b/src/main/java/de/budgetbuddy/backend/transaction/Transaction.java
@@ -108,6 +108,7 @@ public static class Update {
         private String receiver;
         private String description;
         private Double transferAmount;
+        private ArrayList<TransactionFile.Create> attachedFiles = new ArrayList<>();
     }
 
     @Data

diff --git a/src/main/java/de/budgetbuddy/backend/transaction/TransactionController.java b/src/main/java/de/budgetbuddy/backend/transaction/TransactionController.java
@@ -8,6 +8,8 @@
 import de.budgetbuddy.backend.paymentMethod.PaymentMethod;
 import de.budgetbuddy.backend.paymentMethod.PaymentMethodRepository;
 import de.budgetbuddy.backend.subscription.SubscriptionRepository;
+import de.budgetbuddy.backend.transaction.file.TransactionFile;
+import de.budgetbuddy.backend.transaction.file.TransactionFileRepository;
 import de.budgetbuddy.backend.user.User;
 import de.budgetbuddy.backend.user.UserRepository;
 import de.budgetbuddy.backend.user.role.RolePermission;
@@ -28,18 +30,21 @@ public class TransactionController {
     private final PaymentMethodRepository paymentMethodRepository;
     private final SubscriptionRepository subscriptionRepository;
     private final TransactionRepository transactionRepository;
+    private final TransactionFileRepository transactionFileRepository;
     private final TransactionService transactionService;
 
     public TransactionController(UserRepository userRepository,
                                  CategoryRepository categoryRepository,
                                  PaymentMethodRepository paymentMethodRepository,
                                  SubscriptionRepository subscriptionRepository,
-                                 TransactionRepository transactionRepository) {
+                                 TransactionRepository transactionRepository,
+                                 TransactionFileRepository transactionFileRepository) {
         this.userRepository = userRepository;
         this.categoryRepository = categoryRepository;
         this.paymentMethodRepository = paymentMethodRepository;
         this.subscriptionRepository = subscriptionRepository;
         this.transactionRepository = transactionRepository;
+        this.transactionFileRepository = transactionFileRepository;
         this.transactionService = new TransactionService(transactionRepository);
     }
 
@@ -185,6 +190,21 @@ public ResponseEntity<ApiResponse<Transaction>> updateTransaction(@RequestBody T
                             "Provided payment-method not found"));
         }
 
+        List<TransactionFile> transactionFiles = new ArrayList<>();
+        if (!payload.getAttachedFiles().isEmpty()) {
+            for (TransactionFile.Create file : payload.getAttachedFiles()) {
+                transactionFiles.add(TransactionFile.builder()
+                        .owner(transactionOwner)
+                        .transaction(transaction)
+                        .fileName(file.getFileName())
+                        .fileSize(file.getFileSize())
+                        .mimeType(file.getMimeType())
+                        .location(file.getFileUrl())
+                        .build());
+            }
+            transactionFileRepository.saveAll(transactionFiles);
+        }
+
         return ResponseEntity
                 .status(HttpStatus.OK)
                 .body(new ApiResponse<>(transactionRepository.save(Transaction.builder()
@@ -196,7 +216,7 @@ public ResponseEntity<ApiResponse<Transaction>> updateTransaction(@RequestBody T
                         .receiver(payload.getReceiver())
                         .description(payload.getDescription())
                         .transferAmount(payload.getTransferAmount())
-                        .attachedFiles(transaction.getAttachedFiles())
+                        .attachedFiles(transactionFiles)
                         .createdAt(transaction.getCreatedAt())
                         .build())));
     }

diff --git a/src/main/java/de/budgetbuddy/backend/transaction/file/TransactionFile.java b/src/main/java/de/budgetbuddy/backend/transaction/file/TransactionFile.java
@@ -48,8 +48,17 @@ public class TransactionFile {
     @Column(name = "location", length = 100)
     private String location;
 
+    @Builder.Default
     @Column(name = "created_at", nullable = false, updatable = false, columnDefinition = "TIMESTAMP DEFAULT CURRENT_TIMESTAMP")
     @ColumnDefault("CURRENT_TIMESTAMP")
     private Date createdAt = new Date();
 
+    @Data
+    public static class Create {
+        private String fileName;
+        private int fileSize;
+        private String mimeType;
+        private String fileUrl;
+    }
+
 }
diff --git a/src/test/java/de/budgetbuddy/backend/transaction/TransactionControllerTests.java b/src/test/java/de/budgetbuddy/backend/transaction/TransactionControllerTests.java
@@ -8,6 +8,7 @@
 import de.budgetbuddy.backend.paymentMethod.PaymentMethod;
 import de.budgetbuddy.backend.paymentMethod.PaymentMethodRepository;
 import de.budgetbuddy.backend.subscription.SubscriptionRepository;
+import de.budgetbuddy.backend.transaction.file.TransactionFileRepository;
 import de.budgetbuddy.backend.user.User;
 import de.budgetbuddy.backend.user.UserRepository;
 import de.budgetbuddy.backend.user.role.Role;
@@ -37,6 +38,7 @@ public class TransactionControllerTests {
     private final CategoryRepository categoryRepository;
     private final PaymentMethodRepository paymentMethodRepository;
     private final TransactionRepository transactionRepository;
+    private final TransactionFileRepository transactionFileRepository;
     private final TransactionService transactionService;
     private final TransactionController transactionController;
     private final ObjectMapper objectMapper = new ObjectMapper().registerModule(new JavaTimeModule());
@@ -48,7 +50,14 @@ public class TransactionControllerTests {
         this.paymentMethodRepository = Mockito.mock(PaymentMethodRepository.class);
         SubscriptionRepository subscriptionRepository = Mockito.mock(SubscriptionRepository.class);
         this.transactionRepository = Mockito.mock(TransactionRepository.class);
-        this.transactionController = new TransactionController(userRepository, categoryRepository, paymentMethodRepository, subscriptionRepository, transactionRepository);
+        this.transactionFileRepository = Mockito.mock(TransactionFileRepository.class);
+        this.transactionController = new TransactionController(
+                userRepository,
+                categoryRepository,
+                paymentMethodRepository,
+                subscriptionRepository,
+                transactionRepository,
+                transactionFileRepository);
         this.transactionService = new TransactionService(transactionRepository);
     }