import logging


class Register:
    def __init__(self, registry_name):
        self._dict = {}
        self._name = registry_name

    def __setitem__(self, key, value):
        if not callable(value):
            raise Exception(f"Value of a Registry must be a callable!\nValue: {value}")
        if key is None:
            key = value.__name__
        if key in self._dict:
            logging.warning("Key %s already in registry %s." % (key, self._name))
        self._dict[key] = value

    def register(self, target):
        """Decorator to register a function or class."""

        def add(key, value):
            self[key] = value
            return value

        if callable(target):
            # @reg.register
            return add(None, target)
        # @reg.register('alias')
        return lambda x: add(target, x)

    def __getitem__(self, key):
        return self._dict[key]

    def __contains__(self, key):
        return key in self._dict

    def keys(self):
        return self._dict.keys()

    def keys_str(self):
        return ",".join(self._dict.keys())

    def items(self):
        """Returns all registered items as a list of tuples (key, value)."""
        return list(self._dict.items())

    def dict(self):
        return self._dict


"""
python -m src.common.register
"""

if __name__ == "__main__":
    functions = Register("llm_function_calling")

    @functions.register("test")
    def test():
        print("test")

    @functions.register
    def test1():
        print("test")

    @functions.register
    def test2():
        print("test")

    print(functions.items())
