package eu.dnetlib.lbs.clients;

import java.util.Date;
import java.util.Objects;

import org.apache.lucene.search.join.ScoreMode;
import org.elasticsearch.action.search.SearchType;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.junit.Ignore;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.elasticsearch.annotations.Document;
import org.springframework.data.elasticsearch.core.ElasticsearchTemplate;
import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
import org.springframework.data.elasticsearch.core.query.SearchQuery;
import org.springframework.data.util.CloseableIterator;

import eu.dnetlib.lbs.elasticsearch.Event;
import eu.dnetlib.lbs.subscriptions.MapCondition;
import eu.dnetlib.lbs.subscriptions.NotificationFrequency;
import eu.dnetlib.lbs.subscriptions.NotificationMode;
import eu.dnetlib.lbs.subscriptions.Subscription;
import eu.dnetlib.lbs.utils.DateParser;

@Ignore
// @RunWith(SpringJUnit4ClassRunner.class)
// @ContextConfiguration(locations = { "classpath:/applicationContext-test-queries.xml" })
public class IndexClientTest {

	private static final String topic = "ENRICH/MORE/PID";
	private static final Date fromDate = DateParser.parse("2016-01-31");

	private static final String indexName = Event.class.getAnnotation(Document.class).indexName();
	private static final String indexType = Event.class.getAnnotation(Document.class).type();

	@Autowired
	private ElasticsearchTemplate elasticsearchTemplate;

	@Test
	public void testSearchTopic() {
		System.out.println("Start searching");

		final SearchQuery searchQuery = new NativeSearchQueryBuilder()
				.withQuery(QueryBuilders.boolQuery()
						.must(QueryBuilders.matchQuery("topic", topic))
						.must(QueryBuilders.rangeQuery("creationDate").from(fromDate)))
				.withSearchType(SearchType.DEFAULT)
				.withIndices(indexName)
				.withTypes(indexType)
				.withPageable(PageRequest.of(0, 10))
				.build();

		int count = 0;
		final CloseableIterator<Event> it = this.elasticsearchTemplate.stream(searchQuery, Event.class);
		while (it.hasNext()) {
			System.out.println(" > " + it.next());
			count++;
		}
		System.out.println("SIZE: " + count);

	}

	@Test
	@Ignore
	public void testSearchSubscription() {
		System.out.println("Start searching");

		final Subscription s = new Subscription();
		s.setSubscriptionId("sub-db0b35d4-1b0f-4660-a849-34fbec8fb6f7");
		s.setSubscriber("artini@isti.cnr.it");
		s.setTopic("ENRICH/MORE/OPENACCESS_VERSION");
		s.setFrequency(NotificationFrequency.daily);
		s.setMode(NotificationMode.MOCK);
		s.setConditions(
				"[{\"field\":\"target_datasource_name\",\"fieldType\":\"STRING\",\"operator\":\"EXACT\",\"listParams\":[{\"value\":\"Research Papers in Economics\"}]},{\"field\":\"trust\",\"fieldType\":\"FLOAT\",\"operator\":\"RANGE\",\"listParams\":[{\"value\":\"0\",\"otherValue\":\"1\"}]}]");

		final BoolQueryBuilder mapQuery = QueryBuilders.boolQuery();

		s.getConditionsAsList().stream()
				.map(MapCondition::asQueryBuilder)
				.filter(Objects::nonNull)
				.forEach(mapQuery::must);

		final SearchQuery searchQuery = new NativeSearchQueryBuilder()
				.withQuery(QueryBuilders.boolQuery()
						.must(QueryBuilders.matchQuery("topic", s.getTopic()))
						.must(QueryBuilders.rangeQuery("creationDate").from(s.getLastNotificationDate()))
						.must(QueryBuilders.nestedQuery("map", mapQuery, ScoreMode.None)))
				.withSearchType(SearchType.DEFAULT)
				.withIndices(indexName)
				.withTypes(indexType)
				.withPageable(PageRequest.of(0, 10))
				.build();

		int count = 0;
		final CloseableIterator<Event> it = this.elasticsearchTemplate.stream(searchQuery, Event.class);
		while (it.hasNext()) {
			System.out.println(" > " + it.next());
			count++;
		}
		System.out.println("SIZE: " + count);

	}
}