92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"text/template"
|
||
|
|
"unicode"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
var (
|
||
|
|
enbasCLISchemaFilepath string
|
||
|
|
executorsFilePath string
|
||
|
|
)
|
||
|
|
|
||
|
|
flag.StringVar(&enbasCLISchemaFilepath, "path-to-enbas-cli-schema", "", "The path to the Enbas CLI schema file")
|
||
|
|
flag.StringVar(&executorsFilePath, "path-to-enbas-executors", "", "The path to the executors Go file")
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
schema, err := readSchemaFile(enbasCLISchemaFilepath)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("ERROR: Unable to read the schema file: %v.\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := generateExecutors(schema, executorsFilePath); err != nil {
|
||
|
|
fmt.Printf("ERROR: Unable to generate the executors: %v.\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := runGoImports(executorsFilePath); err != nil {
|
||
|
|
fmt.Printf("ERROR: Unable to run goimports: %v.\n", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func readSchemaFile(path string) (enbasCLISchema, error) {
|
||
|
|
file, err := os.Open(path)
|
||
|
|
if err != nil {
|
||
|
|
return enbasCLISchema{}, fmt.Errorf("unable to open the schema file: %w", err)
|
||
|
|
}
|
||
|
|
defer file.Close()
|
||
|
|
|
||
|
|
var schema enbasCLISchema
|
||
|
|
|
||
|
|
if err := json.NewDecoder(file).Decode(&schema); err != nil {
|
||
|
|
return enbasCLISchema{}, fmt.Errorf("unable to decode the JSON data: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return schema, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func generateExecutors(schema enbasCLISchema, output string) error {
|
||
|
|
funcMap := template.FuncMap{
|
||
|
|
"capitalise": capitalise,
|
||
|
|
"getFlagType": schema.Flags.getType,
|
||
|
|
"getFlagDescription": schema.Flags.getDescription,
|
||
|
|
}
|
||
|
|
|
||
|
|
tmpl := template.Must(template.New("executor-template").Funcs(funcMap).Parse(executorsFileTemplate))
|
||
|
|
|
||
|
|
file, err := os.Create(output)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("unable to create the output file: %w", err)
|
||
|
|
}
|
||
|
|
defer file.Close()
|
||
|
|
|
||
|
|
if err := tmpl.Execute(file, schema.Commands); err != nil {
|
||
|
|
return fmt.Errorf("unable to generate the code from the template: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func runGoImports(path string) error {
|
||
|
|
imports := exec.Command("goimports", "-w", path)
|
||
|
|
|
||
|
|
if err := imports.Run(); err != nil {
|
||
|
|
return fmt.Errorf("received an error after running goimports: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func capitalise(str string) string {
|
||
|
|
runes := []rune(str)
|
||
|
|
|
||
|
|
runes[0] = unicode.ToUpper(runes[0])
|
||
|
|
|
||
|
|
return string(runes)
|
||
|
|
}
|