alexminichino/trawell

package com.trawell.controllers;

import javax.servlet.http.HttpSession;

import com.trawell.models.Itinerary;
import com.trawell.models.User;
import com.trawell.services.ItineraryService;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;


/**
 * @author Milione Vincent
 * @author Lamberti Vincenzo
 * RestUsersController andranno mappate tutte le funzionalità relative all'itinerario per la comunicazione REST
 *
 */
@RestController
@RequestMapping(value = "/api")
public class RestItineraryController {
    @Autowired
    ItineraryService dao;

    /**
     * The method maps "/api/itinerary/add" post request that allows to add a new itinerary ad
     * to the logged user's itinerary list.
     * @author Milione Vincent
     * @param itinerary object containing the itinerary's data
     * @param session
     * @return a 200 HttpResponse if insertion was successful, 500 otherwise
     */
    @PostMapping(value = "/itinerary/add", consumes = MediaType.ALL_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Itinerary> add (@RequestBody Itinerary itinerary, HttpSession session){
        User user = (User) session.getAttribute("user");
        Itinerary createdItinerary = null;

        if (user != null) {
            itinerary.setUser(user);
            itinerary.getDestinations().stream().forEach(d -> {d.setItinerary(itinerary);});
            createdItinerary = dao.create(itinerary);
        }
     
        return createdItinerary == null ? new ResponseEntity<Itinerary>(HttpStatus.INTERNAL_SERVER_ERROR) : new ResponseEntity<Itinerary>(HttpStatus.OK);
    }
    
    /**
     * The method maps "/api/itinerary/modify" post request that allows to update a itinerary 
     * from the logged user's itinerary list.
     * @author Milione Vincent
     * @param itinerary object containing the itinerary's data
     * @param session
     * @return a 200 HttpResponse if update was successful, 500 otherwise
     */
    @RequestMapping(value = "/itinerary/modify", method = RequestMethod.POST, consumes = MediaType.ALL_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Itinerary> modify (@RequestBody Itinerary itinerary, HttpSession session) {
        User user = (User) session.getAttribute("user");
        Itinerary updatedItinerary = null;

        if (user != null) {
            itinerary.setUser(user);
            itinerary.getDestinations().stream().forEach(d -> {d.setItinerary(itinerary);});
            updatedItinerary = dao.update(itinerary);
        }

        if (updatedItinerary == null) 
            return new ResponseEntity<Itinerary>(HttpStatus.INTERNAL_SERVER_ERROR);
        else {
            user.getUserItineraries().remove(itinerary);
            user.getUserItineraries().add(updatedItinerary);
            return new ResponseEntity<Itinerary>(HttpStatus.OK);    
        }
}
    

    /**
     * The method maps "/api/itinerary/eliminate/{id}" post request that allows to delete an itinerary
     * from the logged user's itinerary list.
     * @param session
     * @param id
     * @return a 200 HttpResponse if deletion was successful, 500 otherwise
     */
    @RequestMapping(value = "/itinerary/eliminate/{id}", method = RequestMethod.POST, consumes = MediaType.ALL_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Itinerary> delete (HttpSession session, @PathVariable("id") Long id) {
        User user = (User) session.getAttribute("user");

        if (user != null) {
            user.getUserItineraries().remove(new Itinerary (id));
            dao.delete(id);
            return new ResponseEntity<Itinerary>(HttpStatus.OK);
        }
        
        return new ResponseEntity<Itinerary>(HttpStatus.INTERNAL_SERVER_ERROR);
    }

    /**
     * The method maps "/api/itinerary/import/{id}" post request that allows to import a public itinerary
     * @param session
     * @param id
     * @return a 200 HttpResponse if publicizing was successful, 500 otherwise
     */
    @RequestMapping(value = "/itinerary/import/{id}", method = RequestMethod.POST, consumes = MediaType.ALL_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Itinerary> importItinerary (HttpSession session, @PathVariable("id") Long id) {
        User user = (User) session.getAttribute("user");

        if (user != null) {
            Itinerary itinerary = dao.findOne(id);
            itinerary.setUser(user);
            itinerary.setId(null);

            return dao.create(itinerary) == null ? new ResponseEntity<Itinerary>(HttpStatus.OK) : new ResponseEntity<Itinerary>(HttpStatus.INTERNAL_SERVER_ERROR);
        }
        
        return new ResponseEntity<Itinerary>(HttpStatus.INTERNAL_SERVER_ERROR);
    }
    /**
     * The method maps "/api/itinerary/publicize/{id}" post request that allows to publicize an itinerary
     * from the logged user's itinerary list.
     * @param session
     * @param id
     * @return a 200 HttpResponse if publicizing was successful, 500 otherwise
     */
    @RequestMapping(value = "/itinerary/publicize/{id}", method = RequestMethod.POST, consumes = MediaType.ALL_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<Itinerary> publicize (HttpSession session, @PathVariable("id") Long id) {
        User user = (User) session.getAttribute("user");

        if (user != null) {
            Itinerary itinerary = dao.findOne(id);
            itinerary.setUser(null);
            itinerary.setId(null);

            return dao.create(itinerary) == null ? new ResponseEntity<Itinerary>(HttpStatus.OK) : new ResponseEntity<Itinerary>(HttpStatus.INTERNAL_SERVER_ERROR);
        }
        
        return new ResponseEntity<Itinerary>(HttpStatus.INTERNAL_SERVER_ERROR);
    }

}