rest.MomondoRest.java Source code

Introduction

Here is the source code for rest.MomondoRest.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package rest;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import entity.MomondoFlight;
import entity.Passenger;
import entity.Reservation;
import facades.MomondoFacade;
import java.io.IOException;
import java.util.List;
import java.util.concurrent.ExecutionException;
import javax.annotation.security.RolesAllowed;
import javax.ws.rs.Consumes;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Response;
import utility.Post;

/**
 *
 * @author steffen
 */
@Path("momondo/")
public class MomondoRest {

    public static MomondoFacade f = new MomondoFacade();
    //    public static Gson gsonDate = new GsonBuilder().setDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSS'Z'").setPrettyPrinting().create();

    //    public MomondoRest() {
    //
    //        f = new MomondoFacade();        
    //    }

    @GET
    @Path("{from}/{date}/{numtickets}")
    @RolesAllowed("User")
    @Consumes("application/json")
    public Response getFlightsSimple(@PathParam("from") String from, @PathParam("date") String date,
            @PathParam("numtickets") int numtickets) throws InterruptedException, ExecutionException {

        List<MomondoFlight> momondoFlightList = f.getFlightsSimple(from, date, numtickets);

        JsonArray jsonFlights = new JsonArray();

        for (MomondoFlight m : momondoFlightList) {

            JsonObject jo = new JsonObject();
            jo.addProperty("airline", m.getAirline());
            jo.addProperty("flightID", m.getFlightId());
            jo.addProperty("origin", m.getOrigin());
            jo.addProperty("destination", m.getDestination());
            jo.addProperty("date", m.getDate());
            jo.addProperty("numberOfSeats", m.getNumberOfSeats());
            jo.addProperty("totalPrice", m.getTotalPrice());
            jo.addProperty("traveltime", m.getTravelTime());
            jo.addProperty("normalurl", m.getNormalUrl());

            jsonFlights.add(jo);
        }

        return Response.status(Response.Status.OK).entity(jsonFlights.toString()).build();
    }

    @GET
    @Path("{from}/{to}/{date}/{numtickets}")
    @RolesAllowed("User")
    @Consumes("application/json")
    public Response getFlightsAdvanced(@PathParam("from") String from, @PathParam("to") String to,
            @PathParam("date") String date, @PathParam("numtickets") int numtickets)
            throws InterruptedException, ExecutionException {

        List<MomondoFlight> momondoFlightList = f.getFlightsAdvanced(from, to, date, numtickets);

        JsonArray jsonFlights = new JsonArray();

        for (MomondoFlight m : momondoFlightList) {

            JsonObject jo = new JsonObject();
            jo.addProperty("airline", m.getAirline());
            jo.addProperty("flightID", m.getFlightId());
            jo.addProperty("origin", m.getOrigin());
            jo.addProperty("destination", m.getDestination());
            jo.addProperty("date", m.getDate());
            jo.addProperty("numberOfSeats", m.getNumberOfSeats());
            jo.addProperty("totalPrice", m.getTotalPrice());
            jo.addProperty("traveltime", m.getTravelTime());
            jo.addProperty("normalurl", m.getNormalUrl());

            jsonFlights.add(jo);
        }

        return Response.status(Response.Status.OK).entity(jsonFlights.toString()).build();
    }

    @POST
    @Consumes("application/json")
    @Produces("application/json")
    public Response reservationRequest(String jsonString) throws IOException {
        Post p = new Post();
        JsonObject js = new JsonParser().parse(jsonString).getAsJsonObject();
        String url = js.get("url").getAsString();
        js.remove("url");
        String finalJs = js.toString();

        String stringResponse = p.postReservation(url, finalJs);
        return Response.status(Response.Status.OK).entity(stringResponse).build();
    }

    @POST
    @Path("savereservation/{username}")
    @Consumes("application/json")
    public Response saveReservation(@PathParam("username") String userName, String jsonString) throws IOException {

        JsonObject js = new JsonParser().parse(jsonString).getAsJsonObject();

        String flightID = js.get("flightID").getAsString();
        String origin = js.get("Origin").getAsString();
        String destination = js.get("Destination").getAsString();
        String date = js.get("Date").getAsString();
        int flightTime = js.get("FlightTime").getAsInt();
        int numberOfSeats = js.get("numberOfSeats").getAsInt();
        String reserveeName = js.get("ReserveeName").getAsString();

        Reservation reservation = new Reservation(flightID, origin, destination, date, flightTime, numberOfSeats,
                reserveeName);

        JsonArray jsonPassengerArray = js.get("Passengers").getAsJsonArray();
        for (JsonElement jsonPassenger : jsonPassengerArray) {
            String firstname = jsonPassenger.getAsJsonObject().get("firstName").getAsString();
            String lastname = jsonPassenger.getAsJsonObject().get("lastName").getAsString();
            Passenger p = new Passenger(firstname, lastname);
            reservation.addPassengers(p);
            p.setReservation(reservation);

        }

        f.saveReservation(userName, reservation);

        return Response.status(Response.Status.OK).build();
    }

    @GET
    @Path("getreservation/{us}")
    @Consumes("application/json")
    public Response getReservationByUsername(@PathParam("us") String userName) {

        List<Reservation> reservationsList = f.getReservationByUsername(userName);
        JsonArray jsonReservations = new JsonArray();

        for (Reservation r : reservationsList) {

            JsonObject jo = new JsonObject();
            jo.addProperty("flightID", r.getFlightID());
            jo.addProperty("Origin", r.getOrigin());
            jo.addProperty("Destination", r.getDestination());
            jo.addProperty("Date", r.getDate());
            jo.addProperty("FlightTime", r.getFlightTime());
            jo.addProperty("numberOfSeats", r.getNumberOfSeats());
            jo.addProperty("ReserveeName", r.getReserveeName());
            JsonArray jsonPassengers = new JsonArray();

            List<Passenger> passengerList = r.getPassengers();
            for (Passenger p : passengerList) {
                JsonObject jo2 = new JsonObject();
                jo2.addProperty("firstName", p.getFirstname());
                jo2.addProperty("lastName", p.getLastname());
                jsonPassengers.add(jo2);
            }
            jo.add("Passengers", jsonPassengers);
            jsonReservations.add(jo);
        }

        return Response.status(Response.Status.OK).entity(jsonReservations.toString()).build();
    }
}