package org.simantics;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Path;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Platform;
import org.eclipse.core.runtime.Status;
import org.eclipse.equinox.app.IApplication;
import org.eclipse.equinox.app.IApplicationContext;
import org.eclipse.osgi.service.datalocation.Location;
import org.simantics.application.arguments.Arguments;
import org.simantics.application.arguments.IArgumentFactory;
import org.simantics.application.arguments.IArgumentFactory.StringArgumentFactory;
import org.simantics.application.arguments.IArgumentFactory.NoValueArgumentFactory;
import org.simantics.application.arguments.IArguments;
import org.simantics.application.arguments.SimanticsArguments;
import org.simantics.internal.Activator;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Tuukka Lehtonen
 * @since 1.34.0
 */
public class BaselineCreatorApplication implements IApplication {

	private static final Logger LOGGER = LoggerFactory.getLogger(BaselineCreatorApplication.class);

	private static final IArgumentFactory<String> OUTPUT = new StringArgumentFactory("-o");
	private static final IArgumentFactory<Boolean> VERBOSE = new NoValueArgumentFactory("-v");

	IArgumentFactory<?>[] accepted = {
			SimanticsArguments.RECOVERY_POLICY_FIX_ERRORS,
			SimanticsArguments.ONTOLOGY_RECOVERY_POLICY_REINSTALL,
			SimanticsArguments.DISABLE_INDEX,
			SimanticsArguments.DATABASE_ID,
			OUTPUT,
			VERBOSE,
	};

	private static String currentLocalDateTimeStamp() {
		return LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd_HHmm"));
	}

	private static Path constructOutputPath(Path workspace, IArguments parsedArgs) {
		if (parsedArgs.contains(OUTPUT)) {
			return workspace.resolve(parsedArgs.get(OUTPUT));
		} else {
			return workspace.resolve(workspace.getFileName().toString() + "-" + currentLocalDateTimeStamp() + ".zip");
		}
	}

	private static Path getInstanceLocation() throws CoreException, IOException {
		Location l = Platform.getInstanceLocation();
		if (l == null)
			throw new CoreException(new Status(IStatus.ERROR, Activator.PLUGIN_ID,
					"Workspace not defined. Use -data <path> argument to define where to place the baselining workspace."));

		Location instanceLoc = Platform.getInstanceLocation();
		if (instanceLoc == null || instanceLoc.isReadOnly())
			throw new CoreException(new Status(IStatus.ERROR, Activator.PLUGIN_ID,
					"Workspace not defined. Use -data <path> argument to define where to place the baselining workspace."));

		URL workspaceUrl = l.getURL();
		Path workspacePath = new File(workspaceUrl.getPath()).toPath();
		Files.createDirectories(workspacePath);
		return workspacePath;
	}

	@Override
	public Object start(IApplicationContext context) throws Exception {
		try {
			Path workspace = getInstanceLocation();

			String[] args = (String[]) context.getArguments().get("application.args");
			IArguments parsedArgs = Arguments.parse(args, accepted);

			Path output = constructOutputPath(workspace, parsedArgs);

			// Create database and indexes
			IProgressMonitor progress = parsedArgs.contains(VERBOSE)
					? new TimingProgressMonitor()
					: new NullProgressMonitor();
			Simantics.startUpHeadless(parsedArgs, progress);
			Simantics.shutdown(progress);

			// Create the baseline package file
			Path actualOutput = DatabaseBaselines.packageBaseline(workspace, output);
			System.out.println("OK " + actualOutput.toAbsolutePath());

			return IApplication.EXIT_OK;
		} catch (Exception e) {
			LOGGER.error("Baseline creation failed.", e);
			throw (Exception) e;
		} finally {
			System.exit(0);
		}
	}

	@Override
	public void stop() {
	}

}