feat: add participant attributes to token creation (#763)

Author: inickt

autocomplete/fish_autocomplete

@@ -296,6 +296,8 @@ complete -c lk -n '__fish_seen_subcommand_from create' -f -l identity -s i -r -d
 complete -c lk -n '__fish_seen_subcommand_from create' -f -l name -s n -r -d '`NAME` of the participant, used with --join. defaults to identity'
 complete -c lk -n '__fish_seen_subcommand_from create' -f -l room -s r -r -d '`NAME` of the room to join'
 complete -c lk -n '__fish_seen_subcommand_from create' -f -l metadata -r -d '`JSON` metadata to encode in the token, will be passed to participant'
+complete -c lk -n '__fish_seen_subcommand_from create' -f -l attribute -r -d 'set attributes in key=value format, can be used multiple times'
+complete -c lk -n '__fish_seen_subcommand_from create' -l attribute-file -r -d 'read attributes from a `JSON` file'
 complete -c lk -n '__fish_seen_subcommand_from create' -f -l valid-for -r -d '`TIME` that the token is valid for, e.g. "5m", "1h10m" (s: seconds, m: minutes, h: hours)'
 complete -c lk -n '__fish_seen_subcommand_from create' -f -l grant -r -d 'Additional `VIDEO_GRANT` fields. It\'ll be merged with other arguments (JSON formatted)'
 complete -c lk -n '__fish_seen_subcommand_from create' -f -l help -s h -d 'show help'
@@ -315,6 +317,8 @@ complete -c lk -n '__fish_seen_subcommand_from create-token' -f -l name -s n -r
 complete -c lk -n '__fish_seen_subcommand_from create-token' -f -l room -s r -r -d '`NAME` of the room to join'
 complete -c lk -n '__fish_seen_subcommand_from create-token' -f -l room-configuration -r -d 'name of the room configuration to use when creating a room'
 complete -c lk -n '__fish_seen_subcommand_from create-token' -f -l metadata -r -d '`JSON` metadata to encode in the token, will be passed to participant'
+complete -c lk -n '__fish_seen_subcommand_from create-token' -f -l attribute -r -d 'set attributes in key=value format, can be used multiple times'
+complete -c lk -n '__fish_seen_subcommand_from create-token' -l attribute-file -r -d 'read attributes from a `JSON` file'
 complete -c lk -n '__fish_seen_subcommand_from create-token' -f -l valid-for -r -d 'Amount of `TIME` that the token is valid for. i.e. "5m", "1h10m" (s: seconds, m: minutes, h: hours)'
 complete -c lk -n '__fish_seen_subcommand_from create-token' -f -l grant -r -d 'Additional `VIDEO_GRANT` fields. It\'ll be merged with other arguments (JSON formatted)'
 complete -c lk -n '__fish_seen_subcommand_from create-token' -f -l help -s h -d 'show help'

cmd/lk/token.go

@@ -19,6 +19,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"os"
 	"slices"
 	"time"
 
@@ -98,6 +99,15 @@ var (
 							Name:  "metadata",
 							Usage: "`JSON` metadata to encode in the token, will be passed to participant",
 						},
+						&cli.StringSliceFlag{
+							Name:  "attribute",
+							Usage: "set attributes in key=value format, can be used multiple times",
+						},
+						&cli.StringFlag{
+							Name:      "attribute-file",
+							Usage:     "read attributes from a `JSON` file",
+							TakesFile: true,
+						},
 						&cli.StringFlag{
 							Name:  "valid-for",
 							Usage: "`TIME` that the token is valid for, e.g. \"5m\", \"1h10m\" (s: seconds, m: minutes, h: hours)",
@@ -183,6 +193,15 @@ var (
 					Name:  "metadata",
 					Usage: "`JSON` metadata to encode in the token, will be passed to participant",
 				},
+				&cli.StringSliceFlag{
+					Name:  "attribute",
+					Usage: "set attributes in key=value format, can be used multiple times",
+				},
+				&cli.StringFlag{
+					Name:      "attribute-file",
+					Usage:     "read attributes from a `JSON` file",
+					TakesFile: true,
+				},
 				&cli.StringFlag{
 					Name:  "valid-for",
 					Usage: "Amount of `TIME` that the token is valid for. i.e. \"5m\", \"1h10m\" (s: seconds, m: minutes, h: hours)",
@@ -202,6 +221,29 @@ func createToken(ctx context.Context, c *cli.Command) error {
 	metadata := c.String("metadata")
 	validFor := c.String("valid-for")
 	roomPreset := c.String("room-preset")
+	participantAttributes, err := parseKeyValuePairs(c, "attribute")
+	if err != nil {
+		return fmt.Errorf("failed to parse participant attributes: %w", err)
+	}
+
+	if attrFile := c.String("attribute-file"); attrFile != "" {
+		fileData, err := os.ReadFile(attrFile)
+		if err != nil {
+			return fmt.Errorf("failed to read attribute file: %w", err)
+		}
+
+		var fileAttrs map[string]string
+		if err := json.Unmarshal(fileData, &fileAttrs); err != nil {
+			return fmt.Errorf("failed to parse attribute file as JSON: %w", err)
+		}
+
+		if participantAttributes == nil {
+			participantAttributes = make(map[string]string)
+		}
+		for key, value := range fileAttrs {
+			participantAttributes[key] = value
+		}
+	}
 
 	// required only for join, will be generated if not provided
 	participant := c.String("identity")
@@ -323,7 +365,7 @@ func createToken(ctx context.Context, c *cli.Command) error {
 		}
 	}
 
-	_, err := requireProjectWithOpts(ctx, c, ignoreURL)
+	_, err = requireProjectWithOpts(ctx, c, ignoreURL)
 	if err != nil {
 		return err
 	}
@@ -346,6 +388,9 @@ func createToken(ctx context.Context, c *cli.Command) error {
 	if metadata != "" {
 		at.SetMetadata(metadata)
 	}
+	if len(participantAttributes) != 0 {
+		at.SetAttributes(participantAttributes)
+	}
 	if roomPreset != "" {
 		at.SetRoomPreset(roomPreset)
 	}