noxfile.py

"""
nox configuration for cookiecutter project template.
"""

from __future__ import annotations

import contextlib
import functools
import os
import subprocess
import tempfile
from pathlib import Path

import nox

CI = os.environ.get("CI") is not None

ROOT = Path(".")
PYTHON_VERSIONS = ["3.11"]
PYTHON_DEFAULT_VERSION = PYTHON_VERSIONS[-1]

nox.options.default_venv_backend = "venv"
nox.options.reuse_existing_virtualenvs = True


# In CI, use Python interpreter provided by GitHub Actions
if CI:
    nox.options.force_venv_backend = "none"


MD_PATHS = ["*.md"]


@functools.lru_cache
def _list_files() -> list[Path]:
    file_list = []
    for cmd in (
        ["git", "ls-files"],
        ["git", "ls-files", "--others", "--exclude-standard"],
    ):
        cmd_result = subprocess.run(cmd, check=True, text=True, capture_output=True)
        file_list.extend(cmd_result.stdout.splitlines())
    file_paths = [Path(p) for p in file_list]
    return file_paths


def list_files(suffix: str | None = None) -> list[Path]:
    """List all non-files not-ignored by git."""
    file_paths = _list_files()
    if suffix is not None:
        file_paths = [p for p in file_paths if p.suffix == suffix]
    return file_paths


def run_readable(session, mode="check"):
    session.run(
        "docker",
        "run",
        "--platform",
        "linux/amd64",
        "--rm",
        "-v",
        f"{ROOT.absolute()}:/data",
        "-w",
        "/data",
        "ghcr.io/bobheadxi/readable:v0.5.0@sha256:423c133e7e9ca0ac20b0ab298bd5dbfa3df09b515b34cbfbbe8944310cc8d9c9",
        mode,
        *MD_PATHS,
        external=True,
    )


def run_shellcheck(session, mode="check"):
    shellcheck_cmd = [
        "docker",
        "run",
        "--platform",
        "linux/amd64",  # while this image is multi-arch, we cannot use digest with multi-arch images
        "--rm",
        "-v",
        f"{ROOT.absolute()}:/mnt",
        "-w",
        "/mnt",
        "-q",
        "koalaman/shellcheck:0.9.0@sha256:a527e2077f11f28c1c1ad1dc784b5bc966baeb3e34ef304a0ffa72699b01ad9c",
    ]

    files = list_files(suffix=".sh")
    if not files:
        session.log("No shell files found")
        return
    shellcheck_cmd.extend(files)

    if mode == "fmt":
        with tempfile.NamedTemporaryFile(mode="w+") as diff_file:
            session.run(
                *shellcheck_cmd,
                "--format=diff",
                external=True,
                stdout=diff_file,
                success_codes=[0, 1],
            )
            diff_file.seek(0)
            diff = diff_file.read()
            if len(diff.splitlines()) > 1:  # ignore single-line message
                session.log("Applying shellcheck patch:\n%s", diff)
                subprocess.run(
                    ["patch", "-p1"],
                    input=diff,
                    text=True,
                    check=True,
                )

    session.run(*shellcheck_cmd, external=True)


@nox.session(name="format", python=PYTHON_DEFAULT_VERSION)
def format_(session):
    """Lint the code and apply fixes in-place whenever possible."""
    session.run("pip", "install", "-e", ".[format]")
    session.run("ruff", "check", "--fix", ".")
    run_shellcheck(session, mode="fmt")
    run_readable(session, mode="fmt")
    session.run("ruff", "format", ".")


@nox.session(python=PYTHON_DEFAULT_VERSION)
def lint(session):
    """Run linters in readonly mode."""
    session.run("pip", "install", "-e", ".[lint]")
    session.run("ruff", "check", "--diff", ".")
    session.run("codespell", ".")
    run_shellcheck(session, mode="check")
    run_readable(session, mode="check")
    session.run("ruff", "format", "--diff", ".")


@contextlib.contextmanager
def crufted_project(session):
    session.run("pip", "install", "-e", ".")
    tmpdir = crufted_project.tmpdir
    if not tmpdir:
        session.notify("cleanup_crufted_project")
        crufted_project.tmpdir = tmpdir = tempfile.TemporaryDirectory(prefix="rt-crufted_")
        session.log("Creating project in %s", tmpdir.name)
        session.run("cruft", "create", ".", "--output-dir", tmpdir.name, "--no-input")
        project_path = Path(tmpdir.name) / "project"
        with session.chdir(project_path):
            session.run("git", "init", external=True)
            session.run("./setup-dev.sh", external=True)
    else:
        project_path = Path(tmpdir.name) / "project"
    with session.chdir(project_path):
        yield project_path


crufted_project.tmpdir = None


def rm_root_owned(session, dirpath):
    assert not ROOT.is_relative_to(dirpath)  # sanity check before we nuke dirpath
    session.run("docker", "run", "--rm", "-v", f"{dirpath}:/tmpdir/", "alpine:3.18.0", "rm", "-rf", "/tmpdir/project")


@contextlib.contextmanager
def docker_up(session):
    session.run("docker", "compose", "up", "-d")
    yield
    session.run("docker", "compose", "down", "-v", "--remove-orphans")


@nox.session(python=PYTHON_DEFAULT_VERSION, tags=["crufted_project"])
def lint_crufted_project(session):
    with crufted_project(session):
        session.run("nox", "-s", "lint")  # TODO: RT-49 re-enable 'type_check'


@nox.session(python=PYTHON_DEFAULT_VERSION, tags=["crufted_project"])
def test_crufted_project(session):
    with crufted_project(session):
        with docker_up(session):
            session.run("nox", "-s", "test")


@nox.session(python=PYTHON_DEFAULT_VERSION)
def cleanup_crufted_project(session):
    if crufted_project.tmpdir:
        # workaround for docker compose creating root-owned files
        rm_root_owned(session, crufted_project.tmpdir.name)
        crufted_project.tmpdir.cleanup()
        crufted_project.tmpdir = None