uoa-monitor-service/src/main/java/eu/dnetlib/uoamonitorservice/controllers/StakeholderController.java

package eu.dnetlib.uoamonitorservice.controllers;

//import com.fasterxml.jackson.core.type.TypeReference;
//import com.fasterxml.jackson.databind.ObjectMapper;
import eu.dnetlib.uoamonitorservice.dao.*;
import eu.dnetlib.uoamonitorservice.entities.*;
import eu.dnetlib.uoamonitorservice.handlers.EntityNotFoundException;
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.Date;
import java.util.List;

@RestController
@CrossOrigin(origins = "*")
public class StakeholderController {
    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 IndicatorDAO indicatorDAO;

    @Autowired
    private TopicController topicController;

    @RequestMapping(value = "/build-stakeholder", method = RequestMethod.POST)
    public Stakeholder<Topic<Category<SubCategory<Indicator>>>> buildFullStakeholder(@RequestBody Stakeholder<Topic<Category<SubCategory<Indicator>>>> stakeholderFull) {
        log.debug("build stakeholder");

        Stakeholder<String> stakeholder = new Stakeholder<>(stakeholderFull);

        List<String> topics = new ArrayList<>();
        List<Topic<Category<SubCategory<Indicator>>>> topicsFull = new ArrayList<>();
        for(Topic topic : stakeholderFull.getTopics()) {
            Topic<Category<SubCategory<Indicator>>> topicFull = topicController.buildTopic(topic);
            topicsFull.add(topicFull);
            topics.add(topicFull.getId());
        }
        stakeholderFull.setTopics(topicsFull);
        stakeholder.setTopics(topics);

        Stakeholder<String> stakeholderSaved = stakeholderDAO.save(stakeholder);
        stakeholderFull.setId(stakeholderSaved.getId());
        return stakeholderFull;
        //return null;
    }

    public Stakeholder setFullEntities(Stakeholder<String> stakeholder) {
        Stakeholder<Topic> stakeholderFull = new Stakeholder<>(stakeholder);

        List<Topic> topics = new ArrayList<>();

        for (String topicId: (List<String>)stakeholder.getTopics()) {
            Topic<String> topic = topicDAO.findById(topicId);
            Topic<Category> topicFull = new Topic<Category>(topic);

            List<Category> categories = new ArrayList<>();

            for(String categoryId : topic.getCategories()) {
                Category<String> category = categoryDAO.findById(categoryId);
                Category<SubCategory> categoryFull = new Category<SubCategory>(category);

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

                for(String subCategoryId : category.getSubCategories()) {
                    SubCategory<String> subCategory = subCategoryDAO.findById(subCategoryId);
                    SubCategory subCategoryFull = new SubCategory<Indicator>(subCategory);

                    List<Indicator> charts = new ArrayList<>();
                    for(String indicatorId : subCategory.getCharts()) {
                        charts.add(indicatorDAO.findById(indicatorId));
                    }
                    subCategoryFull.setCharts(charts);

                    List<Indicator> numbers = new ArrayList<>();
                    for(String indicatorId : subCategory.getNumbers()) {
                        numbers.add(indicatorDAO.findById(indicatorId));
                    }
                    subCategoryFull.setNumbers(numbers);

                    subCategories.add(subCategoryFull);
                }

                categoryFull.setSubCategories(subCategories);
                categories.add(categoryFull);
            }

            topicFull.setCategories(categories);
            topics.add(topicFull);
        }

        stakeholderFull.setTopics(topics);
        return stakeholderFull;
    }

    @RequestMapping(value = "/stakeholder/all", method = RequestMethod.GET)
    public List<Stakeholder> getAllStakeholders(@RequestParam(required = false) String type) {
        List<Stakeholder> stakeholders;
        if(type == null) {
            stakeholders = stakeholderDAO.findAll();
        } else {
            stakeholders = stakeholderDAO.findByType(type);
        }

        List<Stakeholder> stakeholdersFull = new ArrayList<>();
        for(Stakeholder stakeholder : stakeholders) {
            stakeholdersFull.add(this.setFullEntities(stakeholder));
        }

        return stakeholdersFull;
    }

    @RequestMapping(value = "/stakeholder/default", method = RequestMethod.GET)
    public List<Stakeholder> getAllDefaultStakeholders(@RequestParam(required = false) String type) {
        List<Stakeholder> stakeholders;
        if(type == null) {
            stakeholders = stakeholderDAO.findByIsDefaultProfile(true);
        } else {
            stakeholders = stakeholderDAO.findByIsDefaultProfileAndType(true, type);
        }

        List<Stakeholder> stakeholdersFull = new ArrayList<>();
        for(Stakeholder stakeholder : stakeholders) {
            stakeholdersFull.add(this.setFullEntities(stakeholder));
        }
        return stakeholdersFull;
    }

    @RequestMapping(value = "/stakeholder", method = RequestMethod.GET)
    public List<Stakeholder> getAllRealStakeholders(@RequestParam(required = false) String type) {
        List<Stakeholder> stakeholders;
        if(type == null) {
            stakeholders = stakeholderDAO.findByIsDefaultProfile(false);
        } else {
            stakeholders = stakeholderDAO.findByIsDefaultProfileAndType(false, type);
        }

        List<Stakeholder> stakeholdersFull = new ArrayList<>();
        for(Stakeholder stakeholder : stakeholders) {
            stakeholdersFull.add(this.setFullEntities(stakeholder));
        }
        log.debug(new Date());

        return stakeholdersFull;
    }

