initial
This commit is contained in:
@@ -0,0 +1,81 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"edmand46/arpack/generator"
|
||||
"edmand46/arpack/parser"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func main() {
|
||||
in := flag.String("in", "", "input Go file with struct definitions")
|
||||
outGo := flag.String("out-go", "", "output directory for generated Go code")
|
||||
outCS := flag.String("out-cs", "", "output directory for generated C# code")
|
||||
namespace := flag.String("cs-namespace", "Arpack.Messages", "C# namespace")
|
||||
flag.Parse()
|
||||
|
||||
if *in == "" {
|
||||
log.Fatal("arpack: -in is required")
|
||||
}
|
||||
if *outGo == "" && *outCS == "" {
|
||||
log.Fatal("arpack: at least one of -out-go or -out-cs is required")
|
||||
}
|
||||
|
||||
schema, err := parser.ParseSchemaFile(*in)
|
||||
if err != nil {
|
||||
log.Fatalf("arpack: parse error: %v", err)
|
||||
}
|
||||
msgs := schema.Messages
|
||||
if len(msgs) == 0 {
|
||||
log.Fatalf("arpack: no structs found in %s", *in)
|
||||
}
|
||||
|
||||
baseName := strings.TrimSuffix(filepath.Base(*in), ".go")
|
||||
|
||||
if *outGo != "" {
|
||||
pkgName := filepath.Base(*outGo)
|
||||
if pkgName == "." || pkgName == "" {
|
||||
pkgName = msgs[0].PackageName
|
||||
}
|
||||
|
||||
src, err := generator.GenerateGoSchema(schema, pkgName)
|
||||
if err != nil {
|
||||
log.Fatalf("arpack: Go generation error: %v", err)
|
||||
}
|
||||
|
||||
outPath := filepath.Join(*outGo, baseName+"_gen.go")
|
||||
if err := os.MkdirAll(*outGo, 0755); err != nil {
|
||||
log.Fatalf("arpack: mkdir %s: %v", *outGo, err)
|
||||
}
|
||||
if err := os.WriteFile(outPath, src, 0644); err != nil {
|
||||
log.Fatalf("arpack: write %s: %v", outPath, err)
|
||||
}
|
||||
|
||||
fmt.Printf("arpack: wrote %s\n", outPath)
|
||||
}
|
||||
|
||||
if *outCS != "" {
|
||||
src, err := generator.GenerateCSharpSchema(schema, *namespace)
|
||||
if err != nil {
|
||||
log.Fatalf("arpack: C# generation error: %v", err)
|
||||
}
|
||||
|
||||
outPath := filepath.Join(*outCS, toTitle(baseName)+".gen.cs")
|
||||
if err := os.MkdirAll(*outCS, 0755); err != nil {
|
||||
log.Fatalf("arpack: mkdir %s: %v", *outCS, err)
|
||||
}
|
||||
if err := os.WriteFile(outPath, src, 0644); err != nil {
|
||||
log.Fatalf("arpack: write %s: %v", outPath, err)
|
||||
}
|
||||
|
||||
fmt.Printf("arpack: wrote %s\n", outPath)
|
||||
}
|
||||
}
|
||||
|
||||
func toTitle(s string) string {
|
||||
return strings.ToUpper(s[:1]) + strings.ToLower(s[1:])
|
||||
}
|
||||
Reference in New Issue
Block a user