mirror of
https://github.com/zed-industries/zed.git
synced 2024-12-24 17:28:40 +00:00
WIP: start on live_kit_server
This commit is contained in:
parent
caeae38e3a
commit
5d433b1666
10 changed files with 144 additions and 75 deletions
3
.gitmodules
vendored
Normal file
3
.gitmodules
vendored
Normal file
|
@ -0,0 +1,3 @@
|
|||
[submodule "crates/live_kit_server/protocol"]
|
||||
path = crates/live_kit_server/protocol
|
||||
url = https://github.com/livekit/protocol
|
16
Cargo.lock
generated
16
Cargo.lock
generated
|
@ -3203,7 +3203,11 @@ version = "0.1.0"
|
|||
dependencies = [
|
||||
"anyhow",
|
||||
"hmac 0.12.1",
|
||||
"hyper",
|
||||
"jwt",
|
||||
"prost 0.8.0",
|
||||
"prost-build",
|
||||
"prost-types 0.8.0",
|
||||
"serde",
|
||||
"sha2 0.10.6",
|
||||
]
|
||||
|
@ -4363,7 +4367,7 @@ dependencies = [
|
|||
"multimap",
|
||||
"petgraph",
|
||||
"prost 0.9.0",
|
||||
"prost-types",
|
||||
"prost-types 0.9.0",
|
||||
"regex",
|
||||
"tempfile",
|
||||
"which",
|
||||
|
@ -4395,6 +4399,16 @@ dependencies = [
|
|||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "prost-types"
|
||||
version = "0.8.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "603bbd6394701d13f3f25aada59c7de9d35a6a5887cfc156181234a44002771b"
|
||||
dependencies = [
|
||||
"bytes 1.2.1",
|
||||
"prost 0.8.0",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "prost-types"
|
||||
version = "0.9.0"
|
||||
|
|
|
@ -36,7 +36,7 @@ fn main() {
|
|||
let live_kit_secret = std::env::var("LIVE_KIT_SECRET").unwrap();
|
||||
|
||||
cx.spawn(|mut cx| async move {
|
||||
let user1_token = live_kit_server::create_token(
|
||||
let user1_token = live_kit_server::token::create(
|
||||
&live_kit_key,
|
||||
&live_kit_secret,
|
||||
"test-room",
|
||||
|
@ -46,7 +46,7 @@ fn main() {
|
|||
let room1 = Room::new();
|
||||
room1.connect(&live_kit_url, &user1_token).await.unwrap();
|
||||
|
||||
let user2_token = live_kit_server::create_token(
|
||||
let user2_token = live_kit_server::token::create(
|
||||
&live_kit_key,
|
||||
&live_kit_secret,
|
||||
"test-room",
|
||||
|
|
|
@ -12,5 +12,11 @@ doctest = false
|
|||
anyhow = "1.0.38"
|
||||
hmac = "0.12"
|
||||
jwt = "0.16"
|
||||
hyper = "0.14"
|
||||
prost = "0.8"
|
||||
prost-types = "0.8"
|
||||
serde = { version = "1.0", features = ["derive", "rc"] }
|
||||
sha2 = "0.10"
|
||||
sha2 = "0.10"
|
||||
|
||||
[build-dependencies]
|
||||
prost-build = "0.9"
|
||||
|
|
5
crates/live_kit_server/build.rs
Normal file
5
crates/live_kit_server/build.rs
Normal file
|
@ -0,0 +1,5 @@
|
|||
fn main() {
|
||||
prost_build::Config::new()
|
||||
.compile_protos(&["protocol/livekit_room.proto"], &["protocol"])
|
||||
.unwrap();
|
||||
}
|
1
crates/live_kit_server/protocol
Submodule
1
crates/live_kit_server/protocol
Submodule
|
@ -0,0 +1 @@
|
|||
Subproject commit 8645a138fb2ea72c4dab13e739b1f3c9ea29ac84
|
36
crates/live_kit_server/src/api.rs
Normal file
36
crates/live_kit_server/src/api.rs
Normal file
|
@ -0,0 +1,36 @@
|
|||
use crate::token;
|
||||
use hyper::{client::HttpConnector, Request, Uri};
|
||||
|
||||
pub struct Client {
|
||||
http: hyper::Client<HttpConnector>,
|
||||
uri: Uri,
|
||||
key: String,
|
||||
secret: String,
|
||||
}
|
||||
|
||||
impl Client {
|
||||
pub fn new(uri: Uri, key: String, secret: String) -> Self {
|
||||
assert!(uri.scheme().is_some(), "base uri must have a scheme");
|
||||
assert!(uri.authority().is_some(), "base uri must have an authority");
|
||||
Self {
|
||||
http: hyper::Client::new(),
|
||||
uri: uri,
|
||||
key,
|
||||
secret,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn create_room(&self) {
|
||||
// let mut uri = url.clone();
|
||||
// uri.set_path_and_query()
|
||||
|
||||
let uri = Uri::builder()
|
||||
.scheme(self.uri.scheme().unwrap().clone())
|
||||
.authority(self.uri.authority().unwrap().clone())
|
||||
.path_and_query("twirp/livekit.RoomService/CreateRoom")
|
||||
.build();
|
||||
|
||||
// token::create(api_key, secret_key, room_name, participant_name)
|
||||
self.http.request(req)
|
||||
}
|
||||
}
|
|
@ -1,71 +1,3 @@
|
|||
use anyhow::Result;
|
||||
use hmac::{Hmac, Mac};
|
||||
use jwt::SignWithKey;
|
||||
use serde::Serialize;
|
||||
use sha2::Sha256;
|
||||
use std::{
|
||||
ops::Add,
|
||||
time::{Duration, SystemTime, UNIX_EPOCH},
|
||||
};
|
||||
|
||||
static DEFAULT_TTL: Duration = Duration::from_secs(6 * 60 * 60); // 6 hours
|
||||
|
||||
#[derive(Default, Serialize)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
struct ClaimGrants<'a> {
|
||||
iss: &'a str,
|
||||
sub: &'a str,
|
||||
iat: u64,
|
||||
exp: u64,
|
||||
nbf: u64,
|
||||
jwtid: &'a str,
|
||||
video: VideoGrant<'a>,
|
||||
}
|
||||
|
||||
#[derive(Default, Serialize)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
struct VideoGrant<'a> {
|
||||
room_create: Option<bool>,
|
||||
room_join: Option<bool>,
|
||||
room_list: Option<bool>,
|
||||
room_record: Option<bool>,
|
||||
room_admin: Option<bool>,
|
||||
room: Option<&'a str>,
|
||||
can_publish: Option<bool>,
|
||||
can_subscribe: Option<bool>,
|
||||
can_publish_data: Option<bool>,
|
||||
hidden: Option<bool>,
|
||||
recorder: Option<bool>,
|
||||
}
|
||||
|
||||
pub fn create_token(
|
||||
api_key: &str,
|
||||
secret_key: &str,
|
||||
room_name: &str,
|
||||
participant_name: &str,
|
||||
) -> Result<String> {
|
||||
let secret_key: Hmac<Sha256> = Hmac::new_from_slice(secret_key.as_bytes())?;
|
||||
|
||||
let now = SystemTime::now();
|
||||
|
||||
let claims = ClaimGrants {
|
||||
iss: api_key,
|
||||
sub: participant_name,
|
||||
iat: now.duration_since(UNIX_EPOCH).unwrap().as_secs(),
|
||||
exp: now
|
||||
.add(DEFAULT_TTL)
|
||||
.duration_since(UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_secs(),
|
||||
nbf: 0,
|
||||
jwtid: participant_name,
|
||||
video: VideoGrant {
|
||||
room: Some(room_name),
|
||||
room_join: Some(true),
|
||||
can_publish: Some(true),
|
||||
can_subscribe: Some(true),
|
||||
..Default::default()
|
||||
},
|
||||
};
|
||||
Ok(claims.sign_with_key(&secret_key)?)
|
||||
}
|
||||
mod api;
|
||||
mod proto;
|
||||
mod token;
|
||||
|
|
1
crates/live_kit_server/src/proto.rs
Normal file
1
crates/live_kit_server/src/proto.rs
Normal file
|
@ -0,0 +1 @@
|
|||
include!(concat!(env!("OUT_DIR"), "/livekit.rs"));
|
71
crates/live_kit_server/src/token.rs
Normal file
71
crates/live_kit_server/src/token.rs
Normal file
|
@ -0,0 +1,71 @@
|
|||
use anyhow::Result;
|
||||
use hmac::{Hmac, Mac};
|
||||
use jwt::SignWithKey;
|
||||
use serde::Serialize;
|
||||
use sha2::Sha256;
|
||||
use std::{
|
||||
ops::Add,
|
||||
time::{Duration, SystemTime, UNIX_EPOCH},
|
||||
};
|
||||
|
||||
static DEFAULT_TTL: Duration = Duration::from_secs(6 * 60 * 60); // 6 hours
|
||||
|
||||
#[derive(Default, Serialize)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
struct ClaimGrants<'a> {
|
||||
iss: &'a str,
|
||||
sub: &'a str,
|
||||
iat: u64,
|
||||
exp: u64,
|
||||
nbf: u64,
|
||||
jwtid: &'a str,
|
||||
video: VideoGrant<'a>,
|
||||
}
|
||||
|
||||
#[derive(Default, Serialize)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
struct VideoGrant<'a> {
|
||||
room_create: Option<bool>,
|
||||
room_join: Option<bool>,
|
||||
room_list: Option<bool>,
|
||||
room_record: Option<bool>,
|
||||
room_admin: Option<bool>,
|
||||
room: Option<&'a str>,
|
||||
can_publish: Option<bool>,
|
||||
can_subscribe: Option<bool>,
|
||||
can_publish_data: Option<bool>,
|
||||
hidden: Option<bool>,
|
||||
recorder: Option<bool>,
|
||||
}
|
||||
|
||||
pub fn create(
|
||||
api_key: &str,
|
||||
secret_key: &str,
|
||||
room_name: &str,
|
||||
participant_name: &str,
|
||||
) -> Result<String> {
|
||||
let secret_key: Hmac<Sha256> = Hmac::new_from_slice(secret_key.as_bytes())?;
|
||||
|
||||
let now = SystemTime::now();
|
||||
|
||||
let claims = ClaimGrants {
|
||||
iss: api_key,
|
||||
sub: participant_name,
|
||||
iat: now.duration_since(UNIX_EPOCH).unwrap().as_secs(),
|
||||
exp: now
|
||||
.add(DEFAULT_TTL)
|
||||
.duration_since(UNIX_EPOCH)
|
||||
.unwrap()
|
||||
.as_secs(),
|
||||
nbf: 0,
|
||||
jwtid: participant_name,
|
||||
video: VideoGrant {
|
||||
room: Some(room_name),
|
||||
room_join: Some(true),
|
||||
can_publish: Some(true),
|
||||
can_subscribe: Some(true),
|
||||
..Default::default()
|
||||
},
|
||||
};
|
||||
Ok(claims.sign_with_key(&secret_key)?)
|
||||
}
|
Loading…
Reference in a new issue