from argparse import Namespace
from typing import Any

from mdrsclient.api import FoldersApi
from mdrsclient.commands.base import BaseCommand
from mdrsclient.exceptions import IllegalArgumentException
from mdrsclient.models import FolderAccessLevel


class ChaclCommand(BaseCommand):
    @classmethod
    def register(cls, parsers: Any) -> None:
        chacl_parser = parsers.add_parser("chacl", help="change the folder access level")
        chacl_parser.add_argument("access_level", help="access level (private, cbs_open, pw_open)")
        chacl_parser.add_argument("-r", "--recursive", help="change access levels recursively", action="store_true")
        chacl_parser.add_argument("-p", "--password", help="password to set when access level is `pw_open`")
        chacl_parser.add_argument("remote_path", help="remote folder path (remote:/lab/path/)")
        chacl_parser.set_defaults(func=cls.func)

    @classmethod
    def func(cls, args: Namespace) -> None:
        remote_path = str(args.remote_path)
        access_level = FolderAccessLevel.key2id(str(args.access_level))
        if access_level is None:
            raise IllegalArgumentException(
                "Invalid `access_level` parameter. must be `private`, `cbs_open` or `pw_open`."
            )
        password = str(args.password) if args.password else None
        is_recursive = bool(args.recursive)
        cls.chacl(remote_path, access_level, is_recursive, password)

    @classmethod
    def chacl(cls, remote_path: str, access_level: int, is_recursive: bool, password: str | None) -> None:
        remote = remote_path.split(":", 1)[0] if ":" in remote_path else ""
        from mdrsclient.client import MdrsClient

        client = MdrsClient.from_remote(remote)
        client.chacl(remote_path, access_level, is_recursive, password)