add attributes to Claims and AccessToken

Author: spencerbart

livekit-api/src/access_token.rs

@@ -13,10 +13,7 @@
 // limitations under the License.
 
 use std::{
-    env,
-    fmt::Debug,
-    ops::Add,
-    time::{Duration, SystemTime, UNIX_EPOCH},
+    collections::HashMap, env, fmt::Debug, ops::Add, time::{Duration, SystemTime, UNIX_EPOCH}
 };
 
 use jsonwebtoken::{self, DecodingKey, EncodingKey, Header};
@@ -146,6 +143,7 @@ pub struct Claims {
     pub sip: SIPGrants,
     pub sha256: String, // Used to verify the integrity of the message body
     pub metadata: String,
+    pub attributes: HashMap<String, String>,
     pub room_config: Option<livekit_protocol::RoomConfiguration>,
 }
 
@@ -182,6 +180,7 @@ impl AccessToken {
                 sip: SIPGrants::default(),
                 sha256: Default::default(),
                 metadata: Default::default(),
+                attributes: HashMap::new(),
                 room_config: Default::default(),
             },
         }
@@ -229,6 +228,11 @@ impl AccessToken {
         self
     }
 
+    pub fn with_attributes(mut self, attributes: HashMap<String, String>) -> Self {
+        self.claims.attributes = attributes;
+        self
+    }
+
     pub fn with_sha256(mut self, sha256: &str) -> Self {
         self.claims.sha256 = sha256.to_owned();
         self