mirror of https://github.com/nmasse-itix/liche.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
1.2 KiB
81 lines
1.2 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/kr/text"
|
|
)
|
|
|
|
func stringSetToSlice(s2b map[string]bool) []string {
|
|
ss := make([]string, 0, len(s2b))
|
|
|
|
for s := range s2b {
|
|
ss = append(ss, s)
|
|
}
|
|
|
|
return ss
|
|
}
|
|
|
|
func printToStderr(xs ...interface{}) {
|
|
fmt.Fprintln(os.Stderr, xs...)
|
|
}
|
|
|
|
func fail(err error) {
|
|
s := err.Error()
|
|
printToStderr(color.RedString(strings.ToUpper(s[:1]) + s[1:]))
|
|
os.Exit(1)
|
|
}
|
|
|
|
func indent(s string) string {
|
|
return text.Indent(s, "\t")
|
|
}
|
|
|
|
func listDirectory(d string, fc chan<- string) error {
|
|
return filepath.Walk(d, func(f string, i os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
b, err := regexp.MatchString("(^\\.)|(/\\.)", f)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !i.IsDir() && !b && isMarkupFile(f) {
|
|
fc <- f
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func findMarkupFiles(fs []string, recursive bool, fc chan<- string) {
|
|
for _, f := range fs {
|
|
i, err := os.Stat(f)
|
|
|
|
if err != nil {
|
|
fail(err)
|
|
}
|
|
|
|
if i.IsDir() && recursive {
|
|
err := listDirectory(f, fc)
|
|
|
|
if err != nil {
|
|
fail(err)
|
|
}
|
|
|
|
} else if i.IsDir() {
|
|
fail(fmt.Errorf("%v is not a file", f))
|
|
} else {
|
|
fc <- f
|
|
}
|
|
}
|
|
|
|
close(fc)
|
|
}
|
|
|