package com.github.markusbernhardt.proxy.selector.misc;

import java.io.IOException;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.SocketAddress;
import java.net.URI;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import com.github.markusbernhardt.proxy.selector.direct.NoProxySelector;

/*****************************************************************************
 * This is a facade for a list of ProxySelecor objects. You can register
 * different ProxySelectors per Protocol.
 * 
 * @author Markus Bernhardt, Copyright 2016
 * @author Bernd Rosstauscher, Copyright 2009
 ****************************************************************************/

public class ProtocolDispatchSelector extends ProxySelector {

	private Map<String, ProxySelector> selectors;
	private ProxySelector fallbackSelector;

	/*************************************************************************
	 * Constructor
	 ************************************************************************/

	public ProtocolDispatchSelector() {
		super();
		this.selectors = new ConcurrentHashMap<String, ProxySelector>();
		this.fallbackSelector = NoProxySelector.getInstance();
	}

	/*************************************************************************
	 * Sets a selector responsible for the given protocol.
	 * 
	 * @param protocol
	 *            the name of the protocol.
	 * @param selector
	 *            the selector to use.
	 ************************************************************************/

	public void setSelector(String protocol, ProxySelector selector) {
		if (protocol == null) {
			throw new NullPointerException("Protocol must not be null.");
		}
		if (selector == null) {
			throw new NullPointerException("Selector must not be null.");
		}
		this.selectors.put(protocol, selector);
	}

	/*************************************************************************
	 * Removes the selector installed for the given protocol.
	 * 
	 * @param protocol
	 *            the protocol name.
	 * @return the old selector that is removed.
	 ************************************************************************/

	public ProxySelector removeSelector(String protocol) {
		return this.selectors.remove(protocol);
	}

	/*************************************************************************
	 * Gets the selector installed for the given protocol.
	 * 
	 * @param protocol
	 *            the protocol name.
	 * @return the selector for that protocol, null if none is currently set.
	 ************************************************************************/

	public ProxySelector getSelector(String protocol) {
		return this.selectors.get(protocol);
	}

	/*************************************************************************
	 * Sets the fallback selector that is always called when no matching
	 * protocol selector was found..
	 * 
	 * @param selector
	 *            the selector to use.
	 ************************************************************************/

	public void setFallbackSelector(ProxySelector selector) {
		if (selector == null) {
			throw new NullPointerException("Selector must not be null.");
		}
		this.fallbackSelector = selector;
	}

	/*************************************************************************
	 * connectFailed
	 * 
	 * @see java.net.ProxySelector#connectFailed(java.net.URI,
	 *      java.net.SocketAddress, java.io.IOException)
	 ************************************************************************/

	@Override
	public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
		ProxySelector selector = this.fallbackSelector;
		String protocol = uri.getScheme();
		if (protocol != null && this.selectors.get(protocol) != null) {
			selector = this.selectors.get(protocol);
		}
		selector.connectFailed(uri, sa, ioe);
	}

	/*************************************************************************
	 * select
	 * 
	 * @see java.net.ProxySelector#select(java.net.URI)
	 ************************************************************************/

	@Override
	public List<Proxy> select(URI uri) {
		ProxySelector selector = this.fallbackSelector;
		String protocol = uri.getScheme();
		if (protocol != null && this.selectors.get(protocol) != null) {
			selector = this.selectors.get(protocol);
		}
		return selector.select(uri);
	}

  /*************************************************************************
   * Gets the size of the selector map.
   * 
   * @return the size of the selector map.
   ************************************************************************/

  public int size() {
    return this.selectors.size();
  }

}
