bflows-bandi-be/src/main/java/net/gepafin/tendermanagement/dao/DocumentDao.java

package net.gepafin.tendermanagement.dao;

import java.util.stream.Collectors;

import lombok.extern.slf4j.Slf4j;
import net.gepafin.tendermanagement.enums.DocumentSourceTypeEnum;
import net.gepafin.tendermanagement.repositories.ApplicationRepository;
import net.gepafin.tendermanagement.web.rest.api.errors.CustomValidationException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import net.gepafin.tendermanagement.config.Translator;
import net.gepafin.tendermanagement.constants.GepafinConstant;
import net.gepafin.tendermanagement.entities.CallEntity;
import net.gepafin.tendermanagement.entities.DocumentEntity;
import net.gepafin.tendermanagement.enums.DocumentTypeEnum;
import net.gepafin.tendermanagement.model.response.DocumentResponseBean;
import net.gepafin.tendermanagement.model.response.UploadFileOnAmazonS3Response;
import net.gepafin.tendermanagement.repositories.DocumentRepository;
import net.gepafin.tendermanagement.service.AmazonS3Service;
import net.gepafin.tendermanagement.service.CallService;
import net.gepafin.tendermanagement.web.rest.api.errors.ResourceNotFoundException;
import net.gepafin.tendermanagement.web.rest.api.errors.Status;

import java.util.ArrayList;
import java.util.List;

@Slf4j
@Component
public class DocumentDao {

	@Autowired
	private AmazonS3Service amazonS3Service;

	@Autowired
	private DocumentRepository documentRepository;

	@Autowired
	private CallDao callDao;

	@Autowired
	private CallService callService;

	@Autowired
	private S3PathConfig s3ConfigBean;

	@Autowired
	private ApplicationRepository applicationFormRepository;
	
    @Value("${aws.s3.url.folder}")
    private String s3Folder;

	public List<DocumentResponseBean> uploadFiles(List<MultipartFile> files, Long sourceId, DocumentSourceTypeEnum sourceType, DocumentTypeEnum fileType) {
		List<DocumentEntity> documentEntities = new ArrayList<>();
		Long source = resolveSourceId(sourceId, sourceType);
		for (MultipartFile file : files) {
			UploadFileOnAmazonS3Response uploadFileOnAmazonS3Response = uploadFileOnAmazonS3(file, sourceType, sourceId);
			if (uploadFileOnAmazonS3Response != null) {
				DocumentEntity documentEntity = new DocumentEntity();
				documentEntity.setFileName(uploadFileOnAmazonS3Response.getFileName());
				documentEntity.setSource(sourceType.getValue());
				documentEntity.setSourceId(source);
				documentEntity.setType(fileType.getValue());
				documentEntity.setFilePath(uploadFileOnAmazonS3Response.getFilePath());
				documentEntity.setIsDeleted(false);
				documentEntities.add(documentEntity);
			}
		}
		documentRepository.saveAll(documentEntities);
		return documentEntities.stream().map(callDao::convertToDocumentResponseBean).collect(Collectors.toList());
	}
	private UploadFileOnAmazonS3Response uploadFileOnAmazonS3(MultipartFile file, DocumentSourceTypeEnum type, Long sourceId) {

        Long applicationId = 0L;
		Long callId = sourceId;
		if (type == DocumentSourceTypeEnum.APPLICATION) {
			applicationId = sourceId;
			callId = applicationFormRepository.findCallIdById(applicationId);
		}
		try {
			String s3Path = generateS3Path(type, callId, applicationId);
			log.info("Generated S3 path {}", s3Path);
			return amazonS3Service.uploadFileOnAmazonS3(s3Path, file);
		} catch (Exception e) {
			throw new CustomValidationException(Status.VALIDATION_ERROR, Translator.toLocale(GepafinConstant.UPLOAD_ERROR_S3));
		}
	}
	public String generateS3Path(DocumentSourceTypeEnum typeOfDocument, Long callId, Long applicationId) {

		try {
			return s3ConfigBean.generateDocumentPath(typeOfDocument, callId, applicationId);
		} catch (IllegalArgumentException e) {
			throw new CustomValidationException(Status.VALIDATION_ERROR, Translator.toLocale(GepafinConstant.S3_PATH_GENERATION_ERROR_MSG));
		}
	}
	private Long resolveSourceId(Long sourceId, DocumentSourceTypeEnum sourceType) {
		if (sourceType == DocumentSourceTypeEnum.CALL) {
			CallEntity callEntity = callService.validateCall(sourceId);
			return callEntity.getId();
		}
//		else if (sourceType == SourceTypeEnum.APPLICATION) {
//			ApplicationEntity applicationEntity = applicationService.validateApplication(sourceId);
//			return applicationEntity.getId(); // Assuming ApplicationEntity has getId()
//		}
//
		return sourceId;
	}

	public void deleteFile(Long documentId) {
		DocumentEntity documentEntity = validateDocument(documentId);
//        String fileName= Utils.extractFileName(documentEntity.getFilePath());
//        deleteFileOnAmazonS3(fileName);
		documentEntity.setIsDeleted(true);
		documentRepository.save(documentEntity);
	}


	public DocumentEntity validateDocument(Long id) {
		return documentRepository.findById(id).orElseThrow(() -> new ResourceNotFoundException(Status.NOT_FOUND,
				Translator.toLocale(GepafinConstant.DOCUMENT_NOT_FOUND)));
	}

	public DocumentResponseBean updateDocument(Long documentId, MultipartFile file, DocumentTypeEnum documentTypeEnum) {
		DocumentEntity documentEntity = validateDocument(documentId);
		String type = documentEntity.getSource();
		UploadFileOnAmazonS3Response uploadFileOnAmazonS3Response = updateFileOnAmazonS3(file, DocumentSourceTypeEnum.valueOf(type), documentEntity.getSourceId());
        if (uploadFileOnAmazonS3Response != null) {
			documentEntity.setFileName(uploadFileOnAmazonS3Response.getFileName());
			documentEntity.setFilePath(uploadFileOnAmazonS3Response.getFilePath());
			documentEntity.setType(documentTypeEnum.getValue());
			documentEntity.setSource(documentEntity.getSource());
			documentEntity.setSourceId(documentEntity.getSourceId());
			documentRepository.save(documentEntity);
		}
		return callDao.convertToDocumentResponseBean(documentEntity);
	}
	private UploadFileOnAmazonS3Response updateFileOnAmazonS3(MultipartFile file, DocumentSourceTypeEnum type, Long id) {

		try {
			Long callId;
			Long applicationId;
			if(type.equals("APPLICATION")){
				callId = applicationFormRepository.findCallIdById(id);
				applicationId = id;
			}else{
				callId = id;
				applicationId = 0L;
			}
            String s3Path = generateS3Path(type, callId, applicationId);
			log.info("Generated S3 path {}", s3Path);
			return amazonS3Service.uploadFileOnAmazonS3(s3Path, file);
		} catch (Exception e) {
			throw new CustomValidationException(Status.VALIDATION_ERROR, Translator.toLocale(GepafinConstant.UPLOAD_ERROR_S3));
		}
	}
	public DocumentResponseBean getDocument(Long documentId) {
		DocumentEntity documentEntity = validateDocument(documentId);
		return callDao.convertToDocumentResponseBean(documentEntity);
	}
}