tab-daemon 0.5.4

the daemon module for the tab terminal multiplexer
Documentation 
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use crate::{message::tab_assignment::AssignTab, prelude::*};
use crate::{
    message::{
        tab::TabRecv,
        tab_manager::{TabManagerRecv, TabManagerSend},
    },
    state::tab::TabsState,
};
use anyhow::Context;

use std::{
    collections::HashMap,
    sync::atomic::{AtomicUsize, Ordering},
};
use tab_api::tab::{TabId, TabMetadata};

/// Manages the currently running tabs.  This is a point-of-contact between the tab-command and tab-pty clients.
///
/// - Serves 'create tab' requests from the tab-command client.
/// - Spawns tab-pty processes (OS processes), and issues offers of tab assignment to connected pty clients.
/// - Terminates tabs when requested by the tab-command client.
pub struct TabManagerService {
    _recv: Lifeline,
}

static TAB_ID_COUNTER: AtomicUsize = AtomicUsize::new(0);
// working on a bug here where all the ptys disconnect, and TabRecv goes dead.
impl Service for TabManagerService {
    type Bus = ListenerBus;
    type Lifeline = anyhow::Result<Self>;

    fn spawn(bus: &Self::Bus) -> Self::Lifeline {
        let _recv = {
            let mut rx = bus.rx::<TabManagerRecv>()?;

            let mut tx = bus.tx::<TabManagerSend>()?;
            let mut tx_tabs = bus.tx::<TabRecv>()?;
            let mut tx_tabs_state = bus.tx::<TabsState>()?;
            let mut tx_assign_tab = bus.tx::<AssignTab>()?;

            let mut tabs: HashMap<TabId, TabMetadata> = HashMap::new();

            Self::try_task("recv", async move {
                'msg: while let Some(msg) = rx.recv().await {
                    match msg {
                        TabManagerRecv::CreateTab(create) => {
                            for tab in tabs.values() {
                                if tab.name == create.name {
                                    continue 'msg;
                                }
                            }

                            debug!("recieved request to create tab {}", &create.name);
                            let id = TAB_ID_COUNTER.fetch_add(1, Ordering::SeqCst) as u16;
                            let tab_id = TabId(id);
                            let tab_metadata = TabMetadata::create(tab_id, create);

                            tx_assign_tab.send(AssignTab(tab_metadata.clone())).await?;

                            tabs.insert(tab_id, tab_metadata);
                            tx_tabs_state.send(TabsState::new(&tabs)).await?;
                        }
                        TabManagerRecv::CloseTab(close) => {
                            Self::close_tab(
                                close,
                                &mut tabs,
                                &mut tx,
                                &mut tx_tabs,
                                &mut tx_tabs_state,
                            )
                            .await?;
                        }
                    }
                }
                Ok(())
            })
        };

        Ok(Self { _recv })
    }
}

impl TabManagerService {
    async fn close_tab(
        id: TabId,
        tabs: &mut HashMap<TabId, TabMetadata>,
        tx: &mut impl Sender<TabManagerSend>,
        tx_close: &mut impl Sender<TabRecv>,
        tx_tabs_state: &mut impl Sender<TabsState>,
    ) -> anyhow::Result<()> {
        info!("TabManager terminating tab {}", id);
        tabs.remove(&id);

        tx.send(TabManagerSend::TabTerminated(id))
            .await
            .context("tx TabTerminated")
            .ok();
        tx_close.send(TabRecv::Terminate(id)).await.ok();
        tx_tabs_state
            .send(TabsState::new(&tabs))
            .await
            .context("tx_tabs_state TabsState")
            .ok();

        Ok(())
    }
}