#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题七 · 步骤 1：检查 sandbox / ssrf_proxy / plugin_daemon 服务状态。"""
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()
        out, err = d.sh(
            "cd %s && docker compose ps | grep -E 'sandbox|ssrf_proxy|plugin_daemon'" % C.DIFY_DOCKER_DIR
        )
        print(out or err)
        out2, _ = d.sh(
            "docker logs %s --tail 80 2>&1 | grep -iE 'shell|DifyShellLayer|provider'" % C.PLUGIN_CONTAINER
        )
        print("\n=== DifyShellLayer 相关日志 ===")
        print(out2.strip() or "(无匹配)")
        return 0
    finally:
        d.close()


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