zed/crates/live_kit_server/src/api.rs

use crate::{proto, token};
use anyhow::{anyhow, Result};
use prost::Message;
use reqwest::header::CONTENT_TYPE;
use std::{future::Future, sync::Arc};

#[derive(Clone)]
pub struct Client {
    http: reqwest::Client,
    url: Arc<str>,
    key: Arc<str>,
    secret: Arc<str>,
}

impl Client {
    pub fn new(mut url: String, key: String, secret: String) -> Self {
        if url.ends_with('/') {
            url.pop();
        }

        Self {
            http: reqwest::Client::new(),
            url: url.into(),
            key: key.into(),
            secret: secret.into(),
        }
    }

    pub fn url(&self) -> &str {
        &self.url
    }

    pub fn create_room(&self, name: String) -> impl Future<Output = Result<proto::Room>> {
        self.request(
            "twirp/livekit.RoomService/CreateRoom",
            token::VideoGrant {
                room_create: Some(true),
                ..Default::default()
            },
            proto::CreateRoomRequest {
                name,
                ..Default::default()
            },
        )
    }

    pub fn delete_room(&self, name: String) -> impl Future<Output = Result<()>> {
        let response = self.request(
            "twirp/livekit.RoomService/DeleteRoom",
            token::VideoGrant {
                room_create: Some(true),
                ..Default::default()
            },
            proto::DeleteRoomRequest { room: name },
        );
        async move {
            let _: proto::DeleteRoomResponse = response.await?;
            Ok(())
        }
    }

    pub fn remove_participant(
        &self,
        room: String,
        identity: String,
    ) -> impl Future<Output = Result<()>> {
        let response = self.request(
            "twirp/livekit.RoomService/RemoveParticipant",
            token::VideoGrant {
                room_admin: Some(true),
                room: Some(&room),
                ..Default::default()
            },
            proto::RoomParticipantIdentity {
                room: room.clone(),
                identity,
            },
        );
        async move {
            let _: proto::RemoveParticipantResponse = response.await?;
            Ok(())
        }
    }

    pub fn room_token(&self, room: &str, identity: &str) -> Result<String> {
        token::create(
            &self.key,
            &self.secret,
            Some(identity),
            token::VideoGrant {
                room: Some(room),
                room_join: Some(true),
                can_publish: Some(true),
                can_subscribe: Some(true),
                ..Default::default()
            },
        )
    }

    fn request<Req, Res>(
        &self,
        path: &str,
        grant: token::VideoGrant,
        body: Req,
    ) -> impl Future<Output = Result<Res>>
    where
        Req: Message,
        Res: Default + Message,
    {
        let client = self.http.clone();
        let token = token::create(&self.key, &self.secret, None, grant);
        let url = format!("{}/{}", self.url, path);
        async move {
            let token = token?;
            let response = client
                .post(&url)
                .header(CONTENT_TYPE, "application/protobuf")
                .bearer_auth(token)
                .body(body.encode_to_vec())
                .send()
                .await?;
            if response.status().is_success() {
                Ok(Res::decode(response.bytes().await?)?)
            } else {
                Err(anyhow!(
                    "POST {} failed with status code {:?}, {:?}",
                    url,
                    response.status(),
                    response.text().await
                ))
            }
        }
    }
}