fix: reject non-YAML extensions for --sqlconfig flag

Author: dlevy-msft-sql

README.md

@@ -84,6 +84,17 @@ sqlcmd config connection-strings
 sqlcmd config view
 ```
 
+#### Custom configuration files
+
+The `--sqlconfig` flag specifies a custom configuration file. The file must be YAML format with a `.yaml` or `.yml` extension:
+
+```
+sqlcmd config --sqlconfig ./myproject.yaml add-endpoint --name ep1434 --address localhost --port 1434
+sqlcmd config --sqlconfig ./myproject.yaml view
+```
+
+The default file (`~/.sqlcmd/sqlconfig`) has no extension and is also YAML.
+
 ### Versions
 
 To see all version tags to choose from (2017, 2019, 2022 etc.), and install a specific version, run:

internal/config/config.go

@@ -4,13 +4,16 @@
 package config
 
 import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
 	. "github.com/microsoft/go-sqlcmd/cmd/modern/sqlconfig"
 	"github.com/microsoft/go-sqlcmd/internal/io/file"
 	"github.com/microsoft/go-sqlcmd/internal/io/folder"
 	"github.com/microsoft/go-sqlcmd/internal/pal"
-	"os"
-	"path/filepath"
-	"testing"
 )
 
 var config Sqlconfig
@@ -24,6 +27,11 @@ func SetFileName(name string) {
 		panic("name is empty")
 	}
 
+	if ext := strings.ToLower(filepath.Ext(name)); ext != "" && ext != ".yaml" && ext != ".yml" {
+		checkErr(fmt.Errorf("configuration file %q has unsupported extension %q (must be .yaml or .yml)", name, ext))
+		return
+	}
+
 	filename = name
 
 	file.CreateEmptyIfNotExists(filename)

internal/config/config_test.go

@@ -380,6 +380,30 @@ func TestConfig_GetCurrentContextEndPointNotFoundPanic(t *testing.T) {
 	})
 }
 
+func TestSetFileName_RejectsNonYAMLExtension(t *testing.T) {
+	var gotErr error
+	originalCallback := errorCallback
+	errorCallback = func(err error) { gotErr = err }
+	t.Cleanup(func() { errorCallback = originalCallback })
+
+	SetFileName("config.json")
+	assert.Error(t, gotErr)
+	assert.Contains(t, gotErr.Error(), ".json")
+
+	gotErr = nil
+	SetFileName("config.toml")
+	assert.Error(t, gotErr)
+	assert.Contains(t, gotErr.Error(), ".toml")
+
+	gotErr = nil
+	SetFileName(pal.FilenameInUserHomeDotDirectory(".sqlcmd", "sqlconfig-test.yaml"))
+	assert.NoError(t, gotErr)
+
+	gotErr = nil
+	SetFileName(pal.FilenameInUserHomeDotDirectory(".sqlcmd", "sqlconfig-test.yml"))
+	assert.NoError(t, gotErr)
+}
+
 func TestConfig_DeleteContextThatDoesNotExist(t *testing.T) {
 	assert.Panics(t, func() {
 		contextOrdinal("does-not-exist")