import json

from bureau import Bureau, add_command, add_api


class InhumanResources(Bureau):
    """
    This is a core functional bureau of the Screenless office it keeps track
    of all published methods provided by all other bureaus and can print
    the menu for the whole system or an individual item
    """

    name = "Inhuman Resources"
    prefix = "IR"
    version = 0

    def __init__(self):
        Bureau.__init__(self)
        self.menu = {}
        # update_commands(self)

    @add_api("addbureau", "Register Bureau")
    def add_bureau(self, data):
        """
        Register Bureau with Inhuman Resources

        data = { prefix: "IR",
                 bureauname: "Inhuman Resources",
                 desc: "Keep track of public resources provided by bureaus"
               }
        """
        d = json.loads(data)
        try:
            name = d["name"]
            prefix = d["prefix"]
            desc = d["desc"]
        except KeyError as e:
            print("cannot add invalid bureau:", str(e))
            return
        print("added menu")
        self.menu[prefix] = {"name": name,
                             "desc": desc,
                             "commands": {},
                             "apis": {}}

    @add_api("addcommand", "Register Command")
    def add_cmd(self, data):
        d = json.loads(data)
        try:
            prefix = d["prefix"]
            cmd = d["cmd"]
            cmdname = d["cmdname"]
            desc = d["desc"]
        except KeyError as e:
            print("cannot add invalid command:", str(e))
            return
        if prefix not in self.menu:
            # TODO: this should throw some kind of error message/log
            print("error: cannot add command ", cmd, "to non-existent prefix",
                  prefix)
        else:
            self.menu[prefix]["commands"][cmd] = {"name": cmdname,
                                                  "desc": desc}

    @add_api("addapi", "Register API Method")
    def add_api_method(self, data):
        d = json.loads(data)
        try:
            prefix = d["prefix"]
            cmd = d["api"]
            cmdname = d["apiname"]
            desc = d["desc"]
        except KeyError as e:
            print("cannot add invalid command:", str(e))
            return
        if prefix not in self.menu:
            # TODO: this should throw some kind of error message/log
            print("error: cannot add command ", cmd, "to non-existent prefix",
                  prefix)
        else:
            self.menu[prefix]["apis"][cmd] = {"name": cmdname,
                                              "desc": desc}

    @add_command("menu", "Print Menu")
    def print_menu(self):
        """
        Prints the menu of commands for all operational bureaus.
        """
        print(self.menu)


if __name__ == "__main__":
    hr = InhumanResources()
    hr.run()