import shutil
from collections.abc import Callable
from pathlib import Path

from clan_lib.errors import ClanError


def get_sops_folder(flake_dir: Path) -> Path:
    return flake_dir / "sops"


def gen_sops_subfolder(subdir: str) -> Callable[[Path], Path]:
    def folder(flake_dir: Path) -> Path:
        return flake_dir / "sops" / subdir

    return folder


sops_secrets_folder = gen_sops_subfolder("secrets")
sops_users_folder = gen_sops_subfolder("users")
sops_machines_folder = gen_sops_subfolder("machines")
sops_groups_folder = gen_sops_subfolder("groups")


def list_objects(path: Path, is_valid: Callable[[str], bool]) -> list[str]:
    if not path.exists():
        return []
    return [f.name for f in path.iterdir() if is_valid(f.name)]


def remove_object(path: Path, name: str) -> list[Path]:
    paths_to_commit = []
    try:
        shutil.rmtree(path / name)
        paths_to_commit.append(path / name)
    except FileNotFoundError as e:
        msg = f"{name} not found in {path}"
        raise ClanError(msg) from e
    if not next(path.iterdir(), None):
        path.rmdir()
    return paths_to_commit