import lldb


class Resolver:
    got_files = 0
    func_list = []

    def __init__(self, bkpt, extra_args, dict):
        self.bkpt = bkpt
        self.extra_args = extra_args

        Resolver.func_list = []
        Resolver.got_files = 0

    def __callback__(self, sym_ctx):
        sym_name = "not_a_real_function_name"
        sym_item = self.extra_args.GetValueForKey("symbol")
        if sym_item.IsValid():
            sym_name = sym_item.GetStringValue(1000)
        else:
            print("Didn't have a value for key 'symbol'")

        if sym_ctx.compile_unit.IsValid():
            Resolver.got_files = 1
        else:
            Resolver.got_files = 2

        if sym_ctx.function.IsValid():
            Resolver.got_files = 3
            func_name = sym_ctx.function.GetName()
            Resolver.func_list.append(func_name)
            if sym_name == func_name:
                self.bkpt.AddLocation(sym_ctx.function.GetStartAddress())
            return

        if sym_ctx.module.IsValid():
            sym = sym_ctx.module.FindSymbol(sym_name, lldb.eSymbolTypeCode)
            if sym.IsValid():
                self.bkpt.AddLocation(sym.GetStartAddress())

    def get_short_help(self):
        return "I am a python breakpoint resolver"


class ResolverModuleDepth(Resolver):
    def __get_depth__(self):
        return lldb.eSearchDepthModule


class ResolverCUDepth(Resolver):
    def __get_depth__(self):
        return lldb.eSearchDepthCompUnit


class ResolverFuncDepth(Resolver):
    def __get_depth__(self):
        return lldb.eSearchDepthFunction


class ResolverBadDepth(Resolver):
    def __get_depth__(self):
        return lldb.kLastSearchDepthKind + 1