137 lines
4.1 KiB
Python
137 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Claude Code 启动时显示系统信息和统计
|
|
"""
|
|
import json
|
|
import sys
|
|
import os
|
|
import subprocess
|
|
import platform
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
def get_file_count(directory, pattern="*"):
|
|
"""获取目录下匹配模式的文件数量"""
|
|
try:
|
|
return len(list(Path(directory).rglob(pattern)))
|
|
except:
|
|
return 0
|
|
|
|
def get_repo_stats():
|
|
"""获取 Git 仓库统计信息"""
|
|
stats = {}
|
|
try:
|
|
# 获取提交数量
|
|
result = subprocess.run(
|
|
["git", "rev-list", "--count", "HEAD"],
|
|
capture_output=True, text=True
|
|
)
|
|
if result.returncode == 0:
|
|
stats['commits'] = result.stdout.strip()
|
|
|
|
# 获取最新提交
|
|
result = subprocess.run(
|
|
["git", "log", "-1", "--format=%h %s"],
|
|
capture_output=True, text=True
|
|
)
|
|
if result.returncode == 0:
|
|
stats['latest'] = result.stdout.strip()
|
|
|
|
# 获取分支数量
|
|
result = subprocess.run(
|
|
["git", "branch", "-a"],
|
|
capture_output=True, text=True
|
|
)
|
|
if result.returncode == 0:
|
|
branches = [line for line in result.stdout.split('\n') if line.strip()]
|
|
stats['branches'] = len(branches)
|
|
|
|
except Exception as e:
|
|
stats['error'] = str(e)
|
|
|
|
return stats
|
|
|
|
def display_system_info():
|
|
"""显示系统信息"""
|
|
# ANSI 颜色
|
|
CYAN = '\033[96m'
|
|
GREEN = '\033[92m'
|
|
YELLOW = '\033[93m'
|
|
BLUE = '\033[94m'
|
|
BOLD = '\033[1m'
|
|
END = '\033[0m'
|
|
|
|
print(f"\n{CYAN}{'━' * 70}{END}", file=sys.stderr)
|
|
print(f"{BOLD}{BLUE}📊 系统信息 & 项目统计{END}", file=sys.stderr)
|
|
print(f"{CYAN}{'━' * 70}{END}", file=sys.stderr)
|
|
|
|
# 系统信息
|
|
print(f"{GREEN}🖥️ 系统:{END} {platform.system()} {platform.release()}", file=sys.stderr)
|
|
print(f"{GREEN}🐍 Python:{END} {platform.python_version()}", file=sys.stderr)
|
|
|
|
# 当前项目信息
|
|
cwd = os.getcwd()
|
|
print(f"{GREEN}📂 项目目录:{END} {cwd}", file=sys.stderr)
|
|
|
|
# 文件统计
|
|
try:
|
|
py_files = get_file_count(cwd, "*.py")
|
|
js_files = get_file_count(cwd, "*.js") + get_file_count(cwd, "*.ts") + get_file_count(cwd, "*.jsx") + get_file_count(cwd, "*.tsx")
|
|
md_files = get_file_count(cwd, "*.md")
|
|
|
|
if py_files > 0:
|
|
print(f"{GREEN}📄 Python 文件:{END} {py_files}", file=sys.stderr)
|
|
if js_files > 0:
|
|
print(f"{GREEN}📄 JS/TS 文件:{END} {js_files}", file=sys.stderr)
|
|
if md_files > 0:
|
|
print(f"{GREEN}📄 Markdown 文件:{END} {md_files}", file=sys.stderr)
|
|
except:
|
|
pass
|
|
|
|
# Git 信息
|
|
try:
|
|
stats = get_repo_stats()
|
|
if stats:
|
|
print(f"{YELLOW}{'─' * 70}{END}", file=sys.stderr)
|
|
if 'commits' in stats:
|
|
print(f"{GREEN}📝 提交数:{END} {stats['commits']}", file=sys.stderr)
|
|
if 'branches' in stats:
|
|
print(f"{GREEN}🌿 分支数:{END} {stats['branches']}", file=sys.stderr)
|
|
if 'latest' in stats:
|
|
print(f"{GREEN}✨ 最新提交:{END} {stats['latest']}", file=sys.stderr)
|
|
except:
|
|
pass
|
|
|
|
print(f"{CYAN}{'━' * 70}{END}\n", file=sys.stderr)
|
|
|
|
def main():
|
|
# 加载输入
|
|
try:
|
|
input_data = json.load(sys.stdin)
|
|
except json.JSONDecodeError as e:
|
|
print(f"Error: Invalid JSON input: {e}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
# 显示系统信息
|
|
display_system_info()
|
|
|
|
# 返回输出
|
|
home_claude_dir = os.path.expanduser("~/.claude")
|
|
context = f"""回答我的问题的时候开头必须是'好的！！！！大先生！！！'
|
|
|
|
重要配置信息：
|
|
- 全局 Claude 配置目录路径：{home_claude_dir}
|
|
- 当你需要修改或查找全局 Claude 配置时，请始终使用此目录：{home_claude_dir}"""
|
|
|
|
output = {
|
|
"hookSpecificOutput": {
|
|
"hookEventName": "SessionStart",
|
|
"message":"你好，大先生",
|
|
"additionalContext": context
|
|
}
|
|
}
|
|
print(json.dumps(output))
|
|
|
|
if __name__ == "__main__":
|
|
main()
|