/*
 * Copyright 2012 Future Systems
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.krakenapps.httpd.impl;

import java.net.InetSocketAddress;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import org.apache.felix.ipojo.annotations.Component;
import org.apache.felix.ipojo.annotations.Invalidate;
import org.apache.felix.ipojo.annotations.Provides;
import org.apache.felix.ipojo.annotations.Requires;
import org.apache.felix.ipojo.annotations.Validate;
import org.krakenapps.api.KeyStoreManager;
import org.krakenapps.api.PrimitiveConverter;
import org.krakenapps.confdb.Config;
import org.krakenapps.confdb.ConfigDatabase;
import org.krakenapps.confdb.ConfigService;
import org.krakenapps.confdb.Predicates;
import org.krakenapps.httpd.HttpConfiguration;
import org.krakenapps.httpd.HttpContext;
import org.krakenapps.httpd.HttpContextRegistry;
import org.krakenapps.httpd.HttpServer;
import org.krakenapps.httpd.HttpService;
import org.krakenapps.httpd.VirtualHost;
import org.osgi.framework.BundleContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Component(name = "http-service")
@Provides
public class HttpServiceImpl implements HttpService {
	private final Logger logger = LoggerFactory.getLogger(HttpServiceImpl.class.getName());

	@Requires
	private ConfigService conf;

	@Requires
	private KeyStoreManager keyStoreManager;

	private BundleContext bc;

	private HttpContextRegistry httpContextRegistry;

	private ConcurrentMap<InetSocketAddress, HttpServer> listeners;

	public HttpServiceImpl(BundleContext bc) {
		this.bc = bc;
		this.listeners = new ConcurrentHashMap<InetSocketAddress, HttpServer>();
		this.httpContextRegistry = new HttpContextRegistryImpl();
	}

	@Validate
	public void start() {
		// load configured servers
		ConfigDatabase db = conf.ensureDatabase("kraken-httpd");
		Collection<HttpConfiguration> configs = db.findAll(HttpConfiguration.class).getDocuments(HttpConfiguration.class);
		for (HttpConfiguration c : configs) {
			try {
				logger.info("kraken httpd: opening http server [{}]", PrimitiveConverter.serialize(c));
				HttpServer server = loadServer(c);
				server.open();

				// build regex patterns
				HttpConfiguration sc = server.getConfiguration();
				for (VirtualHost v : sc.getVirtualHosts())
					v.setHostNames(v.getHostNames());

				logger.trace("kraken httpd: opened http server [{}]", server.getConfiguration().getListenAddress());
			} catch (Throwable t) {
				logger.error("kraken httpd: cannot open server", t);
			}
		}
	}

	@Invalidate
	public void stop() {
		for (HttpServer server : listeners.values())
			server.close();
	}

	@Override
	public HttpContextRegistry getContextRegistry() {
		return httpContextRegistry;
	}

	@Override
	public HttpContext ensureContext(String name) {
		return httpContextRegistry.ensureContext(name);
	}

	@Override
	public HttpContext findContext(String name) {
		return httpContextRegistry.findContext(name);
	}

	@Override
	public Collection<InetSocketAddress> getListenAddresses() {
		return listeners.keySet();
	}

	@Override
	public HttpServer getServer(InetSocketAddress listen) {
		return listeners.get(listen);
	}

	@Override
	public HttpServer createServer(HttpConfiguration config) {
		HttpServer server = loadServer(config);
		ConfigDatabase db = conf.ensureDatabase("kraken-httpd");
		db.add(config, "kraken-httpd", "opened server " + config.getListenAddress());

		return server;
	}

	private HttpServer loadServer(HttpConfiguration config) {
		HttpServer server = new HttpServerImpl(bc, config, httpContextRegistry, keyStoreManager, conf);
		HttpServer old = listeners.putIfAbsent(config.getListenAddress(), server);
		if (old != null)
			throw new IllegalStateException("server already exists: " + config.getListenAddress());
		return server;
	}

	@Override
	public void removeServer(InetSocketAddress listen) {
		HttpServer server = listeners.remove(listen);
		if (server != null) {
			server.close();

			ConfigDatabase db = conf.ensureDatabase("kraken-httpd");
			Map<String, Object> filter = new HashMap<String, Object>();
			filter.put("listen_address", listen.getAddress().getHostAddress());
			filter.put("listen_port", listen.getPort());
			Config c = db.findOne(HttpConfiguration.class, Predicates.field(filter));
			if (c != null)
				db.remove(c, false, "kraken-httpd", "removed server " + listen);
		}
	}
}