Added the possibility of genearting sm, am and registry separately
[sfa.git] / wsdl / gw2wsdl.py
index 257cd69..558ad1f 100755 (executable)
@@ -2,8 +2,6 @@
 #
 # Sapan Bhatia <sapanb@cs.princeton.edu>
 #
-# This code is under preliminary development. I am going to clean it up
-# once it is tested to work.
 # Generates a WSDL for geniwrapper
 
 
@@ -12,12 +10,16 @@ import time
 import pdb
 import xml.dom.minidom
 import xml.dom.ext
-import globals
 import apistub
+import inspect
+
 from types import *
+from optparse import OptionParser
 
-from geni.util.auth import Auth
-from geni.util.parameter import Parameter,Mixed
+from sfa.trust.auth import Auth
+from sfa.util.parameter import Parameter,Mixed
+
+import globals
 
 complex_types = {}
 services = {}
@@ -44,7 +46,6 @@ def fold_complex_type_names(acc, arg):
         acc = [python_is_braindead,name]
     return acc
 
-
 def fold_complex_type(acc, arg):
     global complex_types
     name = name_complex_type(arg)
@@ -60,11 +61,9 @@ def name_complex_type(arg):
     global num_types
     global types
 
-    #if (complex_types.has_key(arg)):
-    #    return complex_types[arg]
-
     types_section = types.getElementsByTagName("xsd:schema")[0]
 
+    #pdb.set_trace()
     if (isinstance(arg, Mixed)):
         inner_types = reduce(fold_complex_type, arg)
         inner_names = reduce(fold_complex_type_names, arg)
@@ -92,17 +91,12 @@ def name_complex_type(arg):
         inner_type = name_complex_type(arg[0]) 
         num_types=num_types+1
         type_name = "Type%d"%num_types
-        complex_type = types_section.appendChild(types.createElement("xsd:complexType"))
+        complex_type = types_section.appendChild(types.createElement("xsd:simpleType"))
         complex_type.setAttribute("name", type_name)
-        complex_content = complex_type.appendChild(types.createElement("xsd:complexContent"))
-        restriction = complex_content.appendChild(types.createElement("restriction"))
-        restriction.setAttribute("base","soapenc:Array")
-        attribute = restriction.appendChild(types.createElement("attribute"))
-        attribute.setAttribute("ref","soapenc:arrayType")
-        attribute.setAttribute("wsdl:arrayType",inner_type+"[]")
+        complex_content = complex_type.appendChild(types.createElement("xsd:list"))
+        complex_content.setAttribute("itemType",inner_type)
         return "xsdl:%s"%type_name
-        
-    elif type(arg) == DictType or arg == DictType or issubclass(arg, dict):
+    elif type(arg) == DictType or arg == DictType or (inspect.isclass(arg) and issubclass(arg, dict)):
         num_types=num_types+1
         type_name = "Type%d"%num_types
         complex_type = types_section.appendChild(types.createElement("xsd:complexType"))
@@ -116,11 +110,13 @@ def name_complex_type(arg):
             element.setAttribute("type",inner_type)
 
         return "xsdl:%s"%type_name 
-
     else:
         return (name_simple_type(arg))
 
 def name_simple_type(arg_type):
+    # A Parameter is reported as an instance, even though it is serialized as a type <>
+    if type(arg_type) == InstanceType:
+        return (name_simple_type(arg_type.type))
     if arg_type == None:
         return "none"
     if arg_type == DictType:
@@ -142,6 +138,7 @@ def param_type(arg):
 
 def add_wsdl_ports_and_bindings (wsdl):
     for method in apistub.methods:
+
         # Skip system. methods
         if "system." in method:
             continue
@@ -228,28 +225,30 @@ def add_wsdl_ports_and_bindings (wsdl):
 
 def add_wsdl_service(wsdl):
     for service in services.keys():
-        service_el = wsdl.firstChild.appendChild(wsdl.createElement("service"))
-        service_el.setAttribute("name", service)
+        global interface_options
+        if (getattr(interface_options,service)):
+            service_el = wsdl.firstChild.appendChild(wsdl.createElement("service"))
+            service_el.setAttribute("name", service)
 
-        for method in services[service]:
-            name=method
-            servport_el = service_el.appendChild(wsdl.createElement("port"))
-            servport_el.setAttribute("name", name + "_port")
-            servport_el.setAttribute("binding", "tns:" + name + "_binding")
+            for method in services[service]:
+                    name=method
+                    servport_el = service_el.appendChild(wsdl.createElement("port"))
+                    servport_el.setAttribute("name", name + "_port")
+                    servport_el.setAttribute("binding", "tns:" + name + "_binding")
 
