__main__.py

import argparse
from pathlib import Path
from typing import Optional, Union
from io import BytesIO

from rsi import (
    Rsi,
    HyphenSplitter,
    NumberSplitter,
    SimpleSplitter,
    UnderscoreSplitter,
)


def main() -> int:
    """
    Entry point for the command line rsi utility script.
    """
    parser = argparse.ArgumentParser()
    subparser = parser.add_subparsers(dest="command")

    _from_dmi = subparser.add_parser("from_dmi", help="Will create an RSI from a BYOND DMI file.")
    _from_dmi.add_argument("input", help="The DMI file to read from.")
    _from_dmi.add_argument("output", help="The RSI to output to.", type=Path)
    _from_dmi.add_argument("-c", "--copyright", help="Specifies the copyright of the new RSI file.")
    _from_dmi.add_argument("-i", "--indents", help="Indents to use for the meta.json file.", type=int)
    _from_dmi.add_argument("-l", "--license", help="Specifies the license of the new RSI file.")
    _from_dmi.add_argument("-s", "--splitter", help="Splitter to use for the rsi if applicable.")

    _new_rsi = subparser.add_parser("new", help="Will create a new RSI at the provided directory.")
    _new_rsi.add_argument("rsi", help="The location of the new RSI. Must not exist yet.", type=Path)
    _new_rsi.add_argument("dimensions", help="The dimensions of the new rsi, in <width>x<height> format, like so: 32x32.")
    _new_rsi.add_argument("-c", "--copyright", action="store", help="Copyright info for this RSI file such as author.", nargs="?")
    _new_rsi.add_argument("-l", "--license", action="store", help="The license of this RSI file, as valid SPDX License Identifier (Google it).", nargs="?")
    _new_rsi.add_argument("--dont-make-parent-dirs", action="store_true", help="Do not create parent directories if they do not exist, instead throw an error.", dest="no_parents")

    args = parser.parse_args()

    if args.command == "from_dmi":
        from_dmi(args.input, args.output, args.license, args.copyright, args.indents, args.splitter)
        return 0

    if args.command == "new":
        return new_rsi(args.rsi, args.dimensions, args.copyright, args.license, not args.no_parents)

    print("No command specified!")
    return 1


def from_dmi(inputf: Union[BytesIO, Path, str],
             output: Path,
             new_license: Optional[str],
             new_copyright: Optional[str],
             indents: Optional[int] = None,
             splitter: Optional[str] = "",) -> None:
    rsi = Rsi.from_dmi(inputf)
    if new_license:
        rsi.license = new_license
    if new_copyright:
        rsi.copyright = new_copyright

    splitter_class = {
        "hyphen": HyphenSplitter,
        "number": NumberSplitter,
        "simple": SimpleSplitter,
        "underscore": UnderscoreSplitter,
    }.get(splitter, None)  # type: ignore

    if splitter_class is None:
        rsi.write(output)
    else:
        splitter_class(rsi).split_to(output, indents)


def new_rsi(loc: Path,
            dimensions: str,
            rsi_copyright: Optional[str],
            rsi_license: Optional[str],
            make_parents: bool) -> int:
    try:
        dimsplit = dimensions.split("x")
        if len(dimsplit) != 2:
            print("Incorrect amount of dimensions passed, expected exactly 2.")
            return 1
        x = int(dimsplit[0])
        y = int(dimsplit[1])

    except ValueError:
        print("Invalid dimensions passed.")
        return 1

    if not loc.parent.exists() and not make_parents:
        print("Parent directories do not exist. Aborting.")
        return 1

    rsi = Rsi((x, y))
    rsi.license = rsi_license
    rsi.copyright = rsi_copyright
    rsi.write(loc, make_parents)

    return 0