80 lines
1.3 KiB
Go
80 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"golang.org/x/tools/imports"
|
|
)
|
|
|
|
var (
|
|
tmpl = template.Must(template.New("var").Parse(`
|
|
// CODE GENERATED. DO NOT EDIT
|
|
package {{.PkgName }}
|
|
var (
|
|
{{ .Name }} = ` + "`" + `{{.Value}}` + "`" + `
|
|
)
|
|
|
|
`))
|
|
)
|
|
|
|
type vars struct {
|
|
PkgName string
|
|
Name string
|
|
Value string
|
|
}
|
|
|
|
func main() {
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
if len(os.Args) < 3 {
|
|
log.Fatalf("Usage: %s file varname [output]", os.Args[0])
|
|
}
|
|
fname := os.Args[1]
|
|
targetVar := os.Args[2]
|
|
pkg := os.Getenv("GOPACKAGE")
|
|
absFilePath, err := filepath.Abs(fname)
|
|
|
|
targetFName := strings.ToLower(targetVar) + ".go"
|
|
if len(os.Args) > 3 {
|
|
targetFName = os.Args[3]
|
|
}
|
|
log.Println(absFilePath, "=>", targetFName)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
b, err := ioutil.ReadFile(fname)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if _, err := os.Stat(absFilePath); err != nil {
|
|
os.Remove(absFilePath)
|
|
}
|
|
buff := bytes.Buffer{}
|
|
err = tmpl.Execute(&buff, vars{
|
|
Name: targetVar,
|
|
Value: string(b),
|
|
PkgName: pkg,
|
|
})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
data, err := imports.Process(targetFName, buff.Bytes(), nil)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
f, err := os.Create(targetFName)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
_, err = f.Write(data)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|