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.apache.log4j.Logger;

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

import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

@RestController

@CrossOrigin(origins = "*")

public class CategoryController {

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

    @Autowired

    private StakeholderDAO stakeholderDAO;

    @Autowired

    private TopicDAO topicDAO;

    @Autowired

    private CategoryDAO categoryDAO;

    @Autowired

    private SubCategoryDAO subCategoryDAO;

    @Autowired

    private SectionDAO sectionDAO;

    @Autowired

    private IndicatorDAO indicatorDAO;

    @Autowired

    private SubCategoryController subCategoryController;

    public Category<SubCategory> buildCategory(Category<SubCategory> categoryFull) {

        Category<String> category = new Category<>(categoryFull);

        List<String> subCategories = new ArrayList<>();

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

        for(SubCategory<Section<Indicator>> subCategory : categoryFull.getSubCategories()) {

            SubCategory<Section<Indicator>> subcategoryFull = subCategoryController.buildSubCategory(subCategory);

            subCategoriesFull.add(subcategoryFull);

            subCategories.add(subcategoryFull.getId());

        }

        categoryFull.setSubCategories(subCategoriesFull);

        category.setSubCategories(subCategories);

        categoryDAO.save(category);

        categoryFull.setId(category.getId());

        return categoryFull;

    }

    @RequestMapping(value = "/{stakeholderId}/{topicId}/save", method = RequestMethod.POST)

