-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
121 lines (97 loc) · 1.98 KB
/
main.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
116
117
118
119
120
121
package main
import (
"bytes"
"flag"
"fmt"
"io"
"io/ioutil"
"os"
"strings"
"sync"
)
func main() {
source := flag.String("source", "./", "source directory")
out := flag.String("out", "./", "output directory")
flag.Parse()
sourceDir := strings.TrimSuffix(*source, "/") + "/"
outDir := strings.TrimSuffix(*out, "/") + "/"
createIfMissing(outDir)
exclude := flag.Args()
files, err := ioutil.ReadDir(sourceDir)
if err != nil {
panic(err)
}
var wg sync.WaitGroup
FILE_ITERATOR:
for _, f := range files {
if !isFile(sourceDir + f.Name()) {
continue FILE_ITERATOR
}
for _, exc := range exclude {
if strings.Contains(f.Name(), exc) {
continue FILE_ITERATOR
}
}
if strings.Contains(f.Name(), "min") {
continue FILE_ITERATOR
}
nameComponents := strings.Split(f.Name(), ".")
extension := nameComponents[len(nameComponents)-1]
j := job{
sourceDir: sourceDir,
outDir: outDir,
fileName: f.Name(),
extension: extension,
}
wg.Add(1)
go func() {
defer wg.Done()
execute(j)
}()
}
wg.Wait()
}
type job struct {
sourceDir string
outDir string
fileName string
extension string
}
func execute(j job) {
content := readFile(j.sourceDir + j.fileName)
writeFile(content, j.fileName, j.extension, j.outDir)
}
func readFile(name string) []byte {
buf := bytes.NewBuffer(nil)
file, err := os.Open(name)
if err != nil {
panic(err)
}
_, err = io.Copy(buf, file)
if err != nil {
panic(err)
}
file.Close()
return buf.Bytes()
}
func writeFile(content []byte, fileName string, extension string, out string) {
s := minify(content)
err := ioutil.WriteFile(out+strings.TrimSuffix(fileName, extension)+"min."+extension, s, 0644)
if err != nil {
panic(err)
}
}
func isFile(path string) bool {
fileInfo, err := os.Stat(path)
if err != nil {
fmt.Println(err)
return false
}
return !fileInfo.IsDir()
}
func createIfMissing(path string) {
_, err := os.Stat(path)
if os.IsNotExist(err) {
os.Mkdir(path, os.ModePerm)
}
}