D-Net

package eu.dnetlib.uoamonitorservice.controllers;

import eu.dnetlib.uoamonitorservice.dao.*;

import eu.dnetlib.uoamonitorservice.entities.*;

import eu.dnetlib.uoamonitorservice.handlers.EntityNotFoundException;

import eu.dnetlib.uoamonitorservice.handlers.PathNotValidException;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.security.access.prepost.PreAuthorize;

import java.util.ArrayList;

@RestController

@CrossOrigin(origins = "*")

public class SubCategoryController {

    private final Logger log = Logger.getLogger(this.getClass());

    @Autowired

    @Autowired

    @Autowired

    private TopicDAO topicDAO;

    @Autowired

    private CategoryDAO categoryDAO;

    @Autowired

    private SubCategoryDAO subCategoryDAO;

    @Autowired

    @Autowired

    private SectionController sectionController;

    public SubCategory<Section<Indicator>> buildSubCategory(SubCategory<Section<Indicator>> subcategoryFull) {

        List<Section<Indicator>> sectionChartsFull = new ArrayList<>();

        for(Section section : subcategoryFull.getCharts()) {

            Section<Indicator> sectionFull = sectionController.buildSection(section);

            sectionChartsFull.add(sectionFull);

            sectionCharts.add(sectionFull.getId());

        subCategory.setCharts(sectionCharts);

        List<Section<Indicator>> sectionNumbersFull = new ArrayList<>();

        for(Section section : subcategoryFull.getNumbers()) {

            Section<Indicator> sectionFull = sectionController.buildSection(section);

            sectionNumbersFull.add(sectionFull);

            sectionNumbers.add(sectionFull.getId());

        subCategory.setNumbers(sectionNumbers);

        subCategory.setCreationDate(date);

        subCategory.setUpdateDate(date);

        subcategoryFull.setCreationDate(date);

        subcategoryFull.setUpdateDate(date);

    }

                                                           @PathVariable("topicId") String topicId,

                                                           @PathVariable("categoryId") String categoryId,

                                                           @RequestBody SubCategory<Section<Indicator>> subcategoryFull) {

        Date date = new Date();

        subCategory.setUpdateDate(date);

        subcategoryFull.setUpdateDate(date);

        if(subcategoryFull.getId() != null) {

            oldSubcategory = subCategoryDAO.findById(subcategoryFull.getId());

            subCategory.setCreationDate(date);

            subcategoryFull.setCreationDate(date);

//        for(Indicator chart : subcategoryFull.getCharts()) {

//            charts.add(chart.getId());

//        }

//        subCategory.setCharts(charts);

//

//        List<String> numbers = new ArrayList<>();

//        for(Indicator numbr : subcategoryFull.getNumbers()) {

//            numbers.add(numbr.getId());

//        }

//        subCategory.setNumbers(numbers);

        for(Section chartSection : subcategoryFull.getCharts()) {

            chartSections.add(chartSection.getId());

        }

        subCategory.setCharts(chartSections);

        for(Section numberSection : subcategoryFull.getNumbers()) {

            numberSections.add(numberSection.getId());

        }

        subCategory.setNumbers(numberSections);

        if(stakeholder.getDefaultId() == null) {

            if(subcategoryFull.getId() == null) {

            } else {

            subCategoryDAO.save(subCategory);

        int index = subcategories.indexOf(subCategory.getId());

        if(index == -1) {

            subcategories.add(subCategory.getId());

            categoryDAO.save(category);

            log.debug("Subcategory saved!");

        }

        numberSections = null;

        subCategory = null;

        return subcategoryFull;

    }

        log.debug("On save default subCategory");

        List<Category> categories = categoryDAO.findByDefaultId(categoryId);

        for(Category category : categories) {

            SubCategory subCategoryNew = new SubCategory();

            subCategoryNew.copyFromDefault(subCategory);

            subCategoryDAO.save(subCategoryNew);

            List<String> subCategories = category.getSubCategories();

            subCategories.add(subCategoryNew.getId());

            categoryDAO.save(category);

        }

    }

        List<SubCategory> subCategories = subCategoryDAO.findByDefaultId(subCategory.getId());

        boolean changed = false;

        for(SubCategory subCategoryBasedOnDefault : subCategories) {

                    && (oldSubcategory.getName() == null || oldSubcategory.getName().equals(subCategoryBasedOnDefault.getName()))) {

                subCategoryBasedOnDefault.setName(subCategory.getName());

            }

                    && (oldSubcategory.getDescription() == null || oldSubcategory.getDescription().equals(subCategoryBasedOnDefault.getDescription()))) {

                subCategoryBasedOnDefault.setDescription(subCategory.getDescription());

            }

            if(!changed) {

                continue;

//            subCategoryBasedOnDefault.setName(subCategory.getName());

//            subCategoryBasedOnDefault.setDescription(subCategory.getDescription());

        }

    }

    public boolean deleteSubCategory(@PathVariable("stakeholderId") String stakeholderId,

                                     @PathVariable("topicId") String topicId,

                                     @PathVariable("categoryId") String categoryId,

                                     @RequestParam(required = false) String children) {

        if(subcategory != null) {

            Stakeholder<String> stakeholder = stakeholderDAO.findById(stakeholderId);

            List<String> roles = rolesUtils.getRoles();

            if(subcategory.getDefaultId() != null && !rolesUtils.hasCreateAndDeleteAuthority(roles, stakeholder.getType())) {

                // EXCEPTION - Access denied

                throw new AccessDeniedException("Delete subcategory: You are not authorized to delete a default SubCategory in stakeholder with id: "+stakeholderId);

            }

            int index = subcategories.indexOf(subcategoryId);

            if(index != -1) {

                if(category.getDefaultId() == null && children != null) {

                    onDeleteDefaultSubCategory(subcategoryId, categoryId, children);

//                    Section<String> chartSection = sectionDAO.findById(chartSectionId);

//                    if (chartSection == null) {

//                        // EXCEPTION - Section not found

//                        throw new EntityNotFoundException("Delete SubCategory: Section with id: "+chartSectionId+" not found (section exists in subcategory: "+subcategoryId+")");

//                    }

//

//                    for (String chartId : chartSection.getIndicators()) {

//                        indicatorDAO.delete(chartId);

//                    }

//                    subcategory.setCharts(null);

//                    sectionDAO.delete(chartSectionId);

//                }

//

//                for(String numberSectionId : subcategory.getNumbers()) {

//                    Section<String> numberSection = sectionDAO.findById(numberSectionId);

//                    if (numberSection == null) {

//                        // EXCEPTION - Section not found

//                        throw new EntityNotFoundException("Delete SubCategory: Section with id: "+numberSectionId+" not found (section exists in subcategory: "+subcategoryId+")");

//                    }

//

//                    for (String numberId : numberSection.getIndicators()) {

//                        indicatorDAO.delete(numberId);

//                    }

//                    subcategory.setNumbers(null);

//                    sectionDAO.delete(numberSectionId);

//                }

                subcategory.setNumbers(null);

                categoryDAO.save(category);

                log.debug("Subcategory deleted!");

                throw new PathNotValidException("Delete subcategory: Subcategory with id: "+subcategoryId+" not found in Category: "+categoryId);

            throw new EntityNotFoundException("Delete subcategory: SubCategory with id: "+subcategoryId+" not found");

        return true;

    }

        if(children.equals("delete")) {

            List<Category> categories = categoryDAO.findByDefaultId(defaultCategoryId);

            List<SubCategory> subCategories = subCategoryDAO.findByDefaultId(defaultSubCategoryId);

            for(Category category : categories) {

                Iterator<SubCategory> subCategoriesIterator = subCategories.iterator();

                while(subCategoriesIterator.hasNext()) {

                    SubCategory subCategory = subCategoriesIterator.next();

                    String subCategoryId = subCategory.getId();

                    if(category.getSubCategories() != null && category.getSubCategories().contains(subCategoryId)) {

                        subCategoriesIterator.remove();

                        category.getSubCategories().remove(subCategoryId);

                        categoryDAO.save(category);

                        sectionController.deleteTree(subCategory);

                        subCategoryDAO.delete(subCategoryId);

                        log.debug("SubCategory with id: "+subCategoryId+" deleted!");

                        break;

                    }

                }

            }

        } else if(children.equals("disconnect")) {

            List<SubCategory> subCategories = subCategoryDAO.findByDefaultId(defaultSubCategoryId);

            for(SubCategory subCategory : subCategories) {

                sectionController.disConnectTree(subCategory);

                subCategory.setDefaultId(null);

                subCategoryDAO.save(subCategory);

                log.debug("DefaultId for SubCategory with id: "+subCategory.getId()+" empty!");

            }

        }

        return true;

    }

    public List<SubCategory> reorderSubCategories(@PathVariable("stakeholderId") String stakeholderId,

                                            @PathVariable("topicId") String topicId,

                                            @PathVariable("categoryId") String categoryId,

                                            @RequestBody List<String> subCategories) {

        log.debug("reorder subCategories");

        log.debug("Stakeholder: "+stakeholderId + " - Topic: "+topicId + " - Category: "+categoryId);

        Category<String> category = checkForExceptions(stakeholderId, topicId, categoryId);

        category.setSubCategories(subCategories);

        categoryDAO.save(category);

        log.debug("SubCategories reordered!");

        List<SubCategory> subCategoriesFull = new ArrayList<>();

        for(String subCategoryId : subCategories) {

            subCategoriesFull.add(subCategoryDAO.findById(subCategoryId));

        }

        return subCategoriesFull;

    }

//    public Boolean toggleSubCategoryStatus(@PathVariable("stakeholderId") String stakeholderId,

//                                           @PathVariable("topicId") String topicId,

//                                           @PathVariable("categoryId") String categoryId,

//                                           @PathVariable("subcategoryId") String subcategoryId) {

//        log.debug("toggle subCategory status (isActive)");

//        log.debug("Stakeholder: "+stakeholderId + " - Topic: "+topicId + " - Category: "+categoryId+ " - SubCategory: "+subcategoryId);

//

//        SubCategory subCategory = subCategoryDAO.findById(subcategoryId);

//        if (subCategory == null) {

//            // EXCEPTION - SubCategory not found

//            throw new EntityNotFoundException("Toggle subCategory status: SubCategory with id: "+subcategoryId+" not found");

//        }

//        subCategory.setIsActive(!subCategory.getIsActive());

//

//        this.toggleSubCategory(stakeholderId, topicId, categoryId, subCategory);

//

//        return subCategory.getIsActive();

//    }

//

//    @RequestMapping(value = "/{stakeholderId}/{topicId}/{categoryId}/{subcategoryId}/toggle-access", method = RequestMethod.POST)

//    public Boolean toggleSubCategoryAccess(@PathVariable("stakeholderId") String stakeholderId,

//                                           @PathVariable("topicId") String topicId,

//                                           @PathVariable("categoryId") String categoryId,

//                                           @PathVariable("subcategoryId") String subcategoryId) {

//        log.debug("toggle subCategory access (isPublic)");

//        log.debug("Stakeholder: "+stakeholderId + " - Topic: "+topicId + " - Category: "+categoryId+ " - SubCategory: "+subcategoryId);

//

//        SubCategory subCategory = subCategoryDAO.findById(subcategoryId);

//        if (subCategory == null) {

//            // EXCEPTION - SubCategory not found

//            throw new EntityNotFoundException("Toggle subCategory access: SubCategory with id: "+subcategoryId+" not found");

//        }

//        subCategory.setIsPublic(!subCategory.getIsPublic());

//

//        this.toggleSubCategory(stakeholderId, topicId, categoryId, subCategory);

//

//        return subCategory.getIsPublic();

//    }

    @RequestMapping(value = "/{stakeholderId}/{topicId}/{categoryId}/{subcategoryId}/change-visibility", method = RequestMethod.POST)

    public Visibility changeTopicVisibility(@PathVariable("stakeholderId") String stakeholderId,

                                            @PathVariable("topicId") String topicId,

                                            @PathVariable("categoryId") String categoryId,

                                            @PathVariable("subcategoryId") String subcategoryId,

                                            @RequestParam("visibility") Visibility visibility) {

        log.debug("change subCategory visibility: "+visibility);

        SubCategory subCategory = subCategoryDAO.findById(subcategoryId);

        if (subCategory == null) {

            // EXCEPTION - SubCategory not found

        this.toggleSubCategory(stakeholderId, topicId, categoryId, subCategory);

    public void toggleSubCategory(String stakeholderId, String topicId, String categoryId, SubCategory subcategory) {

            subCategoryDAO.save(subcategory);

            log.debug("SubCategory toggled!");

        } else {

            // EXCEPTION - SubCategory not found in Stakeholder: stakeholder.getAlias(); -> Topic: topic.getAlias(); -> Category: category.getAlias();

            throw new PathNotValidException("Toggle subCategory: SubCategory with id: "+subcategory.getId()+" not found in Category: "+categoryId);

        }

    }

        Stakeholder<String> stakeholder = stakeholderDAO.findById(stakeholderId);

        if(stakeholder == null) {

        if(!rolesUtils.hasUpdateAuthority(roles, stakeholder.getType(), stakeholder.getAlias())) {

            // EXCEPTION - Access denied

            throw new AccessDeniedException("CheckForExceptions SubCategory: You are not authorized to update stakeholder with id: "+stakeholderId);

        }

        if(topic == null) {

            // EXCEPTION - Topic not found

            throw new EntityNotFoundException("Save indicator: Topic with id: "+topicId+" not found");

        }

        if(!stakeholder.getTopics().contains(topicId)) {

            // EXCEPTION - Topic not found in Stakeholder: stakeholder.getAlias();

            throw new PathNotValidException("Save indicator: Topic with id: " + topicId + " not found in Stakeholder: " + stakeholderId);

        }

        Category<String> category = categoryDAO.findById(categoryId);

        if(category == null) {

            // EXCEPTION - Category not found

            throw new EntityNotFoundException("Save indicator: Category with id: "+categoryId+" not found");

        }

        if(!topic.getCategories().contains(categoryId)) {

            // EXCEPTION - Category not found in Stakeholder: stakeholder.getAlias(); -> Topic: topic.getAlias();

            throw new PathNotValidException("Save indicator: Category with id: "+categoryId+" not found in Topic: "+topicId);

        }

//        SubCategory<String> subcategory = subCategoryDAO.findById(subcategoryId);

//        if(subcategory == null) {

//            // EXCEPTION - SubCategory not found

//            throw new EntityNotFoundException("Save indicator: SubCategory with id: "+subcategoryId+" not found");

//        }

//

//        if (!category.getSubCategories().contains(subcategoryId)) {

//            // EXCEPTION - SubCategory not found in Stakeholder: stakeholder.getAlias(); -> Topic: topic.getAlias(); -> Category: category.getAlias();

//            throw new PathNotValidException("Save indicator: SubCategory with id: "+subcategoryId+" not found in Category: "+categoryId);

//        }

        return  category;

        List<String> subCategories = category.getSubCategories();

        for(String subCategoryId : subCategories) {

            SubCategory subCategory = subCategoryDAO.findById(subCategoryId);

            if (subCategory == null) {

                // EXCEPTION - SubCategory not found

                throw new EntityNotFoundException("SubCategory delete tree: SubCategory with id: "+subCategoryId+" not found (subCategory exists in category: "+category.getId()+")");

            }

            sectionController.deleteTree(subCategory);

            subCategoryDAO.delete(subCategoryId);

        }

    }

    public void disConnectTree(Category category) {

        List<String> subCategories = category.getSubCategories();

        for(String subCategoryId : subCategories) {

            SubCategory subCategory = subCategoryDAO.findById(subCategoryId);

            if (subCategory == null) {

                // EXCEPTION - SubCategory not found

                throw new EntityNotFoundException("SubCategory disconnect tree: SubCategory with id: "+subCategoryId+" not found (subCategory exists in category: "+category.getId()+")");

            }

            sectionController.disConnectTree(subCategory);

            subCategory.setDefaultId(null);

            subCategoryDAO.save(subCategory);

        }

    }