From 058d427fa60143347d13eeb187a7978e67bfe7ff Mon Sep 17 00:00:00 2001
From: Maisem Ali <maisem@tailscale.com>
Date: Sat, 29 Jul 2023 18:45:29 -0700
Subject: [PATCH] tailcfg: add helper to unmarshal PeerCap values

Updates #4217

Signed-off-by: Maisem Ali <maisem@tailscale.com>
---
 tailcfg/tailcfg.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/tailcfg/tailcfg.go b/tailcfg/tailcfg.go
index 505ccc133..68fec3e03 100644
--- a/tailcfg/tailcfg.go
+++ b/tailcfg/tailcfg.go
@@ -1223,6 +1223,25 @@ const (
 // the application via the WhoIs API.
 type PeerCapMap map[PeerCapability][]json.RawMessage
 
+// UnmarshalCapJSON unmarshals each JSON value in cm[cap] as T.
+// If cap does not exist in cm, it returns (nil, nil).
+// It returns an error if the values cannot be unmarshaled into the provided type.
+func UnmarshalCapJSON[T any](cm PeerCapMap, cap PeerCapability) ([]T, error) {
+	vals, ok := cm[cap]
+	if !ok {
+		return nil, nil
+	}
+	out := make([]T, 0, len(vals))
+	for _, v := range vals {
+		var t T
+		if err := json.Unmarshal(v, &t); err != nil {
+			return nil, err
+		}
+		out = append(out, t)
+	}
+	return out, nil
+}
+
 // HasCapability reports whether c has the capability cap.
 // This is used to test for the existence of a capability, especially
 // when the capability has no values.