watch-party/src/watch_session.rs

use serde::{Deserialize, Serialize};
use std::time::Instant;

#[derive(Serialize, Deserialize, Clone)]
pub struct SubtitleTrack {
    pub url: String,
    pub name: String,
}

#[derive(Clone)]
pub struct WatchSession {
    pub video_url: String,
    pub subtitle_tracks: Vec<SubtitleTrack>,

    is_playing: bool,
    playing_from_timestamp: u64,
    playing_from_instant: Instant,
    // TODO: How do we keep track of the current playing time ?
}

#[derive(Serialize)]
pub struct WatchSessionView {
    pub video_url: String,
    pub subtitle_tracks: Vec<SubtitleTrack>,
    pub current_time_ms: u64,
    pub is_playing: bool,
}

impl WatchSession {
    pub fn new(video_url: String, subtitle_tracks: Vec<SubtitleTrack>) -> Self {
        WatchSession {
            video_url,
            subtitle_tracks,
            is_playing: false,
            playing_from_timestamp: 0,
            playing_from_instant: Instant::now(),
        }
    }

    pub fn view(&self) -> WatchSessionView {
        WatchSessionView {
            video_url: self.video_url.clone(),
            subtitle_tracks: self.subtitle_tracks.clone(),
            current_time_ms: self.get_time_ms() as u64,
            is_playing: self.is_playing,
        }
    }

    pub fn get_time_ms(&self) -> u64 {
        if !self.is_playing {
            return self.playing_from_timestamp;
        }

        self.playing_from_timestamp + self.playing_from_instant.elapsed().as_millis() as u64
    }

    pub fn set_time_ms(&mut self, time_ms: u64) {
        self.playing_from_timestamp = time_ms;
        self.playing_from_instant = Instant::now();
    }

    pub fn set_playing(&mut self, playing: bool) {
        self.set_time_ms(self.get_time_ms());
        self.is_playing = playing;
    }
}