tartrazine/cli/enry/main.go

92 lines
1.5 KiB
Go
Raw Normal View History

2016-07-13 17:05:23 +00:00
package main
import (
"encoding/json"
"flag"
"fmt"
"io/ioutil"
"log"
2016-07-13 17:05:23 +00:00
"os"
"path/filepath"
2017-06-08 07:27:27 +00:00
"gopkg.in/src-d/enry.v1"
2016-07-13 17:05:23 +00:00
)
func main() {
flag.Usage = usage
2016-07-13 17:05:23 +00:00
flag.Parse()
2016-07-13 22:38:06 +00:00
root, err := filepath.Abs(flag.Arg(0))
if err != nil {
log.Fatal(err)
2016-07-13 22:38:06 +00:00
}
2016-07-13 17:05:23 +00:00
errors := false
out := make(map[string][]string, 0)
2016-07-13 22:38:06 +00:00
err = filepath.Walk(root, func(path string, f os.FileInfo, err error) error {
if err != nil {
errors = true
log.Println(err)
return filepath.SkipDir
2016-07-13 22:38:06 +00:00
}
relativePath, err := filepath.Rel(root, path)
if err != nil {
errors = true
log.Println(err)
return nil
}
if relativePath == "." {
return nil
}
if f.IsDir() {
relativePath = relativePath + "/"
}
2017-06-13 11:56:07 +00:00
if enry.IsVendor(relativePath) || enry.IsDotFile(relativePath) ||
enry.IsDocumentation(relativePath) || enry.IsConfiguration(relativePath) {
2016-07-13 17:05:23 +00:00
if f.IsDir() {
return filepath.SkipDir
}
return nil
}
if f.IsDir() {
return nil
}
content, err := ioutil.ReadFile(path)
if err != nil {
errors = true
log.Println(err)
return nil
}
2016-07-13 17:05:23 +00:00
2017-06-13 11:56:07 +00:00
language := enry.GetLanguage(filepath.Base(path), content)
if language == enry.OtherLanguage {
return nil
2016-07-13 17:05:23 +00:00
}
out[language] = append(out[language], relativePath)
2016-07-13 17:05:23 +00:00
return nil
})
if err != nil {
log.Fatal(err)
}
2016-07-13 22:38:06 +00:00
data, _ := json.MarshalIndent(out, "", " ")
fmt.Printf("%s\n", data)
2016-07-13 17:05:23 +00:00
}
2016-07-13 22:38:06 +00:00
func usage() {
fmt.Fprintf(
2017-06-14 10:43:26 +00:00
os.Stderr, "enry, A simple (and faster) implementation of github/linguist \nusage: %s <path>\n",
2016-07-13 22:38:06 +00:00
os.Args[0],
)
flag.PrintDefaults()
}