Tools/Addlicense
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #14 from jmccann/custom-license-picked

Browse Source 
Read license from file (take 2)
This commit is contained in:
alex
2018-11-30 19:01:25 +01:00
committed by GitHub
parent 486fb589fd 8bc5e5f694
commit 1e1ee4516a
2 changed files with 38 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
main.go
View File

@@ -20,6 +20,7 @@ import (
"bytes"
"flag"
"fmt"
"html/template"
"io/ioutil"
"log"
"os"
@@ -44,9 +45,10 @@ Flags:
`
var (
holder = flag.String("c", "Google LLC", "copyright holder")
license = flag.String("l", "apache", "license type: apache, bsd, mit")
year = flag.Int("y", time.Now().Year(), "year")
holder = flag.String("c", "Google LLC", "copyright holder")
license = flag.String("l", "apache", "license type: apache, bsd, mit")
licensef = flag.String("f", "", "license file")
year = flag.Int("y", time.Now().Year(), "year")
)
func main() {
@@ -65,6 +67,26 @@ func main() {
Holder: *holder,
}
var t *template.Template
if *licensef != "" {
d, err := ioutil.ReadFile(*licensef)
if err != nil {
log.Printf("license file: %v", err)
os.Exit(1)
}
t, err = template.New("").Parse(string(d))
if err != nil {
log.Printf("license file: %v", err)
os.Exit(1)
}
} else {
t = licenseTemplate[*license]
if t == nil {
log.Printf("unknown license: %s", *license)
os.Exit(1)
}
}
// process at most 1000 files in parallel
ch := make(chan *file, 1000)
done := make(chan struct{})
@@ -73,7 +95,7 @@ func main() {
for f := range ch {
wg.Add(1)
go func(f *file) {
err := addLicense(f.path, f.mode, *license, data)
err := addLicense(f.path, f.mode, t, data)
if err != nil {
log.Printf("%s: %v", f.path, err)
}
@@ -110,30 +132,30 @@ func walk(ch chan<- *file, start string) {
})
}
func addLicense(path string, fmode os.FileMode, typ string, data *copyrightData) error {
func addLicense(path string, fmode os.FileMode, tmpl *template.Template, data *copyrightData) error {
var lic []byte
var err error
switch filepath.Ext(path) {
default:
return nil
case ".c", ".h":
lic, err = prefix(typ, data, "/*", " * ", " */")
lic, err = prefix(tmpl, data, "/*", " * ", " */")
case ".js", ".jsx", ".tsx", ".css", ".tf":
lic, err = prefix(typ, data, "/**", " * ", " */")
lic, err = prefix(tmpl, data, "/**", " * ", " */")
case ".cc", ".cpp", ".cs", ".go", ".hh", ".hpp", ".java", ".m", ".mm", ".proto", ".rs", ".scala", ".swift", ".dart":
lic, err = prefix(typ, data, "", "// ", "")
lic, err = prefix(tmpl, data, "", "// ", "")
case ".py", ".sh", ".yaml", ".yml":
lic, err = prefix(typ, data, "", "# ", "")
lic, err = prefix(tmpl, data, "", "# ", "")
case ".el", ".lisp":
lic, err = prefix(typ, data, "", ";; ", "")
lic, err = prefix(tmpl, data, "", ";; ", "")
case ".erl":
lic, err = prefix(typ, data, "", "% ", "")
lic, err = prefix(tmpl, data, "", "% ", "")
case ".hs", ".sql":
lic, err = prefix(typ, data, "", "-- ", "")
lic, err = prefix(tmpl, data, "", "-- ", "")
case ".html", ".xml":
lic, err = prefix(typ, data, "<!--", " ", "-->")
lic, err = prefix(tmpl, data, "<!--", " ", "-->")
case ".php":
lic, err = prefix(typ, data, "<?php", "// ", "?>")
lic, err = prefix(tmpl, data, "<?php", "// ", "?>")
}
if err != nil || lic == nil {
return err

9
tmpl.go
View File

@@ -36,14 +36,9 @@ type copyrightData struct {
Holder string
}
// prefix will execute a license template l with data d
// prefix will execute a license template t with data d
// and prefix the result with top, middle and bottom.
func prefix(l string, d *copyrightData, top, mid, bot string) ([]byte, error) {
t := licenseTemplate[l]
if t == nil {
return nil, fmt.Errorf("unknown license: %s", l)
}
func prefix(t *template.Template, d *copyrightData, top, mid, bot string) ([]byte, error) {
var buf bytes.Buffer
if err := t.Execute(&buf, d); err != nil {
return nil, err