From 9399ceec367135a133de8231557e53d75043c4b4 Mon Sep 17 00:00:00 2001
From: dirgantarasiahaan <siahaandirgantara@gmail.com>
Date: Tue, 30 May 2023 11:30:24 +0700
Subject: [PATCH] fix validasi filename

---
 .../repository/DrpDokumenRepository.java      |  3 +++
 .../DrpPengadaanDokumenRepository.java        |  3 +++
 .../PostDrpUploadDokumenPendukungService.java | 19 +++++++++++++------
 .../drp/PostDrpUploadDokumenRKAPService.java  |  9 +++++++++
 4 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/iconplus/smartproc/repository/DrpDokumenRepository.java b/src/main/java/com/iconplus/smartproc/repository/DrpDokumenRepository.java
index 6f13bb5..fe3e208 100644
--- a/src/main/java/com/iconplus/smartproc/repository/DrpDokumenRepository.java
+++ b/src/main/java/com/iconplus/smartproc/repository/DrpDokumenRepository.java
@@ -5,9 +5,12 @@ import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.stereotype.Repository;
 
 import java.util.List;
+import java.util.Optional;
 
 @Repository
 public interface DrpDokumenRepository extends JpaRepository<DrpDokumen, Long> {
 
     List<DrpDokumen> findByDrpIdAndIsDeleteFalse(Long drpId);
+
+    Optional<DrpDokumen> findByFilenameAndIsDeleteFalse(String filename);
 }
diff --git a/src/main/java/com/iconplus/smartproc/repository/DrpPengadaanDokumenRepository.java b/src/main/java/com/iconplus/smartproc/repository/DrpPengadaanDokumenRepository.java
index a3ccf03..2bc9b84 100644
--- a/src/main/java/com/iconplus/smartproc/repository/DrpPengadaanDokumenRepository.java
+++ b/src/main/java/com/iconplus/smartproc/repository/DrpPengadaanDokumenRepository.java
@@ -5,10 +5,13 @@ import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.stereotype.Repository;
 
 import java.util.List;
+import java.util.Optional;
 
 @Repository
 public interface DrpPengadaanDokumenRepository extends JpaRepository<DrpPengadaanDokumen, Long> {
 
     List<DrpPengadaanDokumen> findByDrpIdAndIsDeleteFalse(Long drpId);
 
+    Optional<DrpPengadaanDokumen> findByFilenameAndIsDeleteFalse(String filename);
+
 }
diff --git a/src/main/java/com/iconplus/smartproc/service/drp/PostDrpUploadDokumenPendukungService.java b/src/main/java/com/iconplus/smartproc/service/drp/PostDrpUploadDokumenPendukungService.java
index 276a462..bb50d0c 100644
--- a/src/main/java/com/iconplus/smartproc/service/drp/PostDrpUploadDokumenPendukungService.java
+++ b/src/main/java/com/iconplus/smartproc/service/drp/PostDrpUploadDokumenPendukungService.java
@@ -1,11 +1,11 @@
 package com.iconplus.smartproc.service.drp;
 
-import com.iconplus.smartproc.helper.model.EmptyResponse;
+import com.iconplus.smartproc.exception.BusinessException;
 import com.iconplus.smartproc.helper.service.BaseService;
-import com.iconplus.smartproc.model.entity.DrpDokumen;
 import com.iconplus.smartproc.model.request.DrpDokumenRequest;
 import com.iconplus.smartproc.model.response.DrpDokumenResponse;
-import com.iconplus.smartproc.repository.DrpDokumenRepository;
+import com.iconplus.smartproc.repository.DrpPengadaanDokumenRepository;
+import com.iconplus.smartproc.util.Constants;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
@@ -20,9 +20,9 @@ public class PostDrpUploadDokumenPendukungService implements BaseService<DrpDoku
     @Value("${config.file-upload.pendukung}")
     private String pathFileUploadPendukung;
 
-    private final DrpDokumenRepository drpDokumenRepository;
-    public PostDrpUploadDokumenPendukungService(DrpDokumenRepository drpDokumenRepository) {
-        this.drpDokumenRepository = drpDokumenRepository;
+    private final DrpPengadaanDokumenRepository drpPengadaanDokumenRepository;
+    public PostDrpUploadDokumenPendukungService(DrpPengadaanDokumenRepository drpPengadaanDokumenRepository) {
+        this.drpPengadaanDokumenRepository = drpPengadaanDokumenRepository;
     }
 
     @Override
@@ -31,6 +31,13 @@ public class PostDrpUploadDokumenPendukungService implements BaseService<DrpDoku
         MultipartFile file = input.getFile();
         String fileName = file.getOriginalFilename();
 
+        var drpDokumen= drpPengadaanDokumenRepository.findByFilenameAndIsDeleteFalse(fileName);
+        if (drpDokumen.isPresent()) {
+            throw new BusinessException(Constants.ERR_CODE_10002,
+                    Constants.ERR_TTL_10002,
+                    String.format(Constants.ERR_MSG_10002, "Filename", fileName));
+        }
+
         File path = new File(pathFileUploadPendukung + fileName);
 
         path.createNewFile();
diff --git a/src/main/java/com/iconplus/smartproc/service/drp/PostDrpUploadDokumenRKAPService.java b/src/main/java/com/iconplus/smartproc/service/drp/PostDrpUploadDokumenRKAPService.java
index 86aa82b..73110df 100644
--- a/src/main/java/com/iconplus/smartproc/service/drp/PostDrpUploadDokumenRKAPService.java
+++ b/src/main/java/com/iconplus/smartproc/service/drp/PostDrpUploadDokumenRKAPService.java
@@ -1,11 +1,13 @@
 package com.iconplus.smartproc.service.drp;
 
+import com.iconplus.smartproc.exception.BusinessException;
 import com.iconplus.smartproc.helper.model.EmptyResponse;
 import com.iconplus.smartproc.helper.service.BaseService;
 import com.iconplus.smartproc.model.entity.DrpDokumen;
 import com.iconplus.smartproc.model.request.DrpDokumenRequest;
 import com.iconplus.smartproc.model.response.DrpDokumenResponse;
 import com.iconplus.smartproc.repository.DrpDokumenRepository;
+import com.iconplus.smartproc.util.Constants;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
@@ -31,6 +33,13 @@ public class PostDrpUploadDokumenRKAPService implements BaseService<DrpDokumenRe
         MultipartFile file = input.getFile();
         String fileName = file.getOriginalFilename();
 
+        var drpDokumen= drpDokumenRepository.findByFilenameAndIsDeleteFalse(fileName);
+        if (drpDokumen.isPresent()) {
+            throw new BusinessException(Constants.ERR_CODE_10002,
+                    Constants.ERR_TTL_10002,
+                    String.format(Constants.ERR_MSG_10002, "Filename", fileName));
+        }
+
         File path = new File(pathFileUploadRkap + fileName);
 
         path.createNewFile();