"""SSH 上去看 llm_settings + enrichment_loop 状态(v2: 不用复杂 quote)"""
import os
import sys
import paramiko

HOST = "207.57.129.228"
PORT = 19717
USER = "root"
PASS = os.environ.get("REMOTE_PASS", "")

if not PASS:
    print("REMOTE_PASS not set", file=sys.stderr)
    sys.exit(1)

c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect(HOST, port=PORT, username=USER, password=PASS,
          timeout=30, allow_agent=False, look_for_keys=False)
print("✓ 连上\n")


def run(label, cmd, timeout=30):
    print(f"\n=== {label} ===")
    print(f"$ {cmd[:200]}{'...' if len(cmd) > 200 else ''}")
    try:
        si, so, se = c.exec_command(cmd, timeout=timeout)
        out = so.read().decode(errors="replace")
        err = se.read().decode(errors="replace")
        rc = so.channel.recv_exit_status()
    except Exception as e:
        print(f"[exception] {type(e).__name__}: {e}")
        return
    if out.strip(): print(out.rstrip())
    if err.strip(): print(f"[stderr] {err.rstrip()}")
    print(f"-> rc={rc}")


# 1) 写一个 shell 文件到远程,然后用 bash file.sh 调(避开 quote)
#    把 SQL 都拼到 /tmp/check_llm.sh
shell_script = r"""#!/bin/bash
set -a
. /srv/news/.env
set +a
cd /srv/news
echo "--- llm_settings ---"
docker compose exec -T postgres psql -U "$POSTGRES_USER" -d "$POSTGRES_DB" \
  -c "SELECT id, enabled, chat_model, image_model, interval_sec, updated_at FROM llm_settings;"
echo "--- worker asyncio tasks ---"
docker compose exec -T worker python <<'PYEOF'
import asyncio
async def main():
    for t in asyncio.all_tasks():
        if not t.done():
            print(f'name={t.get_name()!r} done={t.done()}')
asyncio.run(main())
PYEOF
echo "--- worker 日志(enrich 关键字) ---"
docker compose logs --tail=500 worker 2>&1 | grep -iE "enrich|llm_settings|enrichment_loop|Traceback" | head -40
"""

# 写入远程
si, so, se = c.exec_command("cat > /tmp/check_llm.sh <<'EOFCAT'\n" + shell_script + "\nEOFCAT", timeout=10)
so.read()
se.read()
so.channel.recv_exit_status()
print("✓ shell 脚本已写入 /tmp/check_llm.sh\n")

# 执行
run("llm_settings + asyncio + 日志", "bash /tmp/check_llm.sh", timeout=60)

c.close()
print("\n✓ 完成")