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

from ..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]:
    objs: list[str] = []
    if not path.exists():
        return objs
    for f in os.listdir(path):
        if is_valid(f):
            objs.append(f)
    return objs


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:
        raise ClanError(f"{name} not found in {path}")
    if not os.listdir(path):
        os.rmdir(path)
    return paths_to_commit