shell-cell 1.6.3

Shell-Cell. CLI app to spawn and manage containerized shell environments
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

mod ui;
use std::sync::mpsc::{Receiver, RecvTimeoutError};

use crate::{
    buildkit::{BuildKitD, container_info::SCellContainerInfo, image_info::SCellImageInfo},
    cli::{
        MIN_FPS,
        ls::app::{AppInner, AppItemSuperTrait, ls::LsState},
    },
};

/// Holds the state when the user is viewing the inspect overlay.
pub struct LoadingState<Item> {
    buildkit: BuildKitD,
    rx: Receiver<color_eyre::Result<Vec<Item>>>,
}

impl LoadingState<SCellContainerInfo> {
    /// Creates a new [`LoadingState<SCellContainerInfo>`], spawning an async task
    /// that fetches the current Shell-Cell image list.
    pub fn load(buildkit: BuildKitD) -> AppInner<SCellContainerInfo> {
        let (tx, rx) = std::sync::mpsc::channel();
        tokio::spawn({
            let buildkit = buildkit.clone();
            async move {
                let result = async {
                    let res = buildkit.list_containers().await;
                    tokio::time::sleep(std::time::Duration::from_millis(500)).await;
                    res
                }
                .await;
                drop(tx.send(result));
            }
        });

        AppInner::Loading(Self { buildkit, rx })
    }
}

impl LoadingState<SCellImageInfo> {
    /// Creates a new [`LoadingState<SCellImageInfo>`], spawning an async task
    /// that fetches the current Shell-Cell image list.
    pub fn load(buildkit: BuildKitD) -> AppInner<SCellImageInfo> {
        let (tx, rx) = std::sync::mpsc::channel();

        tokio::spawn({
            let buildkit = buildkit.clone();
            async move {
                let result = async {
                    let res = buildkit.list_images().await;
                    tokio::time::sleep(std::time::Duration::from_millis(500)).await;
                    res
                }
                .await;
                drop(tx.send(result));
            }
        });

        AppInner::Loading(Self { buildkit, rx })
    }
}

impl<Item: Clone + AppItemSuperTrait> LoadingState<Item> {
    /// Polls the background stop task for completion and returns the next app state.
    ///
    /// - [`AppInner::Loading`] — still loading (self is returned back)
    /// - [`AppInner::Ls`] — stop succeeded; contains the refreshed item list
    pub fn try_recv(self) -> color_eyre::Result<AppInner<Item>> {
        match self.rx.recv_timeout(MIN_FPS) {
            Ok(Ok(items)) => Ok(LsState::ls(items, self.buildkit)),
            Ok(Err(e)) => {
                color_eyre::eyre::bail!(e)
            },
            Err(RecvTimeoutError::Timeout) => Ok(AppInner::Loading(self)),
            Err(RecvTimeoutError::Disconnected) => {
                color_eyre::eyre::bail!(
                    "StoppingState channel cannot be disconnected without returning a result"
                )
            },
        }
    }
}