钩子

.trae/hooks.json

{
  "version": 1,
  "hooks": {
    "PreToolUse": [
      {
        "matcher": "RunCommand|Read|Write|Edit",
        "hooks": [
          {
            "type": "command",
            "command": "python3 .trae/hooks/block-env.py",
            "timeout": 30
          }
        ]
      }
    ]
  }
}

.trae/hooks/block-env.py

它只有一件事要做：一旦拿不准，就让它输出 deny，绝不以异常退出。

#!/usr/bin/env python3
"""读工具调用参数，命中 .env 就拒绝。

TraeCode 没有 failClosed：脚本崩溃或超时，这条钩子就等于不存在。
所以任何一处不确定，都要走输出 deny 这条路，不要让它以异常退出。
"""
import json
import os
import sys

DENY = {
    "hookSpecificOutput": {
        "hookEventName": "PreToolUse",
        "permissionDecision": "deny",
        "permissionDecisionReason": "密钥只放在 .env，不能读，也不能提交。",
    }
}

def mentions_env(value):
    if isinstance(value, dict):
        return any(mentions_env(v) for v in value.values())
    if isinstance(value, list):
        return any(mentions_env(v) for v in value)
    if isinstance(value, str):
        for part in value.replace("\\", "/").split():
            if os.path.basename(part.strip("\"'")) == ".env":
                return True
    return False

def main():
    try:
        data = json.load(sys.stdin)
    except Exception:
        json.dump(DENY, sys.stdout, ensure_ascii=False)
        return 0

    if mentions_env(data):
        json.dump(DENY, sys.stdout, ensure_ascii=False)
    return 0

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