1 files changed, 144 insertions, 0 deletions
diff --git a/vid-app-common/src/main/java/org/onap/vid/services/CategoryParameterServiceImpl.java b/vid-app-common/src/main/java/org/onap/vid/services/CategoryParameterServiceImpl.java
new file mode 100644
index 000000000..02d3521dc
--- /dev/null
+++ b/vid-app-common/src/main/java/org/onap/vid/services/CategoryParameterServiceImpl.java
@@ -0,0 +1,144 @@
+package org.onap.vid.services;
+
+import java.io.IOException;
+import java.util.*;
+import java.util.stream.Collectors;
+
+import org.openecomp.portalsdk.core.logging.logic.EELFLoggerDelegate;
+import org.openecomp.portalsdk.core.service.DataAccessService;
+import org.onap.vid.category.AddCategoryOptionResponse;
+import org.onap.vid.category.CategoryParameterOptionRep;
+import org.onap.vid.model.CategoryParameter;
+import org.onap.vid.model.CategoryParameterOption;
+import org.onap.vid.category.AddCategoryOptionsRequest;
+import org.onap.vid.category.CategoryParametersResponse;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.onap.vid.model.CategoryParameter.Family;
+
+import javax.ws.rs.ForbiddenException;
+
+
+@Service
+public class CategoryParameterServiceImpl implements CategoryParameterService {
+
+    @Autowired
+    private DataAccessService dataAccessService;
+
+    private static final EELFLoggerDelegate LOG = EELFLoggerDelegate.getLogger(CategoryParameterServiceImpl.class);
+
+    public static class UnfoundedCategoryException extends RuntimeException {
+
+        public UnfoundedCategoryException(String message) {
+            super(message);
+        }
+    }
+
+    public static class UnfoundedCategoryOptionException extends RuntimeException {
+
+        public UnfoundedCategoryOptionException(String message) {
+            super(message);
+        }
+    }
+
+    public static class AlreadyExistOptionNameException extends RuntimeException {
+
+        public AlreadyExistOptionNameException(String message) {
+            super(message);
+        }
+    }
+
+    @Override
+    public CategoryParametersResponse getCategoryParameters(Family familyName) throws IOException {
+        List<CategoryParameter> categoryParameters = dataAccessService.getList(CategoryParameter.class, String.format(" where family = '%s' ",familyName), null, null);
+        return convertToCategoryParametersResponse(categoryParameters);
+    }
+
+    private CategoryParametersResponse convertToCategoryParametersResponse(List<CategoryParameter> categoryParameters) {
+        Comparator<CategoryParameterOptionRep> comparator = Comparator.comparing(CategoryParameterOptionRep::getName, String.CASE_INSENSITIVE_ORDER);
+        Map<String, List<CategoryParameterOptionRep>> categoryParametersMap = categoryParameters.stream().collect(Collectors.toMap(
+                CategoryParameter::getName,
+                x ->  x.getOptions().stream().map(opt -> new CategoryParameterOptionRep(opt.getAppId(), opt.getName())).sorted(comparator).collect(Collectors.toList())));
+        return new CategoryParametersResponse(categoryParametersMap);
+    }
+
+    @Override
+    public AddCategoryOptionResponse createCategoryParameterOptions(String categoryName, AddCategoryOptionsRequest optionsRequest) throws IOException, UnfoundedCategoryException {
+
+        AddCategoryOptionResponse response = new AddCategoryOptionResponse(new ArrayList<>());
+        CategoryParameter categoryParameter = getCategoryParameter(categoryName);
+        Set<String> categoryOptions = categoryParameter.getOptions().stream().map(CategoryParameterOption::getName).collect(Collectors.toSet());
+        for (String optionName : optionsRequest.options) {
+            if (categoryOptions.contains(optionName)) {
+                response.getErrors().add(String.format("Option %s already exist for category %s", optionName, categoryName));
+                continue;
+            }
+            String appId = categoryParameter.isIdSupported() ? UUID.randomUUID().toString() : optionName;
+            CategoryParameterOption categoryParameterOption = new CategoryParameterOption(appId, optionName, categoryParameter);
+            dataAccessService.saveDomainObject(categoryParameterOption, null);
+        }
+
+        return response;
+    }
+
+    private CategoryParameter getCategoryParameter( String categoryName) {
+        List<CategoryParameter> categoryParameters = dataAccessService.getList(CategoryParameter.class, String.format(" where name = '%s' ", categoryName), null, null);
+        if (categoryParameters.size() != 1) {
+            String msg = "There is no category parameter with name " + categoryName;
+            LOG.debug(msg);
+            throw new UnfoundedCategoryException(msg);
+        }
+
+
+        return categoryParameters.get(0);
+    }
+
+    @Override
+    public AddCategoryOptionResponse updateCategoryParameterOption(String categoryName, CategoryParameterOptionRep option) {
+        AddCategoryOptionResponse response = new AddCategoryOptionResponse(new ArrayList<>());
+        CategoryParameter categoryParameter = getCategoryParameter(categoryName);
+        if (!categoryParameter.isIdSupported()) {
+            String msg = "Updating option name for category: " + categoryName + ", is not allowed";
+            LOG.debug(msg);
+            throw new ForbiddenException(msg);
+        }
+        Optional<CategoryParameterOption> categoryParameterOptionOptional = categoryParameter.getOptions().stream().filter(x->x.getAppId().equals(option.getId())).findFirst();
+        if (!categoryParameterOptionOptional.isPresent()) {
+            String msg = "There is no option with id "+option.getId() + " for category " + categoryName;
+            LOG.debug(msg);
+            throw new UnfoundedCategoryOptionException(msg);
+        }
+        CategoryParameterOption categoryParameterOption = categoryParameterOptionOptional.get();
+        Optional<CategoryParameterOption> alreadyExistOptionWithName = categoryParameter.getOptions().stream().filter(x->x.getName().equals(option.getName())).findFirst();
+        if (alreadyExistOptionWithName.isPresent() && !alreadyExistOptionWithName.get().getAppId().equals(categoryParameterOption.getAppId())) {
+            String msg = "Option with name "+option.getName() + " already exist for category " + categoryName;
+            LOG.debug(msg);
+            throw new AlreadyExistOptionNameException(msg);
+        }
+
+        categoryParameterOption.setName(option.getName());
+        dataAccessService.saveDomainObject(categoryParameterOption, null);
+
+        return response;
+    }
+
+    @Override
+    public void deleteCategoryOption(String categoryName, CategoryParameterOption option) throws IOException {
+        List<CategoryParameter> categoryParameters = dataAccessService.getList(CategoryParameter.class, String.format(" where name = '%s'", categoryName), null, null);
+        if (categoryParameters.size() != 1) {
+            String msg = "There is no category parameter with name " + categoryName;
+            LOG.debug(msg);
+            throw new UnfoundedCategoryException(msg);
+        }
+        CategoryParameter categoryParameter = categoryParameters.get(0);
+        Set<CategoryParameterOption> categoryOptions = categoryParameter.getOptions();
+        for (CategoryParameterOption categoryOption: categoryOptions) {
+            if(categoryOption.getName().equals(option.getName()))
+            {
+                dataAccessService.deleteDomainObject(categoryOption, null);
+            }
+        }
+    }
+
+}
+