source: gs3-extensions/atea-nlp-tools/trunk/src/koreromaori-proxy/src/main/java/org/atea/nlptools/koreromaoriinterface/TranscriptionServlet.java@ 35239

package org.atea.nlptools.koreromaoriinterface;

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.List;

import javax.servlet.ServletException;
import javax.servlet.annotation.MultipartConfig;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.google.gson.FieldNamingPolicy;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.reflect.TypeToken;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.atea.nlptools.koreromaoriinterface.models.MyTranscriptionResponse;
import org.atea.nlptools.koreromaoriinterface.models.TranscriptionResult;
import org.atea.nlptools.koreromaoriinterface.services.ReoTuhituhiApiService;

@MultipartConfig
public class TranscriptionServlet extends HttpServlet
{
    private static final long serialVersionUID = 1L;

    private static final Logger logger = LogManager.getLogger(TranscriptionServlet.class);
    private static final Type transcriptionListType = new TypeToken<List<TranscriptionResult>>(){}.getType();

    private final Gson jsonSerialiser;
    private final ReoTuhituhiApiService transcriptionService;

    public TranscriptionServlet()
    {
        jsonSerialiser = new GsonBuilder().setFieldNamingPolicy(FieldNamingPolicy.LOWER_CASE_WITH_UNDERSCORES).create();
        this.transcriptionService = new ReoTuhituhiApiService(jsonSerialiser);
    }

    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
        throws IOException
    {
        response.sendError(403);
    }
    
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException
    {
        logger.trace("POST request received.");

        response.setContentType("application/json; charset=UTF-8");
        PrintWriter writer = response.getWriter();

        String audioFileKeysParameter = request.getParameter("audioFileKeys");
        if (audioFileKeysParameter == null) {
            response.sendError(400, "Form data was incorrect: missing audioFileKeys part.");
            return;
        }
        String[] audioFileKeys = audioFileKeysParameter.split("\\|");

        ArrayList<InputStream> audioFileStreams = new ArrayList<InputStream>(audioFileKeys.length);
        for (String audioFileKey : audioFileKeys)
        {
            audioFileStreams.add(request.getPart(audioFileKey).getInputStream());
        }

        try
        {
            List<TranscriptionResult> results = transcriptionService.GetTranscriptions(audioFileStreams);
            List<MyTranscriptionResponse> responses = new ArrayList<MyTranscriptionResponse>(results.size());

            for (TranscriptionResult result : results) {
                responses.add(MyTranscriptionResponse.FromTranscriptionResult(result, "placeholder")); // TODO: Insert correct file name
            }

            String json = jsonSerialiser.toJson(responses, transcriptionListType);
            writer.append(json);
        }
        catch (Exception ex)
        {
            response.sendError(500, "Call to the Reo Tuhituhi API failed." + ex);
            logger.error("Failed to complete API call", ex);
        }
    }
}