private Binding binding;

    private Element types;

    private Definition definition;

    private WSDLFactory factory;

   

   

    public void fixBindingOperation(String name,BindingOperation operation){

        BindingOutput out = new BindingOutputImpl();

        SOAPBodyImpl bodyImpl = new SOAPBodyImpl();

        bodyImpl.setUse("literal");

        bodyImpl.setNamespaceURI("http://extreme.indiana.edu/xregistry2/2007_02_21");

        operation.setBindingOutput(out);

    }

    public void fixOperation(Message message,Operation op){

        Output output = new OutputImpl();

        output.setMessage(message);

        op.setOutput(output);

    }

   

   

    public Element createType(String name,Document doc){

        Element ele = doc.createElement("xs:element");

        ele.setAttribute("name",name+"Response");

        ele.appendChild(doc.createElement("xs:complexType"));

        return ele;

    }

//    <wsdl:part name="part1" element="ns:listGroupsResponse" />

//</wsdl:message>

   

    public Message createMessage(String name){

        Message message =  new MessageImpl();

        message.setQName(new QName("http://extreme.indiana.edu/xregistry2/2007_02_21",name+"ResponseMessage","msgns"));

        Part part = new PartImpl();

        part.setElementName(new QName("http://extreme.indiana.edu/xregistry2/2007_02_21",name+"Response","ns"));

        message.addPart(part);

        return message;

    }

   

   

   

    public static void main(String[] args) throws Exception{

        WSDLFixer fixer = new WSDLFixer();

        fixer.fix();

//            javax.xml.parsers.DocumentBuilderFactory.newInstance();

//        javax.xml.parsers.DocumentBuilder builder =  domfactory.newDocumentBuilder();

//       

//

//        DOMImplementation dImpl = builder.getDOMImplementation();

    }

   

    public void fix() throws Exception{

       

       

        String targetNs = "http://extreme.indiana.edu/xregistry2/2007_02_21";

        //Document doc = dImpl.createDocument(targetNs, "factoryServices", null);

       

       

        String wsdlAsStr = Utils.readFile("src/xregistry.wsdl");

      

        WSDLReader reader = WSDLFactory.newInstance().newWSDLReader();

        Definition olddefinition = reader.readWSDL(null, new InputSource(new StringReader(wsdlAsStr)));

        //Definition definition = reader.readWSDL(null, new InputSource(new StringReader(wsdlAsStr)));;

        Definition definition = WSDLFactory.newInstance().newDefinition();

       

        Iterator ns = olddefinition.getNamespaces().keySet().iterator();

        while(ns.hasNext()){

            String prefix = (String)ns.next();

            String nsURI = (String)olddefinition.getNamespaces().get(prefix);

            definition.addNamespace(prefix,nsURI);

        }

        Iterator bs = olddefinition.getBindings().values().iterator();

        while(bs.hasNext()){

            definition.addBinding((Binding)bs.next());

        }

       

        Iterator pts = olddefinition.getPortTypes().values().iterator();

        while(pts.hasNext()){

            definition.addPortType((PortType)pts.next());

        }

       

        Iterator msgs = olddefinition.getMessages().values().iterator();

        while(msgs.hasNext()){

            definition.addMessage((Message)msgs.next());

        }

       

        Iterator ss = olddefinition.getServices().values().iterator();

        while(ss.hasNext()){

            definition.addService((javax.wsdl.Service)ss.next());

        }

        definition.setTypes(olddefinition.getTypes());

       

       

       

        Binding binding = definition.getBinding(new QName(targetNs,"IXregistrySOAP11Binding"));

        SchemaImpl impl;

        Document doc = null;

       

        Iterator it = definition.getTypes().getExtensibilityElements().iterator();

        if(it.hasNext()){

            impl = (SchemaImpl)it.next();

            types = impl.getElement();

           

            doc = types.getOwnerDocument();

           

           

        }

       

       // System.out.println(binding.toString());

       

       

       

       

       

       

       

       

       

       

        Iterator ops = binding.getPortType().getOperations().iterator();

        while(ops.hasNext()){

            Operation op = (Operation)ops.next();

            if(op.getOutput() == null){

               

                String name = op.getName();

                //System.out.println("Fixed ="+name);

                //System.out.println("<wsdl:message name=\""+name+"ResponseMessage\">\n<wsdl:part name=\"part1\" element=\"ns:"+name+"Response\" />\n</wsdl:message>");

                System.out.println("<wsdl:operation name=\""+name+"\">\n<soap:operation soapAction=\"urn:addCapability\" style=\"document\" />\n<wsdl:input><soap:body use=\"literal\" namespace=\"http://extreme.indiana.edu/xregistry2/2007_02_21\" /></wsdl:input></wsdl:operation>");

           

               

                Element ele = createType(name, doc);

                types.appendChild(ele);

                Message message = createMessage(name);

                definition.addMessage(message);

                fixOperation(message, op);

                fixBindingOperation(name, binding.getBindingOperation(name, null, null));

            }

        }

       

       

       

        WSDLWriter w =  WSDLFactory.newInstance().newWSDLWriter();

       

        w.writeWSDL(definition, System.out);

       

    }

   

   

   

//       

//    }