Support stdin and stdout from piped process

pkg/cmd/cmd.go

@@ -38,9 +38,32 @@ func Execute(version string) {
 	}
 }
 
+func printPlan(input string) {
+	// If we have input
+	if len(input) > 0 {
+		plan, err := parser.Parse(input)
+		if err != nil {
+			if err == parser.ErrParseFailure {
+				os.Stderr.WriteString(color.RedString("Failed to parse plan. Returning original input.\n")) // nolint: gosec
+				fmt.Println(input)
+				return
+			}
+		}
+
+		// plan will be nil if the parser panicked (potentially due to unrecognized
+		// character or sequences) so we return the original input.
+		if plan == nil {
+			os.Stderr.WriteString(color.RedString("Failed to parse plan. Returning original input.\n")) // nolint: gosec
+			fmt.Println(input)
+			return
+		}
+
+		printer.PrettyPrint(plan)
+	}
+}
+
 func runScenery(cmd *cobra.Command, args []string) {
 	var input string
-	var foundInput bool
 
 	if noColor {
 		color.NoColor = true
@@ -51,13 +74,28 @@ func runScenery(cmd *cobra.Command, args []string) {
 	if (stat.Mode() & os.ModeCharDevice) == 0 {
 		var lines []string
 		scanner := bufio.NewScanner(os.Stdin)
+		insidePlanBlock := false
+
+		// Loop through input until separator
 		for scanner.Scan() {
-			lines = append(lines, scanner.Text())
+			text := scanner.Text()
+			if strings.Contains(text, "------------------------------------------------------------------------") {
+				if insidePlanBlock {
+					// If we detect it again, its the end of the plan block
+					insidePlanBlock = false
+					input = strings.Join(lines, "\n")
+					printPlan(input)
+				} else {
+					// Otherwise we are entering the plan block
+					insidePlanBlock = true
+				}
+				fmt.Println(text)
+			} else if insidePlanBlock {
+				lines = append(lines, text)
+			} else {
+				fmt.Println(text)
+			}
 		}
-
-		input = strings.Join(lines, "\n")
-		foundInput = true
-
 	} else if len(args) == 1 {
 		fileContents, err := ioutil.ReadFile(args[0])
 		if err != nil {
@@ -66,29 +104,11 @@ func runScenery(cmd *cobra.Command, args []string) {
 		}
 
 		input = string(fileContents)
-		foundInput = true
-	}
-
-	if foundInput {
-		plan, err := parser.Parse(input)
-		if err != nil {
-			if err == parser.ErrParseFailure {
-				os.Stderr.WriteString(color.RedString("Failed to parse plan. Returning original input.\n")) // nolint: gosec
-				fmt.Println(input)
-				return
-			}
-		}
-
-		// plan will be nil if the parser panicked (potentially due to unrecognized
-		// character or sequences) so we return the original input.
-		if plan == nil {
-			os.Stderr.WriteString(color.RedString("Failed to parse plan. Returning original input.\n")) // nolint: gosec
-			fmt.Println(input)
-			return
-		}
-
-		printer.PrettyPrint(plan)
+		printPlan(input)
 	} else {
+		// If no stdin or arguments, print usage
 		cmd.Usage() // nolint: gosec
+		os.Exit(0)
 	}
+
 }

test/main.tf

@@ -0,0 +1,23 @@
+provider "aws" {}
+
+data "aws_iam_policy_document" "write" {
+  statement {
+    sid    = "ECRGetAuthorizationToken"
+    effect = "Allow"
+
+    actions = [
+      "ecr:InitiateLayerUpload",
+      "ecr:UploadLayer*",
+      "ecr:CompleteLayerUpload",
+      "ecr:PutImage",
+    ]
+
+    resources = ["*"]
+  }
+}
+
+resource "aws_iam_policy" "write" {
+  name        = "testpolicy-deleteme"
+  description = "Allow IAM Users to pull from ECR"
+  policy      = "${data.aws_iam_policy_document.write.json}"
+}

test/subproc.sh

@@ -0,0 +1,5 @@
+#/bin/bash
+
+echo "Enter some text:"
+read -r input
+echo "you typed: $input"

test/wrapper.sh

@@ -0,0 +1,12 @@
+#/bin/bash
+
+set -x
+
+pushd test
+trap popd EXIT
+
+echo "asdasd"
+
+echo "!23123"
+
+terraform plan -input=true -lock=false