package org.argeo.slc.web.mvc.provisioning;

import java.util.Iterator;
import java.util.Set;
import java.util.SortedMap;
import java.util.StringTokenizer;
import java.util.TreeMap;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.argeo.slc.SlcException;
import org.argeo.slc.build.BasicNameVersion;
import org.argeo.slc.build.ModularDistribution;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;

public class ModularDistributionInterceptor extends HandlerInterceptorAdapter {
	private Set<ModularDistribution> modularDistributions;

	public final static String RELEASE = "RELEASE";
	public final static String LATEST = "LATEST";

	@Override
	public boolean preHandle(HttpServletRequest request,
			HttpServletResponse response, Object handler) throws Exception {
		String path = request.getPathInfo();
		StringTokenizer stS = new StringTokenizer(path, "/");
		String distributionName = stS.nextToken();
		String distributionVersion = stS.nextToken();

		SortedMap<String, ModularDistribution> choices = new TreeMap<String, ModularDistribution>();
		distribs: for (Iterator<ModularDistribution> it = modularDistributions
				.iterator(); it.hasNext();) {
			ModularDistribution md = it.next();
			if (md.getName().equals(distributionName)) {
				if (distributionVersion.equals(RELEASE)
						&& md.getVersion().contains("SNAPSHOT"))
					continue distribs;

				else if (distributionVersion.equals(LATEST))
					choices.put(md.getVersion(), md);
				else if (distributionVersion.equals(md.getVersion())) {
					choices.put(md.getVersion(), md);
					break distribs;
				}
			}
		}

		if (choices.size() == 0)
			throw new SlcException("Cannot find distribution for "
					+ new BasicNameVersion(distributionName,
							distributionVersion));

		ModularDistribution modularDistribution = choices.get(choices
				.firstKey());

		request.setAttribute("modularDistribution", modularDistribution);

		return true;
	}

	public void setModularDistributions(
			Set<ModularDistribution> modularDistributions) {
		this.modularDistributions = modularDistributions;
	}

}