package de.intevation.lada.rest;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.logging.Logger;

import javax.enterprise.context.RequestScoped;
import javax.inject.Inject;
import javax.inject.Named;
import javax.ws.rs.Consumes;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.PUT;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.UriInfo;

import org.apache.commons.io.IOUtils;
import org.jboss.resteasy.plugins.providers.multipart.InputPart;
import org.jboss.resteasy.plugins.providers.multipart.MultipartFormDataInput;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import de.intevation.lada.auth.Authentication;
import de.intevation.lada.auth.AuthenticationException;
import de.intevation.lada.auth.AuthenticationResponse;
import de.intevation.lada.auth.Authorization;
import de.intevation.lada.data.LProbeRepository;
import de.intevation.lada.data.QueryBuilder;
import de.intevation.lada.data.Repository;
import de.intevation.lada.data.importer.Importer;
import de.intevation.lada.model.LProbe;
import de.intevation.lada.model.LProbeInfo;
import de.intevation.lada.utils.QueryTools;

/**
* This class produces a RESTful service to read, write and update
* LProbe objects.
*
* @author <a href="mailto:rrenkert@intevation.de">Raimund Renkert</a>
*/
@Path("/proben")
@RequestScoped
public class LProbeService {

    /**
     * The Repository for LProbe.
     */
    @Inject
    @Named("lproberepository")
    private Repository repository;

    @Inject
    @Named("lmessungrepository")
    private Repository messungRepository;

    /**
     * The authorization module.
     */
    @Inject
    @Named("ldapauth")
    private Authentication authentication;

    @Inject
    @Named("dataauthorization")
    private Authorization authorization;

    /**
     * The logger for this class.
     */
    @Inject
    private Logger log;

    /**
     * Request a LProbe via its id.
     *
     * @param id        The LProbe id
     * @param header    THe HTTP header containing authorization information.
     * @return Response object.
     */
    @GET
    @Path("/{id}")
    @Produces("text/json")
    public Response findById(
        @PathParam("id") String id,
        @Context HttpHeaders header
    ) {
        try {
            AuthenticationResponse auth =
                authentication.authorizedGroups(header);
            Response response =
                repository.findById(LProbeInfo.class, id);
            @SuppressWarnings("unchecked")
            List<LProbeInfo> probe = (List<LProbeInfo>)response.getData();
            if (probe.isEmpty()) {
                return new Response(false, 601, new ArrayList<LProbeInfo>());
            }
            String nbId = probe.get(0).getNetzbetreiberId();
            String mstId = probe.get(0).getMstId();
            if (auth.getNetzbetreiber().contains(nbId) ||
                probe.get(0).getFertig()) {
                if (auth.getMst().contains(mstId)) {
                    return response;
                }
                return response;
            }
            return new Response(false, 698, new ArrayList<LProbe>());
        }
        catch(AuthenticationException ae) {
            return new Response(false, 699, new ArrayList<LProbe>());
        }
    }

    /**
     * Request LProbe via a filter.
     *
     * Query parameters are used for the filter in form of key-value pairs.
     * This filter can take the three parameters
     *   mstId=$MSTID (String)
     *   umwId=$UWBID (String)
     *   begin=$PROBEENTNAHMEBEGIN (Timestamp)
     *
     * @param info      The URL query parameters.
     * @param header    The HTTP header containing authorization information.
     * @return Response object.
     */
    @GET
    @Produces("text/json")
    public Response filter(
        @Context UriInfo info,
        @Context HttpHeaders header
    ) {
        try {
            AuthenticationResponse auth =
                authentication.authorizedGroups(header);
            QueryBuilder<LProbeInfo> builder =
                new QueryBuilder<LProbeInfo>(
                    repository.getEntityManager(),
                    LProbeInfo.class);

            MultivaluedMap<String, String> params = info.getQueryParameters();
            if (params.isEmpty() || !params.containsKey("qid")) {
                return repository.filter(builder.getQuery());
            }
            String qid = params.getFirst("qid");
            JSONObject query = QueryTools.getQueryById(qid);
            List<String> filters = new ArrayList<String>();
            List<String> results = new ArrayList<String>();
            String sql = "";
            try {
                sql = query.getString("sql");
                JSONArray jFilters = query.getJSONArray("filters");
                for (int i = 0; i < jFilters.length(); i++) {
                    JSONObject jFilter = jFilters.getJSONObject(i);
                    filters.add(jFilter.getString("dataIndex"));
                }
                JSONArray jResults = query.getJSONArray("result");
                for (int i = 0; i < jResults.length(); i++) {
                    JSONObject jResult = jResults.getJSONObject(i);
                    results.add(jResult.getString("dataIndex"));
                }
            }
            catch (JSONException e) {
                return new Response(false, 603, new ArrayList<LProbeInfo>());
            }

            if (sql == null || sql.length() == 0) {
                return new Response(false, 603, new ArrayList<LProbeInfo>());
            }
            LProbeRepository lpr = (LProbeRepository)repository;
            String subselect = "(select * from l_probe_info where ";
            List<String> netzbetreiberIds = auth.getNetzbetreiber();
            boolean first = true;
            for (String netzbetreiberId: netzbetreiberIds) {
                if (first) {
                    subselect += "netzbetreiber_id = '" + netzbetreiberId + "' ";
                    first = false;
                }
                else {
                    subselect += "or netzbetreiber_id = '" + netzbetreiberId + "' ";
                }
            }
            subselect += "or fertig = true) as lp";
            sql = sql.replace("l_probe", subselect);
            return lpr.filterFree(sql, filters, results, params);
        }
        catch(AuthenticationException ae) {
            return new Response(false, 699, new ArrayList<LProbe>());
        }
    }

    /**
     * Update a LProbe object.
     *
     * @param probe     A LProbeInfo object wrapping the LProbe object.
     * @param header    The HTTP header containing authorization information.
     * @return Response object.
     */
    @PUT
    @Path("/{id}")
    @Produces("text/json")
    @Consumes("application/json")
    public Response update(LProbeInfo probe, @Context HttpHeaders header) {
        try {
            if(authentication.hasAccess(header, probe.getProbeId())) {
                LProbe p = probe.toLProbe();
                p.setProbeId(probe.getProbeId());
                return repository.update(p);
            }
            return new Response(false, 698, new ArrayList<LProbeInfo>());
        }
        catch(AuthenticationException ae) {
            return new Response(false, 699, new ArrayList<LProbeInfo>());
        }
    }

    /**
     * Create a new LProbe object.
     *
     * @param probe     A LProbeInfo object wrapping the LProbe object.
     * @param header    The HTTP header containing authorization information.
     * @return Response object.
     */
    @POST
    @Produces("text/json")
    @Consumes("application/json")
    public Response create(LProbeInfo probe, @Context HttpHeaders header) {
        try {
            AuthenticationResponse auth =
                authentication.authorizedGroups(header);
            if (auth.getNetzbetreiber().contains(probe.getNetzbetreiberId()) &&
                auth.getMst().contains(probe.getMstId())) {
                LProbe p = probe.toLProbe();
                return repository.create(p);
            }
            return new Response(false, 698, new ArrayList<LProbeInfo>());
        }
        catch(AuthenticationException ae) {
            return new Response(false, 699, new ArrayList<LProbeInfo>());
        }
    }
}