import jinja2
import os
import sys
import yaml
import pdb

# add the parent directory to sys.path
import os,sys,inspect
currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
parentdir = os.path.dirname(currentdir)
sys.path.append(parentdir)

"""
{'derived_from': 'tosca.nodes.Root', 'capabilities': {'scalable': {'type': 'tosca.capabilities.Scalable'},
'service': {'type': 'tosca.capabilities.xos.Service'}}, 'properties': {'icon_url': {'required': False,
'type': 'string'}, 'public_key': {'required': False, 'type': 'string'}, 'kind': {'default': 'generic',
'type': 'string'}, 'published': {'default': True, 'type': 'boolean'}, 'view_url': {'required': False, 'type': 'string'}, 'enabled': {'default': True, 'type': 'boolean'}, 'versionNumber': {'required': False, 'type': 'string'}}}
"""

class ToscaDocumenter(object):
    def __init__(self, fn="./custom_types/xos.yaml", templatedir="./doctemplates/html", templatename="toscadoctemplate.html", destfn="tosca_reference.html"):
        self.env = jinja2.Environment(loader=jinja2.FileSystemLoader(templatedir))

        self.node_types = {}
        self.root_types = yaml.load(file("definitions/TOSCA_definition_1_0.yaml").read())
        for x in self.root_types.keys():
            if x in ["tosca.nodes.Compute", "tosca.nodes.network.Network"]:
                self.node_types[x] = self.root_types[x]

        self.custom_types = yaml.load(file(fn).read())
        self.node_types.update(self.custom_types.get("node_types"))

        self.destfn = destfn
        self.templatename = templatename

    def run(self):
        node_types=[]
        for k in sorted(self.node_types.keys()):
            nt = self.node_types[k]
            nt["node_type_name"] = k

            derived_from = nt.get("derived_from","")

            if derived_from.startswith("tosca.nodes"):
                nt["node_type_kind"] = "node"
            elif derived_from.startswith("tosca.capabilities"):
                nt["node_type_kind"] = "capability"
            elif derived_from.startswith("tosca.relationships"):
                nt["node_type_kind"] = "relationship"

            node_types.append(nt)

        template = self.env.get_template(self.templatename)

        self.destf = open(self.destfn,"w")
        self.destf.write(template.render(node_types=node_types))

def main():
    ToscaDocumenter().run()

if __name__=="__main__":
    main()