zed/crates/workspace/src/shared_screen.rs

use crate::{
    item::ItemEvent,
    persistence::model::{ItemId, WorkspaceId},
    Item, ItemNavHistory, Pane, Workspace,
};
use anyhow::{anyhow, Result};
use call::participant::{Frame, RemoteVideoTrack};
use client::{PeerId, User};
use futures::StreamExt;
use gpui::{
    elements::*,
    geometry::{rect::RectF, vector::vec2f},
    Entity, ModelHandle, MouseButton, RenderContext, Task, View, ViewContext, ViewHandle,
    WeakViewHandle,
};
use project::Project;
use settings::Settings;
use smallvec::SmallVec;
use std::{
    path::PathBuf,
    sync::{Arc, Weak},
};

pub enum Event {
    Close,
}

pub struct SharedScreen {
    track: Weak<RemoteVideoTrack>,
    frame: Option<Frame>,
    pub peer_id: PeerId,
    user: Arc<User>,
    nav_history: Option<ItemNavHistory>,
    _maintain_frame: Task<()>,
}

impl SharedScreen {
    pub fn new(
        track: &Arc<RemoteVideoTrack>,
        peer_id: PeerId,
        user: Arc<User>,
        cx: &mut ViewContext<Self>,
    ) -> Self {
        let mut frames = track.frames();
        Self {
            track: Arc::downgrade(track),
            frame: None,
            peer_id,
            user,
            nav_history: Default::default(),
            _maintain_frame: cx.spawn(|this, mut cx| async move {
                while let Some(frame) = frames.next().await {
                    this.update(&mut cx, |this, cx| {
                        this.frame = Some(frame);
                        cx.notify();
                    })
                }
                this.update(&mut cx, |_, cx| cx.emit(Event::Close));
            }),
        }
    }
}

impl Entity for SharedScreen {
    type Event = Event;
}

impl View for SharedScreen {
    fn ui_name() -> &'static str {
        "SharedScreen"
    }

    fn render(&mut self, cx: &mut RenderContext<Self>) -> ElementBox {
        enum Focus {}

        let frame = self.frame.clone();
        MouseEventHandler::<Focus>::new(0, cx, |_, cx| {
            Canvas::new(move |bounds, _, cx| {
                if let Some(frame) = frame.clone() {
                    let size = constrain_size_preserving_aspect_ratio(
                        bounds.size(),
                        vec2f(frame.width() as f32, frame.height() as f32),
                    );
                    let origin = bounds.origin() + (bounds.size() / 2.) - size / 2.;
                    cx.scene.push_surface(gpui::mac::Surface {
                        bounds: RectF::new(origin, size),
                        image_buffer: frame.image(),
                    });
                }
            })
            .contained()
            .with_style(cx.global::<Settings>().theme.shared_screen)
            .boxed()
        })
        .on_down(MouseButton::Left, |_, cx| cx.focus_parent_view())
        .boxed()
    }
}

impl Item for SharedScreen {
    fn deactivated(&mut self, cx: &mut ViewContext<Self>) {
        if let Some(nav_history) = self.nav_history.as_ref() {
            nav_history.push::<()>(None, cx);
        }
    }

    fn tab_content(
        &self,
        _: Option<usize>,
        style: &theme::Tab,
        _: &gpui::AppContext,
    ) -> gpui::ElementBox {
        Flex::row()
            .with_child(
                Svg::new("icons/disable_screen_sharing_12.svg")
                    .with_color(style.label.text.color)
                    .constrained()
                    .with_width(style.icon_width)
                    .aligned()
                    .contained()
                    .with_margin_right(style.spacing)
                    .boxed(),
            )
            .with_child(
                Label::new(
                    format!("{}'s screen", self.user.github_login),
                    style.label.clone(),
                )
                .aligned()
                .boxed(),
            )
            .boxed()
    }

    fn project_path(&self, _: &gpui::AppContext) -> Option<project::ProjectPath> {
        Default::default()
    }

    fn project_entry_ids(&self, _: &gpui::AppContext) -> SmallVec<[project::ProjectEntryId; 3]> {
        Default::default()
    }

    fn is_singleton(&self, _: &gpui::AppContext) -> bool {
        false
    }

    fn set_nav_history(&mut self, history: ItemNavHistory, _: &mut ViewContext<Self>) {
        self.nav_history = Some(history);
    }

    fn clone_on_split(&self, cx: &mut ViewContext<Self>) -> Option<Self> {
        let track = self.track.upgrade()?;
        Some(Self::new(&track, self.peer_id, self.user.clone(), cx))
    }

    fn can_save(&self, _: &gpui::AppContext) -> bool {
        false
    }

    fn save(
        &mut self,
        _: ModelHandle<project::Project>,
        _: &mut ViewContext<Self>,
    ) -> Task<Result<()>> {
        Task::ready(Err(anyhow!("Item::save called on SharedScreen")))
    }

    fn save_as(
        &mut self,
        _: ModelHandle<project::Project>,
        _: PathBuf,
        _: &mut ViewContext<Self>,
    ) -> Task<Result<()>> {
        Task::ready(Err(anyhow!("Item::save_as called on SharedScreen")))
    }

    fn reload(
        &mut self,
        _: ModelHandle<project::Project>,
        _: &mut ViewContext<Self>,
    ) -> Task<Result<()>> {
        Task::ready(Err(anyhow!("Item::reload called on SharedScreen")))
    }

    fn to_item_events(event: &Self::Event) -> Vec<ItemEvent> {
        match event {
            Event::Close => vec![ItemEvent::CloseItem],
        }
    }

    fn serialized_item_kind() -> Option<&'static str> {
        None
    }

    fn deserialize(
        _project: ModelHandle<Project>,
        _workspace: WeakViewHandle<Workspace>,
        _workspace_id: WorkspaceId,
        _item_id: ItemId,
        _cx: &mut ViewContext<Pane>,
    ) -> Task<Result<ViewHandle<Self>>> {
        unreachable!("Shared screen can not be deserialized")
    }
}