package org.argeo.init;

import java.lang.System.Logger;
import java.lang.System.Logger.Level;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;

import org.argeo.api.init.InitConstants;
import org.argeo.api.init.RuntimeManager;
import org.argeo.init.logging.ThinLoggerFinder;
import org.argeo.init.osgi.OsgiRuntimeContext;
import org.argeo.internal.init.InternalState;

/**
 * Dynamically configures and launches multiple runtimes, coordinated by a main
 * one.
 */
public class RuntimeManagerMain {
	private final static Logger logger = System.getLogger(RuntimeManagerMain.class.getName());

	private final static String ENV_STATE_DIRECTORY = "STATE_DIRECTORY";
	private final static String ENV_CONFIGURATION_DIRECTORY = "CONFIGURATION_DIRECTORY";
	private final static String ENV_CACHE_DIRECTORY = "CACHE_DIRECTORY";

	private final static long RUNTIME_SHUTDOWN_TIMEOUT = 60 * 1000;

	private Map<String, String> configuration = new HashMap<>();

	RuntimeManagerMain(Path configArea, Path stateArea, Path cacheArea) {
		RuntimeManager.loadDefaults(configuration);

		configuration.put(InitConstants.PROP_OSGI_USE_SYSTEM_PROPERTIES, "false");

		configuration.put(InitConstants.PROP_ARGEO_CONFIG_AREA, configArea.toString());
		configuration.put(InitConstants.PROP_ARGEO_STATE_AREA, stateArea.toString());
		configuration.put(InitConstants.PROP_ARGEO_CACHE_AREA, cacheArea.toString());

		configuration.put(InitConstants.PROP_OSGI_SHARED_CONFIGURATION_AREA, configArea.toUri().toString());
		configuration.put(InitConstants.PROP_OSGI_SHARED_CONFIGURATION_AREA_RO, "true");

		configuration.put(InitConstants.PROP_OSGI_CONFIGURATION_AREA,
				cacheArea.resolve(RuntimeManager.OSGI_STORAGE_DIRNAME).toUri().toString());
		configuration.put(InitConstants.PROP_OSGI_INSTANCE_AREA,
				stateArea.resolve(RuntimeManager.DATA).toUri().toString());

		// TODO find a cleaner way to configure Jackrabbit indexes
		configuration.put("argeo.node.repo.indexesBase", cacheArea.resolve("indexes").toString());

		logger.log(Level.TRACE, () -> "Runtime manager configuration: " + configuration);
	}

	public void run() {
		OsgiRuntimeContext managerRuntimeContext = new OsgiRuntimeContext(OsgiRuntimeContext.loadFrameworkFactory(),
				configuration);
		try {
			managerRuntimeContext.run();
			InternalState.setMainRuntimeContext(managerRuntimeContext);

			// shutdown on exit
			Runtime.getRuntime().addShutdownHook(new Thread(() -> shutdown(), "Runtime shutdown"));

//			BundleContext bc = managerRuntimeContext.getFramework().getBundleContext();
//			// uninstall init as a bundle since it will be available via OSGi system
//			OsgiBoot.uninstallBundles(bc, SYMBOLIC_NAME_INIT);
//			bc.registerService(RuntimeManager.class, this, new Hashtable<>(configuration));
			logger.log(Level.DEBUG, "Registered runtime manager");

			managerRuntimeContext.waitForStop(0);
		} catch (InterruptedException e) {
			e.printStackTrace();
			System.exit(1);
		}

	}

	protected void shutdown() {
		// shutdown manager runtime
		try {
			InternalState.getMainRuntimeContext().close();
			InternalState.getMainRuntimeContext().waitForStop(RUNTIME_SHUTDOWN_TIMEOUT);
//			logger.log(Logger.Level.INFO, "Argeo Init stopped with PID " + ProcessHandle.current().pid());
			System.out.flush();
		} catch (Exception e) {
			e.printStackTrace();
			Runtime.getRuntime().halt(1);
		}
	}

	public static void main(String[] args) {
		ThinLoggerFinder.reloadConfiguration();
		logger.log(Logger.Level.DEBUG, () -> "Argeo Init starting with PID " + ProcessHandle.current().pid());

		Path writableArea = getLocalPath(InitConstants.PROP_ARGEO_STATE_AREA, ENV_STATE_DIRECTORY, null);
		Path configArea = getLocalPath(InitConstants.PROP_ARGEO_CONFIG_AREA, ENV_CONFIGURATION_DIRECTORY, null);
		Path cacheArea = getLocalPath(InitConstants.PROP_ARGEO_CACHE_AREA, ENV_CACHE_DIRECTORY, writableArea);
		RuntimeManagerMain runtimeManager = new RuntimeManagerMain(configArea, writableArea, cacheArea);
		runtimeManager.run();
	}

	private static Path getLocalPath(String systemProperty, String environmentVariable, Path defaultPath) {
		String prop = System.getProperty(systemProperty);
		if (prop != null)
			return Paths.get(prop);
		String env = System.getenv().get(environmentVariable);
		if (env != null)
			return Paths.get(env);
		if (defaultPath != null)
			return defaultPath;
		throw new IllegalStateException("No local path set with system property " + systemProperty
				+ " or environment variable " + environmentVariable);
		// TODO allocate a temporary directory? or defaults based on working directory ?
	}
}