96 lines
1.9 KiB
Go
96 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/fs"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
)
|
|
|
|
const (
|
|
stateGlobal = iota
|
|
stateTemplate
|
|
stateGen
|
|
)
|
|
|
|
func main() {
|
|
db, err := ioutil.ReadFile(os.Args[2])
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var data map[string]interface{}
|
|
if err := json.Unmarshal(db, &data); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = filepath.WalkDir(os.Args[1], func(path string, d fs.DirEntry, err error) error {
|
|
if d.IsDir() {
|
|
return nil
|
|
}
|
|
if filepath.Ext(path) != ".go" {
|
|
return nil
|
|
}
|
|
fb, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
lines := strings.Split(string(fb), "\n")
|
|
|
|
outLines := make([]string, 0, len(lines))
|
|
var templateLines []string
|
|
|
|
state := stateGlobal
|
|
|
|
for i, line := range lines {
|
|
ln := i+1
|
|
switch state {
|
|
case stateGlobal:
|
|
outLines = append(outLines, line)
|
|
if line == `/* inline-gen template` {
|
|
state = stateTemplate
|
|
fmt.Printf("template section start %s:%d\n", path, ln)
|
|
}
|
|
case stateTemplate:
|
|
outLines = append(outLines, line) // output all template lines
|
|
|
|
if line == `inline-gen start */` {
|
|
state = stateGen
|
|
fmt.Printf("generated section start %s:%d\n", path, ln)
|
|
continue
|
|
}
|
|
templateLines = append(templateLines, line)
|
|
case stateGen:
|
|
if line != `//inline-gen end` {
|
|
continue
|
|
}
|
|
state = stateGlobal
|
|
fmt.Printf("inline gen:\n")
|
|
fmt.Println(strings.Join(templateLines, "\n"))
|
|
|
|
tpl, err := template.New("").Parse(strings.Join(templateLines, "\n"))
|
|
if err != nil {
|
|
fmt.Printf("%s:%d: parsing template: %s\n", path, ln, err)
|
|
os.Exit(1)
|
|
}
|
|
var b bytes.Buffer
|
|
err = tpl.Execute(&b, data)
|
|
|
|
outLines = append(outLines, strings.Split(b.String(), "\n")...)
|
|
fmt.Println("inline gen-ed:\n", b.String())
|
|
|
|
outLines = append(outLines, line)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|