    public Category<SubCategory> saveCategory(@PathVariable("stakeholderId") String stakeholderId,

                                              @PathVariable("topicId") String topicId,

                                              @RequestBody Category<SubCategory> categoryFull) {

        log.debug("save category");

        log.debug("Alias: "+categoryFull.getAlias() + " - Id: "+categoryFull.getId() + " - Stakeholder: "+stakeholderId + " - Topic: "+topicId);

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

        if(stakeholder != null) {

            Category<String> oldCategory = null;

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

                oldCategory = categoryDAO.findById(categoryFull.getId());

            }

            Topic<String> topic = topicDAO.findById(topicId);

            if(topic != null) {

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

                    // if category not exists (no id), create a new default subcategory, identical to category

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

                        SubCategory<String> subCategory = new SubCategory<>();

                        subCategory.createOverviewSubCategory(categoryFull);

                        subCategoryDAO.save(subCategory);

                        List<SubCategory> subCategories = categoryFull.getSubCategories();

                        subCategories.add(subCategory);

                    }

                    Category<String> category = new Category<>(categoryFull);

                    List<String> subCategories = new ArrayList<>();

                    for(SubCategory subCategory : categoryFull.getSubCategories()) {

                        subCategories.add(subCategory.getId());

                    }

                    category.setSubCategories(subCategories);

                    categoryDAO.save(category);

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

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

                            onSaveDefaultCategory(category, topicId);

                        } else {

                            onUpdateDefaultCategory(category, oldCategory);

                        }

                    }

                    List<String> categories = topic.getCategories();

                    int index = categories.indexOf(category.getId());

                    if(index == -1) {

                        categories.add(category.getId());

                        topicDAO.save(topic);

                        log.debug("Category saved!");

                        categoryFull.setId(category.getId());

                    }

                    subCategories = null;

                    category = null;

                } else {

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

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

                }

            } else {

                // EXCEPTION - Topic not found

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

            }

        } else {

            // EXCEPTION - Stakeholder not found

            throw new EntityNotFoundException("Save category: Stakeholder with id: "+stakeholderId+" not found");

        }

        return categoryFull;

    }

    public void onSaveDefaultCategory(Category<String> category, String topicId) {

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

        List<Topic> topics = topicDAO.findByDefaultId(topicId);

        for(Topic topic : topics) {

            Category categoryNew = new Category();

            categoryNew.copyFromDefault(category);

            categoryDAO.save(categoryNew);

            List<String> categories = topic.getCategories();

            categories.add(categoryNew.getId());

            topicDAO.save(topic);

        }

        String subCategoryOverviewId = category.getSubCategories().get(0);

        SubCategory subCategoryOverview = subCategoryDAO.findById(subCategoryOverviewId);

        subCategoryController.onSaveDefaultSubCategory(subCategoryOverview, category.getId());

    }

    public void onUpdateDefaultCategory(Category category, Category oldCategory) {

        log.debug("On update default category");

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

        boolean changed = false;

        for(Category categoryBasedOnDefault : categories) {

            if(category.getName() != null && !category.getName().equals(categoryBasedOnDefault.getName())

                    && (oldCategory.getName() == null || oldCategory.getName().equals(categoryBasedOnDefault.getName()))) {

                categoryBasedOnDefault.setName(category.getName());

                changed = true;

            }

            if(category.getDescription() != null && !category.getDescription().equals(categoryBasedOnDefault.getDescription())

                    && (oldCategory.getDescription() == null || oldCategory.getDescription().equals(categoryBasedOnDefault.getDescription()))) {

                categoryBasedOnDefault.setDescription(category.getDescription());

                changed = true;

            }

            if(!changed) {

//                break;

                continue;

            }

//            categoryBasedOnDefault.setName(category.getName());

//            categoryBasedOnDefault.setDescription(category.getDescription());

            categoryDAO.save(categoryBasedOnDefault);

        }

    }

    @RequestMapping(value = "/{stakeholderId}/{topicId}/{categoryId}/delete", method = RequestMethod.DELETE)

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

                                  @PathVariable("topicId") String topicId,

                                  @PathVariable("categoryId") String categoryId,

                                  @RequestParam(required = false) String children) {

        log.debug("delete category");

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

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

        if(stakeholder != null) {

            Topic<String> topic = topicDAO.findById(topicId);

            if(topic != null) {

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

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

                    if(category != null) {

                        List<String> categories = topic.getCategories();

                        int index = categories.indexOf(categoryId);

                        if(index != -1) {

                            // this category belongs in default profile

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

                                onDeleteDefaultCategory(categoryId, topicId, children);

                            }

//                            for(String subCategoryId : category.getSubCategories()) {

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

//                                if(subcategory == null) {

//                                    // EXCEPTION - SubCategory not found

//                                    throw new EntityNotFoundException("Delete category: SubCategory with id: "+subCategoryId+" not found (subcategory exists in category: "+categoryId+")");

//                                }

//

//                                for(String chartSectionId : subcategory.getCharts()) {

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

//                                    if (chartSection == null) {

//                                        // EXCEPTION - Section not found

//                                        throw new EntityNotFoundException("Delete topic: 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 topic: 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);

//                                }

//

//                                subCategoryDAO.delete(subCategoryId);

//                            }

                            subCategoryController.deleteTree(category);

                            category.setSubCategories(null);

                            categories.remove(index);

                            topicDAO.save(topic);

                            categoryDAO.delete(categoryId);

                            log.debug("Category deleted!");

                        } else {

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

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

                        }

                    } else {

                        // EXCEPTION - Category not found

                        throw new EntityNotFoundException("Delete category: Category with id: "+categoryId+" not found");

                    }

                } else {

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

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

                }

            } else {

                // EXCEPTION - Topic not found

                throw new EntityNotFoundException("Delete category: Topic with id: "+topicId+" not found");

            }

        } else {

            // EXCEPTION - Stakeholder not found

            throw new EntityNotFoundException("Delete category: Stakeholder with id: "+stakeholderId+" not found");

        }

        return true;

    }

    public boolean onDeleteDefaultCategory(String defaultCategoryId, String defaultTopicId, String children) {

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

            List<Topic> topics = topicDAO.findByDefaultId(defaultTopicId);

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

            for(Topic topic : topics) {

                Iterator<Category> categoriesIterator = categories.iterator();

                while(categoriesIterator.hasNext()) {

                    Category category = categoriesIterator.next();

                    String categoryId = category.getId();

                    if(topic.getCategories() != null && topic.getCategories().contains(categoryId)) {

                        categoriesIterator.remove();

                        topic.getCategories().remove(categoryId);

                        topicDAO.save(topic);

                        subCategoryController.deleteTree(category);

                        categoryDAO.delete(categoryId);

                        log.debug("Category with id: "+categoryId+" deleted!");

                        break;

                    }

                }

            }

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

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

            for(Category category : categories) {

                subCategoryController.disConnectTree(category);

                category.setDefaultId(null);

                categoryDAO.save(category);

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

            }

        }

        return true;

    }

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

    public Boolean toggleCategoryStatus(@PathVariable("stakeholderId") String stakeholderId,

                                        @PathVariable("topicId") String topicId,

                                        @PathVariable("categoryId") String categoryId) {

        log.debug("toggle category status (isActive)");

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

        Category category = categoryDAO.findById(categoryId);

        if (category == null) {

            // EXCEPTION - Category not found

            throw new EntityNotFoundException("Toggle category status: Category with id: "+categoryId+" not found");

        }

        category.setIsActive(!category.getIsActive());

        this.toggleCategory(stakeholderId, topicId, category);

        return category.getIsActive();

    }

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

    public Boolean toggleCategoryAccess(@PathVariable("stakeholderId") String stakeholderId,

                                        @PathVariable("topicId") String topicId,

                                        @PathVariable("categoryId") String categoryId) {

        log.debug("toggle category access (isPublic)");

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

        Category category = categoryDAO.findById(categoryId);

        if (category == null) {

            // EXCEPTION - Category not found

            throw new EntityNotFoundException("Toggle category access: Category with id: "+categoryId+" not found");

        }

        category.setIsPublic(!category.getIsPublic());

        this.toggleCategory(stakeholderId, topicId, category);

        return category.getIsPublic();

    }

    public void toggleCategory(String stakeholderId, String topicId, Category category) {

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

        if (stakeholder != null) {

            Topic<String> topic = topicDAO.findById(topicId);

            if (topic != null) {

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

                    if (topic.getCategories().contains(category.getId())) {

                        categoryDAO.save(category);

                        log.debug("Category toggled!");

                    } else {

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

                        throw new PathNotValidException("Toggle category: Category with id: "+category.getId()+" not found in Topic: "+topicId);

                    }

                } else {

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

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

                }

            } else {

                // EXCEPTION - Topic not found

                throw new EntityNotFoundException("Toggle category: Topic with id: "+topicId+" not found");

            }

        } else {

            // EXCEPTION - Stakeholder not found

            throw new EntityNotFoundException("Toggle category: Stakeholder with id: "+stakeholderId+" not found");

        }

    }

    public void deleteTree(Topic topic) {

        List<String> categories = topic.getCategories();

        for(String categoryId : categories) {

            Category category = categoryDAO.findById(categoryId);

            if (category == null) {

                // EXCEPTION - Category not found

                throw new EntityNotFoundException("Category delete tree: Category with id: "+categoryId+" not found (category exists in topic: "+topic.getId()+")");

            }

            subCategoryController.deleteTree(category);

            categoryDAO.delete(categoryId);

        }

    }

    public void disConnectTree(Topic topic) {

        List<String> categories = topic.getCategories();

        for(String categoryId : categories) {

            Category category = categoryDAO.findById(categoryId);

            if (category == null) {

                // EXCEPTION - Category not found

                throw new EntityNotFoundException("Category disconnect tree: Category with id: "+categoryId+" not found (category exists in topic: "+topic.getId()+")");

            }

            subCategoryController.disConnectTree(category);

            category.setDefaultId(null);

            categoryDAO.save(category);

        }

    }

}