1use crate::{
  2    item::{Item, ItemEvent},
  3    ItemNavHistory, WorkspaceId,
  4};
  5use anyhow::Result;
  6use call::participant::{Frame, RemoteVideoTrack};
  7use client::{proto::PeerId, User};
  8use futures::StreamExt;
  9use gpui::{
 10    elements::*,
 11    geometry::{rect::RectF, vector::vec2f},
 12    platform::MouseButton,
 13    AppContext, Entity, Task, View, ViewContext,
 14};
 15use settings::Settings;
 16use smallvec::SmallVec;
 17use std::{
 18    borrow::Cow,
 19    sync::{Arc, Weak},
 20};
 21
 22pub enum Event {
 23    Close,
 24}
 25
 26pub struct SharedScreen {
 27    track: Weak<RemoteVideoTrack>,
 28    frame: Option<Frame>,
 29    pub peer_id: PeerId,
 30    user: Arc<User>,
 31    nav_history: Option<ItemNavHistory>,
 32    _maintain_frame: Task<Result<()>>,
 33}
 34
 35impl SharedScreen {
 36    pub fn new(
 37        track: &Arc<RemoteVideoTrack>,
 38        peer_id: PeerId,
 39        user: Arc<User>,
 40        cx: &mut ViewContext<Self>,
 41    ) -> Self {
 42        let mut frames = track.frames();
 43        Self {
 44            track: Arc::downgrade(track),
 45            frame: None,
 46            peer_id,
 47            user,
 48            nav_history: Default::default(),
 49            _maintain_frame: cx.spawn(|this, mut cx| async move {
 50                while let Some(frame) = frames.next().await {
 51                    this.update(&mut cx, |this, cx| {
 52                        this.frame = Some(frame);
 53                        cx.notify();
 54                    })?;
 55                }
 56                this.update(&mut cx, |_, cx| cx.emit(Event::Close))?;
 57                Ok(())
 58            }),
 59        }
 60    }
 61}
 62
 63impl Entity for SharedScreen {
 64    type Event = Event;
 65}
 66
 67impl View for SharedScreen {
 68    fn ui_name() -> &'static str {
 69        "SharedScreen"
 70    }
 71
 72    fn render(&mut self, cx: &mut ViewContext<Self>) -> AnyElement<Self> {
 73        enum Focus {}
 74
 75        let frame = self.frame.clone();
 76        MouseEventHandler::<Focus, _>::new(0, cx, |_, cx| {
 77            Canvas::new(move |scene, bounds, _, _, _| {
 78                if let Some(frame) = frame.clone() {
 79                    let size = constrain_size_preserving_aspect_ratio(
 80                        bounds.size(),
 81                        vec2f(frame.width() as f32, frame.height() as f32),
 82                    );
 83                    let origin = bounds.origin() + (bounds.size() / 2.) - size / 2.;
 84                    scene.push_surface(gpui::platform::mac::Surface {
 85                        bounds: RectF::new(origin, size),
 86                        image_buffer: frame.image(),
 87                    });
 88                }
 89            })
 90            .contained()
 91            .with_style(cx.global::<Settings>().theme.shared_screen)
 92        })
 93        .on_down(MouseButton::Left, |_, _, cx| cx.focus_parent())
 94        .into_any()
 95    }
 96}
 97
 98impl Item for SharedScreen {
 99    fn tab_tooltip_text(&self, _: &AppContext) -> Option<Cow<str>> {
100        Some(format!("{}'s screen", self.user.github_login).into())
101    }
102    fn deactivated(&mut self, cx: &mut ViewContext<Self>) {
103        if let Some(nav_history) = self.nav_history.as_ref() {
104            nav_history.push::<()>(None, cx);
105        }
106    }
107
108    fn tab_content<V: View>(
109        &self,
110        _: Option<usize>,
111        style: &theme::Tab,
112        _: &AppContext,
113    ) -> gpui::AnyElement<V> {
114        Flex::row()
115            .with_child(
116                Svg::new("icons/disable_screen_sharing_12.svg")
117                    .with_color(style.label.text.color)
118                    .constrained()
119                    .with_width(style.type_icon_width)
120                    .aligned()
121                    .contained()
122                    .with_margin_right(style.spacing),
123            )
124            .with_child(
125                Label::new(
126                    format!("{}'s screen", self.user.github_login),
127                    style.label.clone(),
128                )
129                .aligned(),
130            )
131            .into_any()
132    }
133
134    fn set_nav_history(&mut self, history: ItemNavHistory, _: &mut ViewContext<Self>) {
135        self.nav_history = Some(history);
136    }
137
138    fn clone_on_split(
139        &self,
140        _workspace_id: WorkspaceId,
141        cx: &mut ViewContext<Self>,
142    ) -> Option<Self> {
143        let track = self.track.upgrade()?;
144        Some(Self::new(&track, self.peer_id, self.user.clone(), cx))
145    }
146
147    fn to_item_events(event: &Self::Event) -> SmallVec<[ItemEvent; 2]> {
148        match event {
149            Event::Close => smallvec::smallvec!(ItemEvent::CloseItem),
150        }
151    }
152}