senslog1/src/main/java/cz/hsrs/servlet/provider/DataService.java

package cz.hsrs.servlet.provider;

import java.io.IOException;
import java.io.PrintWriter;
import java.sql.SQLException;
import java.util.List;

import javax.naming.AuthenticationException;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;

import cz.hsrs.db.DBJsonUtils;
import cz.hsrs.db.model.UnitPosition;
import cz.hsrs.db.model.UnitTrack;
import cz.hsrs.db.model.composite.LastPosition;
import cz.hsrs.db.model.composite.RealUnit;
import cz.hsrs.db.model.custom.UnitPositionSimple;
import cz.hsrs.db.util.UnitUtil;
import cz.hsrs.db.util.UserUtil;
import cz.hsrs.db.util.UtilFactory;
import cz.hsrs.servlet.feeder.ServiceParameters;
import cz.hsrs.servlet.security.LoginUser;

/**
 * Servlet implementation class DataService
 * 
 * /DataService?Operation=GetDataByUserName&user=pepa&limit=100
 * /DataService?Operation=GetUnitsList
 * /DataService?Operation=GetUnits
 * /DataService?Operation=GetUnitsGeoJson
 * /DataService?Operation=GetUnitTypes
 */
public class DataService extends DBServlet {

    private static final long serialVersionUID = 1L;

    private static final String ASC = "ASC";
    private static final String DESC = "DESC";

    public static final String GET_TRACK = "GetTracks";
    public static final String GET_LAST_POSTION = "GetLastPositions";
    public static final String GET_LAST_POSTION_WITH_STATUS = "GetLastPositionsWithStatus";
    public static final String GET_UNITS = "GetUnits";
    public static final String GET_RECENT_TRACK = "GetRecentTracks";
    public static final String GET_POSITIONS = "GetPositions";
    public static final String GET_POSITIONS_RANGE = "GetPositionsDay";
    public static final String GET_UNITS_LIST = "GetUnitsList";
    public static final String GET_UNITS_GEOJSON = "GetUnitsGeoJson";
    public static final String GET_UNIT_TYPES = "GetUnitTypes";
    
    private UtilFactory db;

    public DataService() {
        super();
    }


    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

        /* CORS Filtering */
        String originDomain = request.getHeader("origin");
        originDomain = originDomain == null ? "*" : originDomain;
        response.addHeader("Access-Control-Allow-Origin", originDomain);
        response.addHeader("Access-Control-Allow-Headers", "origin, content-type, accept, authorization");
        response.addHeader("Access-Control-Allow-Credentials", "true");
        response.addHeader("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS, HEAD");

        request.setCharacterEncoding("UTF-8");
        response.setCharacterEncoding("UTF-8");
        
