From c8b627565f44a71b9c7fa1aca414ecbdfc802ab4 Mon Sep 17 00:00:00 2001
From: Kyle Isom <coder@kyleisom.net>
Date: Fri, 14 Oct 2016 09:51:29 -0700
Subject: [PATCH] jlp: support reading JSON from stdin.

---
 cmd/jlp/jlp.go | 35 +++++++++++++++++++++++++++++++----
 1 file changed, 31 insertions(+), 4 deletions(-)

diff --git a/cmd/jlp/jlp.go b/cmd/jlp/jlp.go
index c1415e9..e0bba6b 100644
--- a/cmd/jlp/jlp.go
+++ b/cmd/jlp/jlp.go
@@ -6,12 +6,21 @@ import (
 	"flag"
 	"fmt"
 	"io/ioutil"
+	"os"
 
 	"github.com/kisom/goutils/lib"
 )
 
 func prettify(file string, validateOnly bool) error {
-	in, err := ioutil.ReadFile(file)
+	var in []byte
+	var err error
+
+	if file == "-" {
+		in, err = ioutil.ReadAll(os.Stdin)
+	} else {
+		in, err = ioutil.ReadFile(file)
+	}
+
 	if err != nil {
 		lib.Warn(err, "ReadFile")
 		return err
@@ -28,7 +37,12 @@ func prettify(file string, validateOnly bool) error {
 		return nil
 	}
 
-	err = ioutil.WriteFile(file, buf.Bytes(), 0644)
+	if file == "-" {
+		_, err = os.Stdout.Write(buf.Bytes())
+	} else {
+		err = ioutil.WriteFile(file, buf.Bytes(), 0644)
+	}
+
 	if err != nil {
 		lib.Warn(err, "WriteFile")
 	}
@@ -37,7 +51,15 @@ func prettify(file string, validateOnly bool) error {
 }
 
 func compact(file string, validateOnly bool) error {
-	in, err := ioutil.ReadFile(file)
+	var in []byte
+	var err error
+
+	if file == "-" {
+		in, err = ioutil.ReadAll(os.Stdin)
+	} else {
+		in, err = ioutil.ReadFile(file)
+	}
+
 	if err != nil {
 		lib.Warn(err, "ReadFile")
 		return err
@@ -54,7 +76,12 @@ func compact(file string, validateOnly bool) error {
 		return nil
 	}
 
-	err = ioutil.WriteFile(file, buf.Bytes(), 0644)
+	if file == "-" {
+		_, err = os.Stdout.Write(buf.Bytes())
+	} else {
+		err = ioutil.WriteFile(file, buf.Bytes(), 0644)
+	}
+
 	if err != nil {
 		lib.Warn(err, "WriteFile")
 	}