package org.webslinger.servlet;

 import java.io.File;

 import java.io.FileInputStream;

 import java.io.IOException;

 import java.net.URL;

 import java.util.Collection;

 import java.util.ArrayList;

 import java.util.HashMap;

 import java.util.HashSet;

 import java.util.Iterator;

 import java.util.LinkedHashSet;

 import java.util.List;

 import java.util.Set;

 import javax.management.JMException;

 import javax.naming.NamingException;

 import javax.servlet.ServletConfig;

 import javax.servlet.ServletContext;

 import javax.servlet.ServletException;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 import javax.xml.parsers.ParserConfigurationException;

 import org.w3c.dom.Document;

 import org.w3c.dom.Element;

 import org.xml.sax.SAXException;

 import org.webslinger.AbstractMappingWebslingerServletContextFactory;

 import org.webslinger.WebslingerServletContext;

 import org.webslinger.collections.ArrayUtil;

 import org.webslinger.container.WebslingerContainer;

 import org.webslinger.servlet.WebslingerServlet;

 import org.webslinger.util.GeneratedResult;

 import org.webslinger.util.TTLCachedObject;

 import org.webslinger.util.TTLObject;

 import org.webslinger.xml.XmlUtil;

 public class XmlContextMapper extends AbstractMappingWebslingerServletContextFactory {

     protected File mappingFile;

     protected final HashSet<String> suffixes = new HashSet<String>();

     protected ServletContext servletContext;

     protected final HashMap<String, String> hostToServer = new HashMap<String, String>();

     protected final HashMap<String, Layout> servers = new HashMap<String, Layout>();

     public void init(ServletConfig config) throws ServletException, IOException {

         servletContext = config.getServletContext();

         try {

             Document doc = XmlUtil.read(servletContext.getResourceAsStream("/WEB-INF/mapping.xml"));

             for (Element e: XmlUtil.getElementsNamed(doc.getDocumentElement(), "suffix")) {

                 suffixes.add(e.getAttribute("value"));

             }

             for (Element e: XmlUtil.getElementsNamed(doc.getDocumentElement(), "server")) {

                 LinkedHashSet<String> bases = new LinkedHashSet<String>();

                 for (Element e2: XmlUtil.getElementsNamed(e, "base")) {

                     bases.add(e2.getAttribute("target"));

                 }

                 servers.put(e.getAttribute("id"), newMappingLayout(e.getAttribute("id"), e.getAttribute("target"), bases.toArray(new String[bases.size()])));

             }

             for (Element e: XmlUtil.getElementsNamed(doc.getDocumentElement(), "host")) {

                 hostToServer.put(e.getAttribute("name"), e.getAttribute("id"));

             }

             System.err.println("suffixes=" + suffixes);

             System.err.println("servers=" + servers);

             System.err.println("hostToServer=" + hostToServer);

         } catch (Exception e) {

             throw new ServletException(e.getMessage(), e);

         }

         super.init(config, config.getInitParameter("module-base"));

     }

     protected Layout[] getStartLayouts() throws ServletException, IOException {

         return new Layout[0];

     }

     public void destroy() {

         super.destroy();

     }

 /*

     protected interface Layout {

         String getTarget() throws Exception;

         String[] getBases() throws Exception;

     }

 */

     protected Set<String> getSuffixes() {

         return suffixes;

     }

     protected MappingLayout newMappingLayout(String id, String target, String[] bases) {

         return new MappingLayout(id, target, bases);

     }

     protected Layout lookupLayout(String hostName, String contextPath) {

         String id = hostToServer.get(hostName);

         if (id == null) return null;

         return servers.get(id);

     }

     protected class MappingLayout implements Layout {

         private final String id;

         private final String target;

         private final String[] bases;

         private int hashCode;

         protected MappingLayout(String id, String target, String[] bases) {

             this.id = id;

             this.target = target;

             this.bases = bases;

             hashCode = target.hashCode() ^ ArrayUtil.hashCodeHelper(bases);

         }

         public String getId() {

             return id;

         }

         public String getTarget() {

             return target;

         }

         public String[] getBases() {

             return bases;

         }

         public int hashCode() {

             return hashCode;

         }

         public boolean equals(Object o) {

             if (!(o instanceof MappingLayout)) return false;

             MappingLayout other = (MappingLayout) o;

             if (!target.equals(other.target)) return false;

             return ArrayUtil.equalsHelper(bases, other.bases);

         }

     }

 }