"""Prints code page tables."""


import sys

from escpos import printer
from escpos.constants import (
    CODEPAGE_CHANGE,
    CTL_CR,
    CTL_FF,
    CTL_HT,
    CTL_LF,
    CTL_VT,
    ESC,
)


def main():
    """Init printer and print codepage tables."""
    dummy = printer.Dummy()

    dummy.hw("init")

    for codepage in sys.argv[1:] or ["USA"]:
        dummy.set(height=2, width=2)
        dummy._raw(codepage + "\n\n\n")
        print_codepage(dummy, codepage)
        dummy._raw("\n\n")

    dummy.cut()

    print(dummy.output)


def print_codepage(printer, codepage):
    """Print a codepage."""
    if codepage.isdigit():
        codepage = int(codepage)
        printer._raw(CODEPAGE_CHANGE + bytes((codepage,)))
        printer._raw("after")
    else:
        printer.charcode(codepage)

    sep = ""

    # Table header
    printer.set(font="b")
    printer._raw(f"  {sep.join(map(lambda s: hex(s)[2:], range(0, 16)))}\n")
    printer.set()

    # The table
    for x in range(0, 16):
        # First column
        printer.set(font="b")
        printer._raw(f"{hex(x)[2:]} ")
        printer.set()

        for y in range(0, 16):
            byte = bytes(
                (x * 16 + y),
            )

            if byte in (ESC, CTL_LF, CTL_FF, CTL_CR, CTL_HT, CTL_VT):
                byte = " "

            printer._raw(byte)
            printer._raw(sep)
        printer._raw("\n")


if __name__ == "__main__":
    main()