/
main.go
98 lines (85 loc) · 1.82 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
package main
import (
"encoding/json"
"flag"
"fmt"
"log"
"net/http"
"os"
"path/filepath"
"strings"
"github.com/cycloidio/inframap/provider"
)
var (
dry bool
branch string
)
func init() {
flag.BoolVar(&dry, "dry", false, "Makes a dry run")
flag.StringVar(&branch, "branch", "master", "Branch from tfdocs to use")
}
func main() {
flag.Parse()
for _, pr := range provider.TypeValues() {
if pr == provider.Raw || pr == provider.IM {
continue
}
url := fmt.Sprintf("https://raw.githubusercontent.com/cycloidio/tfdocs/%s/assets/%s/icons.json", branch, pr)
res, err := http.Get(url)
if err != nil {
log.Fatal(err)
}
if res.StatusCode != 200 {
log.Fatalf("expecting a 200 status code but got %d for provider %s", res.StatusCode, pr)
}
icons := make(map[string]string)
err = json.NewDecoder(res.Body).Decode(&icons)
if err != nil {
log.Fatal(err)
}
res.Body.Close()
paths := make(map[string]bool)
for _, p := range icons {
if p == "" {
continue
}
paths[p] = false
}
rootDir := fmt.Sprintf("assets/icons/%s/", pr)
_, err = os.Stat(rootDir)
if os.IsNotExist(err) {
fmt.Printf("missing icons for %s\n", pr)
continue
}
err = filepath.Walk(rootDir, func(p string, fi os.FileInfo, err error) error {
if err != nil {
return err
}
if !fi.IsDir() {
// We need to remove the rootDir as it's not on tfdocs
// and also replace the .png for .svg
np := strings.Replace(
strings.TrimPrefix(p, rootDir),
".png", ".svg", -1,
)
if _, ok := paths[np]; !ok {
fmt.Printf("remove %s %s\n", pr, p)
if !dry {
os.Remove(p)
}
} else {
paths[np] = true
}
}
return nil
})
if err != nil {
log.Fatal(err)
}
for p, ok := range paths {
if !ok {
fmt.Printf("missing %s %s\n", pr, p)
}
}
}
}