-            soapaddress = servport_el.appendChild(wsdl.createElement("soap:address"))
-            soapaddress.setAttribute("location", "%s/%s" % (globals.plc_ns,service))
+                    soapaddress = servport_el.appendChild(wsdl.createElement("soap:address"))
+                    soapaddress.setAttribute("location", "%s/%s" % (globals.plc_ns,service))
 
 
 def get_wsdl_definitions():
     wsdl_text_header = """
         <wsdl:definitions
         name="geniwrapper_autogenerated"
-        targetNamespace="%s/2009/06/sfa.wsdl"
+        targetNamespace="%s/2009/07/sfa.wsdl"
         xmlns="http://schemas.xmlsoap.org/wsdl/"
         xmlns:xsd="http://www.w3.org/2001/XMLSchema"
-        xmlns:xsdl="http://%s/2009/06/schema"
-        xmlns:tns="%s/2009/06/sfa.wsdl"
+        xmlns:xsdl="%s/2009/07/schema"
+        xmlns:tns="%s/2009/07/sfa.wsdl"
         xmlns:soap="http://schemas.xmlsoap.org/wsdl/soap/"
         xmlns:soapenc="http://schemas.xmlsoap.org/wsdl/soap/encoding"
         xmlns:wsdl="http://schemas.xmlsoap.org/wsdl/"/>
@@ -263,28 +262,43 @@ def get_wsdl_definitions_and_types():
     wsdl_text_header = """
     <wsdl:definitions
         name="geniwrapper_autogenerated"
-        targetNamespace="%s/2009/06/sfa.wsdl"
+        targetNamespace="%s/2009/07/sfa.wsdl"
         xmlns="http://schemas.xmlsoap.org/wsdl/"
         xmlns:xsd="http://www.w3.org/2001/XMLSchema"
-        xmlns:xsdl="http://%s/2009/06/schema"
-        xmlns:tns="%s/2009/06/sfa.wsdl"
+        xmlns:xsdl="%s/2009/07/schema"
+        xmlns:tns="%s/2009/07/sfa.wsdl"
         xmlns:soap="http://schemas.xmlsoap.org/wsdl/soap/"
         xmlns:soapenc="http://schemas.xmlsoap.org/wsdl/soap/encoding"
         xmlns:wsdl="http://schemas.xmlsoap.org/wsdl/">
         <types>
-            <xsd:schema xmlns="http://www.w3.org/2001/XMLSchema" targetNamespace="%s/2009/06/schema"/>
+            <xsd:schema xmlns="http://www.w3.org/2001/XMLSchema" targetNamespace="%s/2009/07/schema"/>
         </types>
     </wsdl:definitions> """ % (globals.plc_ns, globals.plc_ns, globals.plc_ns, globals.plc_ns)
     wsdl = xml.dom.minidom.parseString(wsdl_text_header)
     return wsdl
 
-    
-types = get_wsdl_definitions_and_types()
-
-wsdl = get_wsdl_definitions()
-add_wsdl_ports_and_bindings(wsdl)
-wsdl_types = wsdl.importNode(types.getElementsByTagName("types")[0], True)
-wsdl.firstChild.appendChild(wsdl_types)
-add_wsdl_service(wsdl)
-
-xml.dom.ext.PrettyPrint(wsdl)
+def main():
+    global types
+    global interface_options
+
+    parser = OptionParser()
+    parser.add_option("-r", "--registry", dest="registry", action="store_true", 
+                              help="Generate registry.wsdl", metavar="FILE")
+    parser.add_option("-s", "--slice-manager",
+                              action="store_true", dest="slicemgr", 
+                              help="Generate sm.wsdl")
+    parser.add_option("-a", "--aggregate", action="store_true", dest="aggregate",
+                              help="Generate am.wsdl")
+
+    (interface_options, args) = parser.parse_args()
+    types = get_wsdl_definitions_and_types()
+    wsdl = get_wsdl_definitions()
+    add_wsdl_ports_and_bindings(wsdl)
+    wsdl_types = wsdl.importNode(types.getElementsByTagName("types")[0], True)
+    wsdl.firstChild.appendChild(wsdl_types)
+    add_wsdl_service(wsdl)
+
+    xml.dom.ext.PrettyPrint(wsdl)
+
+if __name__ == "__main__":
+        main()