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
|
package main
//go:generate go run gen.go
import (
"bytes"
"compress/zlib"
"encoding/base64"
"fmt"
"io"
"os"
"path/filepath"
"strings"
"text/template"
)
func main() {
tmpls := make(map[string][]string)
err := filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
if err != nil {
fmt.Fprintf(os.Stderr, "Failed to access path %q: %v\n", path, err)
return err
}
details := strings.SplitN(path, ".", 2)
if len(details) != 2 || details[1] != "sql" {
//fmt.Fprintf(os.Stderr, "Skipping non-template: %+v \n", info.Name())
return nil
}
pathDetails := strings.SplitN(details[0], "/", 2)
driver := pathDetails[0]
//name := pathDetails[1]
fmt.Fprintf(os.Stderr, "Processing file: %s\n", path)
input, err := os.Open(path)
if err != nil {
return err
}
var out bytes.Buffer
encoder := base64.NewEncoder(base64.StdEncoding, &out)
compressor := zlib.NewWriter(encoder)
_, err = io.Copy(compressor, input)
input.Close()
compressor.Close()
encoder.Close()
if err != nil {
return err
}
tmpls[driver] = append(tmpls[driver], out.String())
return nil
})
if err != nil {
fmt.Fprintf(os.Stderr, "Failed to load templates: %s\n", err)
}
// Now generate our encoded template file
metaTmpl, err := template.New("meta").Parse(tmplData)
if err != nil {
fmt.Printf("Failed to parse template: %s", err)
} else {
metaTmpl.Execute(os.Stdout, tmpls)
}
}
var tmplData = `package main
// Automatically generated, don't bother editing.
import (
"bytes"
"strings"
"compress/zlib"
"encoding/base64"
"fmt"
"io"
)
// migrations holds a set of base64 encoded SQL scripts.
var migrations = map[string][]string{
{{- range $driver, $schema := . }}
"{{$driver}}": []string{
{{- range $b := $schema }}
"{{ $b }}",
{{ end }}
},
{{ end }}
}
func decodeMigrations(driver string) ([]string, error) {
data, ok := migrations[driver]
if !ok {
return nil, fmt.Errorf("no migrations for driver %q", driver)
}
out := make([]string, len(data))
for i, b := range data {
decoder := base64.NewDecoder(base64.StdEncoding, strings.NewReader(b))
decompressor, err := zlib.NewReader(decoder)
if err != nil {
return nil, fmt.Errorf("unable to decode migration: %w", err)
}
var buf bytes.Buffer
_, err = io.Copy(&buf, decompressor)
if err != nil {
return nil, fmt.Errorf("failed to decompress migration: %w", err)
}
out[i] = buf.String()
}
return out, nil
}
`
|
