From 48b481b68d29238c6120bed893d96e6bc980fb96 Mon Sep 17 00:00:00 2001
From: Mike Farah <mikefarah@gmail.com>
Date: Mon, 20 Mar 2023 09:16:20 +1100
Subject: [PATCH] Unknown file type should default to yaml, Fixes #1609

---
 acceptance_tests/inputs-format-auto.sh | 17 +++++++++++++++++
 cmd/utils.go                           | 23 +++++++++++++++++++----
 2 files changed, 36 insertions(+), 4 deletions(-)

diff --git a/acceptance_tests/inputs-format-auto.sh b/acceptance_tests/inputs-format-auto.sh
index 302f3025..a5e6897c 100755
--- a/acceptance_tests/inputs-format-auto.sh
+++ b/acceptance_tests/inputs-format-auto.sh
@@ -2,6 +2,7 @@
 
 setUp() {
   rm test*.yml 2>/dev/null || true
+  rm test*.tfstate 2>/dev/null || true
   rm test*.json 2>/dev/null || true
   rm test*.properties 2>/dev/null || true
   rm test*.csv 2>/dev/null || true
@@ -29,6 +30,22 @@ EOM
   assertEquals "$expected" "$X"
 }
 
+testInputTfstate() {
+  cat >test.tfstate <<EOL
+{ "mike" : { "things": "cool" } }
+EOL
+
+  read -r -d '' expected << EOM
+{"mike": {"things": "cool"}}
+EOM
+
+  X=$(./yq test.tfstate)
+  assertEquals "$expected" "$X"
+
+  X=$(./yq ea test.tfstate)
+  assertEquals "$expected" "$X"
+}
+
 testInputJsonOutputYaml() {
   cat >test.json <<EOL
 { "mike" : { "things": "cool" } }
diff --git a/cmd/utils.go b/cmd/utils.go
index c8ba95fa..a447f050 100644
--- a/cmd/utils.go
+++ b/cmd/utils.go
@@ -10,6 +10,10 @@ import (
 	"gopkg.in/op/go-logging.v1"
 )
 
+func isAutomaticOutputFormat() bool {
+	return outputFormat == "" || outputFormat == "auto" || outputFormat == "a"
+}
+
 func initCommand(cmd *cobra.Command, args []string) (string, []string, error) {
 	cmd.SilenceUsage = true
 
@@ -60,10 +64,20 @@ func initCommand(cmd *cobra.Command, args []string) (string, []string, error) {
 	if inputFormat == "" || inputFormat == "auto" || inputFormat == "a" {
 
 		inputFormat = yqlib.FormatFromFilename(inputFilename)
-		if outputFormat == "" || outputFormat == "auto" || outputFormat == "a" {
-			outputFormat = yqlib.FormatFromFilename(inputFilename)
+
+		_, err := yqlib.InputFormatFromString(inputFormat)
+		if err != nil {
+			// unknown file type, default to yaml
+			yqlib.GetLogger().Debug("Unknown file format extension '%v', defaulting to yaml", inputFormat)
+			inputFormat = "yaml"
+			if isAutomaticOutputFormat() {
+				outputFormat = "yaml"
+			}
+		} else if isAutomaticOutputFormat() {
+			// automatic input worked, we can do it for output too unless specified
+			outputFormat = inputFormat
 		}
-	} else if outputFormat == "" || outputFormat == "auto" || outputFormat == "a" {
+	} else if isAutomaticOutputFormat() {
 		// backwards compatibility -
 		// before this was introduced, `yq -pcsv things.csv`
 		// would produce *yaml* output.
@@ -80,7 +94,8 @@ func initCommand(cmd *cobra.Command, args []string) (string, []string, error) {
 	if err != nil {
 		return "", nil, err
 	}
-	yqlib.GetLogger().Debug("Using outputformat %v", outputFormat)
+	yqlib.GetLogger().Debug("Using input format %v", inputFormat)
+	yqlib.GetLogger().Debug("Using output format %v", outputFormat)
 
 	if outputFormatType == yqlib.YamlOutputFormat ||
 		outputFormatType == yqlib.PropsOutputFormat {