package org.argeo.cms.internal.kernel;

import java.io.IOException;
import java.io.InputStream;
import java.io.Writer;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.Dictionary;
import java.util.List;
import java.util.SortedMap;
import java.util.TreeMap;

import javax.naming.InvalidNameException;
import javax.naming.directory.Attributes;
import javax.naming.directory.BasicAttributes;
import javax.naming.ldap.LdapName;
import javax.naming.ldap.Rdn;
import javax.websocket.server.ServerEndpointConfig;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.argeo.api.NodeConstants;
import org.argeo.cms.CmsException;
import org.argeo.cms.internal.http.InternalHttpConstants;
import org.argeo.cms.websocket.CmsWebSocketConfigurator;
import org.argeo.naming.AttributesDictionary;
import org.argeo.naming.LdifParser;
import org.argeo.naming.LdifWriter;
import org.argeo.osgi.useradmin.UserAdminConf;
import org.eclipse.equinox.http.jetty.JettyConfigurator;
import org.osgi.framework.BundleContext;
import org.osgi.framework.FrameworkUtil;
import org.osgi.service.cm.Configuration;
import org.osgi.service.cm.ConfigurationAdmin;
import org.osgi.service.cm.ConfigurationEvent;
import org.osgi.service.cm.ConfigurationListener;

/** Manages the LDIF-based deployment configuration. */
class DeployConfig implements ConfigurationListener {
	private final Log log = LogFactory.getLog(getClass());
	private final BundleContext bc = FrameworkUtil.getBundle(getClass()).getBundleContext();

	private static Path deployConfigPath = KernelUtils.getOsgiInstancePath(KernelConstants.DEPLOY_CONFIG_PATH);
	private SortedMap<LdapName, Attributes> deployConfigs = new TreeMap<>();
	private final DataModels dataModels;

	public DeployConfig(ConfigurationAdmin configurationAdmin, DataModels dataModels, boolean isClean) {
		this.dataModels = dataModels;
		// ConfigurationAdmin configurationAdmin =
		// bc.getService(bc.getServiceReference(ConfigurationAdmin.class));
		try {
			boolean isFirstInit = false;
			if (!isInitialized()) { // first init
				isFirstInit = true;
				firstInit();
			}
			init(configurationAdmin, isClean, isFirstInit);
		} catch (IOException e) {
			throw new CmsException("Could not init deploy configs", e);
		}
		// FIXME check race conditions during initialization
		// bc.registerService(ConfigurationListener.class, this, null);
	}

	private void firstInit() throws IOException {
		log.info("## FIRST INIT ##");
		Files.createDirectories(deployConfigPath.getParent());

		// FirstInit firstInit = new FirstInit();
		InitUtils.prepareFirstInitInstanceArea();

		if (!Files.exists(deployConfigPath))
			deployConfigs = new TreeMap<>();
		else// config file could have juste been copied by preparation
			try (InputStream in = Files.newInputStream(deployConfigPath)) {
				deployConfigs = new LdifParser().read(in);
			}
		save();
	}

