feat(coderd/httpapi): add QueryParamParser.JSONStringMap

johnstcn · johnstcn · commit 3184814c18ba · 2025-02-18T12:48:23.000Z
diff --git a/coderd/httpapi/queryparams.go b/coderd/httpapi/queryparams.go
@@ -2,6 +2,7 @@ package httpapi
 
 import (
 	"database/sql"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"net/url"
@@ -257,6 +258,23 @@ func (p *QueryParamParser) Strings(vals url.Values, def []string, queryParam str
 	})
 }
 
+func (p *QueryParamParser) JSONStringMap(vals url.Values, def map[string]string, queryParam string) map[string]string {
+	v, err := parseQueryParam(p, vals, func(v string) (map[string]string, error) {
+		var m map[string]string
+		if err := json.NewDecoder(strings.NewReader(v)).Decode(&m); err != nil {
+			return nil, err
+		}
+		return m, nil
+	}, def, queryParam)
+	if err != nil {
+		p.Errors = append(p.Errors, codersdk.ValidationError{
+			Field:  queryParam,
+			Detail: fmt.Sprintf("Query param %q must be a valid JSON object: %s", queryParam, err.Error()),
+		})
+	}
+	return v
+}
+
 // ValidEnum represents an enum that can be parsed and validated.
 type ValidEnum interface {
 	// Add more types as needed (avoid importing large dependency trees).
diff --git a/coderd/httpapi/queryparams_test.go b/coderd/httpapi/queryparams_test.go
@@ -473,6 +473,70 @@ func TestParseQueryParams(t *testing.T) {
 		testQueryParams(t, expParams, parser, parser.UUIDs)
 	})
 
+	t.Run("JSONStringMap", func(t *testing.T) {
+		t.Parallel()
+
+		expParams := []queryParamTestCase[map[string]string]{
+			{
+				QueryParam: "valid_map",
+				Value:      `{"key1": "value1", "key2": "value2"}`,
+				Expected: map[string]string{
+					"key1": "value1",
+					"key2": "value2",
+				},
+			},
+			{
+				QueryParam: "empty",
+				Value:      "{}",
+				Default:    map[string]string{},
+				Expected:   map[string]string{},
+			},
+			{
+				QueryParam: "no_value",
+				NoSet:      true,
+				Default:    map[string]string{},
+				Expected:   map[string]string{},
+			},
+			{
+				QueryParam: "default",
+				NoSet:      true,
+				Default:    map[string]string{"key": "value"},
+				Expected:   map[string]string{"key": "value"},
+			},
+			{
+				QueryParam: "null",
+				Value:      "null",
+				Expected:   map[string]string(nil),
+			},
+			{
+				QueryParam: "undefined",
+				Value:      "undefined",
+				Expected:   map[string]string(nil),
+			},
+			{
+				QueryParam:            "invalid_map",
+				Value:                 `{"key1": "value1", "key2": "value2"`, // missing closing brace
+				Expected:              map[string]string(nil),
+				Default:               map[string]string{},
+				ExpectedErrorContains: `Query param "invalid_map" must be a valid JSON object: unexpected EOF`,
+			},
+			{
+				QueryParam:            "incorrect_type",
+				Value:                 `{"key1": 1, "key2": true}`,
+				Expected:              map[string]string(nil),
+				ExpectedErrorContains: `Query param "incorrect_type" must be a valid JSON object: json: cannot unmarshal number into Go value of type string`,
+			},
+			{
+				QueryParam:            "multiple_keys",
+				Values:                []string{`{"key1": "value1"}`, `{"key2": "value2"}`},
+				Expected:              map[string]string(nil),
+				ExpectedErrorContains: `Query param "multiple_keys" provided more than once, found 2 times.`,
+			},
+		}
+		parser := httpapi.NewQueryParamParser()
+		testQueryParams(t, expParams, parser, parser.JSONStringMap)
+	})
+
 	t.Run("Required", func(t *testing.T) {
 		t.Parallel()
 
