wds
37a5d8d804
2025-10-21 - add the shell history feature - add the logging feature - refactor the codebase for better maintainability
108 lines
3.5 KiB
Python
108 lines
3.5 KiB
Python
import os
|
||
from typing import List
|
||
from autoterminal.utils.logger import logger
|
||
|
||
|
||
def clean_command(command: str) -> str:
|
||
"""清理命令字符串"""
|
||
# 移除可能的引号和多余空格
|
||
command = command.strip()
|
||
if command.startswith('"') and command.endswith('"'):
|
||
command = command[1:-1]
|
||
if command.startswith("'") and command.endswith("'"):
|
||
command = command[1:-1]
|
||
return command.strip()
|
||
|
||
|
||
def get_shell_history(count: int = 20) -> List[str]:
|
||
"""
|
||
获取系统 Shell 历史命令
|
||
|
||
Args:
|
||
count: 获取最近的命令数量
|
||
|
||
Returns:
|
||
最近执行的 Shell 命令列表
|
||
"""
|
||
history_commands = []
|
||
|
||
try:
|
||
# 尝试从环境变量获取历史文件路径
|
||
histfile = os.getenv('HISTFILE')
|
||
|
||
# 如果没有 HISTFILE，根据 SHELL 推断
|
||
if not histfile or not os.path.exists(histfile):
|
||
home_dir = os.path.expanduser("~")
|
||
shell = os.getenv('SHELL', '')
|
||
|
||
# 根据当前 Shell 类型优先尝试对应的历史文件
|
||
possible_files = []
|
||
if 'zsh' in shell:
|
||
possible_files = [
|
||
os.path.join(home_dir, ".zsh_history"),
|
||
os.path.join(home_dir, ".zhistory"),
|
||
os.path.join(home_dir, ".bash_history"),
|
||
]
|
||
else: # bash 或其他
|
||
possible_files = [
|
||
os.path.join(home_dir, ".bash_history"),
|
||
os.path.join(home_dir, ".zsh_history"),
|
||
os.path.join(home_dir, ".zhistory"),
|
||
]
|
||
|
||
for file_path in possible_files:
|
||
if os.path.exists(file_path):
|
||
histfile = file_path
|
||
break
|
||
|
||
if histfile and os.path.exists(histfile):
|
||
logger.debug(f"读取 Shell 历史文件: {histfile}")
|
||
|
||
with open(histfile, 'r', encoding='utf-8', errors='ignore') as f:
|
||
lines = f.readlines()
|
||
|
||
# 过滤和清理命令
|
||
for line in lines:
|
||
line = line.strip()
|
||
|
||
# 跳过空行
|
||
if not line:
|
||
continue
|
||
|
||
# 处理 zsh 扩展历史格式 (: timestamp:duration;command)
|
||
if line.startswith(':'):
|
||
parts = line.split(';', 1)
|
||
if len(parts) > 1:
|
||
line = parts[1].strip()
|
||
|
||
# 过滤敏感命令（包含密码、密钥等）
|
||
sensitive_keywords = [
|
||
'password',
|
||
'passwd',
|
||
'secret',
|
||
'key',
|
||
'token',
|
||
'api_key',
|
||
'api-key']
|
||
if any(keyword in line.lower() for keyword in sensitive_keywords):
|
||
continue
|
||
|
||
# 过滤重复命令（保持顺序，只保留最后一次出现）
|
||
if line in history_commands:
|
||
history_commands.remove(line)
|
||
|
||
history_commands.append(line)
|
||
|
||
# 返回最近的 N 条命令
|
||
result = history_commands[-count:] if len(
|
||
history_commands) > count else history_commands
|
||
logger.debug(f"成功获取 {len(result)} 条 Shell 历史命令")
|
||
return result
|
||
else:
|
||
logger.warning("未找到 Shell 历史文件")
|
||
return []
|
||
|
||
except Exception as e:
|
||
logger.warning(f"获取 Shell 历史失败: {e}")
|
||
return []
|