122 lines
2 KiB
Go
122 lines
2 KiB
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
|
|
}
|