# ssh_manager/list_hosts.py

import os
import glob
import socket
import asyncio
from tabulate import tabulate
from collections import OrderedDict

from .utils import print_warning, print_error, Colors
"""
This file is responsible for listing all SSH hosts discovered in ~/.ssh/conf/<label>/config.
We've modified it to show a "Conf Directory" column instead of "IdentityFile."
"""

async def check_ssh_port(ip_address, port):
    """
    Attempt to open an SSH connection to see if the port is open.
    Returns True if successful, False otherwise.
    """
    try:
        reader, writer = await asyncio.wait_for(
            asyncio.open_connection(ip_address, port), timeout=1
        )
        writer.close()
        await writer.wait_closed()
        return True
    except:
        return False

def load_config_file(file_path):
    """
    Parse a single SSH config file and return a list of host blocks.
    Each block is an OrderedDict with keys like 'Host', 'HostName', etc.
    """
    blocks = []
    host_data = None

    try:
        with open(file_path, 'r') as f:
            lines = f.readlines()
    except Exception as e:
        print_error(f"Error reading SSH config file {file_path}: {e}")
        return blocks

    for line in lines:
        stripped_line = line.strip()
        if not stripped_line or stripped_line.startswith('#'):
            continue

        if stripped_line.lower().startswith('host '):
            host_labels = stripped_line.split()[1:]
            for label in host_labels:
                if '*' not in label:
                    if host_data:
                        blocks.append(host_data)
                    host_data = OrderedDict({'Host': label})
                    break
        elif host_data:
            if ' ' in stripped_line:
                key, value = stripped_line.split(None, 1)
                host_data[key] = value.strip()

    if host_data:
        blocks.append(host_data)
    return blocks

async def check_host(host):
    """
    Given a host block, resolve IP, check SSH port, etc.
    Returns a row for tabulate containing:
      1) Host label
      2) User
      3) Port (color-coded if open)
      4) HostName
      5) IP Address (color-coded if resolved)
      6) Conf Directory (green if has IdentityFile, else no color)
    """
    host_label    = host.get('Host', 'N/A')
    hostname      = host.get('HostName', 'N/A')
    user          = host.get('User', 'N/A')
    port          = int(host.get('Port', '22'))
    identity_file = host.get('IdentityFile', 'N/A')

    # Resolve IP
    try:
        ip_address = socket.gethostbyname(hostname)
        colored_ip = f"{Colors.GREEN}{ip_address}{Colors.RESET}"
    except socket.error:
        ip_address = None
        colored_ip = f"{Colors.RED}N/A{Colors.RESET}"

    # Check port
    if ip_address:
        port_open = await check_ssh_port(ip_address, port)
        colored_port = (
            f"{Colors.GREEN}{port}{Colors.RESET}"
            if port_open
            else f"{Colors.RED}{port}{Colors.RESET}"
        )
    else:
        colored_port = f"{Colors.RED}{port}{Colors.RESET}"

    # Conf Directory = ~/.ssh/conf/<host_label>
    conf_path = f"~/.ssh/conf/{host_label}"
    # If there's an IdentityFile, we color this path green
    if identity_file != 'N/A':
        conf_path_display = f"{Colors.GREEN}{conf_path}{Colors.RESET}"
    else:
        conf_path_display = conf_path

    return [
        host_label,
        user,
        colored_port,
        hostname,
        colored_ip,
        conf_path_display
    ]

async def list_hosts(conf_dir):
    """
    List out all hosts found in ~/.ssh/conf/*/config.
    Shows columns: No., Host, User, Port, HostName, IP Address, Conf Directory
    """
    pattern    = os.path.join(conf_dir, "*", "config")
    conf_files = sorted(glob.glob(pattern))

    all_host_blocks = []
    for conf_file in conf_files:
        blocks = load_config_file(conf_file)
        all_host_blocks.extend(blocks)

    # Prepare table
    headers = ["No.", "Host", "User", "Port", "HostName", "IP Address", "Conf Directory"]
    if not all_host_blocks:
        print_warning("No hosts found. The server list is empty.")
        print("\nSSH Conf Subdirectory Host List")
        print(tabulate([], headers=headers, tablefmt="grid"))
        return

    tasks   = [check_host(h) for h in all_host_blocks]
    results = await asyncio.gather(*tasks)

    table = [[idx + 1] + row for idx, row in enumerate(results)]
    print("\nSSH Conf Subdirectory Host List")
    print(tabulate(table, headers=headers, tablefmt="grid"))