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
|
//go:build ignore
// +build ignore
// Autogenerates wrappers from templates to prevent too much duplicated code
// between the code for different modes.
package main
import (
"bytes"
"fmt"
"go/format"
"io/ioutil"
"strings"
"text/template"
)
type Instance struct {
Bits int
}
func (m Instance) Pkg() string {
return strings.ToLower(m.Name())
}
func (m Instance) Name() string {
return fmt.Sprintf("SIKEp%d", m.Bits)
}
func (m Instance) Field() string {
return fmt.Sprintf("Fp%d", m.Bits)
}
var (
Instances = []Instance{
{Bits: 434},
{Bits: 503},
{Bits: 751},
}
TemplateWarning = "// Code generated from"
)
func main() {
generatePackageFiles()
}
// Generates instance/sike.go from templates/pkg.templ.go
func generatePackageFiles() {
tl, err := template.ParseFiles("templates/pkg.templ.go")
if err != nil {
panic(err)
}
for _, mode := range Instances {
buf := new(bytes.Buffer)
err := tl.Execute(buf, mode)
if err != nil {
panic(err)
}
// Formating output code
code, err := format.Source(buf.Bytes())
if err != nil {
panic("error formating code")
}
res := string(code)
offset := strings.Index(res, TemplateWarning)
if offset == -1 {
panic("Missing template warning in pkg.templ.go")
}
err = ioutil.WriteFile(mode.Pkg()+"/sike.go", []byte(res[offset:]), 0o644)
if err != nil {
panic(err)
}
}
}
|