Merge pull request #43 from stefannica/custom-cli

Customized goa CLI code generator

Author: stefannica

cmd/fuseml_core-cli/http.go

@@ -7,7 +7,7 @@ import (
 	"time"
 
 	cli "github.com/fuseml/fuseml-core/gen/http/cli/fuseml_core"
-	"github.com/goccy/go-yaml"
+	yaml "github.com/goccy/go-yaml"
 	goahttp "goa.design/goa/v3/http"
 	goa "goa.design/goa/v3/pkg"
 )
@@ -54,7 +54,7 @@ func responseDecoder(resp *http.Response) goahttp.Decoder {
 	case ct == "", ct == "application/x-yaml", ct == "text/x-yaml":
 		fallthrough
 	case strings.HasSuffix(ct, "+yaml"):
-		return yaml.NewDecoder(resp.Body)
+		return yaml.NewDecoder(resp.Body, yaml.Strict())
 	default:
 		return goahttp.ResponseDecoder(resp)
 	}

cmd/fuseml_core-cli/main.go

@@ -2,13 +2,14 @@ package main
 
 import (
 	"context"
-	"encoding/json"
 	"flag"
 	"fmt"
 	"net/url"
 	"os"
 	"strings"
 
+	"github.com/goccy/go-yaml"
+
 	codeset "github.com/fuseml/fuseml-core/gen/codeset"
 	gitc "github.com/fuseml/fuseml-core/pkg/client"
 	goa "goa.design/goa/v3/pkg"
@@ -92,7 +93,7 @@ func main() {
 	}
 
 	if data != nil {
-		m, _ := json.MarshalIndent(data, "", "    ")
+		m, _ := yaml.Marshal(data)
 		fmt.Println(string(m))
 	}
 	// Pushing the code must be done from client, before passing the request to the server

design/api.go

@@ -1,6 +1,7 @@
 package design
 
 import (
+	_ "github.com/fuseml/fuseml-core/design/cli"
 	. "goa.design/goa/v3/dsl"
 )