import os
from argparse import Namespace
from typing import Any

from mdrsclient.api import FilesApi, FoldersApi
from mdrsclient.commands.base import BaseCommand
from mdrsclient.exceptions import IllegalArgumentException
from mdrsclient.models.file import find_file


class RmCommand(BaseCommand):
    @classmethod
    def register(cls, parsers: Any) -> None:
        rm_parser = parsers.add_parser("rm", help="remove the file or folder")
        rm_parser.add_argument(
            "-r", "--recursive", help="remove folders and their contents recursive", action="store_true"
        )
        rm_parser.add_argument("remote_path", help="remote file path (remote:/lab/path/file)")
        rm_parser.set_defaults(func=cls.func)

    @classmethod
    def func(cls, args: Namespace) -> None:
        remote_path = str(args.remote_path)
        is_recursive = bool(args.recursive)
        cls.rm(remote_path, is_recursive)

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

        client = MdrsClient.from_remote(remote)
        client.rm(remote_path, is_recursive)