    @RequestMapping(value = "/stakeholder/{alias}", method = RequestMethod.GET)
    public Stakeholder getStakeholder(@PathVariable("alias") String alias) {
        Stakeholder<String> stakeholder = stakeholderDAO.findByAlias(alias);
        if(stakeholder == null) {
            // EXCEPTION - Stakeholder not found
            throw new EntityNotFoundException("Get stakeholder: Stakeholder with alias: "+alias+" not found");
        }
        return this.setFullEntities(stakeholder);
    }

    @RequestMapping(value = "/save", method = RequestMethod.POST)
    public Stakeholder<Topic> saveStakeholder(@RequestBody Stakeholder<Topic> stakeholderFull) {
        log.debug("save stakeholder");

//        if(stakeholderFull == null) {
//            log.debug("stakeholder null");
//            // EXCEPTION - Parameter for Stakeholder is not accepted
//        }

        Stakeholder<String> stakeholder = new Stakeholder<>(stakeholderFull);

        List<String> topics = new ArrayList<>();
        for(Topic topic : stakeholderFull.getTopics()) {
            topics.add(topic.getId());
        }
        stakeholder.setTopics(topics);

        Stakeholder<String> stakeholderSaved = stakeholderDAO.save(stakeholder);
        stakeholderFull.setId(stakeholderSaved.getId());

        topics = null;
        stakeholder = null;
        stakeholderSaved = null;

        return stakeholderFull;
    }


    @RequestMapping(value = "/{stakeholderId}/delete", method = RequestMethod.DELETE)
    public boolean deleteStakeholder(@PathVariable("stakeholderId") String stakeholderId) {
        log.debug("delete stakeholder");

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

        if(stakeholder != null) {

            for(String topicId : stakeholder.getTopics()) {
                Topic<String> topic = topicDAO.findById(topicId);
                if (topic == null) {
                    // EXCEPTION - Topic not found
                    throw new EntityNotFoundException("Delete stakeholder: Topic with id: "+topicId+" not found (topic exists in stakeholder: "+stakeholderId+")");
                }

                for (String categoryId : topic.getCategories()) {
                    Category<String> category = categoryDAO.findById(categoryId);
                    if (category == null) {
                        // EXCEPTION - Category not found
                        throw new EntityNotFoundException("Delete stakeholder: Category with id: "+categoryId+" not found (category exists in topic: "+topicId+")");
                    }

                    for (String subCategoryId : category.getSubCategories()) {
                        SubCategory<String> subcategory = subCategoryDAO.findById(subCategoryId);
                        if (subcategory == null) {
                            // EXCEPTION - SubCategory not found
                            throw new EntityNotFoundException("Delete stakeholder: SubCategory with id: "+subCategoryId+" not found (subcategory exists in category: "+categoryId+")");
                        }

                        for (String chartId : subcategory.getCharts()) {
                            indicatorDAO.delete(chartId);
                        }
                        subcategory.setCharts(null);

                        for (String numberId : subcategory.getNumbers()) {
                            indicatorDAO.delete(numberId);
                        }
                        subcategory.setNumbers(null);

                        subCategoryDAO.delete(subCategoryId);
                    }
                    category.setSubCategories(null);
                    categoryDAO.delete(categoryId);
                }
                topic.setCategories(null);
                topicDAO.delete(topicId);
            }
            stakeholder.setTopics(null);
            stakeholderDAO.delete(stakeholderId);
            log.debug("Stakeholder deleted!");
        } else {
            // EXCEPTION - Stakeholder not found
            throw new EntityNotFoundException("Delete stakeholder: Stakeholder with id: "+stakeholderId+" not found");
        }
        return true;
    }


    // The following are not supposed to be used
//    @RequestMapping(value = "/stakeholder/dates", method = RequestMethod.GET)
//    public List<Date> getAllStakeholderDates() {
//        List<Stakeholder> profiles = stakeholderDAO.findAll();
//        List<Date> profileDates = new ArrayList<>();
//
//        int i=0;
//        for(Stakeholder profile : profiles) {
//            log.debug(profile.getCreationDate());
//            profileDates.add(profile.getCreationDate());
//            log.debug(profileDates.get(i));
//            i++;
//        }
//        return profileDates;
//    }
//
//    @RequestMapping(value = "/stakeholder/dates1", method = RequestMethod.GET)
//    public List<String> getAllStakeholderDates1() {
//        List<Stakeholder> profiles = stakeholderDAO.findAll();
//        List<String> profileDates = new ArrayList<>();
//
//        for(Stakeholder profile : profiles) {
//            log.debug(profile.getCreationDate().toString());
//            profileDates.add(profile.getCreationDate().toString());
//        }
//        return profileDates;
//    }
//
//    @RequestMapping(value = "/stakeholder/dates2", method = RequestMethod.GET)
//    public List<String> getAllStakeholderDates2() {
//        List<Stakeholder> profiles = stakeholderDAO.findAll();
//        List<String> profileDates = new ArrayList<>();
//        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
//
//        for(Stakeholder profile : profiles) {
//            log.debug(format.format(profile.getCreationDate()));
//            profileDates.add(format.format(profile.getCreationDate()));
//        }
//        return profileDates;
//    }
}