bb/ls/ls.go

package ls
import(
	"os"
	"fmt"
	"flag"
	"strings"
)

var(
	dirFlag bool
	foldLvl int
)

func IsDir(p string) (bool, error) {
	finfo, e := os.Stat(p)
	if e != nil {
		return false, e
	}
	return finfo.IsDir(), e
}

func ReadDir(p string) ([]os.FileInfo, error) {
	f, e := os.Open(p)
	if e != nil {
		return nil, e
	}
	l, e := f.Readdir(-1)
	if e != nil {
			return nil, e
	}
	f.Close()

	return l, nil
}

func Stat(p string) (os.FileInfo, error) {
	f, e := os.Open(p)
	if e != nil {
		return nil, e
	}
	s, e := f.Stat()
	f.Close()
	return s, nil
}

func ls(p string, fold int) error {
	isDir, e := IsDir(p)
	if e != nil {
		return e
	}

	pp := strings.TrimRight(p, "/")

	if !isDir || dirFlag || fold<1 {
		fmt.Println(pp);
	}else{
		l, e := ReadDir(pp)
		if e!=nil {
			return e
		}
		for _, f := range l {
			s := pp+"/"+f.Name()
			if b, _:=IsDir(s) ; b {
				fmt.Println(s)
			}
			if 0<fold {
				ls(s, fold-1)
			}
		}
	}
	
	return nil
}

func Run(args []string) int {
	status := 0
	arg0 := args[0]
	args = args[1:]
	flagSet := flag.NewFlagSet(arg0, flag.ExitOnError)
	flagSet.IntVar(&foldLvl, "r", 1, "List recursively with choosing deepness, can't be negative or zero.")
	flagSet.BoolVar(&dirFlag, "d", false, "List directory as usual file, doesn't work with with recursive level not equal 1. ")
	flagSet.Usage = func() {
		fmt.Fprintf(os.Stderr, "Usage of %s: %s [files]\n", arg0, arg0)
		flagSet.PrintDefaults()
	}
	flagSet.Parse(args)
	if foldLvl<1 {
		flagSet.Usage()
		return 1
	}
	
	if foldLvl!=1 && dirFlag {
		flagSet.Usage()
		return 1
	}
	args = flagSet.Args()
	if len(args) == 0 {
		l, e := ReadDir(".")
		if e != nil {
			status = 1
			fmt.Fprintf(os.Stderr, "%s: %s.\n", arg0, e)
		} else {
			for _, f := range l {
				e := ls(f.Name(), foldLvl-1)
				if e!=nil {
					status = 1
					fmt.Fprintf(os.Stderr, "%s: %s.\n", arg0, e)
				}
			}
		}
	} else {
		for _, p := range args {
			e := ls(p, foldLvl)
			if e != nil {
				status = 1
				fmt.Fprintf(os.Stderr, "%s: %s.\n", arg0, e)
			}
			
		}
	}
	return status
}