MainController.java

package ac.uk.bristol.cs.santa.grotto;

import ac.uk.bristol.cs.santa.grotto.business.GeoLookup;
import ac.uk.bristol.cs.santa.grotto.business.data.*;
import ac.uk.bristol.cs.santa.grotto.business.route.Location;
import ac.uk.bristol.cs.santa.grotto.business.route.LocationRoutePlanning;
import ac.uk.bristol.cs.santa.grotto.rest.GrottoDTO;
import com.google.maps.errors.ApiException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.util.Pair;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.config.annotation.ViewControllerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import javax.validation.Valid;
import java.io.IOException;
import java.security.Principal;
import java.util.ArrayList;
import java.util.stream.Collectors;

/**
 * Created by csxds on 24/11/2017.
 */
@Controller
public class MainController extends WebMvcConfigurerAdapter {

    private static final Logger logger = LoggerFactory.getLogger(MainController.class);

    @Autowired
    private UserRepository userRepository;

    /**
     * view controllers without logic
     *
     * @param registry
     */
    @Override
    public void addViewControllers(ViewControllerRegistry registry) {
//        registry.addViewController("/contact");
        registry.addViewController("/").setViewName("index");
        registry.addViewController("/login");
        registry.addViewController("/terms").setViewName("terms");
    }

    @RequestMapping("/login-error.html")
    public String loginError(Model model) {
        model.addAttribute("loginError", true);
        return "login";
    }

    @Autowired
    private GeoLookup geoLookup;

    @Autowired
    private GrottoRepository grottoRepository;

    @Autowired
    private EventRepository eventRepository;

    @GetMapping("/grotto/add")
    public String addGrotto(Model model) {
        model.addAttribute("grotto", new Grotto());
        return "grotto_form";
    }


    @PostMapping("/grotto")
    public String submitGrotto(@ModelAttribute Grotto grotto) {
        grottoRepository.save(grotto);
        return "grotto_view";
    }


    @GetMapping("/grotto/{id}")
    public String viewGrotto(@PathVariable Long id, Model model) {
        model.addAttribute("grotto", grottoRepository.findOne(id));
        return "grotto_view";
    }


    @GetMapping("/event/add")
    public String addEvent(Model model) {
        model.addAttribute("event", new Event());
        model.addAttribute("grottos", grottoRepository.findAll());
        return "event_form";
    }

    @Autowired
    private ContactRequestRepository contactRepository;


    @GetMapping("/contact")
    public String showContactForm(ContactRequest contact) {
        return "contact";
    }

    @PostMapping(value = "/contact")
    public String submitContact(@Valid ContactRequest contact, BindingResult binding, RedirectAttributes attr) {
        if (binding.hasErrors()) {
            return "/contact";
        }
        contactRepository.save(contact);
        attr.addFlashAttribute("message", "Thank you for your message. We'll be in touch ASAP");
        return "redirect:/contact";
    }


    @PostMapping("/event")
    public String submitEvent(@ModelAttribute Event event) {
        eventRepository.save(event);
        return "event_view";
    }

    @GetMapping("/event/{id}")
    public String viewEvent(@PathVariable Long id, Model model) {
        model.addAttribute("event", eventRepository.findOne(id));
        return "event_view";
    }

    @GetMapping("/eventbooking/add")
    public String addEventBooking(Model model) {
        model.addAttribute("eventbooking", new EventBooking());
        model.addAttribute("events", eventRepository.findAll());
        return "eventbooking_form";
    }

    @Autowired
    private
    EventBookingRepository eventBookingRepository;

    /**
     * Create booking for logged in user.
     * <p>
     * We get the logged in user from Spring and then look up the UserAccount for this user.
     *
     * @param eventbooking
     * @param principal
     * @param model
     * @return
     */
    @PostMapping("/eventbooking")
    public String submitEventBooking(@ModelAttribute EventBooking eventbooking, Principal principal, Model model) {

        UserAccount userAccount = userRepository.findByUserName(principal.getName());
        eventbooking.setUserAccount(userAccount);
        eventBookingRepository.save(eventbooking);

        model.addAttribute("eventbooking", eventbooking);

        return "eventbooking_view";
    }

    @GetMapping("/eventbooking/{id}")
    public String viewEventBooking(@PathVariable Long id, Model model) {
        model.addAttribute("eventbooking", eventBookingRepository.findOne(id));
        return "eventbooking_view";
    }


    @RequestMapping(value = "/geolookup", method = RequestMethod.POST)
    @ResponseBody
    public String geolookup(@RequestBody String address) throws InterruptedException, ApiException, IOException {
        return String.valueOf(geoLookup.latLngFromAddress(address));
    }

    @Autowired
    private LocationRoutePlanning routePlanning;


    @RequestMapping(value = "/grottoroute", method = RequestMethod.POST)
    @ResponseBody
    public String grottoRoute(@RequestBody GrottoDTO[] grottos) throws InterruptedException, ApiException, IOException {
        ArrayList<Location> locations = new ArrayList<>();
        for (GrottoDTO grotto : grottos) {
            Pair<Double, Double> latLong = geoLookup.latLngFromAddress(grotto.getAddress());
            locations.add(new Location(grotto.getName(), latLong.getFirst(), latLong.getSecond()));
        }
        ArrayList<Location> tour = routePlanning.computeOptimalTour(locations, 1000);
        return tour.stream()
                .map(Location::getName)
                .collect(Collectors.joining(","));
    }

}