import os
import shutil
from pathlib import Path
from typing import Callable

from ..dirs import get_clan_flake_toplevel
from ..errors import ClanError


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


def gen_sops_subfolder(subdir: str) -> Callable[[], Path]:
    def folder() -> Path:
        return get_clan_flake_toplevel() / "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]) -> None:
    if not path.exists():
        return
    for f in os.listdir(path):
        if is_valid(f):
            print(f)


def remove_object(path: Path, name: str) -> None:
    try:
        shutil.rmtree(path / name)
    except FileNotFoundError:
        raise ClanError(f"{name} not found in {path}")
    if not os.listdir(path):
        os.rmdir(path)