1 files changed, 173 insertions, 0 deletions
diff --git a/cmd/binapi-generator/main.go b/cmd/binapi-generator/main.go
new file mode 100644
index 0000000..8045212
--- /dev/null
+++ b/cmd/binapi-generator/main.go
@@ -0,0 +1,173 @@
+// Copyright (c) 2018 Cisco and/or its affiliates.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at:
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package main
+
+import (
+	"bufio"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"github.com/bennyscetbun/jsongo"
+)
+
+var (
+	inputFile       = flag.String("input-file", "", "Input JSON file.")
+	inputDir        = flag.String("input-dir", ".", "Input directory with JSON files.")
+	outputDir       = flag.String("output-dir", ".", "Output directory where package folders will be generated.")
+	includeAPIVer   = flag.Bool("include-apiver", false, "Whether to include VlAPIVersion in generated file.")
+	debug           = flag.Bool("debug", false, "Turn on debug mode.")
+	continueOnError = flag.Bool("continue-onerror", false, "Wheter to continue with next file on error.")
+)
+
+func logf(f string, v ...interface{}) {
+	if *debug {
+		log.Printf(f, v...)
+	}
+}
+
+func main() {
+	flag.Parse()
+
+	if *inputFile == "" && *inputDir == "" {
+		fmt.Fprintln(os.Stderr, "ERROR: input-file or input-dir must be specified")
+		os.Exit(1)
+	}
+
+	if *inputFile != "" {
+		// process one input file
+		if err := generateFromFile(*inputFile, *outputDir); err != nil {
+			fmt.Fprintf(os.Stderr, "ERROR: code generation from %s failed: %v\n", *inputFile, err)
+			os.Exit(1)
+		}
+	} else {
+		// process all files in specified directory
+		files, err := getInputFiles(*inputDir)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "ERROR: code generation failed: %v\n", err)
+			os.Exit(1)
+		}
+		for _, file := range files {
+			if err := generateFromFile(file, *outputDir); err != nil {
+				fmt.Fprintf(os.Stderr, "ERROR: code generation from %s failed: %v\n", file, err)
+				if *continueOnError {
+					continue
+				}
+				os.Exit(1)
+			}
+		}
+	}
+}
+
+// getInputFiles returns all input files located in specified directory
+func getInputFiles(inputDir string) (res []string, err error) {
+	files, err := ioutil.ReadDir(inputDir)
+	if err != nil {
+		return nil, fmt.Errorf("reading directory %s failed: %v", inputDir, err)
+	}
+	for _, f := range files {
+		if strings.HasSuffix(f.Name(), inputFileExt) {
+			res = append(res, filepath.Join(inputDir, f.Name()))
+		}
+	}
+	return res, nil
+}
+
+// generateFromFile generates Go package from one input JSON file
+func generateFromFile(inputFile, outputDir string) error {
+	logf("generating from file: %q", inputFile)
+	defer logf("--------------------------------------")
+
+	ctx, err := getContext(inputFile, outputDir)
+	if err != nil {
+		return err
+	}
+
+	// read input file contents
+	ctx.inputData, err = readFile(inputFile)
+	if err != nil {
+		return err
+	}
+	// parse JSON data into objects
+	jsonRoot, err := parseJSON(ctx.inputData)
+	if err != nil {
+		return err
+	}
+	ctx.packageData, err = parsePackage(ctx, jsonRoot)
+	if err != nil {
+		return err
+	}
+
+	// create output directory
+	packageDir := filepath.Dir(ctx.outputFile)
+	if err := os.MkdirAll(packageDir, 0777); err != nil {
+		return fmt.Errorf("creating output directory %q failed: %v", packageDir, err)
+	}
+	// open output file
+	f, err := os.Create(ctx.outputFile)
+	if err != nil {
+		return fmt.Errorf("creating output file %q failed: %v", ctx.outputFile, err)
+	}
+	defer f.Close()
+
+	// generate Go package code
+	w := bufio.NewWriter(f)
+	if err := generatePackage(ctx, w); err != nil {
+		return err
+	}
+
+	// go format the output file (fail probably means the output is not compilable)
+	cmd := exec.Command("gofmt", "-w", ctx.outputFile)
+	if output, err := cmd.CombinedOutput(); err != nil {
+		return fmt.Errorf("gofmt failed: %v\n%s", err, string(output))
+	}
+
+	// count number of lines in generated output file
+	cmd = exec.Command("wc", "-l", ctx.outputFile)
+	if output, err := cmd.CombinedOutput(); err != nil {
+		log.Printf("wc command failed: %v\n%s", err, string(output))
+	} else {
+		logf("generated lines: %s", output)
+	}
+
+	return nil
+}
+
+// readFile reads content of a file into memory
+func readFile(inputFile string) ([]byte, error) {
+	inputData, err := ioutil.ReadFile(inputFile)
+	if err != nil {
+		return nil, fmt.Errorf("reading data from file failed: %v", err)
+	}
+
+	return inputData, nil
+}
+
+// parseJSON parses a JSON data into an in-memory tree
+func parseJSON(inputData []byte) (*jsongo.JSONNode, error) {
+	root := jsongo.JSONNode{}
+
+	if err := json.Unmarshal(inputData, &root); err != nil {
+		return nil, fmt.Errorf("unmarshalling JSON failed: %v", err)
+	}
+
+	return &root, nil
+}