-
Notifications
You must be signed in to change notification settings - Fork 93
/
copyright.go
115 lines (108 loc) · 3.04 KB
/
copyright.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
// Copyright (C) 2017, 2018, 2019 EGAAS S.A.
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or (at
// your option) any later version.
//
// This program is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
package main
import (
"bytes"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"strings"
)
// The program inserts copyright notice at the beginning of .go files.
const (
codeGen = `// Code generated`
)
var (
copyright []byte
)
// ProcessDir proceeds the specified directory and inserts the copyright at the beginning of the files
func ProcessDir(dir string, recurse bool) {
files, err := ioutil.ReadDir(dir)
if err != nil {
fmt.Println(err)
}
path, err := filepath.Abs(dir)
if err != nil {
fmt.Println(err)
return
}
path = path[strings.Index(path, `go-apla`)+7:]
for _, file := range files {
fname := file.Name()
fullName := filepath.Join(dir, fname)
if strings.HasSuffix(fname, `.go`) {
fmt.Printf(fullName)
if fdata, err := ioutil.ReadFile(fullName); err != nil {
fmt.Println(err)
os.Exit(1)
} else {
var prefix string
if strings.HasPrefix(string(fdata[:len(codeGen)]), codeGen) {
fmt.Println(`...Code generated`)
continue
}
if bytes.Equal(fdata[:4], []byte(`// +`)) {
lines := strings.Split(string(fdata[:256]), "\n")
for _, line := range lines {
if strings.HasPrefix(line, `// +`) {
prefix += line + "\r\n"
} else {
break
}
}
}
if len(prefix) > 0 {
prefix += "\r\n"
}
off := bytes.IndexByte(fdata, 0xa)
if len(fdata) <= len(copyright) || !bytes.Equal(fdata[off+1:off+1+len(copyright)], copyright) {
off := bytes.Index(fdata, []byte(`package`))
if off == -1 {
fmt.Println(`...package has not been found`)
} else {
out := append([]byte(prefix), copyright...)
out = append(out, fdata[off:]...)
if err := ioutil.WriteFile(fullName, out, 0644); err == nil {
fmt.Println(`...Overwrited`)
} else {
fmt.Println(`...` + err.Error())
}
}
} else {
fmt.Println(`...Skipped`)
}
}
}
if recurse && file.IsDir() {
ProcessDir(fullName, recurse)
}
}
}
func main() {
var err error
if copyright, err = ioutil.ReadFile(`copyright.txt`); err == nil && len(copyright) > 0 {
if copyright[len(copyright)-1] != 0xa {
copyright = append(copyright, 0xa)
}
ProcessDir("../..", false)
ProcessDir("../../cmd", true)
ProcessDir("../../packages", true)
ProcessDir("../../tools", true)
} else {
fmt.Println(err)
}
}