go语言文件正则表达式搜索

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"regexp"
)

func main() {
	// 命令行参数
	args := os.Args

	// 检查参数
	if len(args) == 1 {
		fmt.Println("ff is a file find tool. use like bottom")
		fmt.Println("ff [dir] [regexp]")
		return
	}

	if len(args) < 3 {
		fmt.Println("args < 3")
		return
	}

	fileName := args[1]
	pattern := args[2]

	file, err := os.Open(fileName)

	if err != nil {
		fmt.Println(err)
		return
	}

	fi, err := file.Stat()

	if err != nil {
		fmt.Println(err)
		return
	}

	if !fi.IsDir() {
		fmt.Println(fileName, " is not a dir")
	}

	reg, err := regexp.Compile(pattern)

	if err != nil {
		fmt.Println(err)
		return
	}

	// 遍历目录
	filepath.Walk(fileName,
		func(path string, f os.FileInfo, err error) error {
			if err != nil {
				fmt.Println(err)
				return err
			}

			if f.IsDir() {
				return nil
			}

			// 匹配目录
			matched := reg.MatchString(f.Name())

			if matched {
				fmt.Println(path)
			}

			return nil
		})

}