package org.argeo.init;

import java.io.IOException;
import java.io.InputStream;
import java.io.UncheckedIOException;
import java.lang.System.Logger;
import java.lang.System.Logger.Level;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;
import java.util.Properties;
import java.util.TreeMap;
import java.util.function.Consumer;

import org.argeo.init.logging.ThinLoggerFinder;
import org.argeo.init.osgi.OsgiBoot;
import org.argeo.init.osgi.OsgiRuntimeContext;
import org.osgi.framework.Bundle;
import org.osgi.framework.BundleContext;

public class RuntimeManager {
	private final static Logger logger = System.getLogger(RuntimeManager.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 String JVM_ARGS = "jvm.args";

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

	private Map<String, OsgiRuntimeContext> runtimeContexts = new TreeMap<>();

	RuntimeManager(Path configArea, Path stateArea) {
		loadConfig(configArea, configuration);
		configuration.put(OsgiBoot.PROP_OSGI_CONFIGURATION_AREA, stateArea.toUri().toString());
		this.baseConfigArea = configArea.getParent();
		this.baseStateArea = stateArea.getParent();

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

	}

	public void run() {
		logger.log(Level.DEBUG, "Start OSGi");
		try (OsgiRuntimeContext runtimeContext = new OsgiRuntimeContext(configuration)) {
			runtimeContext.run();

			BundleContext bc = runtimeContext.getFramework().getBundleContext();
			// uninstall init as a bundle since it will be available via OSGi system
			for (Bundle b : bc.getBundles()) {
				if (b.getSymbolicName().equals("org.argeo.init")) {
					b.uninstall();
				}
			}
			bc.registerService(RuntimeManager.class, this, new Hashtable<>(configuration));
			logger.log(Level.DEBUG, "Registered runtime manager");

			runtimeContext.waitForStop(0);
		} catch (Exception e) {
			e.printStackTrace();
			System.exit(1);
		}

	}

	public static void loadConfig(Path dir, Map<String, String> config) {
		try {
			System.out.println("Load from "+dir);
			Path jvmArgsPath = dir.resolve(JVM_ARGS);
			if (!Files.exists(jvmArgsPath)) {
				// load from parent directory
				loadConfig(dir.getParent(), config);
			}

			if (Files.exists(dir))
				for (Path p : Files.newDirectoryStream(dir, "*.ini")) {
					Properties props = new Properties();
					try (InputStream in = Files.newInputStream(p)) {
						props.load(in);
					}
					for (Object key : props.keySet()) {
						config.put(key.toString(), props.getProperty(key.toString()));
					}
				}
		} catch (IOException e) {
			throw new UncheckedIOException("Cannot load configuration from " + dir, e);
		}
	}

	OsgiRuntimeContext loadRuntime(String relPath, Consumer<Map<String, String>> configCallback) {
		stopRuntime(relPath);
		Path stateArea = baseStateArea.resolve(relPath);
		Path configArea = baseConfigArea.resolve(relPath);
		Map<String, String> config = new HashMap<>();
		loadConfig(configArea, config);
		config.put(OsgiBoot.PROP_OSGI_CONFIGURATION_AREA, stateArea.toUri().toString());

		if (configCallback != null)
			configCallback.accept(config);
		OsgiRuntimeContext runtimeContext = new OsgiRuntimeContext(config);
		runtimeContexts.put(relPath, runtimeContext);
		return runtimeContext;
	}

	public void startRuntime(String relPath, Consumer<Map<String, String>> configCallback) {
		OsgiRuntimeContext runtimeContext = loadRuntime(relPath, configCallback);
		runtimeContext.run();
	}

	public void stopRuntime(String relPath) {
		if (!runtimeContexts.containsKey(relPath))
			return;
		RuntimeContext runtimeContext = runtimeContexts.get(relPath);
		try {
			runtimeContext.close();
			runtimeContext.waitForStop(60 * 1000);
		} catch (Exception e) {
			logger.log(Level.ERROR, "Cannot close runtime context " + relPath, e);
		} finally {
			runtimeContexts.remove(relPath);
		}

	}

	public static void main(String[] args) {
		ThinLoggerFinder.reloadConfiguration();
		Map<String, String> env = System.getenv();
//		for (String envName : new TreeSet<>(env.keySet())) {
//			System.out.format("%s=%s%n", envName, env.get(envName));
//		}
		if (args.length < 1)
			throw new IllegalArgumentException("A relative configuration directory must be specified");
		Path configArea = Paths.get(System.getProperty("user.dir"), args[0]);

		// System.out.println("## Start with PID " + ProcessHandle.current().pid());
		// System.out.println("user.dir=" + System.getProperty("user.dir"));

		Path stateArea = Paths.get(env.get(ENV_STATE_DIRECTORY));

		RuntimeManager runtimeManager = new RuntimeManager(configArea, stateArea);
		runtimeManager.run();
	}

}