Fix: Make colon optional for reset (#559)

Author: shueybubbles

pkg/sqlcmd/commands.go

@@ -77,7 +77,7 @@ func newCommands() Commands {
 			name:   "LISTVAR",
 		},
 		"RESET": {
-			regex:  regexp.MustCompile(`(?im)^[ \t]*:RESET(?:[ \t]+(.*$)|$)`),
+			regex:  regexp.MustCompile(`(?im)^[ \t]*?:?RESET(?:[ \t]+(.*$)|$)`),
 			action: resetCommand,
 			name:   "RESET",
 		},

pkg/sqlcmd/commands_test.go

@@ -53,6 +53,8 @@ func TestCommandParsing(t *testing.T) {
 		{` !! dir c:\`, "EXEC", []string{` dir c:\`}},
 		{`!!dir c:\`, "EXEC", []string{`dir c:\`}},
 		{`:XML ON `, "XML", []string{`ON `}},
+		{`:RESET`, "RESET", []string{""}},
+		{`RESET`, "RESET", []string{""}},
 	}
 
 	for _, test := range commands {