forked from kevinburke/go-bindata
/
imports.go
79 lines (69 loc) · 1.47 KB
/
imports.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
package xbindata
import (
"fmt"
"io"
"sort"
"strings"
)
func write_imports(w io.Writer, c *Config, imports ...string) (err error) {
imports = append(
imports,
`bc "github.com/moisespsena-go/xbindata/xbcommon"`,
`github.com/moisespsena-go/io-common`,
)
if c.FileSystem {
imports = append(
imports,
`"github.com/moisespsena-go/xbindata/xbfs"`,
`fsapi "github.com/moisespsena-go/assetfs/assetfsapi"`,
)
}
var (
importsMap = map[string]bool{}
newImports, excludes []string
)
for _, imp := range imports {
if _, ok := importsMap[imp]; !ok {
if imp[0] == '-' {
excludes = append(excludes, imp[0:])
} else {
importsMap[imp] = true
}
}
}
for _, imp := range excludes {
if _, ok := importsMap[imp]; ok {
delete(importsMap, imp)
}
}
for imp, _ := range importsMap {
newImports = append(newImports, imp)
}
imports = newImports
sort.Slice(imports, func(i, j int) bool {
a, b := imports[i], imports[j]
ia, ib := strings.IndexByte(a, ' '), strings.IndexByte(b, ' ')
if ia == ib {
if len(a) < len(b) {
return true
}
if len(a) == len(b) {
return a < b
}
return false
} else if ia > 0 && ib < 0 {
return false
} else if ib > 0 && ia < 0 {
return true
}
return a[0:ia] < b[0:ib]
})
for i, imp := range imports {
if imp[len(imp)-1] != '"' {
imp = `"` + imp + `"`
}
imports[i] = "\t" + imp
}
_, err = fmt.Fprintf(w, "import (\n"+strings.Join(imports, "\n")+"\n)\n")
return
}