#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Langfuse 专题 · 步骤 0：确认 Langfuse / Dify 容器与健康检查。"""
from __future__ import annotations

import sys

import _bootstrap  # noqa: F401
import _config as C
from dify_api import Dify


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        print("=== 容器 ===")
        out, err = d.sh(
            "docker ps --format 'table {{.Names}}\t{{.Status}}' | grep -E 'langfuse|docker-api-1|docker-worker-1'"
        )
        print(out or err)

        print("\n=== Langfuse health ===")
        out, err = d.sh("curl -s %s" % C.LANGFUSE_HEALTH)
        print(out or err)

        for ctr in (C.DIFY_API_CTR, C.DIFY_WORKER):
            print("\n=== %s -> Langfuse ===" % ctr)
            out, err = d.sh(
                "docker exec %s curl -s -m 8 %s" % (ctr, C.LANGFUSE_HEALTH)
            )
            print(out or err)
        return 0
    finally:
        d.close()


if __name__ == "__main__":
    sys.exit(main())