	private void setFromFrameworkProperties(boolean isFirstInit) {
		// node repository
		Dictionary<String, Object> nodeConfig = InitUtils
				.getNodeRepositoryConfig(getProps(NodeConstants.NODE_REPOS_FACTORY_PID, NodeConstants.NODE));
		// node repository is mandatory
		putFactoryDeployConfig(NodeConstants.NODE_REPOS_FACTORY_PID, nodeConfig);

		// additional repositories
		dataModels: for (DataModels.DataModel dataModel : dataModels.getNonAbstractDataModels()) {
			if (NodeConstants.NODE_REPOSITORY.equals(dataModel.getName()))
				continue dataModels;
			Dictionary<String, Object> config = InitUtils.getRepositoryConfig(dataModel.getName(),
					getProps(NodeConstants.NODE_REPOS_FACTORY_PID, dataModel.getName()));
			if (config.size() != 0)
				putFactoryDeployConfig(NodeConstants.NODE_REPOS_FACTORY_PID, config);
		}

		// user admin
		List<Dictionary<String, Object>> userDirectoryConfigs = InitUtils.getUserDirectoryConfigs();
		if (userDirectoryConfigs.size() != 0) {
			List<String> activeCns = new ArrayList<>();
			for (int i = 0; i < userDirectoryConfigs.size(); i++) {
				Dictionary<String, Object> userDirectoryConfig = userDirectoryConfigs.get(i);
				String cn = UserAdminConf.baseDnHash(userDirectoryConfig);
				activeCns.add(cn);
				userDirectoryConfig.put(NodeConstants.CN, cn);
				putFactoryDeployConfig(NodeConstants.NODE_USER_ADMIN_PID, userDirectoryConfig);
			}
			// disable others
			LdapName userAdminFactoryName = serviceFactoryDn(NodeConstants.NODE_USER_ADMIN_PID);
			for (LdapName name : deployConfigs.keySet()) {
				if (name.startsWith(userAdminFactoryName) && !name.equals(userAdminFactoryName)) {
					try {
						Attributes attrs = deployConfigs.get(name);
						String cn = name.getRdn(name.size() - 1).getValue().toString();
						if (!activeCns.contains(cn)) {
							attrs.put(UserAdminConf.disabled.name(), "true");
						}
					} catch (Exception e) {
						throw new CmsException("Cannot disable user directory " + name, e);
					}
				}
			}
		}

		// http server
//		Dictionary<String, Object> webServerConfig = InitUtils
//				.getHttpServerConfig(getProps(KernelConstants.JETTY_FACTORY_PID, NodeConstants.DEFAULT));
//		if (!webServerConfig.isEmpty()) {
//			// TODO check for other customizers
//			webServerConfig.put("customizer.class", "org.argeo.equinox.jetty.CmsJettyCustomizer");
//			putFactoryDeployConfig(KernelConstants.JETTY_FACTORY_PID, webServerConfig);
//		}
		LdapName defaultHttpServiceDn = serviceDn(KernelConstants.JETTY_FACTORY_PID, NodeConstants.DEFAULT);
		if (deployConfigs.containsKey(defaultHttpServiceDn)) {
			// remove old default configs since we have now to start Jetty servlet bridge
			// indirectly
			deployConfigs.remove(defaultHttpServiceDn);
		}

		// SAVE
		save();
		//

		// Explicitly configures Jetty so that the default server is not started by the
		// activator of the Equinox Jetty bundle.
		Dictionary<String, Object> webServerConfig = InitUtils
				.getHttpServerConfig(getProps(KernelConstants.JETTY_FACTORY_PID, NodeConstants.DEFAULT));
		if (!webServerConfig.isEmpty()) {
			webServerConfig.put("customizer.class", KernelConstants.CMS_JETTY_CUSTOMIZER_CLASS);

			// TODO centralise with Jetty extender
			Object webSocketEnabled = webServerConfig.get(InternalHttpConstants.WEBSOCKET_ENABLED);
			if (webSocketEnabled != null && webSocketEnabled.toString().equals("true")) {
				bc.registerService(ServerEndpointConfig.Configurator.class, new CmsWebSocketConfigurator(), null);
				webServerConfig.put(InternalHttpConstants.WEBSOCKET_ENABLED, "true");
			}
		}

		int tryCount = 60;
		try {
			tryGettyJetty: while (tryCount > 0) {
				try {
					JettyConfigurator.startServer(KernelConstants.DEFAULT_JETTY_SERVER, webServerConfig);
					// Explicitly starts Jetty OSGi HTTP bundle, so that it gets triggered if OSGi
					// configuration is not cleaned
					FrameworkUtil.getBundle(JettyConfigurator.class).start();
					break tryGettyJetty;
				} catch (IllegalStateException e) {
					// Jetty may not be ready
					try {
						Thread.sleep(1000);
					} catch (Exception e1) {
						// silent
					}
				}
			}
		} catch (Exception e) {
			log.error("Cannot start default Jetty server with config " + webServerConfig, e);
		}

	}

	private void init(ConfigurationAdmin configurationAdmin, boolean isClean, boolean isFirstInit) throws IOException {

		try (InputStream in = Files.newInputStream(deployConfigPath)) {
			deployConfigs = new LdifParser().read(in);
		}
		if (isClean) {
			setFromFrameworkProperties(isFirstInit);
			for (LdapName dn : deployConfigs.keySet()) {
				Rdn lastRdn = dn.getRdn(dn.size() - 1);
				LdapName prefix = (LdapName) dn.getPrefix(dn.size() - 1);
				if (prefix.toString().equals(NodeConstants.DEPLOY_BASEDN)) {
					if (lastRdn.getType().equals(NodeConstants.CN)) {
						// service
						String pid = lastRdn.getValue().toString();
						Configuration conf = configurationAdmin.getConfiguration(pid);
						AttributesDictionary dico = new AttributesDictionary(deployConfigs.get(dn));
						conf.update(dico);
					} else {
						// service factory definition
					}
				} else {
					// service factory service
					Rdn beforeLastRdn = dn.getRdn(dn.size() - 2);
					assert beforeLastRdn.getType().equals(NodeConstants.OU);
					String factoryPid = beforeLastRdn.getValue().toString();
					Configuration conf = configurationAdmin.createFactoryConfiguration(factoryPid.toString(), null);
					AttributesDictionary dico = new AttributesDictionary(deployConfigs.get(dn));
					conf.update(dico);
				}
			}
		}
		// TODO check consistency if not clean
	}

