golang实现一个linux命令ls命令（命令行工具构建）

news2024/10/6 10:30:25

希望2023可以听到这些话：

恭喜你得到了这份工作
恭喜你的建议被采用了
恭喜你被录取了
恭喜你的考试顺利通过了
恭喜你上岸了
恭喜你升职了
恭喜你加薪了
恭喜你体检结果一切正常

在这篇文章下面许个愿吧！

ls 命令

要实现ls，首先先我们复习一下ls命令。

ls不加任何参数，表示查询当前目录下的文件/文件夹
ls后面加上路径，列出所有，不包含隐藏文件/文件夹
ls -a列出所有，包括以 .开头的隐藏文件

此命令还有很多功能，这一节的重点是命令行程度的编写，就实现上面四个。

os.Args 读取参数

定义如下，一个数组。该数组长度由参数个数决定，第一个参数为程序本身，后面为参数列表。

var Args []string

比如ls -l，Args=["ls","-l"]，我们现在实现ls+目录，或者缺省目录时默认当前目录代码如下。

func main() {
 targetDirPath := "./"
 if len(os.Args) > 1 {
  targetDirPath = os.Args[1]
 }

 if dirList, err := ioutil.ReadDir(targetDirPath); err == nil {
  for _, dirInfo := range dirList {
   fmt.Print(dirInfo.Name() + " ")
  }
 } else {
  fmt.Println(err.Error())
 }
}

使用 os.Agrs 来获取传入程序的参数
使用 ioutil.ReadDir 读取目录,将返回值 [] FileInfo遍历,输出目录/文件信息
将遇到的 error输出到终端

输出

$ go build -o ls main.go
$ ./ls 
main.go folder ls
$ ./ls folder
file1 file2

看，我们已经简单的实现了。但我们会发现，要使用-开头的参数时，用os.Args这个还需要手写更多复杂逻辑去判断输入的到底是路径，还是其他参数，于是下一个包登场了。

pflag

昨天的每日一Go已经讲了flag包，pflag 包与 flag 包的工作原理甚至是代码实现都是类似的，下面是 pflag 相对 flag 的一些优势：

支持更加精细的参数类型：例如， flag 只支持 uint 和 uint64，而 pflag 额外支持 uint8、 uint16、 int32 等类型。
支持更多参数类型： ip、 ip mask、 ip net、 count、以及所有类型的 slice 类型。
兼容标准 flag 库的 Flag 和 FlagSet： pflag 更像是对 flag 的扩展。
原生支持更丰富的功能：支持 shorthand、 deprecated、 hidden 等高级功能。

var a = pflag.BoolP("all", "a", false, "Include directory entries whose names begin with a dot (.).")
 var help = pflag.BoolP("help", "h", false, "Show this help message.")
 pflag.Parse()
 if *help {
  pflag.Usage()
  return
 }
 args := pflag.Args()
 if len(args) == 0 {
  args = append(args, "./")
 }
 if len(args) == 1 {
  ShowPath(args[0], *a)
 } else {
  for _, v := range args {
   fmt.Println(v + ":")
   ShowPath(v, *a)
  }
 }

列出文件夹的方式写到了 ShowPath函数里，第二参数是一个 bool判断是否列出全部文件（包括隐藏文件）感兴趣去看我源代码，这里不是重点。
pflag.BoolP 定义一个 bool类型的参数，支持长短参数形式 --all, -a，返回值是指针。
pflag.Parse() 调用这个函数，把参数值设置到相应指针的指向里。
pflag.Usage() 用法打印。

这样我们就实现了ls的基本功能。测试一下。

$ go build -o ls main.go
$ ./ls
folder go.mod go.sum main.go 
$ ./ls -a
. .. .git folder go.mod go.sum main.go 
$ ./ls ./ folder 
./:
folder go.mod go.sum main.go 
folder:
file1 file2
$ ./ls -h
Usage of ./ls:
  -a, --all    Include directory entries whose names begin with a dot (.).
  -h, --help   Show this help message.