52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"os"
|
|||
|
|
"sort"
|
|||
|
|
|
|||
|
|
"gopkg.in/yaml.v2"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func loadBangs() (bangs map[string]string) {
|
|||
|
|
data, err := ioutil.ReadFile("../bangs.yml")
|
|||
|
|
if err != nil {
|
|||
|
|
panic(fmt.Sprintf("Failed to load bangs.yml: %v", err))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err = yaml.Unmarshal(data, &bangs)
|
|||
|
|
if err != nil {
|
|||
|
|
panic(fmt.Sprintf("Failed to decode bangs: %v", err))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
bangs := loadBangs()
|
|||
|
|
|
|||
|
|
// generating sorted keys to have stable generated code
|
|||
|
|
// this is not optimal, performance-wise but it runs only once at build
|
|||
|
|
// time, so it doesn’t matter that much)
|
|||
|
|
bangNames := []string{}
|
|||
|
|
for bang := range bangs {
|
|||
|
|
bangNames = append(bangNames, bang)
|
|||
|
|
}
|
|||
|
|
sort.Strings(bangNames)
|
|||
|
|
|
|||
|
|
out, err := os.Create("bangs.go")
|
|||
|
|
if err != nil {
|
|||
|
|
panic(fmt.Sprintf("Failed to create bangs.go: %v", err))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
out.Write([]byte("// Code generated from ../../bangs.yml DO NOT EDIT.\n"))
|
|||
|
|
out.Write([]byte("package main \n\nfunc getBang(name string) string {\n\tswitch name {\n"))
|
|||
|
|
|
|||
|
|
for _, bangName := range bangNames {
|
|||
|
|
out.Write([]byte("\t\tcase `" + bangName + "`: return `" + bangs[bangName] + "`\n"))
|
|||
|
|
}
|
|||
|
|
out.Write([]byte("\t\tdefault: return \"\"\n"))
|
|||
|
|
out.Write([]byte("\t}\n}\n"))
|
|||
|
|
}
|