"""
    Generate typesafe unsigned integer wrapping functions
"""

def gen_uwrap_code(uint_type: str, uint_type_abbr: str, wrap_width: int, wrap_max_uint: str) -> str: 
    template = "static inline {uint_type} guf_wrap{wrap_width}_{uint_type_abbr}({uint_type} a) {{ return a & {wrap_max_uint}; }}\n"
    return template.format(uint_type = uint_type, uint_type_abbr = uint_type_abbr, wrap_width = wrap_width, wrap_max_uint = wrap_max_uint)

if __name__ == "__main__":
    uint_types = [
        {"uint_type": "uint_least8_t", "uint_type_abbr": "uint_least8_t",  "wrap_width":   8, "wrap_max_uint": "GUF_UINT8_MAX"},
        {"uint_type": "uint_fast8_t",  "uint_type_abbr": "uint_fast8_t",   "wrap_width":   8, "wrap_max_uint": "GUF_UINT8_MAX"},

        "\n",
        {"uint_type": "uint_least16_t", "uint_type_abbr": "uint_least16_t", "wrap_width": 16, "wrap_max_uint": "GUF_UINT16_MAX"},
        {"uint_type": "uint_fast16_t",  "uint_type_abbr": "uint_fast16_t",  "wrap_width": 16, "wrap_max_uint": "GUF_UINT16_MAX"},

        "\n",
        {"uint_type": "uint_least32_t", "uint_type_abbr": "uint_least32_t", "wrap_width": 32, "wrap_max_uint": "GUF_UINT32_MAX"},
        {"uint_type": "uint_fast32_t",  "uint_type_abbr": "uint_fast32_t",  "wrap_width": 32, "wrap_max_uint": "GUF_UINT32_MAX"},

        "\n",
        {"uint_type": "uint_least64_t", "uint_type_abbr": "uint_least64_t", "wrap_width": 64, "wrap_max_uint": "GUF_UINT64_MAX"},
        {"uint_type": "uint_fast64_t",  "uint_type_abbr": "uint_fast64_t",  "wrap_width": 64, "wrap_max_uint": "GUF_UINT64_MAX"},

        "\n",
        {"uint_type": "unsigned char",       "uint_type_abbr": "uchar",      "wrap_width":  8, "wrap_max_uint": "GUF_UINT8_MAX"},
        {"uint_type": "unsigned short",      "uint_type_abbr": "ushort",     "wrap_width": 16, "wrap_max_uint": "GUF_UINT16_MAX"},
        {"uint_type": "unsigned long",       "uint_type_abbr": "ulong",      "wrap_width": 32, "wrap_max_uint": "GUF_UINT32_MAX"},
        {"uint_type": "unsigned long long",  "uint_type_abbr": "ulong_long", "wrap_width": 64, "wrap_max_uint": "GUF_UINT64_MAX"},
    ]


    code = "// Typesafe unsigned integer wrapping functions (generated with tools/intwrap-gen.py)\n"
    for uint_t in uint_types: 
        if isinstance(uint_t, str):
            code += uint_t
        else:
            code += gen_uwrap_code(**uint_t)

    print(code)