tgpt项目新增-y参数实现自动化Shell命令执行功能

背景介绍

tgpt是一个命令行工具，它能够根据用户提供的自然语言提示生成相应的Shell命令。在之前的版本中，当用户使用-s参数请求生成并执行Shell命令时，系统总会要求用户手动确认是否执行该命令。这种设计虽然安全，但在自动化场景下却显得不够高效。

功能改进

最新提交的代码为tgpt增加了一个-y参数，用于实现Shell命令的自动执行功能。这个改进使得工具在自动化脚本中的使用变得更加便捷。

技术实现细节

参数解析部分

在main.go文件中，开发者新增了一个布尔类型的全局变量yes，用于存储是否自动确认执行命令的状态。同时添加了两个新的命令行参数：

isYes := flag.Bool("y", false, "")
flag.BoolVar(isYes, "yes", false, "Deactivate confirm request.")

当用户指定-y或--yes参数时，程序会将yes变量设置为true，从而跳过后续的确认步骤。

Shell命令处理流程

在shellCommand函数中，程序会先检测当前操作系统环境，确定适用的Shell类型（如bash、cmd、PowerShell等）。然后构造一个专门的提示词，要求AI模型只返回纯文本的Shell命令，不包含任何Markdown格式或额外描述。

自动执行机制

在getCommand函数中，当检测到-y参数被设置时，程序会跳过用户确认环节，直接将生成的命令传递给相应的Shell解释器执行：

if yes {
    userInput ="Y"
    bold.Print("\n");
} else {
    // 原有的用户确认逻辑
}

跨平台兼容性

该功能充分考虑了不同操作系统的差异：

1. Windows系统：自动检测使用cmd还是PowerShell
2. Linux/MacOS：使用用户默认的Shell（通过SHELL环境变量获取）
3. 特殊处理：对于多行命令，保持原有的确认机制不变

安全考虑

虽然-y参数提供了便利，但开发者仍然保留了以下安全措施：

1. 多行命令仍需确认，防止意外执行复杂操作
2. 命令生成过程仍然受到AI模型的安全限制
3. 错误处理机制保持不变，确保执行失败时有明确反馈

使用场景示例

这个功能特别适合以下场景：

1. 自动化脚本：在CI/CD流程中自动执行系统维护命令
2. 批量操作：需要连续执行多个生成的命令时
3. 开发工具链：与其他开发工具集成时减少交互环节

总结

tgpt通过增加-y参数，在保持核心功能不变的前提下，显著提升了工具的自动化能力。这一改进既考虑了用户体验，又兼顾了安全性，体现了开发者对命令行工具设计原则的深刻理解。对于需要频繁使用Shell命令自动化的用户来说，这无疑是一个值得期待的功能升级。