快捷導(dǎo)航

使用Golang編寫一個(gè)簡(jiǎn)單的命令行工具

更新時(shí)間：2023年12月28日 10:39:29 作者：陪我去看海

Cobra是一個(gè)強(qiáng)大的開源工具,能夠幫助我們快速構(gòu)建出優(yōu)雅且功能豐富的命令行應(yīng)用,本文將利用Cobra編寫一個(gè)簡(jiǎn)單的命令行工具,感興趣的可以了解下

前言

每周四，在群里總是會(huì)看到一些瘋狂星期四的小故事，而且個(gè)個(gè)的小故事還挺有趣，但是讓我這樣的人肯定是編不出來了，在想剛好在學(xué)習(xí)Go，是不是可以怎么操作一下，寫個(gè)命令行來獲取周四文案啥的，這樣就不用自己去編了。既然有了想法那就開干

準(zhǔn)備

源碼地址，結(jié)合看最好，下面代碼并不完全，大佬們有好的想法，歡迎issues，pr，也歡迎star，一起整活

了解使用Go寫命令行的庫(kù)
找一個(gè)生成瘋狂星期四文案的接口（后期也可以做自動(dòng)生成）
通過特定的命令來調(diào)用這個(gè)接口并拿到這個(gè)文案

在這里我了解到一個(gè)用Go做命令行的流行庫(kù)cobra，據(jù)官方說這個(gè)被用在了很多大型項(xiàng)目，像Kubernetes, Hugo, GitHub CLI等等

找到的免費(fèi)生成瘋狂星期四文案的接口https://api.jixs.cc/api/wenan-fkxqs/index.php

實(shí)現(xiàn)

首先量化需求，既然已接口的形式的話，其實(shí)無非就是調(diào)用接口，拿到返回值了，只是把這個(gè)步驟放在了用戶命令有對(duì)應(yīng)操作之后執(zhí)行，明確需求了，就可以開始寫代碼了

1.使用Go調(diào)用這個(gè)API，拿到返回值并返回

package cmd

import (
	"io"
	"log"
	"net/http"
)

// Fetch 發(fā)起請(qǐng)求
func Fetch() string {
	client := &http.Client{}
	req, _ := http.NewRequest("GET", "https://api.jixs.cc/api/wenan-fkxqs/index.php", nil)
	res, err := client.Do(req)
	if err != nil {
		log.Fatal("Http get error is ", err)
	}
	if res.StatusCode != http.StatusOK {
		log.Fatal("Http status code is ", res.StatusCode)
	}
	defer func(Body io.ReadCloser) {
		err := Body.Close()
		if err != nil {
			log.Fatal(err)
		}
	}(res.Body)
	bytes, err := io.ReadAll(res.Body)
	if err != nil {
		return ""
	}
	return string(bytes)
}

2.使用Cobra做一個(gè)特定命令行

package cmd

import (
	"fmt"
	"github.com/spf13/cobra"
	"os"
)

var ThursdayCmd = &cobra.Command{
	Use:   "Thursday [文案]",
	Short: "crazy Thursday",
	Long:  `crazy Thursday`,
	Run: func(cmd *cobra.Command, args []string) {
		res := Fetch()
		Print(res)
		fmt.Println(res)
	},
}

var rootCmd = &cobra.Command{
	Use: "ct",
}

func Execute() {
	rootCmd.AddCommand(ThursdayCmd)
	rootCmd.SetHelpCommand(ThursdayCmd)
	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

這樣當(dāng)我們執(zhí)行 go run main.go Thursday 后會(huì)在控制臺(tái)打印出獲取到的文案

但是我使用 GoLand 在這一步有點(diǎn)問題，就是當(dāng)文案特別長(zhǎng)的時(shí)候，他在控制臺(tái)還是會(huì)強(qiáng)行顯示一行，無法復(fù)制到全部，所以這時(shí)想起一種解決方法，那就是將文案存入一個(gè)txt文件中，這樣就可以啦。

3.生成文件夾，將結(jié)果存入

package cmd

import (
	"fmt"
	"log"
	"os"
)

func Print(res string) {
	// 創(chuàng)建文件
	file, err := os.Create("./output/output.txt")
	if err != nil {
		fmt.Println("無法創(chuàng)建文件:", err)
		return
	}
	defer func(file *os.File) {
		err := file.Close()
		if err != nil {
			log.Fatal(err)
		}
	}(file)

	// 將內(nèi)容寫入文件
	_, err = file.WriteString(res)
	if err != nil {
		fmt.Println("無法寫入文件:", err)
		return
	}

	fmt.Println("內(nèi)容已成功寫入文件 output.txt")
}

到此基本上就完成了，算是一個(gè)練習(xí)的小demo吧，可以拿著這個(gè)去和群友歡樂對(duì)線吧！