#!/usr/bin/env python3
"""ssh-exec - 通过SSH远程执行单条命令的工具（支持密码认证）"""

import argparse
import sys
import paramiko


def main():
    parser = argparse.ArgumentParser(
        description="ssh-exec - 通过SSH远程执行单条命令（支持密码认证）",
        epilog="""
参数说明:
  -H, --host       SSH主机地址（默认: 127.0.0.1）
  -P, --port       SSH端口号（默认: 22）
  -u, --user        SSH用户名（必填）
  -p, --password   SSH密码（必填）
  -c, --command    要执行的远程命令（必填）

使用示例:
  %(prog)s -u qing -p 12345678 -c "ls -la"
  %(prog)s -u root -H 192.168.1.100 -P 22 -p mypass -c "df -h"
  %(prog)s -u qing -H 127.0.0.1 -p 12345678 -c "uname -a"
""",
        formatter_class=argparse.RawDescriptionHelpFormatter,
    )
    parser.add_argument("-H", "--host", default="127.0.0.1", help="SSH主机地址（默认: 127.0.0.1）")
    parser.add_argument("-P", "--port", type=int, default=22, help="SSH端口号（默认: 22）")
    parser.add_argument("-u", "--user", required=True, help="SSH用户名（必填）")
    parser.add_argument("-p", "--password", required=True, help="SSH密码（必填）")
    parser.add_argument("-c", "--command", required=True, help="要执行的远程命令（必填）")

    args = parser.parse_args()

    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    try:
        client.connect(
            hostname=args.host,
            port=args.port,
            username=args.user,
            password=args.password,
            timeout=10,
        )

        stdin, stdout, stderr = client.exec_command(args.command, timeout=30)

        exit_code = stdout.channel.recv_exit_status()
        out_text = stdout.read().decode("utf-8", errors="replace")
        err_text = stderr.read().decode("utf-8", errors="replace")

        if out_text:
            print(out_text, end="")
        if err_text:
            print(err_text, end="", file=sys.stderr)

        sys.exit(exit_code)

    except paramiko.AuthenticationException:
        print(f"错误: SSH认证失败（{args.user}@{args.host}:{args.port}）", file=sys.stderr)
        sys.exit(1)
    except paramiko.SSHException as e:
        print(f"错误: SSH连接失败 - {e}", file=sys.stderr)
        sys.exit(1)
    except TimeoutError:
        print(f"错误: 连接超时（{args.host}:{args.port}）", file=sys.stderr)
        sys.exit(1)
    except Exception as e:
        print(f"错误: {e}", file=sys.stderr)
        sys.exit(1)
    finally:
        client.close()


if __name__ == "__main__":
    main()