	@Override
	public void configurationEvent(ConfigurationEvent event) {
		try {
			if (ConfigurationEvent.CM_UPDATED == event.getType()) {
				ConfigurationAdmin configurationAdmin = bc.getService(event.getReference());
				Configuration conf = configurationAdmin.getConfiguration(event.getPid(), null);
				LdapName serviceDn = null;
				String factoryPid = conf.getFactoryPid();
				if (factoryPid != null) {
					LdapName serviceFactoryDn = serviceFactoryDn(factoryPid);
					if (deployConfigs.containsKey(serviceFactoryDn)) {
						for (LdapName dn : deployConfigs.keySet()) {
							if (dn.startsWith(serviceFactoryDn)) {
								Rdn lastRdn = dn.getRdn(dn.size() - 1);
								assert lastRdn.getType().equals(NodeConstants.CN);
								Object value = conf.getProperties().get(lastRdn.getType());
								assert value != null;
								if (value.equals(lastRdn.getValue())) {
									serviceDn = dn;
									break;
								}
							}
						}

						Object cn = conf.getProperties().get(NodeConstants.CN);
						if (cn == null)
							throw new IllegalArgumentException("Properties must contain cn");
						if (serviceDn == null) {
							putFactoryDeployConfig(factoryPid, conf.getProperties());
						} else {
							Attributes attrs = deployConfigs.get(serviceDn);
							assert attrs != null;
							AttributesDictionary.copy(conf.getProperties(), attrs);
						}
						save();
						if (log.isDebugEnabled())
							log.debug("Updated deploy config " + serviceDn(factoryPid, cn.toString()));
					} else {
						// ignore non config-registered service factories
					}
				} else {
					serviceDn = serviceDn(event.getPid());
					if (deployConfigs.containsKey(serviceDn)) {
						Attributes attrs = deployConfigs.get(serviceDn);
						assert attrs != null;
						AttributesDictionary.copy(conf.getProperties(), attrs);
						save();
						if (log.isDebugEnabled())
							log.debug("Updated deploy config " + serviceDn);
					} else {
						// ignore non config-registered services
					}
				}
			}
		} catch (Exception e) {
			log.error("Could not handle configuration event", e);
		}
	}

	void putFactoryDeployConfig(String factoryPid, Dictionary<String, Object> props) {
		Object cn = props.get(NodeConstants.CN);
		if (cn == null)
			throw new IllegalArgumentException("cn must be set in properties");
		LdapName serviceFactoryDn = serviceFactoryDn(factoryPid);
		if (!deployConfigs.containsKey(serviceFactoryDn))
			deployConfigs.put(serviceFactoryDn, new BasicAttributes(NodeConstants.OU, factoryPid));
		LdapName serviceDn = serviceDn(factoryPid, cn.toString());
		Attributes attrs = new BasicAttributes();
		AttributesDictionary.copy(props, attrs);
		deployConfigs.put(serviceDn, attrs);
	}

	void putDeployConfig(String servicePid, Dictionary<String, Object> props) {
		LdapName serviceDn = serviceDn(servicePid);
		Attributes attrs = new BasicAttributes(NodeConstants.CN, servicePid);
		AttributesDictionary.copy(props, attrs);
		deployConfigs.put(serviceDn, attrs);
	}

	void save() {
		try (Writer writer = Files.newBufferedWriter(deployConfigPath)) {
			new LdifWriter(writer).write(deployConfigs);
		} catch (IOException e) {
			// throw new CmsException("Cannot save deploy configs", e);
			log.error("Cannot save deploy configs", e);
		}
	}

	boolean isStandalone(String dataModelName) {
		return getProps(NodeConstants.NODE_REPOS_FACTORY_PID, dataModelName) != null;
	}

	/*
	 * UTILITIES
	 */
	private LdapName serviceFactoryDn(String factoryPid) {
		try {
			return new LdapName(NodeConstants.OU + "=" + factoryPid + "," + NodeConstants.DEPLOY_BASEDN);
		} catch (InvalidNameException e) {
			throw new IllegalArgumentException("Cannot generate DN from " + factoryPid, e);
		}
	}

	private LdapName serviceDn(String servicePid) {
		try {
			return new LdapName(NodeConstants.CN + "=" + servicePid + "," + NodeConstants.DEPLOY_BASEDN);
		} catch (InvalidNameException e) {
			throw new IllegalArgumentException("Cannot generate DN from " + servicePid, e);
		}
	}

	private LdapName serviceDn(String factoryPid, String cn) {
		try {
			return (LdapName) serviceFactoryDn(factoryPid).add(new Rdn(NodeConstants.CN, cn));
		} catch (InvalidNameException e) {
			throw new IllegalArgumentException("Cannot generate DN from " + factoryPid + " and " + cn, e);
		}
	}

	Dictionary<String, Object> getProps(String factoryPid, String cn) {
		Attributes attrs = deployConfigs.get(serviceDn(factoryPid, cn));
		if (attrs != null)
			return new AttributesDictionary(attrs);
		else
			return null;
	}

	static boolean isInitialized() {
		return Files.exists(deployConfigPath);
	}

}