#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""打印 dify_trace_langfuse/langfuse_trace.py 中 message_trace() 附近实现。"""

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:
        d.login()
        path = "/app/api/.venv/lib/python3.12/site-packages/dify_trace_langfuse/langfuse_trace.py"
        out, err = d.sh(
            "docker exec docker-api-1 sh -lc %r" % ("grep -n 'def message_trace' %s | head -5" % path),
            t=240,
        )
        print(out or err)

        # 粗略打印函数附近（message_trace 通常在前 300~500 行）
        cmd = "docker exec docker-api-1 sh -lc %r" % ("sed -n '160,380p' %s" % path)
        out2, err2 = d.sh(cmd, t=240)
        print(out2 or err2)
        return 0
    finally:
        d.close()


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