698 lines
22 KiB
Rust
698 lines
22 KiB
Rust
pub mod call_settings;
|
|
pub mod participant;
|
|
pub mod room;
|
|
|
|
use anyhow::{anyhow, Result};
|
|
use audio::Audio;
|
|
use call_settings::CallSettings;
|
|
use client::{proto, Client, TelemetrySettings, TypedEnvelope, User, UserStore, ZED_ALWAYS_ACTIVE};
|
|
use collections::HashSet;
|
|
use futures::{channel::oneshot, future::Shared, Future, FutureExt};
|
|
use gpui::{
|
|
AppContext, AsyncAppContext, Context, EventEmitter, Model, ModelContext, Subscription, Task,
|
|
WeakModel,
|
|
};
|
|
use postage::watch;
|
|
use project::Project;
|
|
use room::Event;
|
|
use settings::Settings;
|
|
use std::sync::Arc;
|
|
|
|
pub use participant::ParticipantLocation;
|
|
pub use room::Room;
|
|
|
|
pub fn init(client: Arc<Client>, user_store: Model<UserStore>, cx: &mut AppContext) {
|
|
CallSettings::register(cx);
|
|
|
|
let active_call = cx.build_model(|cx| ActiveCall::new(client, user_store, cx));
|
|
cx.set_global(active_call);
|
|
}
|
|
|
|
pub struct OneAtATime {
|
|
cancel: Option<oneshot::Sender<()>>,
|
|
}
|
|
|
|
impl OneAtATime {
|
|
/// spawn a task in the given context.
|
|
/// if another task is spawned before that resolves, or if the OneAtATime itself is dropped, the first task will be cancelled and return Ok(None)
|
|
/// otherwise you'll see the result of the task.
|
|
fn spawn<F, Fut, R>(&mut self, cx: &mut AppContext, f: F) -> Task<Result<Option<R>>>
|
|
where
|
|
F: 'static + FnOnce(AsyncAppContext) -> Fut,
|
|
Fut: Future<Output = Result<R>>,
|
|
R: 'static,
|
|
{
|
|
let (tx, rx) = oneshot::channel();
|
|
self.cancel.replace(tx);
|
|
cx.spawn(|cx| async move {
|
|
futures::select_biased! {
|
|
_ = rx.fuse() => Ok(None),
|
|
result = f(cx).fuse() => result.map(Some),
|
|
}
|
|
})
|
|
}
|
|
|
|
fn running(&self) -> bool {
|
|
self.cancel
|
|
.as_ref()
|
|
.is_some_and(|cancel| !cancel.is_canceled())
|
|
}
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct IncomingCall {
|
|
pub room_id: u64,
|
|
pub calling_user: Arc<User>,
|
|
pub participants: Vec<Arc<User>>,
|
|
pub initial_project: Option<proto::ParticipantProject>,
|
|
}
|
|
|
|
/// Singleton global maintaining the user's participation in a room across workspaces.
|
|
pub struct ActiveCall {
|
|
room: Option<(Model<Room>, Vec<Subscription>)>,
|
|
pending_room_creation: Option<Shared<Task<Result<Model<Room>, Arc<anyhow::Error>>>>>,
|
|
location: Option<WeakModel<Project>>,
|
|
_join_debouncer: OneAtATime,
|
|
pending_invites: HashSet<u64>,
|
|
incoming_call: (
|
|
watch::Sender<Option<IncomingCall>>,
|
|
watch::Receiver<Option<IncomingCall>>,
|
|
),
|
|
client: Arc<Client>,
|
|
user_store: Model<UserStore>,
|
|
_subscriptions: Vec<client::Subscription>,
|
|
}
|
|
|
|
impl EventEmitter<Event> for ActiveCall {}
|
|
|
|
impl ActiveCall {
|
|
fn new(client: Arc<Client>, user_store: Model<UserStore>, cx: &mut ModelContext<Self>) -> Self {
|
|
Self {
|
|
room: None,
|
|
pending_room_creation: None,
|
|
location: None,
|
|
pending_invites: Default::default(),
|
|
incoming_call: watch::channel(),
|
|
_join_debouncer: OneAtATime { cancel: None },
|
|
_subscriptions: vec![
|
|
client.add_request_handler(cx.weak_model(), Self::handle_incoming_call),
|
|
client.add_message_handler(cx.weak_model(), Self::handle_call_canceled),
|
|
],
|
|
client,
|
|
user_store,
|
|
}
|
|
}
|
|
|
|
pub fn channel_id(&self, cx: &AppContext) -> Option<u64> {
|
|
self.room()?.read(cx).channel_id()
|
|
}
|
|
|
|
async fn handle_incoming_call(
|
|
this: Model<Self>,
|
|
envelope: TypedEnvelope<proto::IncomingCall>,
|
|
_: Arc<Client>,
|
|
mut cx: AsyncAppContext,
|
|
) -> Result<proto::Ack> {
|
|
let user_store = this.update(&mut cx, |this, _| this.user_store.clone())?;
|
|
let call = IncomingCall {
|
|
room_id: envelope.payload.room_id,
|
|
participants: user_store
|
|
.update(&mut cx, |user_store, cx| {
|
|
user_store.get_users(envelope.payload.participant_user_ids, cx)
|
|
})?
|
|
.await?,
|
|
calling_user: user_store
|
|
.update(&mut cx, |user_store, cx| {
|
|
user_store.get_user(envelope.payload.calling_user_id, cx)
|
|
})?
|
|
.await?,
|
|
initial_project: envelope.payload.initial_project,
|
|
};
|
|
this.update(&mut cx, |this, _| {
|
|
*this.incoming_call.0.borrow_mut() = Some(call);
|
|
})?;
|
|
|
|
Ok(proto::Ack {})
|
|
}
|
|
|
|
async fn handle_call_canceled(
|
|
this: Model<Self>,
|
|
envelope: TypedEnvelope<proto::CallCanceled>,
|
|
_: Arc<Client>,
|
|
mut cx: AsyncAppContext,
|
|
) -> Result<()> {
|
|
this.update(&mut cx, |this, _| {
|
|
let mut incoming_call = this.incoming_call.0.borrow_mut();
|
|
if incoming_call
|
|
.as_ref()
|
|
.map_or(false, |call| call.room_id == envelope.payload.room_id)
|
|
{
|
|
incoming_call.take();
|
|
}
|
|
})?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn global(cx: &AppContext) -> Model<Self> {
|
|
cx.global::<Model<Self>>().clone()
|
|
}
|
|
|
|
pub fn invite(
|
|
&mut self,
|
|
called_user_id: u64,
|
|
initial_project: Option<Model<Project>>,
|
|
cx: &mut ModelContext<Self>,
|
|
) -> Task<Result<()>> {
|
|
if !self.pending_invites.insert(called_user_id) {
|
|
return Task::ready(Err(anyhow!("user was already invited")));
|
|
}
|
|
cx.notify();
|
|
|
|
if self._join_debouncer.running() {
|
|
return Task::ready(Ok(()));
|
|
}
|
|
|
|
let room = if let Some(room) = self.room().cloned() {
|
|
Some(Task::ready(Ok(room)).shared())
|
|
} else {
|
|
self.pending_room_creation.clone()
|
|
};
|
|
|
|
let invite = if let Some(room) = room {
|
|
cx.spawn(move |_, mut cx| async move {
|
|
let room = room.await.map_err(|err| anyhow!("{:?}", err))?;
|
|
|
|
let initial_project_id = if let Some(initial_project) = initial_project {
|
|
Some(
|
|
room.update(&mut cx, |room, cx| room.share_project(initial_project, cx))?
|
|
.await?,
|
|
)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
room.update(&mut cx, move |room, cx| {
|
|
room.call(called_user_id, initial_project_id, cx)
|
|
})?
|
|
.await?;
|
|
|
|
anyhow::Ok(())
|
|
})
|
|
} else {
|
|
let client = self.client.clone();
|
|
let user_store = self.user_store.clone();
|
|
let room = cx
|
|
.spawn(move |this, mut cx| async move {
|
|
let create_room = async {
|
|
let room = cx
|
|
.update(|cx| {
|
|
Room::create(
|
|
called_user_id,
|
|
initial_project,
|
|
client,
|
|
user_store,
|
|
cx,
|
|
)
|
|
})?
|
|
.await?;
|
|
|
|
this.update(&mut cx, |this, cx| this.set_room(Some(room.clone()), cx))?
|
|
.await?;
|
|
|
|
anyhow::Ok(room)
|
|
};
|
|
|
|
let room = create_room.await;
|
|
this.update(&mut cx, |this, _| this.pending_room_creation = None)?;
|
|
room.map_err(Arc::new)
|
|
})
|
|
.shared();
|
|
self.pending_room_creation = Some(room.clone());
|
|
cx.background_executor().spawn(async move {
|
|
room.await.map_err(|err| anyhow!("{:?}", err))?;
|
|
anyhow::Ok(())
|
|
})
|
|
};
|
|
|
|
cx.spawn(move |this, mut cx| async move {
|
|
let result = invite.await;
|
|
if result.is_ok() {
|
|
this.update(&mut cx, |this, cx| this.report_call_event("invite", cx))?;
|
|
} else {
|
|
// TODO: Resport collaboration error
|
|
}
|
|
|
|
this.update(&mut cx, |this, cx| {
|
|
this.pending_invites.remove(&called_user_id);
|
|
cx.notify();
|
|
})?;
|
|
result
|
|
})
|
|
}
|
|
|
|
pub fn cancel_invite(
|
|
&mut self,
|
|
called_user_id: u64,
|
|
cx: &mut ModelContext<Self>,
|
|
) -> Task<Result<()>> {
|
|
let room_id = if let Some(room) = self.room() {
|
|
room.read(cx).id()
|
|
} else {
|
|
return Task::ready(Err(anyhow!("no active call")));
|
|
};
|
|
|
|
let client = self.client.clone();
|
|
cx.background_executor().spawn(async move {
|
|
client
|
|
.request(proto::CancelCall {
|
|
room_id,
|
|
called_user_id,
|
|
})
|
|
.await?;
|
|
anyhow::Ok(())
|
|
})
|
|
}
|
|
|
|
pub fn incoming(&self) -> watch::Receiver<Option<IncomingCall>> {
|
|
self.incoming_call.1.clone()
|
|
}
|
|
|
|
pub fn accept_incoming(&mut self, cx: &mut ModelContext<Self>) -> Task<Result<()>> {
|
|
if self.room.is_some() {
|
|
return Task::ready(Err(anyhow!("cannot join while on another call")));
|
|
}
|
|
|
|
let call = if let Some(call) = self.incoming_call.1.borrow().clone() {
|
|
call
|
|
} else {
|
|
return Task::ready(Err(anyhow!("no incoming call")));
|
|
};
|
|
|
|
if self.pending_room_creation.is_some() {
|
|
return Task::ready(Ok(()));
|
|
}
|
|
|
|
let room_id = call.room_id.clone();
|
|
let client = self.client.clone();
|
|
let user_store = self.user_store.clone();
|
|
let join = self
|
|
._join_debouncer
|
|
.spawn(cx, move |cx| Room::join(room_id, client, user_store, cx));
|
|
|
|
cx.spawn(|this, mut cx| async move {
|
|
let room = join.await?;
|
|
this.update(&mut cx, |this, cx| this.set_room(room.clone(), cx))?
|
|
.await?;
|
|
this.update(&mut cx, |this, cx| {
|
|
this.report_call_event("accept incoming", cx)
|
|
})?;
|
|
Ok(())
|
|
})
|
|
}
|
|
|
|
pub fn decline_incoming(&mut self, cx: &mut ModelContext<Self>) -> Result<()> {
|
|
let call = self
|
|
.incoming_call
|
|
.0
|
|
.borrow_mut()
|
|
.take()
|
|
.ok_or_else(|| anyhow!("no incoming call"))?;
|
|
report_call_event_for_room("decline incoming", call.room_id, None, &self.client, cx);
|
|
self.client.send(proto::DeclineCall {
|
|
room_id: call.room_id,
|
|
})?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn join_channel(
|
|
&mut self,
|
|
channel_id: u64,
|
|
cx: &mut ModelContext<Self>,
|
|
) -> Task<Result<Option<Model<Room>>>> {
|
|
if let Some(room) = self.room().cloned() {
|
|
if room.read(cx).channel_id() == Some(channel_id) {
|
|
return Task::ready(Ok(Some(room)));
|
|
} else {
|
|
room.update(cx, |room, cx| room.clear_state(cx));
|
|
}
|
|
}
|
|
|
|
if self.pending_room_creation.is_some() {
|
|
return Task::ready(Ok(None));
|
|
}
|
|
|
|
let client = self.client.clone();
|
|
let user_store = self.user_store.clone();
|
|
let join = self._join_debouncer.spawn(cx, move |cx| async move {
|
|
Room::join_channel(channel_id, client, user_store, cx).await
|
|
});
|
|
|
|
cx.spawn(|this, mut cx| async move {
|
|
let room = join.await?;
|
|
this.update(&mut cx, |this, cx| this.set_room(room.clone(), cx))?
|
|
.await?;
|
|
this.update(&mut cx, |this, cx| {
|
|
this.report_call_event("join channel", cx)
|
|
})?;
|
|
Ok(room)
|
|
})
|
|
}
|
|
|
|
pub fn hang_up(&mut self, cx: &mut ModelContext<Self>) -> Task<Result<()>> {
|
|
cx.notify();
|
|
self.report_call_event("hang up", cx);
|
|
|
|
Audio::end_call(cx);
|
|
if let Some((room, _)) = self.room.take() {
|
|
room.update(cx, |room, cx| room.leave(cx))
|
|
} else {
|
|
Task::ready(Ok(()))
|
|
}
|
|
}
|
|
|
|
pub fn share_project(
|
|
&mut self,
|
|
project: Model<Project>,
|
|
cx: &mut ModelContext<Self>,
|
|
) -> Task<Result<u64>> {
|
|
if let Some((room, _)) = self.room.as_ref() {
|
|
self.report_call_event("share project", cx);
|
|
room.update(cx, |room, cx| room.share_project(project, cx))
|
|
} else {
|
|
Task::ready(Err(anyhow!("no active call")))
|
|
}
|
|
}
|
|
|
|
pub fn unshare_project(
|
|
&mut self,
|
|
project: Model<Project>,
|
|
cx: &mut ModelContext<Self>,
|
|
) -> Result<()> {
|
|
if let Some((room, _)) = self.room.as_ref() {
|
|
self.report_call_event("unshare project", cx);
|
|
room.update(cx, |room, cx| room.unshare_project(project, cx))
|
|
} else {
|
|
Err(anyhow!("no active call"))
|
|
}
|
|
}
|
|
|
|
pub fn location(&self) -> Option<&WeakModel<Project>> {
|
|
self.location.as_ref()
|
|
}
|
|
|
|
pub fn set_location(
|
|
&mut self,
|
|
project: Option<&Model<Project>>,
|
|
cx: &mut ModelContext<Self>,
|
|
) -> Task<Result<()>> {
|
|
if project.is_some() || !*ZED_ALWAYS_ACTIVE {
|
|
self.location = project.map(|project| project.downgrade());
|
|
if let Some((room, _)) = self.room.as_ref() {
|
|
return room.update(cx, |room, cx| room.set_location(project, cx));
|
|
}
|
|
}
|
|
Task::ready(Ok(()))
|
|
}
|
|
|
|
fn set_room(
|
|
&mut self,
|
|
room: Option<Model<Room>>,
|
|
cx: &mut ModelContext<Self>,
|
|
) -> Task<Result<()>> {
|
|
if room.as_ref() != self.room.as_ref().map(|room| &room.0) {
|
|
cx.notify();
|
|
if let Some(room) = room {
|
|
if room.read(cx).status().is_offline() {
|
|
self.room = None;
|
|
Task::ready(Ok(()))
|
|
} else {
|
|
let subscriptions = vec![
|
|
cx.observe(&room, |this, room, cx| {
|
|
if room.read(cx).status().is_offline() {
|
|
this.set_room(None, cx).detach_and_log_err(cx);
|
|
}
|
|
|
|
cx.notify();
|
|
}),
|
|
cx.subscribe(&room, |_, _, event, cx| cx.emit(event.clone())),
|
|
];
|
|
self.room = Some((room.clone(), subscriptions));
|
|
let location = self
|
|
.location
|
|
.as_ref()
|
|
.and_then(|location| location.upgrade());
|
|
room.update(cx, |room, cx| room.set_location(location.as_ref(), cx))
|
|
}
|
|
} else {
|
|
self.room = None;
|
|
Task::ready(Ok(()))
|
|
}
|
|
} else {
|
|
Task::ready(Ok(()))
|
|
}
|
|
}
|
|
|
|
pub fn room(&self) -> Option<&Model<Room>> {
|
|
self.room.as_ref().map(|(room, _)| room)
|
|
}
|
|
|
|
pub fn client(&self) -> Arc<Client> {
|
|
self.client.clone()
|
|
}
|
|
|
|
pub fn pending_invites(&self) -> &HashSet<u64> {
|
|
&self.pending_invites
|
|
}
|
|
|
|
pub fn report_call_event(&self, operation: &'static str, cx: &AppContext) {
|
|
if let Some(room) = self.room() {
|
|
let room = room.read(cx);
|
|
report_call_event_for_room(operation, room.id(), room.channel_id(), &self.client, cx);
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn report_call_event_for_room(
|
|
operation: &'static str,
|
|
room_id: u64,
|
|
channel_id: Option<u64>,
|
|
client: &Arc<Client>,
|
|
cx: &AppContext,
|
|
) {
|
|
let telemetry = client.telemetry();
|
|
let telemetry_settings = *TelemetrySettings::get_global(cx);
|
|
|
|
telemetry.report_call_event(telemetry_settings, operation, Some(room_id), channel_id)
|
|
}
|
|
|
|
pub fn report_call_event_for_channel(
|
|
operation: &'static str,
|
|
channel_id: u64,
|
|
client: &Arc<Client>,
|
|
cx: &AppContext,
|
|
) {
|
|
let room = ActiveCall::global(cx).read(cx).room();
|
|
|
|
let telemetry = client.telemetry();
|
|
|
|
let telemetry_settings = *TelemetrySettings::get_global(cx);
|
|
|
|
telemetry.report_call_event(
|
|
telemetry_settings,
|
|
operation,
|
|
room.map(|r| r.read(cx).id()),
|
|
Some(channel_id),
|
|
)
|
|
}
|
|
|
|
struct Call {
|
|
follower_states: HashMap<View<Pane>, FollowerState>,
|
|
active_call: Option<(Model<ActiveCall>, Vec<Subscription>)>,
|
|
parent_workspace: WeakView<Workspace>,
|
|
}
|
|
|
|
impl Call {
|
|
fn new(parent_workspace: WeakView<Workspace>, cx: &mut ViewContext<'_, Workspace>) -> Self {
|
|
let mut active_call = None;
|
|
if cx.has_global::<Model<ActiveCall>>() {
|
|
let call = cx.global::<Model<ActiveCall>>().clone();
|
|
let subscriptions = vec![cx.subscribe(&call, Self::on_active_call_event)];
|
|
active_call = Some((call, subscriptions));
|
|
}
|
|
Self {
|
|
follower_states: Default::default(),
|
|
active_call,
|
|
parent_workspace,
|
|
}
|
|
}
|
|
fn on_active_call_event(
|
|
workspace: &mut Workspace,
|
|
_: Model<ActiveCall>,
|
|
event: &call2::room::Event,
|
|
cx: &mut ViewContext<Workspace>,
|
|
) {
|
|
match event {
|
|
call2::room::Event::ParticipantLocationChanged { participant_id }
|
|
| call2::room::Event::RemoteVideoTracksChanged { participant_id } => {
|
|
workspace.leader_updated(*participant_id, cx);
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[async_trait(?Send)]
|
|
impl CallHandler for Call {
|
|
fn leader_updated(&mut self, leader_id: PeerId, cx: &mut ViewContext<Workspace>) -> Option<()> {
|
|
cx.notify();
|
|
|
|
let (call, _) = self.active_call.as_ref()?;
|
|
let room = call.read(cx).room()?.read(cx);
|
|
let participant = room.remote_participant_for_peer_id(leader_id)?;
|
|
let mut items_to_activate = Vec::new();
|
|
|
|
let leader_in_this_app;
|
|
let leader_in_this_project;
|
|
match participant.location {
|
|
call2::ParticipantLocation::SharedProject { project_id } => {
|
|
leader_in_this_app = true;
|
|
leader_in_this_project = Some(project_id)
|
|
== self
|
|
.parent_workspace
|
|
.update(cx, |this, cx| this.project.read(cx).remote_id())
|
|
.log_err()
|
|
.flatten();
|
|
}
|
|
call2::ParticipantLocation::UnsharedProject => {
|
|
leader_in_this_app = true;
|
|
leader_in_this_project = false;
|
|
}
|
|
call2::ParticipantLocation::External => {
|
|
leader_in_this_app = false;
|
|
leader_in_this_project = false;
|
|
}
|
|
};
|
|
|
|
for (pane, state) in &self.follower_states {
|
|
if state.leader_id != leader_id {
|
|
continue;
|
|
}
|
|
if let (Some(active_view_id), true) = (state.active_view_id, leader_in_this_app) {
|
|
if let Some(item) = state.items_by_leader_view_id.get(&active_view_id) {
|
|
if leader_in_this_project || !item.is_project_item(cx) {
|
|
items_to_activate.push((pane.clone(), item.boxed_clone()));
|
|
}
|
|
} else {
|
|
log::warn!(
|
|
"unknown view id {:?} for leader {:?}",
|
|
active_view_id,
|
|
leader_id
|
|
);
|
|
}
|
|
continue;
|
|
}
|
|
// todo!()
|
|
// if let Some(shared_screen) = self.shared_screen_for_peer(leader_id, pane, cx) {
|
|
// items_to_activate.push((pane.clone(), Box::new(shared_screen)));
|
|
// }
|
|
}
|
|
|
|
for (pane, item) in items_to_activate {
|
|
let pane_was_focused = pane.read(cx).has_focus(cx);
|
|
if let Some(index) = pane.update(cx, |pane, _| pane.index_for_item(item.as_ref())) {
|
|
pane.update(cx, |pane, cx| pane.activate_item(index, false, false, cx));
|
|
} else {
|
|
pane.update(cx, |pane, mut cx| {
|
|
pane.add_item(item.boxed_clone(), false, false, None, &mut cx)
|
|
});
|
|
}
|
|
|
|
if pane_was_focused {
|
|
pane.update(cx, |pane, cx| pane.focus_active_item(cx));
|
|
}
|
|
}
|
|
|
|
None
|
|
}
|
|
|
|
fn shared_screen_for_peer(
|
|
&self,
|
|
peer_id: PeerId,
|
|
pane: &View<Pane>,
|
|
cx: &mut ViewContext<Workspace>,
|
|
) -> Option<Box<dyn ItemHandle>> {
|
|
let (call, _) = self.active_call.as_ref()?;
|
|
let room = call.read(cx).room()?.read(cx);
|
|
let participant = room.remote_participant_for_peer_id(peer_id)?;
|
|
let track = participant.video_tracks.values().next()?.clone();
|
|
let user = participant.user.clone();
|
|
todo!();
|
|
// for item in pane.read(cx).items_of_type::<SharedScreen>() {
|
|
// if item.read(cx).peer_id == peer_id {
|
|
// return Box::new(Some(item));
|
|
// }
|
|
// }
|
|
|
|
// Some(Box::new(cx.build_view(|cx| {
|
|
// SharedScreen::new(&track, peer_id, user.clone(), cx)
|
|
// })))
|
|
}
|
|
|
|
fn follower_states_mut(&mut self) -> &mut HashMap<View<Pane>, FollowerState> {
|
|
&mut self.follower_states
|
|
}
|
|
fn follower_states(&self) -> &HashMap<View<Pane>, FollowerState> {
|
|
&self.follower_states
|
|
}
|
|
fn room_id(&self, cx: &AppContext) -> Option<u64> {
|
|
Some(self.active_call.as_ref()?.0.read(cx).room()?.read(cx).id())
|
|
}
|
|
fn hang_up(&self, mut cx: AsyncWindowContext) -> Result<Task<Result<()>>> {
|
|
let Some((call, _)) = self.active_call.as_ref() else {
|
|
bail!("Cannot exit a call; not in a call");
|
|
};
|
|
|
|
call.update(&mut cx, |this, cx| this.hang_up(cx))
|
|
}
|
|
fn active_project(&self, cx: &AppContext) -> Option<WeakModel<Project>> {
|
|
ActiveCall::global(cx).read(cx).location().cloned()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use gpui::TestAppContext;
|
|
|
|
use crate::OneAtATime;
|
|
|
|
#[gpui::test]
|
|
async fn test_one_at_a_time(cx: &mut TestAppContext) {
|
|
let mut one_at_a_time = OneAtATime { cancel: None };
|
|
|
|
assert_eq!(
|
|
cx.update(|cx| one_at_a_time.spawn(cx, |_| async { Ok(1) }))
|
|
.await
|
|
.unwrap(),
|
|
Some(1)
|
|
);
|
|
|
|
let (a, b) = cx.update(|cx| {
|
|
(
|
|
one_at_a_time.spawn(cx, |_| async {
|
|
assert!(false);
|
|
Ok(2)
|
|
}),
|
|
one_at_a_time.spawn(cx, |_| async { Ok(3) }),
|
|
)
|
|
});
|
|
|
|
assert_eq!(a.await.unwrap(), None);
|
|
assert_eq!(b.await.unwrap(), Some(3));
|
|
|
|
let promise = cx.update(|cx| one_at_a_time.spawn(cx, |_| async { Ok(4) }));
|
|
drop(one_at_a_time);
|
|
|
|
assert_eq!(promise.await.unwrap(), None);
|
|
}
|
|
}
|