D-Net

package eu.dnetlib.uoaadmintools.controllers;

import eu.dnetlib.uoaadmintools.dao.CommunityDAO;

import eu.dnetlib.uoaadmintools.dao.DivIdDAO;

import eu.dnetlib.uoaadmintools.dao.PageDAO;

import eu.dnetlib.uoaadmintools.entities.*;

import org.apache.log4j.Logger;

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

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

import java.util.*;

@RestController

@CrossOrigin(origins = "*")

public class DivIdController {

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

    @Autowired

    private DivIdDAO divIdDAO;

    @Autowired

    private PageDAO pageDAO;

    @Autowired

    private DivHelpContentController divHelpContentController;

    DivIdController() {}

    @RequestMapping(value = "/div", method = RequestMethod.GET)

    public List<DivId> getDivIds(@RequestParam(required = false) String page,

                                 @RequestParam(required = false) String name) {

        List<DivId> divIds = null;

        if(page != null && name != null) {

            DivId divId = divIdDAO.findByPagesContainingAndName(page, name);

            if(divId != null) {

                divIds = new ArrayList<>();

                divIds.add(divId);

            }

        } else if(page != null) {

            divIds = divIdDAO.findByPagesContaining(page);

        } else if(name != null) {

            DivId divId = divIdDAO.findByName(name);

            if(divId != null) {

                divIds = new ArrayList<>();

                divIds.add(divId);

            }

        } else {

            divIds = divIdDAO.findAll();

        }

        return divIds;

    }

    @RequestMapping(value = "/divFull", method = RequestMethod.GET)

    public List<DivIdResponse> getDivIdsFull(@RequestParam(required = false) String page,

                                             @RequestParam(required = false) String name) {

        List<DivId> divIds = this.getDivIds(page, name);

        List<DivIdResponse> divIdResponses = new ArrayList<>();

        for(DivId divId : divIds) {

            DivIdResponse divIdResponse = new DivIdResponse(divId);

            List<Page> pages = new ArrayList<>();

            for(String pageId : divId.getPages()) {

                pages.add(pageDAO.findById(pageId));

            }

            divIdResponse.setPages(pages);

            divIdResponses.add(divIdResponse);

        }

        return divIdResponses;

    }

    @RequestMapping(value = "/div/{id}", method = RequestMethod.GET)

    public DivId getDivId(@PathVariable(value = "id") String id) {

        return divIdDAO.findById(id);

    }

    @RequestMapping(value = "/divFull/{id}", method = RequestMethod.GET)

    public DivIdResponse getDivIdFull(@PathVariable(value = "id") String id) {

        DivId divId = divIdDAO.findById(id);

        return divIdResponseFromDivId(divId);

    }

    public DivIdResponse divIdResponseFromDivId(DivId divId) {

        DivIdResponse divIdResponse = new DivIdResponse(divId);

        List<Page> pages = new ArrayList<>();

        for(String pageId : divId.getPages()) {

            pages.add(pageDAO.findById(pageId));

        }

        divIdResponse.setPages(pages);

        return divIdResponse;

    }

    @RequestMapping(value = "/div", method = RequestMethod.DELETE)

    public void deleteAllDivIds() {

        divIdDAO.deleteAll();

    }

    @RequestMapping(value = "/div/save", method = RequestMethod.POST)

    public DivIdResponse insertDivId(@RequestBody DivIdResponse divIdResponse) {

        DivId divId = this.getDivIdByDivIdResponse(divIdResponse);

        DivId savedDivId = divIdDAO.save(divId);

        divIdResponse.setId(savedDivId.getId());

        return this.getDivIdFull(divId.getId());

    }

    private DivId getDivIdByDivIdResponse(DivIdResponse divIdResponse) {

        DivId divId = new DivId();

        divId.setId(divIdResponse.getId());

        divId.setName(divIdResponse.getName());

        List<Page> fullPages = divIdResponse.getPages();

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

        for(Page page : fullPages) {

            pages.add(page.getId());

        }

        divId.setPages(pages);

        return divId;

    }

    @RequestMapping(value = "/div/update", method = RequestMethod.POST)

    public DivIdResponse updateDivId(@RequestBody DivIdResponse divIdResponse) {

        /*

        DivId divId = divIdDAO.findById(_divId.getId());

        divId.setName(_divId.getName());

        divId.setPages(_divId.getPages());

        divIdDAO.save(divId);

        return this.getDivIdFull(divId.getId());

        */

        DivId divId = this.getDivIdByDivIdResponse(divIdResponse);

        divIdDAO.save(divId);

        return divIdResponse;

    }

    @RequestMapping(value = "/div/delete", method = RequestMethod.POST)

    public Boolean deleteDivIds(@RequestBody List<String> divIds) throws Exception {

        for (String id: divIds) {

            DivId divId = divIdDAO.findById(id);

            // delete div contents related to this divId

            List<DivHelpContentResponse> divHelpContentResponses = divHelpContentController.getDivHelpContents(null, null, divId.getName(), null);

            for(DivHelpContentResponse divHelpContentResponse : divHelpContentResponses) {

                divHelpContentController.deleteDivHelpContent(divHelpContentResponse.getId());

            }

            divIdDAO.delete(id);

        }

        return true;

    }

    @RequestMapping(value = "/div/{id}", method = RequestMethod.DELETE)

    public void deleteDivId(@PathVariable(value = "id") String id) {

        divIdDAO.delete(id);

    }

    @RequestMapping(value = "/div/pages", method = RequestMethod.GET)

    public Set<String> getDivIdsPages() {

        List<DivId> divIds = null;

        Set<String> hasCommunityPageDivIds = new HashSet<>();

        divIds = divIdDAO.findAll();

        for(DivId divId : divIds) {

            for(String pageId : divId.getPages()) {

                hasCommunityPageDivIds.add(pageId);

            }

        }

        return hasCommunityPageDivIds;

    }

}