1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
use std::{
    sync::{Arc, Mutex},
    io,
};

use crate::{
    message_queue::MessageQueue,
    project::Project,
    imfs::Imfs,
    session_id::SessionId,
    rbx_session::RbxSession,
    snapshot_reconciler::InstanceChanges,
    fs_watcher::FsWatcher,
};

/// Contains all of the state for a Rojo live-sync session.
pub struct LiveSession {
    pub project: Arc<Project>,
    pub session_id: SessionId,
    pub message_queue: Arc<MessageQueue<InstanceChanges>>,
    pub rbx_session: Arc<Mutex<RbxSession>>,
    pub imfs: Arc<Mutex<Imfs>>,
    _fs_watcher: FsWatcher,
}

impl LiveSession {
    pub fn new(project: Arc<Project>) -> io::Result<LiveSession> {
        let imfs = {
            let mut imfs = Imfs::new();
            imfs.add_roots_from_project(&project)?;

            Arc::new(Mutex::new(imfs))
        };
        let message_queue = Arc::new(MessageQueue::new());

        let rbx_session = Arc::new(Mutex::new(RbxSession::new(
            Arc::clone(&project),
            Arc::clone(&imfs),
            Arc::clone(&message_queue),
        )));

        let fs_watcher = FsWatcher::start(
            Arc::clone(&imfs),
            Arc::clone(&rbx_session),
        );

        let session_id = SessionId::new();

        Ok(LiveSession {
            project,
            session_id,
            message_queue,
            rbx_session,
            imfs,
            _fs_watcher: fs_watcher,
        })
    }

    pub fn get_project(&self) -> &Project {
        &self.project
    }
}