import dataclasses
import os
from argparse import Namespace, _SubParsersAction

from pydantic import parse_obj_as

from mdrsclient.api import FileApi, FolderApi
from mdrsclient.commands.base import BaseCommand
from mdrsclient.exceptions import IllegalArgumentException
from mdrsclient.models import File, FolderSimple


class MvCommand(BaseCommand):
    @classmethod
    def register(cls, parsers: _SubParsersAction) -> None:
        command = cls()
        mv_parser = parsers.add_parser("mv", help="move or rename the file or folder")
        mv_parser.add_argument("src_path", help="source remote path (remote:/lab/path/src)")
        mv_parser.add_argument("dest_path", help="destination remote path (remote:/lab/path/dest)")
        mv_parser.set_defaults(func=command.mv)

    def mv(self, args: Namespace) -> None:
        (s_remote, s_laboratory_name, s_path) = self._parse_remote_host_with_path(args.src_path)
        (d_remote, d_laboratory_name, d_path) = self._parse_remote_host_with_path(args.dest_path)
        if s_remote != d_remote:
            raise IllegalArgumentException("Remote host mismatched.")
        if s_laboratory_name != d_laboratory_name:
            raise IllegalArgumentException("Laboratory mismatched.")
        s_path = s_path.rstrip("/")
        s_dirname = os.path.dirname(s_path)
        s_basename = os.path.basename(s_path)
        if d_path.endswith("/"):
            d_dirname = d_path
            d_basename = s_basename
        else:
            d_dirname = os.path.dirname(d_path)
            d_basename = os.path.basename(d_path)
        connection = self._create_connection(s_remote)
        laboratory = self._find_laboratory(connection, s_laboratory_name)
        s_parent_folder = self._find_folder(connection, laboratory, s_dirname)
        d_parent_folder = self._find_folder(connection, laboratory, d_dirname)
        s_file = s_parent_folder.find_file(s_basename)
        if s_file is not None:
            # source is file
            d_file = d_parent_folder.find_file(d_basename)
            if d_file is not None:
                raise IllegalArgumentException(f"File `{d_basename}` already exists.")
            d_sub_folder = d_parent_folder.find_sub_folder(d_basename)
            if d_sub_folder is not None:
                raise IllegalArgumentException(f"Cannot overwrite non-folder `{d_basename}` with folder `{d_path}`.")
            file_api = FileApi(connection)
            if s_parent_folder.id != d_parent_folder.id:
                file_api.move(s_file, d_parent_folder.id)
            if d_basename != s_basename:
                d_file_dict = dataclasses.asdict(s_file) | {"name": d_basename}
                d_file = parse_obj_as(File, d_file_dict)
                file_api.update(d_file, None)
        else:
            s_folder = s_parent_folder.find_sub_folder(s_basename)
            if s_folder is None:
                raise IllegalArgumentException(f"File or folder `{s_basename}` not found.")
            # source is folder
            if d_parent_folder.find_file(d_basename) is not None:
                raise IllegalArgumentException(f"Cannot overwrite non-folder `{d_basename}` with folder `{s_path}`.")
            d_folder = d_parent_folder.find_sub_folder(d_basename)
            if d_folder is not None:
                if d_folder.id == s_folder.id:
                    raise IllegalArgumentException(f"`{s_path}` and `{s_path}` are the same folder.")
                raise IllegalArgumentException(f"Cannot move `{s_path}` to `{d_path}`: Folder not empty.")
            folder_api = FolderApi(connection)
            if s_parent_folder.id != d_parent_folder.id:
                folder_api.move(s_folder, d_parent_folder.id)
            if s_basename != d_basename:
                d_folder_dict = dataclasses.asdict(s_folder) | {"name": d_basename}
                d_folder = parse_obj_as(FolderSimple, d_folder_dict)
                folder_api.update(d_folder)