83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"alox.protogen/proto"
|
|
"github.com/knadh/koanf"
|
|
"github.com/knadh/koanf/parsers/yaml"
|
|
"github.com/knadh/koanf/providers/file"
|
|
)
|
|
|
|
//go:embed templates/*
|
|
var embeddedTemplates embed.FS
|
|
|
|
var appConfig = koanf.New(".")
|
|
|
|
func main() {
|
|
confPath := flag.String("c", "testdata/conf.yaml", "Config File")
|
|
outFileName := flag.String("o", "", "Name of the Output File")
|
|
inputFile := flag.String("i", "testdata/prot1.json", "Name of the Input File")
|
|
|
|
flag.Parse()
|
|
|
|
if err := appConfig.Load(file.Provider(*confPath), yaml.Parser()); err != nil {
|
|
log.Printf("error loading config: %v", err)
|
|
} else {
|
|
log.Printf("CONF: %v", appConfig)
|
|
}
|
|
|
|
file, err := os.ReadFile(*inputFile)
|
|
if err != nil {
|
|
log.Printf("ERROR: %v", err)
|
|
return
|
|
}
|
|
var cfg proto.Proto
|
|
json.Unmarshal(file, &cfg)
|
|
|
|
// if outFileName is set use it, if is not set set it to default value
|
|
if *outFileName != "" {
|
|
cfg.FileName = *outFileName
|
|
} else {
|
|
*outFileName = "build/proto"
|
|
}
|
|
|
|
// If filename not set in json take it from cli or default
|
|
if cfg.FileName != "" {
|
|
cfg.FileName = *outFileName
|
|
}
|
|
|
|
proto.InitTemplates(embeddedTemplates, "templates/*.tmpl")
|
|
|
|
header, err := proto.CreateCStyleHeader(&cfg)
|
|
if err != nil {
|
|
log.Printf("ERROR: %v", err)
|
|
}
|
|
|
|
if err != nil {
|
|
log.Printf("ERROR: %v", err)
|
|
return
|
|
}
|
|
|
|
err = os.WriteFile(fmt.Sprintf("%s.h", *outFileName), []byte(header), 0666)
|
|
if err != nil {
|
|
log.Printf("ERROR: %v", err)
|
|
return
|
|
}
|
|
|
|
source, err := proto.CreateCStyleSource(&cfg)
|
|
if err != nil {
|
|
log.Printf("ERROR: %v", err)
|
|
}
|
|
err = os.WriteFile(fmt.Sprintf("%s.c", *outFileName), []byte(source), 0666)
|
|
if err != nil {
|
|
log.Printf("ERROR: %v", err)
|
|
return
|
|
}
|
|
}
|