        response.addHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON);
        
        /*
         * /DataService?Operation=GetUnits&user=telemetry&unit_id=356173060488215
         * /DataService?Operation=GetTracks&user=telemetry&limit=500
         * /DataService?Operation=GetLastPositions&user=telemetry
         * /DataService?Operation=GetLastPositionsWithStatus&user=telemetry
         * /DataService?Operation=GetRecentTracks&user=telemetry
         * /DataService?Operation=GetPositions&user=telemetry&limit=500
         * /DataService?Operation=GetPositionsDay&user=telemetry&unit_id=356173060488215&fromTime=2016-02-01&toTime=2016-02-04&ordering=desc
         * /DataService?Operation=GetUnitsList&user=telemetry
         * /DataService?Operation=GetUnitsGeoJson
         */
        PrintWriter out = response.getWriter();

        RequestParameters params = new RequestParameters(request);
        String userId = request.getAttribute("userId").toString();

        try {
            switch (request.getParameter(ServiceParameters.OPERATION)) {
                case GET_UNITS: {
                    DBJsonUtils.writeJSON(out, new RealUnit(), db.userUtil.getLastPositionsByUserNameRes(userId, params.getUnitId()));
                } break;
                case GET_TRACK: {
                    DBJsonUtils.writeJSON(out, new UnitTrack(), db.userUtil.getTracksByUserName(userId, params.limit));
                } break;
                case GET_LAST_POSTION: {
                    DBJsonUtils.writeJSON(out, db.userUtil.getLastPositionsByUserName(userId));
                } break;
                case GET_LAST_POSTION_WITH_STATUS: {
                    List<LastPosition> posList = db.userUtil.getLastPositionWithStatus(userId);
                    DBJsonUtils.writeJSON(out, posList);
                } break;
                case GET_RECENT_TRACK: {
                    DBJsonUtils.writeJSON(out, new UnitTrack(), db.userUtil.getTracksByUserName(userId, 1000));
                } break;
                case GET_POSITIONS: {
                    DBJsonUtils.writeJSON(out, new UnitPosition(), db.userUtil.getPositionsByUserName(userId, params.limit));
                } break;
                case GET_POSITIONS_RANGE: {
                    DBJsonUtils.writeJSON(out, new UnitPositionSimple(), db.userUtil.getPositionsTimeRangeByUserName(userId, params.fromTime, params.toTime, params.getUnitId(), params.getOrdering()));
                } break;
                case GET_UNITS_LIST: {
                    DBJsonUtils.writeJSON(out, db.userUtil.getUnitsByUser(userId));
                } break;
                case GET_UNITS_GEOJSON:{
                	out.write(UserUtil.getUnitsGeoJson(userId));
                	//DBJsonUtils.writeJSON(out, UserUtil.getUnitsGeoJson(userId));
                } break;
                case GET_UNIT_TYPES:{
                	out.write(UnitUtil.getUnitTypes());
                } break;
                default:
                    throw new NullPointerException("No operation specified.");
            }
        } catch (SQLException e) {
            solveGetException(e, out);
        }
    }

    @Override
    public void init() throws ServletException {
        super.init();
        
        try {
            db = new UtilFactory();
        } catch (Exception e) {
            throw new ServletException(e);
        }
    }
    

    /**
     * Request Parameter subclass
     * Parses parameter from the request
     * @author jezekjan
     *
     */
    static class RequestParameters {
        private String user;
        private String fromTime;
        private String toTime;
        private Integer limit;
        private Long unitId;
        private String ordering;

        RequestParameters(HttpServletRequest request) throws NullPointerException{
            Object limO = request.getParameter("limit");
            if (limO != null) {
                limit = Integer.parseInt(limO.toString());
            }

            Object fromTimeO = request.getParameter("fromTime");
            if (fromTimeO != null) {
                fromTime = fromTimeO.toString();
            }
            Object toTimeO = request.getParameter("toTime");
            if (toTimeO != null) {
                toTime = toTimeO.toString();
            }
            Object orderingO = request.getParameter("ordering");
            if (orderingO != null) {
                String orderingS = orderingO.toString();
                if(orderingS.isEmpty()){
                    ordering = ASC;
                } else{
                    ordering = orderingS.equalsIgnoreCase(DESC) ? DESC : ASC;
                }
            }else{
                ordering = ASC;
            }
            Object unitO = request.getParameter("unit_id");
            unitId = unitO != null ? Long.parseLong(unitO.toString()) : null;
        }

        public Long getUnitId() {
            return unitId;
        }

        public void setUnitId(long unitId) {
            this.unitId = unitId;
        }

        public void setUser(String user) {
            this.user = user;
        }

        public Integer getLimit() {
            return limit;
        }

        public void setLimit(Integer limit) {
            this.limit = limit;
        }

        public String getFromTime() {
            return fromTime;
        }

        public void setFromTime(String fromTime) {
            this.fromTime = fromTime;
        }

        public String getToTime() {
            return toTime;
        }

        public void setToTime(String toTime) {
            this.toTime = toTime;
        }

        public String getOrdering() {
            return ordering;
        }

        public void setOrdering(String ordering) {
            this.ordering = ordering;
        }

        @Override
        public String toString() {
            return "RequestParameters [user=" + user + ", fromTime=" + fromTime
                    + ", toTime=" + toTime + ", limit=" + limit + ", unit_id="
                    + unitId + ", ordering=" + ordering + "]";
        }
    }
}