get_mode

#!/usr/bin/env python3

import argparse

import common_bf


def main() -> None:
    parser = argparse.ArgumentParser(description="Reads the current mode of the BF.")
    parser.add_argument(
        "-i",
        "--id",
        dest="id",
        default=0,
        action="store",
        type=int,
        help="Specify the id of the BF.",
    )
    parser.add_argument(
        "-v",
        "--verbose",
        dest="verbose",
        default=False,
        action="store",
        type=bool,
        help="Enable verbose output. This will show the settings",
    )
    parser.add_argument(
        "-n",
        "--next_boot",
        action="store_true",
        help="Get the next_boot mode of the BF",
    )
    args = parser.parse_args()

    bf = common_bf.find_bf_pci_addresses_or_quit(args.id)

    cfg = [
        "INTERNAL_CPU_MODEL",
        "INTERNAL_CPU_PAGE_SUPPLIER",
        "INTERNAL_CPU_ESWITCH_MANAGER",
        "INTERNAL_CPU_IB_VPORT0",
        "INTERNAL_CPU_OFFLOAD_ENGINE",
    ]
    all_cfg = " ".join(cfg)
    ret = common_bf.run(f"mstconfig -e -d {bf} q {all_cfg}").out

    save_next = False
    settings = {}
    for e in ret.split("\n"):
        if not e:
            continue
        if e.startswith("Configurations:"):
            save_next = True
        elif save_next:
            if "different from default/current" in e:
                save_next = False
                continue
            k, default, current, next_boot = e.lstrip("*").split()
            config = next_boot if args.next_boot else current
            settings[k] = config.split("(")[1].split(")")[0]

    dpu_mode = {
        "INTERNAL_CPU_MODEL": "1",
        "INTERNAL_CPU_PAGE_SUPPLIER": "0",
        "INTERNAL_CPU_ESWITCH_MANAGER": "0",
        "INTERNAL_CPU_IB_VPORT0": "0",
        "INTERNAL_CPU_OFFLOAD_ENGINE": "0",
    }

    nic_mode = {
        "INTERNAL_CPU_MODEL": "1",
        "INTERNAL_CPU_PAGE_SUPPLIER": "1",
        "INTERNAL_CPU_ESWITCH_MANAGER": "1",
        "INTERNAL_CPU_IB_VPORT0": "1",
        "INTERNAL_CPU_OFFLOAD_ENGINE": "1",
    }

    if dpu_mode == settings:
        print("dpu")
    elif nic_mode == settings:
        print("nic")
    else:
        print("unknown")
    if args.verbose:
        print(settings)


if __name__ == "__main__":
    main()