package eu.dnetlib.ariadneplus.elasticsearch;

import eu.dnetlib.ariadneplus.elasticsearch.model.*;
import eu.dnetlib.ariadneplus.reader.ResourceManager;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.http.HttpHost;
import org.elasticsearch.action.bulk.BulkRequest;
import org.elasticsearch.action.bulk.BulkResponse;
import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.common.xcontent.XContentType;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.util.Arrays;

@Service
public class BulkUpload {

    private static final Log log = LogFactory.getLog(BulkUpload.class);

    @Value("${elasticsearch.hostname}")
    private String elasticSearchHostName;

    @Value("${elasticsearch.indexname}")
    private String elasticSearchIndexName;

    private RestHighLevelClient client;

    public void init(String elasticSearchHostName, String elasticSearchIndexName) throws IOException {
        this.elasticSearchIndexName = elasticSearchIndexName;
        client = new RestHighLevelClient(
                RestClient.builder(
                        new HttpHost(elasticSearchHostName,9200,"http")));

    }

    @PostConstruct
    public void init() throws IOException {
        client = new RestHighLevelClient(
                RestClient.builder(
                        new HttpHost(elasticSearchHostName,9200,"http")));

    }

    public void index(ResourceManager manager, boolean isCollection) {
        BulkRequest request = new BulkRequest();
        while (manager.hasNext()){
        	
            try {
            	Object next = manager.next();
                AriadneCatalogEntry ace = ((AriadneCatalogEntry) next);
                if (isCollection) {
                    ace.setResourceType("collection");
                }
                else {
                    ace.setResourceType("dataset");
                    Spatial esSpatial = new Spatial();
                    ace.getSpatial().stream().filter(s -> s.getPlaceName()!=null).forEach(s -> {
                        esSpatial.setPlaceName(s.getPlaceName());
                    });
                    ace.getSpatial().stream().filter(s -> s.getLocation()!=null).forEach(s -> {
                        esSpatial.setLocation(s.getLocation());
                    });
                    ace.getSpatial().clear();
                    ace.setSpatial(Arrays.asList(esSpatial));
                }

                String uniqueIsPartOf = ace.getUniqueIsPartOf();
                if (uniqueIsPartOf!=null) {
                    ace.setIsPartOf(Arrays.asList(uniqueIsPartOf));
                }
                if (ace.getContributor()!=null) {
                    ace.getContributor().clear();
                    ace.setContributor(ace.getCreator());
                }
                Distribution distribution = new Distribution();
                AgentInfo distrPublisher = new AgentInfo();
                distrPublisher.setEmail("");
                distrPublisher.setName("");
                distrPublisher.setType("");
                distribution.setPublisher(Arrays.asList(distrPublisher));
                ace.setDistribution(Arrays.asList(distribution));
                ItemMetadataStructure ims = new ItemMetadataStructure();
                ace.setHasItemMetadataStructure(Arrays.asList(ims));
                MetadataRecord mr = new MetadataRecord();
                Dex dex = new Dex();
                mr.setConformsTo(Arrays.asList(dex));
                ace.setHasMetadataRecord(Arrays.asList(mr));
                ace.setKeyword(Arrays.asList(new String("")));
                AgentInfo sr = new AgentInfo();
                ace.setScientificResponsible(Arrays.asList(sr));
                AgentInfo tr = new AgentInfo();
                ace.setTechnicalResponsible(Arrays.asList(tr));

                AgentInfo testPublisher = new AgentInfo();
                testPublisher.setName("TEST");
                ace.getPublisher().add(testPublisher);
                String[] splits = ace.getIdentifier().split("/");
                request.add(new IndexRequest(elasticSearchIndexName).id(splits[splits.length-1])
                        .source(ace.toJson(),XContentType.JSON));
                BulkResponse bulkResponse = client.bulk(request, RequestOptions.DEFAULT);
                log.info("Indexing to ES completed with status: "+bulkResponse.status());
                if (bulkResponse.hasFailures()) {
                    log.error("FailureMessage: "+bulkResponse.buildFailureMessage());
                }

            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            } catch (NoSuchMethodException e) {
                e.printStackTrace();
            } catch (InstantiationException e) {
                e.printStackTrace();
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            } catch (InvocationTargetException e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}