Go命令行工具开发：提升灵活性与文件处理技巧

### Go 命令行工具开发：提升灵活性与文件处理技巧 在开发命令行工具时，提升工具的灵活性和处理文件的能力是非常重要的。本文将介绍如何通过环境变量增加工具的灵活性，以及如何开发一个基本的 Markdown 预览工具。 #### 1. 通过环境变量增加工具灵活性 在开发待办事项工具时，用户目前还不能选择将待办事项列表保存到哪个文件中。为了增加工具的灵活性，可以使用环境变量。 ##### 1.1 使用环境变量的好处 - 用户可以在 shell 配置中一次性指定选项，避免每次执行命令时都输入该选项。 - 用户可以为不同的环境设置不同的配置。 ##### 1.2 在 Go 中使用环境变量 在 Go 中，可以使用 `os` 包来处理环境和环境变量。以下是具体的操作步骤： 1. **修改 `main.go` 文件**： - 将 `todoFileName` 从常量改为变量，以便在环境变量被定义时可以更改它。 ```go // Default file name var todoFileName = ".todo.json" ``` - 在 `main` 函数中，在实例化 `todo.List` 类型之前，检查环境变量 `TODO_FILENAME` 是否被设置。如果设置了，则将其值赋给 `todoFileName`；否则，`todoFileName` 保持默认值。 ```go // Check if the user defined the ENV VAR for a custom file name if os.Getenv("TODO_FILENAME") != "" { todoFileName = os.Getenv("TODO_FILENAME") } ``` 2. **执行工具并设置环境变量**： ```bash $ ls main.go main_test.go $ export TODO_FILENAME=new-todo.json $ go run main.go -task "Test env vars design" $ ls main.go main_test.go new-todo.json $ cat new-todo.json [{"Task":"Test env vars design","Done":false,"CreatedAt":"2018-03-25T23:08:39.780125489-04:00","CompletedAt":"0001-01-01T00:00:00Z"}] $ go run main.go -list 1: Test env vars design ``` ##### 1.3 从标准输入捕获输入 为了让命令行工具更好地与其他工具交互，可以添加通过标准输入（STDIN）添加新任务的功能。以下是具体的操作步骤： 1. **更新 `main.go` 文件的导入列表**： ```go import ( "bufio" "flag" "fmt" "io" "os" "strings" "pragprog.com/rggo/interacting/todo" ) ``` 2. **创建 `getTask` 函数**： ```go // getTask function decides where to get the description for a new // task from: arguments or STDIN func getTask(r io.Reader, args ...string) (string, error) { if len(args) > 0 { return strings.Join(args, " "), nil } s := bufio.NewScanner(r) s.Scan() if err := s.Err(); err != nil { return "", err } if len(s.Text()) == 0 { return "", fmt.Errorf("Task cannot be blank") } return s.Text(), nil } ``` 3. **更新 `main` 函数**： ```go // Parsing command line flags add := flag.Bool("add", false, "Add task to the ToDo list") list := flag.Bool("list", false, "List all tasks") complete := flag.Int("complete", 0, "Item to be completed") case *add: // When any arguments (excluding flags) are provided, they will be // used as the new task t, err := getTask(os.Stdin, flag.Args()...) if err != nil { fmt.Fprintln(os.Stderr, err) os.Exit(1) } l.Add(t) // Save the new list if err := l.Save(todoFileName); err != nil { fmt.Fprintln(os.Stderr, err) os.Exit(1) } ``` 4. **更新测试套件**： - 在 `main_test.go` 文件的导入列表中添加 `io` 包。 - 更新现有的 `AddNewTask` 测试用例中的参数，将 `-task` 改为 `-add`。 - 添加新的测试用例 `AddNewTaskFromSTDIN` 来测试新功能。 ```go func TestTodoCLI(t *testing.T) { task : ```