EDIT

// $Id$

/**

 * Copyright (C) 2014 EDIT

 * European Distributed Institute of Taxonomy

 * http://www.e-taxonomy.eu

 *

 * The contents of this file are subject to the Mozilla Public License Version 1.1

 * See LICENSE.TXT at the top of this package for the full license terms.

 */

package eu.etaxonomy.taxeditor.editor.view.dataimport;

import java.io.IOException;

import java.io.InputStream;

import java.io.StringWriter;

import java.net.URISyntaxException;

import java.util.ArrayList;

import java.util.Collection;

import net.sf.json.JSONArray;

import net.sf.json.JSONObject;

import org.apache.commons.io.IOUtils;

import org.apache.http.client.ClientProtocolException;

import org.eclipse.swt.widgets.Display;

import eu.etaxonomy.cdm.api.facade.DerivedUnitFacade;

import eu.etaxonomy.cdm.ext.occurrence.OccurenceQuery;

import eu.etaxonomy.cdm.ext.occurrence.bioCase.BioCaseQueryServiceWrapper;

import eu.etaxonomy.cdm.ext.occurrence.gbif.GbifQueryServiceWrapper;

import eu.etaxonomy.cdm.io.common.CdmDefaultImport;

import eu.etaxonomy.cdm.io.specimen.abcd206.in.Abcd206ImportConfigurator;

import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationBase;

import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationType;

import eu.etaxonomy.taxeditor.editor.view.dataimport.transientServices.TransientCdmRepository;

import eu.etaxonomy.taxeditor.store.CdmStore;

/**

 * @author pplitzner

 * @date 25.02.2014

 *

 */

public class SpecimenImportEditorInput extends DataImportEditorInput<SpecimenOrObservationBase<?>> {

    private final OccurenceQuery query;

    private final QueryType queryType;

    /**

     * @param results

     */

    public SpecimenImportEditorInput(OccurenceQuery query, QueryType queryType) {

        super();

        this.query = query;

        this.queryType = queryType;

    }

    public void query(){

        String errorMessage = "Could not execute query " + query;

        Collection<SpecimenOrObservationBase<?>> results = new ArrayList<SpecimenOrObservationBase<?>>();

        try {

            InputStream resultStream;

            switch (queryType) {

            case GBIF:

                resultStream = new GbifQueryServiceWrapper().query(query);

                StringWriter stringWriter = new StringWriter();

                IOUtils.copy(resultStream, stringWriter);

                String jsonString = stringWriter.toString();

                JSONArray resultsArray = JSONObject.fromObject(jsonString).getJSONArray("results");

                for(Object o:resultsArray){

                    //parse every record

                    if(o instanceof JSONObject){

                        DerivedUnitFacade derivedUnitFacade = DerivedUnitFacade.NewInstance(SpecimenOrObservationType.PreservedSpecimen);

                        JSONObject record = (JSONObject)o;

                        if(record.has("locality")){

                            String locality = record.getString("locality");

                            derivedUnitFacade.setLocality(locality);

                        }

                        results.add(derivedUnitFacade.innerDerivedUnit());

                    }

                }

                break;

            case BIOCASE:

                resultStream = new BioCaseQueryServiceWrapper().query(SpecimenImportEditorInput.this.query);

                Abcd206ImportConfigurator configurator = Abcd206ImportConfigurator.NewInstance(resultStream, null, false);

                TransientCdmRepository repo =

                        new TransientCdmRepository(CdmStore.getCurrentApplicationConfiguration());

                configurator.setCdmAppController(repo);

                CdmDefaultImport<Abcd206ImportConfigurator> importer = new CdmDefaultImport<Abcd206ImportConfigurator>();

                importer.invoke(configurator);

                results = repo.getUnits();

                setResults(results);

                break;

            }

        } catch (ClientProtocolException e) {

            logger.error(errorMessage, e);

        } catch (IOException e) {

            logger.error(errorMessage, e);

        } catch (URISyntaxException e) {

            logger.error(errorMessage, e);

        }

        setResults(results);

        Display.getDefault().asyncExec(new Runnable() {

            @Override

            public void run() {

                CdmStore.getContextManager().notifyContextRefresh();

            }

        });

    }

    /* (non-Javadoc)

     * @see org.eclipse.ui.IEditorInput#getName()

     */

    @Override

    public String getName() {

        return queryType + " " + query.toString();

    }

    /* (non-Javadoc)

     * @see org.eclipse.ui.IEditorInput#getToolTipText()

     */

    @Override

    public String getToolTipText() {

        return query.toString();

    }

    /* (non-Javadoc)

     * @see java.lang.Object#hashCode()

     */

    @Override

    public int hashCode() {

        final int prime = 31;

        int result = 1;

        result = prime * result + ((query == null) ? 0 : query.hashCode());

        return result;

    }

    /* (non-Javadoc)

     * @see java.lang.Object#equals(java.lang.Object)

     */

    @Override

    public boolean equals(Object obj) {

        if (this == obj) {

            return true;

        }

        if (obj == null) {

            return false;

        }

        if (getClass() != obj.getClass()) {

            return false;

        }

        SpecimenImportEditorInput other = (SpecimenImportEditorInput) obj;

        if (query == null) {

            if (other.query != null) {

                return false;

            }

        } else if (!query.equals(other.query)) {

            return false;

        }

        return true;

    }

}