
package eu.dnetlib.dhp.oa.provision;

import java.io.File;
import java.nio.file.Path;

import org.apache.commons.io.FileUtils;
import org.apache.solr.client.solrj.embedded.JettyConfig;
import org.apache.solr.client.solrj.impl.CloudSolrClient;
import org.apache.solr.client.solrj.request.CollectionAdminRequest;
import org.apache.solr.client.solrj.request.ConfigSetAdminRequest;
import org.apache.solr.client.solrj.request.QueryRequest;
import org.apache.solr.cloud.MiniSolrCloudCluster;
import org.apache.solr.common.params.CollectionParams;
import org.apache.solr.common.params.CoreAdminParams;
import org.apache.solr.common.params.ModifiableSolrParams;
import org.apache.solr.common.util.NamedList;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.io.TempDir;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public abstract class SolrTest {

	protected static final Logger log = LoggerFactory.getLogger(SolrTest.class);

	protected static final String FORMAT = "test";
	protected static final String DEFAULT_COLLECTION = FORMAT + "-index-openaire";
	protected static final String CONFIG_NAME = "testConfig";

	protected static MiniSolrCloudCluster miniCluster;

	@TempDir
	public static Path workingDir;

	@BeforeAll
	public static void setup() throws Exception {

		// random unassigned HTTP port
		final int jettyPort = 0;
		final JettyConfig jettyConfig = JettyConfig.builder().setPort(jettyPort).build();

		log.info(String.format("working directory: %s", workingDir.toString()));
		System.setProperty("solr.log.dir", workingDir.resolve("logs").toString());

		// create a MiniSolrCloudCluster instance
		miniCluster = new MiniSolrCloudCluster(2, workingDir.resolve("solr"), jettyConfig);

		// Upload Solr configuration directory to ZooKeeper
		String solrZKConfigDir = "src/test/resources/eu/dnetlib/dhp/oa/provision/solr/conf/testConfig";
		File configDir = new File(solrZKConfigDir);

		miniCluster.uploadConfigSet(configDir.toPath(), CONFIG_NAME);

		// override settings in the solrconfig include
		System.setProperty("solr.tests.maxBufferedDocs", "100000");
		System.setProperty("solr.tests.maxIndexingThreads", "-1");
		System.setProperty("solr.tests.ramBufferSizeMB", "100");

		// use non-test classes so RandomizedRunner isn't necessary
		System.setProperty("solr.tests.mergeScheduler", "org.apache.lucene.index.ConcurrentMergeScheduler");
		System.setProperty("solr.directoryFactory", "solr.RAMDirectoryFactory");
		System.setProperty("solr.lock.type", "single");

		log.info(new ConfigSetAdminRequest.List().process(miniCluster.getSolrClient()).toString());
		log
			.info(
				CollectionAdminRequest.ClusterStatus
					.getClusterStatus()
					.process(miniCluster.getSolrClient())
					.toString());

		NamedList<Object> res = createCollection(
			miniCluster.getSolrClient(), DEFAULT_COLLECTION, 4, 2, 20, CONFIG_NAME);
		res.forEach(o -> log.info(o.toString()));

		miniCluster.getSolrClient().setDefaultCollection(DEFAULT_COLLECTION);

		log
			.info(
				CollectionAdminRequest.ClusterStatus
					.getClusterStatus()
					.process(miniCluster.getSolrClient())
					.toString());

	}

	@AfterAll
	public static void shutDown() throws Exception {
		miniCluster.shutdown();
		FileUtils.deleteDirectory(workingDir.toFile());
	}

	protected static NamedList<Object> createCollection(CloudSolrClient client, String name, int numShards,
		int replicationFactor, int maxShardsPerNode, String configName) throws Exception {
		ModifiableSolrParams modParams = new ModifiableSolrParams();
		modParams.set(CoreAdminParams.ACTION, CollectionParams.CollectionAction.CREATE.name());
		modParams.set("name", name);
		modParams.set("numShards", numShards);
		modParams.set("replicationFactor", replicationFactor);
		modParams.set("collection.configName", configName);
		modParams.set("maxShardsPerNode", maxShardsPerNode);
		QueryRequest request = new QueryRequest(modParams);
		request.setPath("/admin/collections");
		return client.request(request);
	}

}
