Merge branch 'develop' of https://github.com/Kitzanos/GEPAFIN-BE into develop

src/main/java/net/gepafin/tendermanagement/dao/FlowDao.java

@@ -0,0 +1,175 @@
+package net.gepafin.tendermanagement.dao;
+
+import net.gepafin.tendermanagement.config.Translator;
+import net.gepafin.tendermanagement.constants.GepafinConstant;
+import net.gepafin.tendermanagement.entities.CallEntity;
+import net.gepafin.tendermanagement.entities.FlowDataEntity;
+import net.gepafin.tendermanagement.entities.FlowDataEntity;
+import net.gepafin.tendermanagement.entities.FlowEdgesEntity;
+import net.gepafin.tendermanagement.model.request.FlowDataRequestBean;
+import net.gepafin.tendermanagement.model.request.FlowEdgesRequestBean;
+import net.gepafin.tendermanagement.model.request.FlowRequestBean;
+import net.gepafin.tendermanagement.model.response.EvaluationCriteriaResponseBean;
+import net.gepafin.tendermanagement.model.response.FlowDataResponseBean;
+import net.gepafin.tendermanagement.model.response.FlowEdgesResponseBean;
+import net.gepafin.tendermanagement.model.response.FlowResponseBean;
+import net.gepafin.tendermanagement.repositories.CallRepository;
+import net.gepafin.tendermanagement.repositories.FlowDataRepository;
+import net.gepafin.tendermanagement.repositories.FlowDataRepository;
+import net.gepafin.tendermanagement.repositories.FlowEdgesRepository;
+import net.gepafin.tendermanagement.repositories.FlowEdgesRepository;
+import net.gepafin.tendermanagement.service.CallService;
+import net.gepafin.tendermanagement.service.FormService;
+import net.gepafin.tendermanagement.web.rest.api.errors.CustomValidationException;
+import net.gepafin.tendermanagement.web.rest.api.errors.Status;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Component
+public class FlowDao {
+
+    @Autowired
+    private FlowDataRepository flowDataRepository;
+
+    @Autowired
+    private FlowEdgesRepository flowEdgesRepository;
+
+    @Autowired
+    private CallRepository callRepository;
+
+    @Autowired
+    private CallService callService;
+
+    @Autowired
+    private FormService formService;
+
+    public FlowResponseBean createFlow(FlowRequestBean flowRequestBean, Long callId) {
+        validateFlowRequestBean(flowRequestBean);
+        CallEntity call = setInitialAndFinalFormInCall(flowRequestBean, callId);
+        checkIfFlowExits(callId);
+        validateFlowRequest(flowRequestBean);
+        List<FlowDataEntity> flowDataEntities = createFlowData(flowRequestBean,call);
+        List<FlowEdgesEntity> flowEdgesEntities = createFlowEdges(flowRequestBean,call);
+        FlowResponseBean flowResponseBean = getFlowByCallId(call.getId());
+        return flowResponseBean;
+    }
+
+    public void validateFlowRequestBean(FlowRequestBean flowRequestBean){
+        if(flowRequestBean.getFlowData()==null || flowRequestBean.getFlowEdges()==null){
+            throw new CustomValidationException(Status.BAD_REQUEST,Translator.toLocale(GepafinConstant.FLOW_REQUEST_NOT_PROPER));
+        }
+    }
+
+    public void checkIfFlowExits(Long callId) {
+        List<FlowDataEntity> flowDataEntities = flowDataRepository.findByCallId(callId);
+        List<FlowEdgesEntity> flowEdgesEntities = flowEdgesRepository.findByCallId(callId);
+        if (!flowDataEntities.isEmpty()  || !flowEdgesEntities.isEmpty()) {
+            throw new CustomValidationException(Status.BAD_REQUEST, Translator.toLocale(GepafinConstant.FLOW_ALREADY_EXISTS));
+        }
+    }
+
+    public void validateFlowRequest(FlowRequestBean flowRequestBean) {
+        formService.validateForm(flowRequestBean.getInitialForm());
+        formService.validateForm(flowRequestBean.getFinalForm());
+        flowRequestBean.getFlowData().forEach(flowData -> formService.validateForm(flowData.getFormId()));
+    }
+
+    private List<FlowEdgesResponseBean> createFlowEdgesResponseBean(List<FlowEdgesEntity> flowEdgesEntities) {
+        List<FlowEdgesResponseBean> flowEdgesResponseBeans = flowEdgesEntities.stream()
+                .map(this::convertFlowEdgesEntityToFlowEdgesResponseBean).collect(Collectors.toList());
+        return flowEdgesResponseBeans;
+    }
+
+    private List<FlowDataResponseBean> createFlowDataResponseBean(List<FlowDataEntity> flowDataEntities) {
+        List<FlowDataResponseBean> flowDataResponseBeans = flowDataEntities.stream()
+                .map(this::convertFlowDataEntityToFlowDataResponseBean).collect(Collectors.toList());
+        return flowDataResponseBeans;
+    }
+
+    public FlowResponseBean setInitialAndFinalFormInFlowResponseBean(FlowResponseBean flowResponseBean, CallEntity call) {
+        flowResponseBean.setInitialForm(call.getInitialForm());
+        flowResponseBean.setFinalForm(call.getFinalForm());
+        return flowResponseBean;
+    }
+
+    private CallEntity setInitialAndFinalFormInCall(FlowRequestBean flowRequestBean, Long callId) {
+        CallEntity call = callService.validateCall(callId);
+        call.setInitialForm(flowRequestBean.getInitialForm());
+        call.setFinalForm(flowRequestBean.getFinalForm());
+        call = callRepository.save(call);
+        return call;
+    }
+
+    public List<FlowDataEntity> createFlowData(FlowRequestBean flowRequestBean, CallEntity call) {
+        List<FlowDataEntity> flowDataEntities = flowRequestBean.getFlowData().stream()
+                .map(flowDataRequestBean -> createFlowDataEntity(flowDataRequestBean, call))
+                .collect(Collectors.toList());
+        return flowDataRepository.saveAll(flowDataEntities);
+    }
+
+    public FlowDataEntity createFlowDataEntity(FlowDataRequestBean flowDataRequestBean,CallEntity call) {
+        FlowDataEntity flowDataEntity = new FlowDataEntity();
+        flowDataEntity.setFormId(flowDataRequestBean.getFormId());
+        flowDataEntity.setChoosenField(flowDataRequestBean.getChosenField());
+        flowDataEntity.setChoosenValue(flowDataRequestBean.getChosenValue());
+        flowDataEntity.setCallId(call.getId());
+        return flowDataEntity;
+    }
+
+    public List<FlowEdgesEntity> createFlowEdges(FlowRequestBean flowRequestBean, CallEntity call) {
+        List<FlowEdgesEntity> flowEdgesEntities = flowRequestBean.getFlowEdges().stream()
+                .map(flowEdgesRequestBean -> createFlowEdgesEntity(flowEdgesRequestBean, call))
+                .collect(Collectors.toList());
+        return flowEdgesRepository.saveAll(flowEdgesEntities);
+    }
+
+    public FlowEdgesEntity createFlowEdgesEntity(FlowEdgesRequestBean flowEdgesRequestBean,CallEntity call) {
+        FlowEdgesEntity flowEdgesEntity = new FlowEdgesEntity();
+        flowEdgesEntity.setSourceId(Long.valueOf(flowEdgesRequestBean.getSource()));
+        flowEdgesEntity.setTargetId(Long.valueOf(flowEdgesRequestBean.getTarget()));
+        flowEdgesEntity.setType(flowEdgesRequestBean.getType());
+        flowEdgesEntity.setCallId(call.getId());
+        return flowEdgesEntity;
+    }
+
+    public FlowDataResponseBean convertFlowDataEntityToFlowDataResponseBean(FlowDataEntity flowDataEntity) {
+        FlowDataResponseBean flowDataResponseBean = new FlowDataResponseBean();
+        flowDataResponseBean.setId(flowDataEntity.getId());
+        flowDataResponseBean.setFormId(flowDataEntity.getFormId());
+        flowDataResponseBean.setChosenField(flowDataEntity.getChoosenField());
+        flowDataResponseBean.setChosenValue(flowDataEntity.getChoosenValue());
+        return flowDataResponseBean;
+    }
+
+    public FlowEdgesResponseBean convertFlowEdgesEntityToFlowEdgesResponseBean(FlowEdgesEntity flowEdgesEntity) {
+        FlowEdgesResponseBean flowEdgesResponseBean = new FlowEdgesResponseBean();
+        flowEdgesResponseBean.setId(flowEdgesEntity.getId());
+        flowEdgesResponseBean.setType(flowEdgesEntity.getType());
+        flowEdgesResponseBean.setSource(String.valueOf(flowEdgesEntity.getSourceId()));
+        flowEdgesResponseBean.setTarget(String.valueOf(flowEdgesEntity.getTargetId()));
+        return flowEdgesResponseBean;
+    }
+
+    public FlowResponseBean getFlowByCallId(Long callId){
+        CallEntity call= callService.validateCall(callId);
+        FlowResponseBean flowResponseBean=new FlowResponseBean();
+        List<FlowDataEntity> flowDataEntities=flowDataRepository.findByCallId(call.getId());
+        List<FlowEdgesEntity> flowEdgesEntities=flowEdgesRepository.findByCallId(call.getId());
+        List<FlowDataResponseBean> flowDataResponseBeans=createFlowDataResponseBean(flowDataEntities);
+        List<FlowEdgesResponseBean> flowEdgesResponseBeans=createFlowEdgesResponseBean(flowEdgesEntities);
+        flowResponseBean.setFlowData(flowDataResponseBeans);
+        flowResponseBean.setFlowEdges(flowEdgesResponseBeans);
+        if(flowResponseBean.getFlowData().isEmpty() || flowResponseBean.getFlowEdges().isEmpty()){
+            return null;
+        }
+        flowResponseBean.setCallId(call.getId());
+        flowResponseBean.setInitialForm(call.getInitialForm());
+        flowResponseBean.setFinalForm(call.getFinalForm());
+        return flowResponseBean;
+    }
+
+}