D-Net

package eu.dnetlib.uoaadmintools.controllers;

import eu.dnetlib.uoaadmintools.dao.CommunityDAO;

import eu.dnetlib.uoaadmintools.dao.CommunitySubscribersDAO;

import eu.dnetlib.uoaadmintools.dao.SubscriberDAO;

import eu.dnetlib.uoaadmintools.entities.Community;

import eu.dnetlib.uoaadmintools.entities.CommunitySubscribers;

import eu.dnetlib.uoaadmintools.entities.Subscriber;

import eu.dnetlib.uoaadmintools.handlers.NotFoundException;

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.List;

/**

 * Created by argirok on 2/3/2018.

 */

@RestController

@CrossOrigin(origins = "*")

public class CommunitySubscribersController {

    @Autowired

    CommunitySubscribersDAO communitySubscriberDAO;

    @Autowired

    SubscriberDAO subscriberDAO;

    @Autowired

    CommunityDAO communityDAO;

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

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

    public List<CommunitySubscribers> getAllCommunitySubscribers(){

        return communitySubscriberDAO.findAll();

    }

    @RequestMapping(value = "/community/{pid}/subscribers", method = RequestMethod.GET)

    public CommunitySubscribers getSubscribersPerCommunity(@PathVariable(value="pid", required = true) String pid) throws NotFoundException {

        CommunitySubscribers communitySubscribers = communitySubscriberDAO.findByPid(pid);

        if(communitySubscribers != null){

            return communitySubscribers;

        }else{

            throw new NotFoundException("Community Subscribers not found");

        }

    }

    @RequestMapping(value = "/community/{pid}/subscribers", method = RequestMethod.POST)

    public CommunitySubscribers addSubscriberInCommunity(@PathVariable(value="pid", required = true) String pid, @RequestBody Subscriber subscriber) throws NotFoundException {

        CommunitySubscribers communitySubscribers = communitySubscriberDAO.findByPid(pid);

        if(communitySubscribers == null){

            throw new NotFoundException("Community Subscribers not found");

        }

        Subscriber savedSubscriber = subscriberDAO.findByEmail(subscriber.getEmail());

        if(savedSubscriber==null){

            savedSubscriber = subscriberDAO.save(subscriber);

        }

        communitySubscribers.getSubscribers().add(savedSubscriber);

        return communitySubscriberDAO.save(communitySubscribers);

    }

    @RequestMapping(value = "/community/{pid}/subscribers/delete", method = RequestMethod.POST)

    public CommunitySubscribers deleteSubscriberFromCommunity(@PathVariable(value="pid", required = true) String pid, @RequestBody List<String> emails) throws NotFoundException {

        CommunitySubscribers communitySubscribers = communitySubscriberDAO.findByPid(pid);

        if(communitySubscribers == null){

            throw new NotFoundException("Community Subscribers not found");

        }

        List<Subscriber> list = new ArrayList<>();

        for(Subscriber s:communitySubscribers.getSubscribers()){

            if(emails.indexOf(s.getEmail())==-1){

             list.add(s);

            }

        }

        communitySubscribers.setSubscribers(list);

        return communitySubscriberDAO.save(communitySubscribers);

    }

    @RequestMapping(value = "/subscriber/{email}/communities", method = RequestMethod.GET)

    public List<Community> getCommunitiesPerSubcriber(@PathVariable(value="email", required = true) String email) {

        List<CommunitySubscribers> communitySubscribers = communitySubscriberDAO.findAll();

        List<Community> list = new ArrayList<>();

        for(CommunitySubscribers s:communitySubscribers){

            if(s.getSubscribers().indexOf(email)!=-1){

                list.add(communityDAO.findByPid(s.getPid()));

            }

        }

        return list;

    }

}