com.jaeksoft.searchlib.webservice.parser.ParserImpl Maven / Gradle / Ivy

Go to download
/**   
 * License Agreement for OpenSearchServer
 *
 * Copyright (C) 2014 Emmanuel Keller / Jaeksoft
 * 
 * http://www.open-search-server.com
 * 
 * This file is part of OpenSearchServer.
 *
 * OpenSearchServer is free software: you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 * OpenSearchServer is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with OpenSearchServer. 
 *  If not, see .
 **/

package com.jaeksoft.searchlib.webservice.parser;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.Response.Status;
import javax.ws.rs.core.UriInfo;

import org.apache.commons.lang3.StringUtils;

import com.jaeksoft.searchlib.ClientFactory;
import com.jaeksoft.searchlib.SearchLibException;
import com.jaeksoft.searchlib.analysis.LanguageEnum;
import com.jaeksoft.searchlib.parser.Parser;
import com.jaeksoft.searchlib.parser.ParserFactory;
import com.jaeksoft.searchlib.parser.ParserResultItem;
import com.jaeksoft.searchlib.parser.ParserType;
import com.jaeksoft.searchlib.parser.ParserTypeEnum;
import com.jaeksoft.searchlib.streamlimiter.StreamLimiter;
import com.jaeksoft.searchlib.streamlimiter.StreamLimiterFile;
import com.jaeksoft.searchlib.streamlimiter.StreamLimiterInputStream;
import com.jaeksoft.searchlib.util.IOUtils;
import com.jaeksoft.searchlib.util.LinkUtils;
import com.jaeksoft.searchlib.webservice.CommonListResult;
import com.jaeksoft.searchlib.webservice.CommonServices;
import com.jaeksoft.searchlib.webservice.NameLinkItem;

public class ParserImpl extends CommonServices implements RestParser {

	@Override
	public CommonListResult list(UriInfo uriInfo, String login,
			String key) {
		try {
			getLoggedUser(login, key);
			ClientFactory.INSTANCE.properties.checkApi();
			List parserTypeList = ParserTypeEnum.INSTANCE.getList();
			ArrayList items = new ArrayList(
					parserTypeList.size());
			for (ParserType parserType : parserTypeList) {
				String name = parserType.getName();
				String link = LinkUtils.concatPath(uriInfo.getRequestUri()
						.getPath(), parserType.simpleName);
				items.add(new NameLinkItem(name, link));
			}
			return new CommonListResult(items);
		} catch (InterruptedException e) {
			throw new CommonServiceException(e);
		} catch (IOException e) {
			throw new CommonServiceException(e);
		} catch (SearchLibException e) {
			throw new CommonServiceException(e);
		}
	}

	private ParserType checkParserType(String parserName) {
		ParserType parserType = ParserTypeEnum.INSTANCE.findByName(parserName);
		if (parserType == null)
			throw new CommonServiceException(Status.NOT_FOUND,
					"Parser not found: " + parserName);
		return parserType;
	}

	private ParserFactory checkParserFactory(ParserType parserType)
			throws ClassNotFoundException, SearchLibException {
		return ParserFactory.create(null, null, parserType.getParserClass()
				.getCanonicalName());
	}

	@Override
	public ParserItemResult get(UriInfo uriInfo, String login, String key,
			String parserName) {
		try {
			getLoggedUser(login, key);
			ClientFactory.INSTANCE.properties.checkApi();
			ParserType parserType = checkParserType(parserName);
			return new ParserItemResult(parserType,
					checkParserFactory(parserType));
		} catch (SearchLibException e) {
			throw new CommonServiceException(e);
		} catch (InterruptedException e) {
			throw new CommonServiceException(e);
		} catch (IOException e) {
			throw new CommonServiceException(e);
		} catch (ClassNotFoundException e) {
			throw new CommonServiceException(e);
		}
	}

	@Override
	public ParserDocumentsResult put(UriInfo uriInfo, String login, String key,
			String parserName, LanguageEnum language, String path,
			InputStream inputStream) {
		StreamLimiter streamLimiter = null;
		try {
			getLoggedUser(login, key);
			ClientFactory.INSTANCE.properties.checkApi();
			ParserType parserType = checkParserType(parserName);
			ParserFactory parserFactory = checkParserFactory(parserType);
			MultivaluedMap parserParams = uriInfo
					.getQueryParameters();
			for (String propKey : parserParams.keySet()) {
				if (!propKey.startsWith("p."))
					continue;
				parserFactory.setUserProperty(propKey.substring(2),
						parserParams.getFirst(propKey));
			}
			if (StringUtils.isEmpty(path) && inputStream == null)
				throw new CommonServiceException(Status.NOT_ACCEPTABLE,
						"You should either provide a path or upload a file");
			Parser parser = (Parser) ParserFactory.create(parserFactory);
			if (StringUtils.isEmpty(path))
				streamLimiter = new StreamLimiterInputStream(
						parser.getSizeLimit(), inputStream, null, null);
			else
				streamLimiter = new StreamLimiterFile(parser.getSizeLimit(),
						new File(path));
			parser.doParserContent(null, streamLimiter, language);
			List parserResultList = parser.getParserResults();
			return new ParserDocumentsResult(parserResultList);
		} catch (SearchLibException e) {
			throw new CommonServiceException(e);
		} catch (InterruptedException e) {
			throw new CommonServiceException(e);
		} catch (IOException e) {
			throw new CommonServiceException(e);
		} catch (ClassNotFoundException e) {
			throw new CommonServiceException(e);
		} finally {
			IOUtils.close(streamLimiter);
		}
	}
}