nlphub/src/main/java/org/gcube/data/analysis/nlphub/NLPUploader.java

package org.gcube.data.analysis.nlphub;

//import static org.gcube.common.authorization.client.Constants.authorizationService;

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;

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

import org.gcube.data.analysis.nlphub.legacy.JsonManager;
import org.gcube.data.analysis.nlphub.legacy.NlpHubException;
import org.gcube.data.analysis.nlphub.nlp.NLpLanguageRecognizer;
import org.gcube.data.analysis.nlphub.nlp.NlpUtils;
import org.gcube.data.analysis.nlphub.session.SessionUtils;
import org.gcube.data.analysis.nlphub.shared.Constants;
import org.gcube.data.analysis.nlphub.workspace.WorkspaceManager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Servlet implementation class NLPUploader
 */
// @WebServlet("/NLPUploader")
@WebServlet(asyncSupported = true, name = "NLPUploader", urlPatterns = { "/nlphub-uploader-servlet" })
@MultipartConfig
public class NLPUploader extends HttpServlet {
	private static final long serialVersionUID = 1L;
	private static final Logger logger = LoggerFactory.getLogger(NLPUploader.class);

	/**
	 * @see HttpServlet#HttpServlet()
	 */
	public NLPUploader() {
		super();
		// TODO Auto-generated constructor stub
	}

	/**
	 * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse
	 *      response)
	 */
	protected void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		doWork(request, response);
	}

	/**
	 * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse
	 *      response)
	 */
	protected void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		doWork(request, response);
	}

	private void doWork(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		logger.debug("NLPUploader");
		SessionUtils sessionUtils=new SessionUtils();
		String token = sessionUtils.getToken(request);
		WorkspaceManager ws = new WorkspaceManager();
		response.setContentType("application/json;charset=UTF-8");
		if (request.getParameter("freetext") == null)
			handleFileUpload(request, response, token, ws);
		else
			handleFreeText(request, response, token, ws);
	}

	private void handleFreeText(HttpServletRequest request, HttpServletResponse response, String token,
			WorkspaceManager ws) throws ServletException, IOException {
		String freeText = request.getParameter("freetext");
		String langParameter=request.getParameter("getlang");
		String dataMiner = request.getParameter("dataminer");
		//dataMiner=java.net.URLDecoder.decode(dataMiner,"UTF-8");
		
		
		freeText = NlpUtils.replaceDirtyCharacters(freeText);

		byte[] content = freeText.getBytes("UTF-8");
		String fileName = generateFileName();
		PrintWriter writer = response.getWriter();
		try {
			
			String link=ws.uploadFile(content, fileName, Constants.DEFAULT_DESCRIPTION);
			
			if (langParameter != null) {
				String sentence = NlpUtils.getLanguageRecognizerDigest(new String(content));
				logger.info(sentence);
				try {
					NLpLanguageRecognizer.run(dataMiner, sentence, token, link, response);
				} catch (NlpHubException ex) {
					writer.println(new JsonManager().getSuccessJsonResponse(Constants.UNAVAILABLE, link));
				}
			} else
				writer.println(new JsonManager().getSuccessJsonResponse("" + link));
		} catch (Exception e) {
			logger.error(e.getLocalizedMessage(),e);
			writer.println(new JsonManager().getErrorJsonResponse(e.getLocalizedMessage()));
		}
	}

	private void handleFileUpload(HttpServletRequest request, HttpServletResponse response, String token,
			WorkspaceManager ws) throws ServletException, IOException {
		int contentLength = request.getContentLength();

		Part filePart = request.getPart("mytxtfile");
		String dataMiner = request.getParameter("dataminer");
		//dataMiner=java.net.URLDecoder.decode(dataMiner,"UTF-8");
		

		String fileName = getFileName(filePart);

		PrintWriter writer = response.getWriter();
		// allocate a buffer of request size
		byte[] buffer = new byte[contentLength];
		byte[] bufferedContent;
		try {
			InputStream fileContent = filePart.getInputStream();
			int offset = 0, len = 256, byteRead = 0;
			byte[] readBuffer = new byte[len];
			while (byteRead > -1) {
				byteRead = fileContent.read(readBuffer, 0, len);
				if (byteRead > 0) {
					System.arraycopy(readBuffer, 0, buffer, offset, byteRead);
					offset += byteRead;
				}
			}

			if (offset < contentLength) {
				bufferedContent = new byte[offset];
				System.arraycopy(buffer, 0, bufferedContent, 0, offset);
			} else
				bufferedContent = buffer;

			String stringContent = new String(bufferedContent);
			stringContent = NlpUtils.replaceDirtyCharacters(stringContent);
						
			String link=ws.uploadFile(stringContent.getBytes(), fileName, Constants.DEFAULT_DESCRIPTION);
			
			String sentence = NlpUtils.getLanguageRecognizerDigest(stringContent);
			logger.info(sentence);
			try {
				NLpLanguageRecognizer.run(dataMiner, sentence, token, link, response);
			} catch (NlpHubException ex) {
				writer.println(new JsonManager().getSuccessJsonResponse(Constants.UNAVAILABLE, link));
			}
			// writer.println(new JsonManager().getSuccessJsonResponse("" +
			// link));
		} catch (Exception e) {
			logger.error(e.getLocalizedMessage(),e);
			writer.println(new JsonManager().getErrorJsonResponse(e.getLocalizedMessage()));
		}

	}

	private String getFileName(Part part) {
		String partHeader = part.getHeader("content-disposition");
		logger.debug("Part Header: " + partHeader);
		for (String content : part.getHeader("content-disposition").split(";")) {
			if (content.trim().startsWith("filename")) {
				return content.substring(content.indexOf('=') + 1).trim().replace("\"", "");
			}
		}
		return null;
	}

	private String generateFileName() {
		long now = System.currentTimeMillis();
		return "auto-nlp-" + now;
	}
}