add kv attrs to sip token

livekit · Jun 17, 2024 · 031b1e9 · 031b1e9
1 parent 05359df
commit 031b1e9
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/auth/accesstoken.go b/auth/accesstoken.go
@@ -72,6 +72,19 @@ func (t *AccessToken) SetMetadata(md string) *AccessToken {
 	return t
 }
 
+func (t *AccessToken) SetAttributes(attrs map[string]string) *AccessToken {
+	if len(attrs) == 0 {
+		return t
+	}
+	if t.grant.Attributes == nil {
+		t.grant.Attributes = make(map[string]string)
+	}
+	for k, v := range attrs {
+		t.grant.Attributes[k] = v
+	}
+	return t
+}
+
 func (t *AccessToken) SetSha256(sha string) *AccessToken {
 	t.grant.Sha256 = sha
 	return t

diff --git a/sip/token.go b/sip/token.go
@@ -21,7 +21,11 @@ import (
 	"github.com/livekit/protocol/livekit"
 )
 
-func BuildSIPToken(apiKey, secret, roomName, participantIdentity, participantName, participantMeta string) (string, error) {
+func BuildSIPToken(
+	apiKey, secret, roomName string,
+	participantIdentity, participantName, participantMeta string,
+	participantAttrs map[string]string,
+) (string, error) {
 	t := true
 	at := auth.NewAccessToken(apiKey, secret).
 		AddGrant(&auth.VideoGrant{
@@ -33,6 +37,7 @@ func BuildSIPToken(apiKey, secret, roomName, participantIdentity, participantNam
 		SetIdentity(participantIdentity).
 		SetName(participantName).
 		SetMetadata(participantMeta).
+		SetAttributes(participantAttrs).
 		SetKind(livekit.ParticipantInfo_SIP).
 		SetValidFor(24 * time.Hour)