/*
 * This file is a part of EpnTAPClient.
 * This program aims to provide EPN-TAP support for software clients, like CASSIS spectrum analyzer.
 * See draft specifications: https://voparis-confluence.obspm.fr/pages/viewpage.action?pageId=559861
 * Copyright (C) 2016 Institut de Recherche en Astrophysique et Planétologie.
 *
 * This program 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. This program 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 this program. If not, see
 * <http://www.gnu.org/licenses/>.
 */

package eu.omp.irap.vespa.epntapclient;

import java.text.ParseException;
import java.util.ArrayList;
import java.util.List;

import eu.omp.irap.vespa.epntapclient.granule.Granule;
import eu.omp.irap.vespa.epntapclient.granule.GranuleCtrl;
import eu.omp.irap.vespa.epntapclient.service.Queries;
import eu.omp.irap.vespa.epntapclient.service.ServiceCore;
import eu.omp.irap.vespa.epntapclient.service.ServiceCtrl;
import eu.omp.irap.vespa.epntapclient.voresource.VOResourceCtrl;
import eu.omp.irap.vespa.epntapclient.voresource.VOResourceException;
import eu.omp.irap.vespa.epntapclient.voresource.model.Resource;
import eu.omp.irap.vespa.epntapclient.votable.model.VOTABLE;
import eu.omp.irap.vespa.votable.controller.CantGetVOTableException;
import eu.omp.irap.vespa.votable.controller.VOTableController;
import eu.omp.irap.vespa.votable.utils.StringJoiner;
import eu.omp.irap.vespa.votable.votabledata.VOTableData;

/**
 * @author N. Jourdane
 */
public class EpnTapConnection implements EpnTapInterface {

	// *** Resource ***

	@Override
	public Resource getEPNVOresource(String ivoid) throws VOResourceException {
		return VOResourceCtrl.getVOresource(ivoid);
	}

	// *** Resources ***

	@Override
	public List<Resource> getEPNVOResources() throws VOResourceException {
		List<String> ivoids = VOResourceCtrl.getVOResources(ServiceCore.EPNCORE);
		return VOResourceCtrl.getVOResources(ivoids);
	}

	@Override
	public List<Resource> getEPNVOResources(List<String> keywords)
			throws VOResourceException {
		List<String> ivoids = VOResourceCtrl.getVOResources(ServiceCore.EPNCORE, keywords);
		return VOResourceCtrl.getVOResources(ivoids);
	}

	// *** Service ***

	@Override
	public VOTABLE getEPNService(String ivoid) throws CantGetVOTableException {
		String query = String.format(Queries.SELECT_ALL_TAP_SERVICES_WHERE_IVOID, ivoid);
		return ServiceCtrl.getVoTable(query);
	}

	@Override
	public VOTABLE getEPNService(String ivoid, List<String> attributes)
			throws CantGetVOTableException {
		String select = StringJoiner.join(attributes);
		String query = String.format(Queries.SELECT_TAP_SERVICES_WHERE_IVOID, select, ivoid);
		return ServiceCtrl.getVoTable(query);
	}

	// *** Services ***

	@Override
	public VOTABLE getEPNServices() throws CantGetVOTableException {
		String query = String.format(Queries.SELECT_ALL_TAP_SERVICES_WHERE_CORE,
				ServiceCore.EPNCORE);
		return ServiceCtrl.getVoTable(query);
	}

	@Override
	public VOTABLE getEPNServices(List<String> attributes) throws CantGetVOTableException {
		String select = StringJoiner.join(attributes);
		String query = String.format(Queries.SELECT_TAP_SERVICES, select);
		return ServiceCtrl.getVoTable(query);
	}

	@Override
	public VOTABLE getEPNServices(List<String> keywords, List<String> attributes)
			throws CantGetVOTableException {
		// TODO: move this code to a new class in ServiceCtrl()
		attributes.add("res_subject");
		String select = StringJoiner.join(attributes);
		List<String> whereList = new ArrayList<>();
		for (String keyword : keywords) {
			whereList.add("res_subject = '" + keyword + "'");
		}
		String where = StringJoiner.join(whereList, " OR ");
		String query = String.format(Queries.SELECT_TAP_SERVICES_WHERE_SUBJECT, select, where);
		return ServiceCtrl.getVoTable(query);
	}

	// *** Getters ***

	@Override
	public String getEPNCoreTableName(String ivoid) throws CantGetVOTableException {
		return (String) ServiceCtrl.getParameter(ivoid, "table_name");
	}

	@Override
	public String getTAPURL(String ivoid) throws CantGetVOTableException {
		return (String) ServiceCtrl.getParameter(ivoid, "access_url");
	}

	// *** Queries ***

	@Override
	public List<Granule> sendADQLQuery(String tapURL, String adqlQuery)
			throws CantGetVOTableException {
		VOTableController voTableCtrl = new VOTableController(tapURL, adqlQuery);
		voTableCtrl.readTable();
		VOTableData data = voTableCtrl.getVOTableData();

		List<Granule> granules;
		try {
			GranuleCtrl gc = new GranuleCtrl(data);
			granules = gc.getGranulesFromVOTable();
		} catch (ParseException e) {
			throw new CantGetVOTableException("Parsing error on a granule.", e);
		}
		return granules;
	}

	@Override
	public List<Granule> sendQuery(String tapURL, String schemaName, Query enumeratedQuery)
			throws CantGetVOTableException {
		String query = String.format(enumeratedQuery.toString(), schemaName);
		VOTableController voTableCtrl = new VOTableController(tapURL, query);
		voTableCtrl.readTable();
		VOTableData data = voTableCtrl.getVOTableData();

		List<Granule> granules;
		try {
			GranuleCtrl gc = new GranuleCtrl(data);
			granules = gc.getGranulesFromVOTable();
		} catch (ParseException e) {
			throw new CantGetVOTableException("Parsing error on a granule.", e);
		}
		return granules;
	}

}