"""
服务器信息收集与校验
"""
from typing import Tuple, Optional
from .interaction import UserInteraction

class ServerInfoCollector:
    def __init__(self, interaction: UserInteraction):
        self.ia = interaction

    def parse_user_host(self, user_host: str) -> Optional[Tuple[str, str]]:
        if '@' in user_host and len(user_host.split('@')) == 2:
            user, host = user_host.split('@', 1)
            if user and host:
                return user, host
        return None

    def validate_port(self, port: str) -> bool:
        return port.isdigit() and 1 <= int(port) <= 65535

    def get_user_host(self, prefill: dict) -> Tuple[Optional[str], Optional[str]]:
        default_uh = f"{prefill.get('username','')}@{prefill.get('host','')}" if prefill.get('username') and prefill.get('host') else ""
        while True:
            user_host_str = self.ia.smart_input("输入服务器地址 (格式: user@host)", default=default_uh)
            if not user_host_str: return None, None
            parsed = self.parse_user_host(user_host_str)
            if parsed:
                return parsed
            self.ia.colored_print("❌ 格式错误，请使用 'user@host' 格式。")

    def get_port(self, prefill: dict) -> Optional[str]:
        return self.ia.smart_input("输入SSH端口", default=str(prefill.get("port", "22")), validator=self.validate_port)

    def get_server_name(self, prefill: dict) -> Optional[str]:
        return self.ia.smart_input("为这个连接设置一个唯一的名称", default=prefill.get('name', ''))