import gdb.printing

from .address import KapiMemoryAddressPrinter
from .device import KapiDevicesDevicePrinter


def build_pretty_printers():
    pp = gdb.printing.RegexpCollectionPrettyPrinter("kapi")
    pp.add_printer(
        "kapi_memory_address", "^kapi::memory::address<.*>$", KapiMemoryAddressPrinter
    )
    pp.add_printer(
        "kapi_devices_device", "^kapi::devices::device$", KapiDevicesDevicePrinter
    )
    return pp


def register_printers(objfile):
    gdb.printing.register_pretty_printer(objfile, build_pretty_printers(), replace=True)