petr-muller/pyff

"""Entry point for the `pyff` command"""

import sys
import logging
import pathlib
from argparse import ArgumentParser
from typing import Callable

from pyff.modules import pyff_module_path
from pyff.packages import pyff_package_path
from pyff.directories import pyff_directory
from pyff.repositories import pyff_git_revision
from pyff.kitchensink import highlight, HIGHLIGHTS

LOGGER = logging.getLogger(__name__)


def _pyff_that(function: Callable, what: str, parser: ArgumentParser = ArgumentParser()) -> None:
    parser.add_argument("old")
    parser.add_argument("new")

    parser.add_argument("--highlight-names", dest="highlight", choices=HIGHLIGHTS, default="color")
    parser.add_argument("--debug", action="store_true", default=False)

    args = parser.parse_args()

    if args.debug:
        logging.basicConfig(
            format="%(levelname)s:%(name)s:%(funcName)s: %(message)s", level=logging.DEBUG
        )

    LOGGER.debug(f"Python Diff: old {what} {args.old} | new {what} {args.new}")
    changes = function(pathlib.Path(args.old), pathlib.Path(args.new), args)

    if changes is None:
        print(
            f"Pyff did not detect any significant difference between "
            f"{what} '{args.old}' and '{args.new}'"
        )
        sys.exit(0)

    print(highlight(str(changes), args.highlight))


def pyffmod() -> None:
    """Entry point for the `pyff` command"""

    def compare(old, new, _):
        """Open two arguments as files and compare them"""
        return pyff_module_path(old, new)

    _pyff_that(compare, "module")


def pyffpkg() -> None:
    """Entry point for the `pyff-package` command"""

    def compare(old, new, _):
        """Compare two packages"""
        return pyff_package_path(old, new)

    _pyff_that(compare, "package")


def pyffdir() -> None:
    """Entry point for the `pyff-dir` command"""

    def compare(old, new, _):
        """Compare two directories"""
        return pyff_directory(old, new)

    _pyff_that(compare, "directory")


def pyffgit() -> None:
    """Entry point for the `pyff-git` command"""
    parser = ArgumentParser()
    parser.add_argument("repository")

    def compare(old, new, args):
        """Compare two revisions in a given Git repo"""
        return pyff_git_revision(args.repository, old, new)

    _pyff_that(compare, "revision", parser)


if __name__ == "__main__":
    pyffmod()