#!/usr/bin/env python3
"""
为代码块添加引用语
根据清华大学出版社规范:
- 代码用"代码如下:"引出
- 命令用"命令如下:"引出
"""

import os
import re

def is_command_block(code_content):
 """判断是否为命令块"""
 # 常见的命令特征
 command_patterns = [
 r'^(npm|pnpm|yarn|bun|node|python|pip|git|docker|curl|wget|ssh|scp)',
 r'^(cd|ls|mkdir|rm|cp|mv|cat|grep|find|chmod|chown)',
 r'^(openclaw|clawhub)',
 r'^\$', # shell提示符
 r'^#', # root提示符
 ]
 
 lines = code_content.strip().split('\n')
 if not lines:
 return False
 
 first_line = lines[0].strip()
 for pattern in command_patterns:
 if re.match(pattern, first_line, re.IGNORECASE):
 return True
 
 return False

def add_code_references(filepath):
 """为文件中的代码块添加引用语"""
 with open(filepath, 'r', encoding='utf-8') as f:
 content = f.read()
 
 # 查找所有代码块
 pattern = r'(^|\n)(```[\w]*\n)(.*?)(^```$)'
 
 def replace_code_block(match):
 prefix = match.group(1)
 code_start = match.group(2)
 code_content = match.group(3)
 code_end = match.group(4)
 
 # 检查前面是否已经有引用语
 before_code = content[:match.start()]
 last_lines = before_code.split('\n')[-3:] # 检查前3行
 
 has_reference = False
 for line in last_lines:
 if '代码如下' in line or '命令如下' in line:
 has_reference = True
 break
 
 if has_reference:
 # 已经有引用语,不添加
 return match.group(0)
 
 # 判断是代码还是命令
 if is_command_block(code_content):
 reference = "命令如下:\n\n"
 else:
 reference = "代码如下:\n\n"
 
 return prefix + reference + code_start + code_content + code_end
 
 new_content = re.sub(pattern, replace_code_block, content, flags=re.MULTILINE | re.DOTALL)
 
 # 统计添加的数量
 added_count = new_content.count('代码如下:') + new_content.count('命令如下:') - \
 content.count('代码如下:') - content.count('命令如下:')
 
 if added_count > 0:
 with open(filepath, 'w', encoding='utf-8') as f:
 f.write(new_content)
 return added_count
 
 return 0

def main():
 """主函数"""
 print("开始为代码块添加引用语...")
 print()
 
 total_added = 0
 processed_files = 0
 
 for root, dirs, files in os.walk("docs"):
 for file in sorted(files):
 if file.endswith(".md") and not file.endswith(".emoji-backup"):
 filepath = os.path.join(root, file)
 added = add_code_references(filepath)
 if added > 0:
 print(f"✓ {filepath}: 添加了 {added} 个引用语")
 total_added += added
 processed_files += 1
 
 print()
 print(f"完成! 共处理 {processed_files} 个文件,添加了 {total_added} 个引用语")

if __name__ == "__main__":
 main()