背景
前面一直在折腾 Go 开发终端命令小工具,在开发的过程中使用效果还不错。
经常使用,被运营小姐姐看到, 然后问说有没有什么小工具,可以检测推广的URL 是否被微信给屏蔽了, 于是接着更新 终端命令小工具, 增加 URL 检测功能。
效果如图
运行命令
pf_tools pf_cwx [urlstring]
代码实现
func GetWxUrlInfo(urlString string) { api := "https://cgi.urlsec.qq.com/index.php?m=url&a=validUrl&url=" + urlString resp, err := http.Get(api) if err != nil { fmt.Println("请求失败:", err) return } defer func(Body io.ReadCloser) { _ = Body.Close() }(resp.Body) out, _ := io.ReadAll(resp.Body) if err != nil { fmt.Println("读取响应失败:", err) return } if err := ui.Init(); err != nil { log.Fatalf("failed to initialize termui: %v", err) } defer ui.Close() table := widgets.NewTable() table.Title = "微信URL安全检测" // table.BorderStyle = ui.NewStyle(ui.ColorRed) table.Rows = [][]string{ []string{"网址", "检测结果 "}, } //fmt.Println(string(out)) urlResponse := &WxUrlInfo{} if err := json.Unmarshal(out, &urlResponse); err != nil { fmt.Println("解析json失败:", err) return } enStr := "网址未被微信屏蔽" if urlResponse.ReCode == 0 { enStr = "网址被微信屏蔽" } table.Rows = append(table.Rows, []string{urlString, enStr}) table.TextStyle = ui.NewStyle(ui.ColorGreen) table.TitleStyle = ui.NewStyle(ui.ColorGreen) table.SetRect(0, 0, 80, 5) ui.Render(table) uiEvents := ui.PollEvents() for { e := <-uiEvents switch e.ID { case "q": return } }}
核心 其实就是 使用 golang 调用 cgi.urlsec.qq.com/index.php 这个接口 用来检测 是否被 屏蔽, 目前来说 检测效果还不错. 至少满足 小姐姐的需求, 再啰嗦一下, Golang 的 cobra 确实不错, 值得一试.
开发过程
因为前面已经,整好了 cobra-cli的环境, 所以在 小工具的开发过程中, 只需要 敲入
cobra-cli add pf_cwx_url
直接会在项目的 cmd 目录中 生成 文件 pfCwxUrl.go,代码如下:
/*Copyright © 2024 NAME HERE <EMAIL ADDRESS>*/package cmdimport ( "fmt" "github.com/spf13/cobra")// pfTestCmd represents the pfTest commandvar pfCwxUrlCmd = &cobra.Command{ Use: "pfCwxUrlCmd", Short: "A brief description of your command", Long: `A longer description that spans multiple lines and likely contains examplesand usage of using your command. For example:Cobra is a CLI library for Go that empowers applications.This application is a tool to generate the needed filesto quickly create a Cobra application.`, Run: func(cmd *cobra.Command, args []string) { fmt.Println("pfCwxUrlCmd called") },}func init() { rootCmd.AddCommand(pfCwxUrlCmd) // Here you will define your flags and configuration settings. // Cobra supports Persistent Flags which will work for this command // and all subcommands, e.g.: // pfTestCmd.PersistentFlags().String("foo", "", "A help for foo") // Cobra supports local flags which will only run when this command // is called directly, e.g.: // pfTestCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")}
然后直接修改 代码中的:
Use: "pf_cwx", Short: "微信域名拦截检测", Long: `检测域名是否被微信拦截`,
Run 方法下就是 应对的业务逻辑, 我这里的结构是如下:
├── LICENSE
├── README.md
├── cmd
│ ├── pfB64.go
│ ├── pfCd.go
│ ├── pfCwxUrl.go
│ ├── pfM.go
│ ├── pfMd5.go
│ ├── pfS.go
│ ├── pfT.go
│ ├── pfTest.go
│ ├── pfWt.go
│ └── root.go
├── go.mod
├── go.sum
├── main.go
├── pak
│ ├── base64_cry.go
│ ├── clear.go
│ ├── md5.go
│ ├── mobile.go
│ ├── phone.dat
│ ├── speak.go
│ ├── stime.go
│ ├── weather.go
│ └── wxurl.go
所有的 业务流程代码是放在 pak 的 所以 Run的代码修改成了下面的:
Run: func(cmd *cobra.Command, args []string) { if len(args) == 0 { _ = cmd.Help() return } if pak.CheckUrl(args[0]) == false { _ = cmd.Help() return } pak.GetWxUrlInfo(args[0]) },
然后 运行
go run main.go pf_cwx https://www.baidu.com
测试没有问题以后直接
go install
本地就可以用命令来愉快的玩耍了
pf_tools pf_cwx "https://www.baidu.com"
最后,附上 项目地址:
🌟 更多精彩内容等你发现！关注【PFinalClub】，成为我们的一员，让我们一起在编程的海洋中探索、学习、成长！
