panel - html terminal


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
108
109
110
111

use ansi_to_html::convert_escaped;
use std::sync::Arc;
use std::time::Duration;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpStream;
use tokio::sync::broadcast;
use tokio::sync::broadcast::error::TryRecvError;
use tokio::task::JoinHandle;

use crate::server::State;
pub struct Process {
    inner: TcpStream,
    input: Option<broadcast::Receiver<String>>,
    html_output: Option<broadcast::Sender<String>>,
    plain_output: Option<broadcast::Sender<String>>,
}

impl Process {
    /// spawns the server
    #[must_use]
    pub async fn spawn() -> Self {
        let stream = TcpStream::connect("localhost:6859").await.unwrap();
        Self {
            inner: stream,
            input: None,
            html_output: None,
            plain_output: None,
        }
    }

    pub fn input(mut self, input: broadcast::Receiver<String>) -> Self {
        self.input = Some(input);
        self
    }

    pub fn html_output(mut self, output: broadcast::Sender<String>) -> Self {
        self.html_output = Some(output);
        self
    }

    pub fn plain_output(mut self, output: broadcast::Sender<String>) -> Self {
        self.plain_output = Some(output);
        self
    }

    pub fn with_state(self, state: &Arc<State>) -> Self {
        self.html_output(state.stdout_html.clone())
            .plain_output(state.stdout_plain.clone())
    }

    pub fn link(mut self) -> JoinHandle<()> {
        define_print!("process");
        let mut input = self.input.unwrap();
        let html_output = self.html_output.unwrap();
        let plain_output = self.plain_output.unwrap();
        tokio::spawn(async move {
            let mut stdout = [0; 4096];
            loop {
                if html_output.receiver_count() + plain_output.receiver_count() == 0 {
                    async_std::task::sleep(Duration::from_millis(500)).await;
                    continue;
                }
                match input.try_recv() {
                    Err(e) => match e {
                        TryRecvError::Closed => fail!("closed"),
                        TryRecvError::Lagged(_) => continue,
                        TryRecvError::Empty => {}
                    },
                    Ok(mut s) => {
                        input!("{s}");
                        s += "\n";
                        self.inner.write_all(s.as_bytes()).await.unwrap();
                        self.inner.flush().await.unwrap();
                    }
                }

                let string = {
                    let n = tokio::select! {
                        n = {self.inner.read(&mut stdout)} => n.unwrap(),
                        _ = async_std::task::sleep(Duration::from_millis(500)) => continue,
                    };
                    String::from_utf8_lossy(&stdout[..n]).into_owned()
                };
                for line in string.lines() {
                    output!("{line}");
                }
                if plain_output.receiver_count() > 0 {
                    let stripped =
                        String::from_utf8_lossy(&strip_ansi_escapes::strip(&string).unwrap())
                            .into_owned();
                    plain_output.send(stripped).unwrap();
                }
                if html_output.receiver_count() > 0 {
                    html_output.send(ansi2html(&string)).unwrap();
                }
                nooutput!();
                async_std::task::sleep(Duration::from_millis(500)).await;
            }
        })
    }
}

/// for dark theme
fn ansi2html(ansi: &str) -> String {
    convert_escaped(ansi)
        .unwrap()
        .replace("#555", "#a4a4a0")
        .replace("#55f", "#7486fd")
        .replace("#fff", "wheat")
        .replace("#a00", "#d05047")
}