infer,translate: implement file destinations

Author: thomasdfischer

README.md

@@ -22,7 +22,7 @@ Common uses cases:
 GraphQL schemas, protobuf schemas, YAML, TOML, and XML:
 
 ```
-$ curl https://example.com/some_endpoint | schema infer --graphql
+$ curl http://piedpiper.tld/some_endpoint | schema infer --graphql
 type People {
     age: Float!
     name: String!
@@ -36,7 +36,7 @@ type Object {
 * Omit `--graphql` to get [JSON Schema](https://json-schema.org):
 
 ```
-$ curl https://example.com/some_endpoint | schema infer
+$ curl http://piedpiper.tld/some_endpoint | schema infer
 {
     "title": "",
     "type": "object",

infer/infer.go

@@ -1,25 +1,45 @@
 package infer
 
 import (
+	"bytes"
 	"fmt"
 	"io"
+	"io/ioutil"
 	"os"
 
 	"github.com/Confbase/schema/example"
 	"github.com/Confbase/schema/jsonsch"
 	"github.com/Confbase/schema/util"
 )
 
-func InferEntry(cfg Config, args []string) {
-	if len(args) == 0 {
+func InferEntry(cfg Config, targets []string) {
+	if len(targets) == 0 {
 		if err := Infer(os.Stdin, os.Stdout, cfg); err != nil {
 			fmt.Fprintf(os.Stderr, "error: %v\n", err)
 			os.Exit(1)
 		}
-	} else {
-		fmt.Fprintf(os.Stderr, "error: not implemented yet\n")
+		return
+	}
+
+	buf, err := ioutil.ReadAll(os.Stdin)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "error: failed to read from stdin\n%v", err)
 		os.Exit(1)
 	}
+
+	for _, t := range targets {
+		f, err := os.OpenFile(t, os.O_RDWR|os.O_CREATE, 0666)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "error: failed to open '%v'\n%v\n", t, err)
+			os.Exit(1)
+		}
+		defer f.Close()
+
+		if err := Infer(bytes.NewReader(buf), f, cfg); err != nil {
+			fmt.Fprintf(os.Stderr, "error: %v\n", err)
+			os.Exit(1)
+		}
+	}
 }
 
 func Infer(r io.Reader, w io.Writer, cfg Config) error {

translate/translate.go

@@ -1,23 +1,43 @@
 package translate
 
 import (
+	"bytes"
 	"fmt"
 	"io"
+	"io/ioutil"
 	"os"
 
 	"github.com/Confbase/schema/util"
 )
 
-func TranslateEntry(cfg Config, args []string) {
-	if len(args) == 0 {
+func TranslateEntry(cfg Config, targets []string) {
+	if len(targets) == 0 {
 		if err := Translate(os.Stdin, os.Stdout, cfg); err != nil {
 			fmt.Fprintf(os.Stderr, "error: %v\n", err)
 			os.Exit(1)
 		}
-	} else {
-		fmt.Fprintf(os.Stderr, "error: not implemented yet\n")
+		return
+	}
+
+	buf, err := ioutil.ReadAll(os.Stdin)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "error: failed to read from stdin\n%v", err)
 		os.Exit(1)
 	}
+
+	for _, t := range targets {
+		f, err := os.OpenFile(t, os.O_RDWR|os.O_CREATE, 0666)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "error: failed to open '%v'\n%v\n", t, err)
+			os.Exit(1)
+		}
+		defer f.Close()
+
+		if err := Translate(bytes.NewReader(buf), f, cfg); err != nil {
+			fmt.Fprintf(os.Stderr, "error: %v\n", err)
+			os.Exit(1)
+		}
+	}
 }
 
 func Translate(r io.Reader, w io.Writer, cfg Config) error {