refactor(render): filewriting in render pkg

main.go

@@ -1,10 +1,9 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
-	"io/ioutil"
 	"log"
-	"path/filepath"
 
 	"github.com/go-clix/cli"
 
@@ -20,19 +19,27 @@ func main() {
 		Short: "Utility to parse and transform Jsonnet code that uses the docsonnet extension",
 	}
 
-	output := root.Flags().StringP("output", "o", "docs", "directory to write the .md files to")
+	dir := root.Flags().StringP("output", "o", "docs", "directory to write the .md files to")
+	outputMd := root.Flags().Bool("md", true, "render as markdown files")
+	outputJSON := root.Flags().Bool("json", false, "print loaded docsonnet as JSON")
+
 	root.Run = func(cmd *cli.Command, args []string) error {
-		pkg, err := docsonnet.Load(args[0])
-		if err != nil {
-			return err
-		}
+		file := args[0]
 
-		data := render.Render(*pkg)
-		for k, v := range data {
-			fmt.Println(k)
-			if err := ioutil.WriteFile(filepath.Join(*output, k), []byte(v), 0644); err != nil {
+		switch {
+		case *outputJSON:
+			model, err := docsonnet.Load(file)
+			if err != nil {
 				return err
 			}
+			data, err := json.MarshalIndent(model, "", "  ")
+			if err != nil {
+				return err
+			}
+
+			fmt.Println(string(data))
+		case *outputMd:
+			return render.To(file, *dir)
 		}
 
 		return nil

pkg/docsonnet/fast.go

@@ -4,7 +4,6 @@ import (
 	"fmt"
 	"log"
 	"strings"
-	"time"
 )
 
 // load docsonnet
@@ -13,10 +12,7 @@ import (
 // - only map[string]interface{} and fields
 // - fields (#...) coming first
 func fastLoad(d DS) Package {
-	start := time.Now()
-
 	pkg := d.Package()
-	fmt.Println("load", pkg.Name)
 
 	pkg.API = make(Fields)
 	pkg.Sub = make(map[string]Package)
@@ -51,7 +47,6 @@ func fastLoad(d DS) Package {
 		}
 	}
 
-	fmt.Println("done load", pkg.Name, time.Since(start))
 	return pkg
 }
 

pkg/docsonnet/load.go

@@ -33,13 +33,13 @@ func Load(filename string) (*Package, error) {
 	// invoke load.libsonnet
 	vm.ExtCode("main", fmt.Sprintf(`(import "%s")`, filename))
 
-	log.Println("evaluating Jsonnet")
+	log.Println("Evaluating Jsonnet")
 	data, err := vm.EvaluateSnippet("load.libsonnet", string(load))
 	if err != nil {
 		return nil, err
 	}
 
-	log.Println("parsing result")
+	log.Println("Transforming result")
 	// parse the result
 	var d DS
 	if err := json.Unmarshal([]byte(data), &d); err != nil {

pkg/render/fs.go

@@ -0,0 +1,32 @@
+package render
+
+import (
+	"io/ioutil"
+	"log"
+	"os"
+	"path/filepath"
+
+	"github.com/sh0rez/docsonnet/pkg/docsonnet"
+)
+
+func To(api, out string) error {
+	pkg, err := docsonnet.Load(api)
+	if err != nil {
+		return err
+	}
+
+	if err := os.MkdirAll(out, os.ModePerm); err != nil {
+		return err
+	}
+
+	log.Println("Rendering .md files")
+	data := Render(*pkg)
+	for k, v := range data {
+		if err := ioutil.WriteFile(filepath.Join(out, k), []byte(v), 0644); err != nil {
+			return err
+		}
+	}
+
+	log.Printf("Success! Rendered %v packages from '%s' to '%s'", len(data), api, out)
+	return nil
+}