Revize b476c240

https://kivprogrammers.atlassian.net/browse/TSP2-58

src/main/java/cz/zcu/fav/kiv/antipatterndetectionapp/v2/controller/management/CategoryController.java

@@ -1 +1 @@
 package cz.zcu.fav.kiv.antipatterndetectionapp.v2.controller.management;
 
+import cz.zcu.fav.kiv.antipatterndetectionapp.model.management.Category;
+import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.AdditionalInformationDto;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.CategoryDto;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.GeneralResponseDto;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.service.management.category.ICategories;
@@ -9 +11 @@
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
 
+import java.util.List;
 import java.util.Map;
 
 /**
@@ -23 +27 @@
     private final ICategories categories;
 
     @PostMapping("/getCategories")
-    public ResponseEntity<GeneralResponseDto<CategoryDto>> getCategories(@RequestBody Map<String, String> requestData) {
-        return categories.getResponse(requestData.get("projectId"));
+    public ResponseEntity<GeneralResponseDto<CategoryDto, AdditionalInformationDto<String>>> getCategories(@RequestBody Map<String, String> requestData) {
+        return categories.findCategoriesForProject(requestData.get("projectId"));
     }
 
     @PostMapping("/changeCategory")
-    public ResponseEntity<String> changeCategory() {
-        //TODO
-        return null;
+    public ResponseEntity<GeneralResponseDto<CategoryDto, AdditionalInformationDto<String>>> changeCategory(@RequestParam(value = "selectedBox", required = false) List<CategoryDto> selectedCategories,
+                                                                                                            @RequestParam(value = "submitType", required = false) Integer submitType,
+                                                                                                            @RequestParam(value = "submitId", required = false) String submitId) {
+        return categories.handleCategoryChangeRequest(selectedCategories, submitType, submitId);
     }
 }

src/main/java/cz/zcu/fav/kiv/antipatterndetectionapp/v2/model/AdditionalInformationDto.java

@@ +1 @@
+package cz.zcu.fav.kiv.antipatterndetectionapp.v2.model;
+
+import lombok.Data;
+import lombok.NonNull;
+
+import java.util.HashMap;
+import java.util.Map;
+
+@Data
+public class AdditionalInformationDto<T> {
+
+    Map<String, T> additionalFields;
+
+    public AdditionalInformationDto() {
+        this.additionalFields = new HashMap<>();
+    }
+
+    public void addIntoCollection(@NonNull String key,
+                                  @NonNull T value) {
+        this.additionalFields.putIfAbsent(key, value);
+    }
+
+}

src/main/java/cz/zcu/fav/kiv/antipatterndetectionapp/v2/model/GeneralResponseDto.java

@@ -7 +7 @@
 
 @Data
 @Builder
-public class GeneralResponseDto<T> {
+public class GeneralResponseDto<T,S> {
     private String message;
     private List<T> responseBody;
-
-    public static <T> GeneralResponseDtoBuilder<T> builder() {
-        return new GeneralResponseDtoBuilder<T>();
+    private S additionalInformation;
+    public static <T, S> GeneralResponseDtoBuilder<T, S> builder() {
+        return new GeneralResponseDtoBuilder<T, S>();
     }
 
 }

src/main/java/cz/zcu/fav/kiv/antipatterndetectionapp/v2/service/management/category/ICategories.java

@@ -1 +1 @@
 package cz.zcu.fav.kiv.antipatterndetectionapp.v2.service.management.category;
 
-import cz.zcu.fav.kiv.antipatterndetectionapp.model.Project;
+import cz.zcu.fav.kiv.antipatterndetectionapp.model.management.Category;
+import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.AdditionalInformationDto;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.CategoryDto;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.GeneralResponseDto;
 import org.springframework.http.ResponseEntity;
+import org.springframework.lang.Nullable;
 
 import java.util.List;
-import java.util.Optional;
 
 public interface ICategories {
 
-    ResponseEntity<GeneralResponseDto<CategoryDto>> getResponse(String projectId);
+    ResponseEntity<GeneralResponseDto<CategoryDto, AdditionalInformationDto<String>>> findCategoriesForProject(String projectId);
+
+    ResponseEntity<GeneralResponseDto<CategoryDto, AdditionalInformationDto<String>>> handleCategoryChangeRequest(@Nullable List<CategoryDto> selectedCategories,
+                                                                                                                  @Nullable Integer submitType,
+                                                                                                                  @Nullable String submitId);
 
 }

src/main/java/cz/zcu/fav/kiv/antipatterndetectionapp/v2/service/management/category/impl/CategoriesImpl.java

@@ -1 +1 @@
 package cz.zcu.fav.kiv.antipatterndetectionapp.v2.service.management.category.impl;
 
+import cz.zcu.fav.kiv.antipatterndetectionapp.Constants;
 import cz.zcu.fav.kiv.antipatterndetectionapp.model.Project;
-import cz.zcu.fav.kiv.antipatterndetectionapp.model.management.Category;
-import cz.zcu.fav.kiv.antipatterndetectionapp.model.management.ProjectInstance;
+import cz.zcu.fav.kiv.antipatterndetectionapp.model.management.*;
+import cz.zcu.fav.kiv.antipatterndetectionapp.model.management.interfaces.DatabaseObject;
 import cz.zcu.fav.kiv.antipatterndetectionapp.service.ProjectService;
+import cz.zcu.fav.kiv.antipatterndetectionapp.service.managment.*;
+import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.AdditionalInformationDto;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.CategoryDto;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.model.GeneralResponseDto;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.service.management.category.ICategories;
 import cz.zcu.fav.kiv.antipatterndetectionapp.v2.utils.converters.CategoryToDto;
 import lombok.NonNull;
 import lombok.RequiredArgsConstructor;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.util.Pair;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.lang.Nullable;
@@ -20 +27 @@
 @Service
 @RequiredArgsConstructor
 public class CategoriesImpl implements ICategories {
-
+    private final Logger logger = LoggerFactory.getLogger(this.getClass().getName());
     private final ProjectService projectService;
-    private CategoryToDto categoryMapper = new CategoryToDto();
+    private final CategoryService categoryService;
+    private final WorkUnitService workUnitService;
+    private final IterationService iterationService;
+    private final ActivityService activityService;
+    private final PhaseService phaseService;
+    private final String GENERAL_ERROR_MESSAGE = "ERROR: Application error occurred";
+    private final String SUBMISSION_ERROR_MESSAGE = "Cannot proceed: Submission went wrong";
+    private final CategoryToDto categoryMapper = new CategoryToDto();
     private final String PROJECT_NOT_FOUND_MESSAGE = "Project not found!";
     private final String CATEGORIES_NOT_FOUND_MESSAGE = "Categories not found for the project!";
     private final String INVALID_PROJECT_ID = "Provided project ID is in invalid format!";
@@ -31 +45 @@
         return projectService.getProjectById(id);
     }
 
+    @Override
+    public ResponseEntity<GeneralResponseDto<CategoryDto, AdditionalInformationDto<String>>> handleCategoryChangeRequest(@Nullable List<CategoryDto> selectedCategories,
+                                                                                                                         @Nullable Integer submitType,
+                                                                                                                         @Nullable String submitId) {
+        List<Category> categories = new ArrayList<>();
+
+        if (selectedCategories != null && submitType != null) { // Selected submit
+            if (submitType <= 0 || submitType > Constants.SUBMIT_TYPES) { // ERROR: Unknown submitType
+                logger.warn("@PostMapping(\"/v2/management/changeCategory\") - Cannot proceed: Unknown submitType " + submitType);
+                return createResponseEntity(HttpStatus.BAD_REQUEST, GENERAL_ERROR_MESSAGE, null, null);
+            }
+        } else if (submitId != null) { //Inline submit
+            Pair<Integer, Long> resultPair = extractTypeAndId(submitId);
+            if (resultPair == null) { // Cannot be extracted
+                logger.warn("@PostMapping(\"/v2/management/changeCategory\") - Cannot proceed: Unknown submitId " + submitId);
+                return createResponseEntity(HttpStatus.BAD_REQUEST, GENERAL_ERROR_MESSAGE, null, null);
+            }
+
+            // Add one category into new list
+            submitType = resultPair.getFirst();
+            Long id = resultPair.getSecond();
+            categories.add(categoryService.getCategoryById(id));
+
+        } else { //Unspecific error
+            logger.warn("@PostMapping(\"/management/segment-category\") - Cannot proceed: Submission went wrong");
+            return createResponseEntity(HttpStatus.BAD_REQUEST, SUBMISSION_ERROR_MESSAGE, null, null);
+        }
+
+        // Process all selected categories
+        Pair<String, Boolean> result = processSelectedCategories(submitType, categories);
+
+        AdditionalInformationDto<String> additionalInformation = new AdditionalInformationDto<>();
+        if (!result.getSecond()) {
+            additionalInformation.addIntoCollection("successMessage", "All selected categories (" + categories.size() + ") were assign");
+        } else {
+            additionalInformation.addIntoCollection("informMessage", "Some categories cannot be assign, look at the log for more information");
+        }
+        additionalInformation.addIntoCollection("message", result.getFirst());
+
+        return createResponseEntity(HttpStatus.OK, null, categoryMapper.convert(categories), additionalInformation);
+    }
+
     private List<CategoryDto> getCategories(Project project) {
 
         if (project == null) {
@@ -49 +105 @@
     }
 
     @Override
-    public ResponseEntity<GeneralResponseDto<CategoryDto>> getResponse(String projectId) {
+    public ResponseEntity<GeneralResponseDto<CategoryDto, AdditionalInformationDto<String>>> findCategoriesForProject(String projectId) {
         // --- The value from the request body is String, so there must be a validation if the string is number or not.
         final Long id = tryParse(projectId);
         if (id == Long.MIN_VALUE)
-            return createResponseEntity(HttpStatus.BAD_REQUEST, INVALID_PROJECT_ID, null);
+            return createResponseEntity(HttpStatus.BAD_REQUEST, INVALID_PROJECT_ID, null, null);
 
         // --- get project by ID
         final Project project = getProjectById(id);
 
         if (project == null)
-            return createResponseEntity(HttpStatus.NOT_FOUND, PROJECT_NOT_FOUND_MESSAGE, null);
+            return createResponseEntity(HttpStatus.NOT_FOUND, PROJECT_NOT_FOUND_MESSAGE, null, null);
 
         // --- Get all categories for the specific project
         final List<CategoryDto> categories = getCategories(project);
 
         if (categories == null)
-            return createResponseEntity(HttpStatus.NOT_FOUND, CATEGORIES_NOT_FOUND_MESSAGE, null);
+            return createResponseEntity(HttpStatus.NOT_FOUND, CATEGORIES_NOT_FOUND_MESSAGE, null, null);
 
-        return createResponseEntity(HttpStatus.OK, null, categories);
+        return createResponseEntity(HttpStatus.OK, null, categories, null);
     }
 
-    private ResponseEntity<GeneralResponseDto<CategoryDto>> createResponseEntity(@NonNull HttpStatus status,
-                                                                                 @Nullable String message,
-                                                                                 @Nullable List<CategoryDto> categories) {
+    private ResponseEntity<GeneralResponseDto<CategoryDto, AdditionalInformationDto<String>>> createResponseEntity(@NonNull HttpStatus status,
+                                                                                                                   @Nullable String message,
+                                                                                                                   @Nullable List<CategoryDto> categories,
+                                                                                                                   @Nullable AdditionalInformationDto<String> additionalInfo) {
 
 
         return new ResponseEntity<>(
                 GeneralResponseDto
-                        .<CategoryDto>builder()
+                        .<CategoryDto, AdditionalInformationDto<String>>builder()
                         .message(message)
                         .responseBody(categories)
+                        .additionalInformation(additionalInfo)
                         .build(), status);
     }
 
@@ -91 +149 @@
         }
     }
 
+    private Pair<Integer, Long> extractTypeAndId(String submitId) {
+        String[] parts = submitId.split(Constants.HTML_DELIMITER);
+        try {
+            Integer type = Integer.parseInt(parts[0]);
+            Long id = Long.parseLong(parts[1]);
+            return Pair.of(type, id);
+
+        } catch (NumberFormatException e) {
+            return null;
+        }
+
+    }
+
+    private Pair<String, Boolean> processSelectedCategories(Integer submitType, List<Category> selectedCategories) {
+
+        boolean errorOccurred = false;
+        StringBuilder message = new StringBuilder();
+        for (Category category : selectedCategories) {
+            message.append("Category: ").append(category.getName()).append(System.getProperty("line.separator"));
+
+            if (category.getWorkUnits().size() == 0) { //Inform user that there aren't any WorkUnits
+                errorOccurred = true;
+                message.append(Constants.INDENT)
+                        .append("cannot be assigned due to non-existent work unit")
+                        .append(System.getProperty("line.separator"))
+                        .append(System.getProperty("line.separator"));
+                continue;
+            }
+
+            DatabaseObject object = getNewObject(category.getWorkUnits(), submitType, category);
+
+            for (WorkUnit workUnit : category.getWorkUnits()) {
+
+                boolean hasAttr = true;
+                switch (submitType) {
+                    case 1:
+                        if (workUnit.getIteration() == null) {
+                            hasAttr = false;
+                            workUnit.setIteration((Iteration) object);
+                            workUnitService.saveWorkUnit(workUnit);
+                        }
+                        break;
+                    case 2:
+                        if (workUnit.getPhase() == null) {
+                            hasAttr = false;
+                            workUnit.setPhase((Phase) object);
+                            workUnitService.saveWorkUnit(workUnit);
+                        }
+                        break;
+                    case 3:
+                        if (workUnit.getActivity() == null) {
+                            hasAttr = false;
+                            workUnit.setActivity((Activity) object);
+                            workUnitService.saveWorkUnit(workUnit);
+                        }
+                        break;
+                }
+
+                if (!hasAttr) { // Transformation of selected category was done
+
+                    message.append(Constants.INDENT)
+                            .append("successfully assigned to WU with id ").append(workUnit.getId())
+                            .append(System.getProperty("line.separator"));
+
+                } else { // WorkUnit already has iteration, inform user
+
+                    errorOccurred = true;
+                    message.append(Constants.INDENT)
+                            .append("cannot be assigned to WU with id ").append(workUnit.getId())
+                            .append(" as it already exists with ").append(workUnit.getIteration().getName())
+                            .append(System.getProperty("line.separator"));
+                }
+            }
+            message.append(System.getProperty("line.separator"));
+        }
+        return Pair.of(message.toString(), errorOccurred);
+    }
+
+    private DatabaseObject getNewObject(Set<WorkUnit> workUnits, Integer submitType, Category category) {
+
+        switch (submitType) {
+            case 1:
+                for (WorkUnit workUnit : workUnits) {
+                    if (workUnit.getIteration() == null) {
+                        Iteration iteration = new Iteration(category.getExternalId(),
+                                category.getName(),
+                                category.getDescription(),
+                                category.getProjectInstance().getProjectId());
+                        return iterationService.saveIteration(iteration);
+                    }
+                }
+                break;
+            case 2:
+                for (WorkUnit workUnit : workUnits) {
+                    if (workUnit.getPhase() == null) {
+                        Phase phase = new Phase(category.getExternalId(),
+                                category.getName(),
+                                category.getDescription(),
+                                category.getProjectInstance().getProjectId());
+                        return phaseService.savePhase(phase);
+                    }
+                }
+                break;
+            case 3:
+                for (WorkUnit workUnit : workUnits) {
+                    if (workUnit.getActivity() == null) {
+                        Activity activity = new Activity(category.getExternalId(),
+                                category.getName(),
+                                category.getDescription(),
+                                category.getProjectInstance().getProjectId());
+                        return activityService.saveActivity(activity);
+                    }
+                }
+                break;
+        }
+        return null;
+    }
+
 }