from types import MethodType

env = ij.get("org.scijava.legacy.service.OpEnvironmentService").env()

op_names={str(name) for info in env.infos() for name in info.names()}

#ns={op[:op.index('.')] for op in op_names if op.find('.') >= 0}

class OpNamespace:
    def __init__(self, env, ns):
        self.env = env
        self.ns = ns

class OpGateway(OpNamespace):
    def __init__(self, env):
        super().__init__(env, 'global')

def add_op(c, op_name):
    if hasattr(c, op_name):
        return

    def f(self, *args, **kwargs):
        print(type(self))
        fqop = op_name if self.ns == 'global' else self.ns + "." + op_name
        run = kwargs.get('run', True)
        b = self.env.op(fqop).input(*args)
        # inplace
        # ops.filter.gauss(image, 5, inplace=0)
        if (inplace:=kwargs.get('inplace', None)) is not None:
            return b.mutate(inplace) if run else b.inplace(inplace)

        # computer
        # ops.filter.gauss(image, 5, out=result)
        if (out:=kwargs.get('out', None)) is not None:
            b=b.output(out)
            return b.compute() if run else b.computer()

        # function
        # gauss_op = ops.filter.gauss(image, 5)
        # result = ops.filter.gauss(image, 5)
        return b.apply() if run else b.function()

    if c == OpGateway:
        # op_name is a global.
        setattr(c, op_name, f)
    else:
        #if isinstance(c, MethodType):
            #print("oh no! - race condition happened")
        m=MethodType(f, c)
        setattr(c, op_name, m)

# ipython HACKS
globals()['add_op'] = add_op
globals()['OpNamespace'] = OpNamespace
globals()['OpGateway'] = OpGateway
globals()['MethodType'] = MethodType

def add_namespace(c, ns, env, op_name):
    if not hasattr(c, ns):
        setattr(c, ns, OpNamespace(env, ns))
    add_op(getattr(c, ns), op_name)

for op in op_names:
    print(op)
    dot = op.find('.')
    if dot >= 0:
        ns=op[:dot]
        op_name=op[dot+1:]
        add_namespace(OpGateway, ns, env, op_name)
    else:
        print(f"registering global op: {op}")
        add_op(OpGateway, op)

ops = OpGateway(env)