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

package org.atea.nlptools.koreromaoriinterface;

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

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 javax.servlet.http.Part;

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.exceptions.ReoTuhituhiException;
import org.atea.nlptools.koreromaoriinterface.models.AudioFilePart;
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 ReoTuhituhiApiService transcriptionService;

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

    @Override
    public void init() 
    {
        Properties prop = new Properties();

        try {
            FileInputStream fis = new FileInputStream("../webapps/gs3-koreromaori/config.properties");
            prop.load(fis);
            fis.close();
        } catch (IOException e) {
            e.printStackTrace();
        }

        String apiEndpoint = prop.getProperty("tuhituhi.api.endpoint");
        String apiKey = prop.getProperty("tuhituhi.api.key");
        
        this.transcriptionService = new ReoTuhituhiApiService(jsonSerialiser, apiEndpoint, apiKey);

        logger.info("Initialised!");
    }

    @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("\\|");

        // Create our own representation of each part
        ArrayList<AudioFilePart> audioFileParts = new ArrayList<AudioFilePart>(audioFileKeys.length);
        for (String audioFileKey : audioFileKeys)
        {
            Part p = request.getPart(audioFileKey);
            AudioFilePart audioFilePart = AudioFilePart.fromPart(p);
            audioFileParts.add(audioFilePart);
        }

        try 
        {
            List<AudioFilePart> results = transcriptionService.getTranscriptions(audioFileParts);

            List<MyTranscriptionResponse> responses = new ArrayList<MyTranscriptionResponse>(results.size());
            for (AudioFilePart result : results) {
                responses.add(MyTranscriptionResponse.FromTranscriptionResult(result.getTranscriptionResult(), result.fileName));
            }

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