package org.codehaus.xfire.gen;import java.io.File;import java.io.IOException;import java.io.InputStream;import java.util.HashMap;import java.util.Iterator;import java.util.Map;import java.util.StringTokenizer;import javax.net.ssl.HostnameVerifier;import javax.net.ssl.HttpsURLConnection;import javax.net.ssl.SSLSession;import org.apache.commons.logging.Log;import org.apache.commons.logging.LogFactory;import org.codehaus.xfire.XFireException;import org.codehaus.xfire.gen.jaxb.JAXBSchemaSupport;import org.codehaus.xfire.gen.jsr181.Jsr181Profile;import org.codehaus.xfire.util.ClassLoaderUtils;import org.codehaus.xfire.util.Resolver;import org.codehaus.xfire.wsdl11.parser.WSDLServiceBuilder;import org.xml.sax.InputSource;import com.sun.codemodel.JCodeModel;import com.sun.tools.xjc.api.ErrorListener;/** * A bean type class which generates client and server stubs from a wsdl. * A simple invocation goes like so: * <pre> * Wsdl11Generator gen = new Wsdl11Generator(); * gen.setWsdl("src/wsdl/service.wsdl"); * gen.setOutputDirectory("target/generated-source"); * gen.generate(); * </pre> * @author Dan Diephouse */public class Wsdl11Generator{    private static final Log log = LogFactory.getLog(Wsdl11Generator.class);        public static final String JAXB = "jaxb";    public static final String XMLBEANS = "xmlbeans";    private String wsdl;    private String baseURI;    private String outputDirectory;    private String destinationPackage;        private JCodeModel codeModel = new JCodeModel();        private String profile;    private String binding = JAXB;    private SchemaSupport support;    private String externalBindings;        private boolean explicitAnnotation;        private boolean overwrite;        private boolean generateServerStubs = true;        private boolean forceBare = false;        private ErrorListener jaxbErrorListener;        public boolean isOverwrite()    {        return overwrite;    }    public void setOverwrite(boolean forceOverwrite)    {        this.overwrite = forceOverwrite;    }        private static void configureSSL()    {                HostnameVerifier hv = new HostnameVerifier()        {            public boolean verify(String arg0, SSLSession arg1)            {                return true;            }        };        HttpsURLConnection.setDefaultHostnameVerifier(hv);    }    @SuppressWarnings("unchecked")    public void generate() throws Exception    {        File dest = new File(outputDirectory);        dest.mkdirs();         if( wsdl.toLowerCase().startsWith("https")){            configureSSL();         }                loadSchemaSupport();        if (baseURI != null && new File(baseURI).exists())        {            baseURI = new File(baseURI).toURI().toString();        }        Resolver resolver = new Resolver(baseURI, wsdl);        if (resolver.getInputStream() == null)        {            throw new XFireException("Could not find wsdl " + wsdl + " with a base URI of " + baseURI                                      + ".");        }                String wsdlUri = resolver.getURI().toString();        if (baseURI == null)        {            baseURI = wsdlUri;        }                log.info("Generating code for WSDL at " + wsdlUri + " with a base URI of " + baseURI);                InputSource source = new InputSource(resolver.getInputStream());        source.setSystemId(wsdlUri);        WSDLServiceBuilder builder = new WSDLServiceBuilder(baseURI, source);        builder.setBindingProvider(support.getBindingProvider());        builder.setForceBare(forceBare);        builder.build();                if (profile == null) profile = Jsr181Profile.class.getName();        PluginProfile profileObj =             (PluginProfile) ClassLoaderUtils.loadClass(profile, getClass()).newInstance();                GenerationContext context = new GenerationContext(codeModel, builder.getDefinition());        context.setOutputDirectory(dest);        context.setWsdlLocation(wsdlUri);        context.setBaseURI(baseURI);        context.setSchemas(builder.getSchemas());        context.setExternalBindings(getExternalBindingFiles());        context.setExplicitAnnotation(isExplicitAnnotation());        context.setServerStubOverwritten(isOverwrite());        context.setDescriptorOverwritten(isOverwrite());        context.setServerStubGenerated(isGenerateServerStubs());                if (binding.equals(JAXB) && jaxbErrorListener != null)         {            ((JAXBSchemaSupport)support).setErrorListener(jaxbErrorListener);        }        support.initialize(context);        // The schema generator may replace our code model.        this.codeModel = context.getCodeModel();                context.setServices(builder.getServices());        context.setDestinationPackage(getDestinationPackage());        context.setSchemaGenerator(support);                for (Iterator<GeneratorPlugin> pitr = profileObj.getPlugins().iterator(); pitr.hasNext();)        {            GeneratorPlugin plugin = pitr.next();                        plugin.generate(context);        }        // Write the code!        codeModel.build(dest);    }    private void loadSchemaSupport()        throws Exception    {        if (support == null)        {            if (binding.equals(JAXB))            {                support = loadSupport("org.codehaus.xfire.gen.jaxb.JAXBSchemaSupport");            }            else if (binding.equals(XMLBEANS))            {                support = loadSupport("org.codehaus.xfire.gen.xmlbeans.XmlBeansSchemaSupport");            }            else            {                throw new Exception("Illegal binding: " + binding);            }        }    }      private Map<String,InputStream> getExternalBindingFiles() throws IOException    {        if (externalBindings == null) return null;                Map<String,InputStream> files = new HashMap<String,InputStream>();                StringTokenizer st = new StringTokenizer(externalBindings, ",");        while (st.hasMoreTokens())         {            String name = st.nextToken();                     Resolver resolver = new Resolver(baseURI, name);            if (resolver.getInputStream() == null)                throw new IllegalStateException("Could not find binding file " + name);                        files.put(resolver.getURI().toString(), resolver.getInputStream());        }        return files;    }    private SchemaSupport loadSupport(String name) throws Exception    {        return (SchemaSupport) ClassLoaderUtils.loadClass(name, getClass()).newInstance();    }    public SchemaSupport getSchemaSupport()    {        return support;    }    public void setSchemaSupport(SchemaSupport support)    {        this.support = support;    }    public String getOutputDirectory()    {        return outputDirectory;    }    public void setOutputDirectory(String outputDirectory)    {        this.outputDirectory = outputDirectory;    }    public JCodeModel getCodeModel()    {        return codeModel;    }    public String getBaseURI()    {        return baseURI;    }    public void setBaseURI(String baseURI)    {        this.baseURI = baseURI;    }    public String getWsdl()    {        return wsdl;    }    public void setWsdl(String wsdl)    {        this.wsdl = wsdl;    }    public String getDestinationPackage()    {        return destinationPackage;    }    public void setDestinationPackage(String destinationPackage)    {        this.destinationPackage = destinationPackage;    }    public String getBinding()    {        return binding;    }    public void setBinding(String binding)    {        this.binding = binding;    }    public String getProfile()    {        return profile;    }    public void setProfile(String profile)    {        this.profile = profile;    }    public void setExternalBindings(String externalBindings)    {        this.externalBindings = externalBindings;    }    public String getExternalBindings()    {        return externalBindings;    }	public boolean isExplicitAnnotation()    {        return explicitAnnotation;    }    public void setExplicitAnnotation(boolean explicitAnnotation)    {        this.explicitAnnotation = explicitAnnotation;    }    public boolean isGenerateServerStubs()    {        return generateServerStubs;    }    public void setGenerateServerStubs(boolean generateServerStubs)    {        this.generateServerStubs = generateServerStubs;    }        public void setJAXBErrorListener(ErrorListener jaxbErrorListener)    {        this.jaxbErrorListener = jaxbErrorListener;    }        public boolean isForceBare() {        return forceBare;    }    public void setForceBare(boolean forceBare) {        this.forceBare = forceBare;    }    }