159 lines
5.6 KiB
Python
159 lines
5.6 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import sys
|
||
import os
|
||
import argparse
|
||
import glob
|
||
|
||
from autoterminal.config.loader import ConfigLoader
|
||
from autoterminal.config.manager import ConfigManager
|
||
from autoterminal.llm.client import LLMClient
|
||
from autoterminal.utils.helpers import clean_command
|
||
from autoterminal.history import HistoryManager
|
||
|
||
def main():
|
||
"""主程序入口"""
|
||
# 解析命令行参数
|
||
parser = argparse.ArgumentParser(description='AutoTerminal - 智能终端工具')
|
||
parser.add_argument('user_input', nargs='*', help='用户输入的自然语言命令')
|
||
parser.add_argument('--api-key', help='API密钥')
|
||
parser.add_argument('--base-url', help='Base URL')
|
||
parser.add_argument('--model', help='模型名称')
|
||
parser.add_argument('--history-count', type=int, help='历史命令数量')
|
||
|
||
args = parser.parse_args()
|
||
|
||
# 合并用户输入
|
||
user_input = ' '.join(args.user_input).strip()
|
||
|
||
# 加载配置
|
||
config_loader = ConfigLoader()
|
||
config = config_loader.get_config()
|
||
|
||
# 命令行参数优先级最高
|
||
if args.api_key:
|
||
config['api_key'] = args.api_key
|
||
if args.base_url:
|
||
config['base_url'] = args.base_url
|
||
if args.model:
|
||
config['model'] = args.model
|
||
|
||
# 获取历史命令数量配置
|
||
history_count = args.history_count or config.get('max_history', 10)
|
||
|
||
# 如果配置不完整，使用配置管理器初始化
|
||
config_manager = ConfigManager()
|
||
if not all([config.get('api_key'), config.get('base_url'), config.get('model')]):
|
||
config = config_manager.get_or_create_config()
|
||
if not config:
|
||
print("错误: 缺少必要的配置参数，请通过命令行参数或配置文件提供API密钥、Base URL和模型名称。")
|
||
return 1
|
||
|
||
# 如果有命令行参数输入，直接处理
|
||
if user_input:
|
||
# 初始化历史管理器
|
||
history_manager = HistoryManager(max_history=history_count)
|
||
|
||
# 获取历史命令
|
||
history = history_manager.get_recent_history(history_count)
|
||
|
||
# 获取当前目录内容
|
||
try:
|
||
current_dir_content = glob.glob("*")
|
||
except Exception as e:
|
||
print(f"警告: 无法获取当前目录内容: {e}")
|
||
current_dir_content = []
|
||
|
||
# 初始化LLM客户端
|
||
try:
|
||
llm_client = LLMClient(config)
|
||
except Exception as e:
|
||
print(f"LLM客户端初始化失败: {e}")
|
||
return 1
|
||
|
||
# 调用LLM生成命令
|
||
try:
|
||
generated_command = llm_client.generate_command(
|
||
user_input=user_input,
|
||
history=history,
|
||
current_dir_content=current_dir_content
|
||
)
|
||
cleaned_command = clean_command(generated_command)
|
||
|
||
# 优化输出格式
|
||
print(f"\033[1;32m$\033[0m {cleaned_command}")
|
||
print("\033[1;37mPress Enter to execute...\033[0m")
|
||
|
||
# 等待用户回车确认执行
|
||
try:
|
||
input()
|
||
|
||
# 在用户的环境中执行命令
|
||
os.system(cleaned_command)
|
||
|
||
# 记录到历史
|
||
history_manager.add_command(user_input, cleaned_command)
|
||
except EOFError:
|
||
print("\n输入已取消。")
|
||
return 0
|
||
except Exception as exec_e:
|
||
print(f"命令执行失败: {exec_e}")
|
||
return 1
|
||
|
||
except Exception as e:
|
||
print(f"命令生成失败: {e}")
|
||
return 1
|
||
|
||
return 0
|
||
else:
|
||
# 处理空输入情况 - 生成基于上下文的推荐命令
|
||
history_manager = HistoryManager(max_history=history_count)
|
||
history = history_manager.get_recent_history(history_count)
|
||
|
||
try:
|
||
current_dir_content = glob.glob("*")
|
||
except Exception as e:
|
||
print(f"警告: 无法获取当前目录内容: {e}")
|
||
current_dir_content = []
|
||
|
||
try:
|
||
llm_client = LLMClient(config)
|
||
except Exception as e:
|
||
print(f"LLM客户端初始化失败: {e}")
|
||
return 1
|
||
|
||
# 获取最后执行的命令以避免重复推荐
|
||
last_executed_command = history_manager.get_last_executed_command()
|
||
|
||
try:
|
||
recommendation = llm_client.generate_command(
|
||
user_input="",
|
||
history=history,
|
||
current_dir_content=current_dir_content,
|
||
last_executed_command=last_executed_command
|
||
)
|
||
cleaned_recommendation = clean_command(recommendation)
|
||
|
||
if cleaned_recommendation.strip():
|
||
print(f"\033[1;34m💡 建议命令:\033[0m {cleaned_recommendation}")
|
||
print("\033[1;37mPress Enter to execute, or Ctrl+C to cancel...\033[0m")
|
||
try:
|
||
input()
|
||
os.system(cleaned_recommendation)
|
||
history_manager.add_command("自动推荐", cleaned_recommendation)
|
||
except EOFError:
|
||
print("\n输入已取消。")
|
||
return 0
|
||
except Exception as exec_e:
|
||
print(f"命令执行失败: {exec_e}")
|
||
return 1
|
||
else:
|
||
print("没有找到相关的命令建议。")
|
||
except Exception as e:
|
||
print(f"命令推荐生成失败: {e}")
|
||
return 1
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|