#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""打印 langfuse_trace.py 中 tool_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"
        )

        def sed(a: int, b: int) -> str:
            cmd = "docker exec docker-api-1 sh -lc %r" % ("sed -n '%d,%dp' %s" % (a, b, path))
            out, err = d.sh(cmd)
            return out or err or ""

        print("===== lines 90-190 =====")
        print(sed(90, 190))
        print("\n===== lines 380-520 =====")
        print(sed(380, 520))
        return 0
    finally:
        d.close()


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