wye 0.1.1

Wye is reverse tee, simple multiplexer like tmux
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

use std::{env, os::fd::AsRawFd};

use nix::{
    pty::openpty,
    sys::termios::{SetArg, cfmakeraw, tcgetattr, tcsetattr},
    unistd::{ForkResult, close},
};
use wye::{
    WYE_SESSION_VAR, child, config, log, parent,
    term::{TerminalModeGuard, get_winsize},
};

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let config = config::parse_config()?;

    if let Ok(session) = env::var(WYE_SESSION_VAR) {
        return log::log_already_in_session(session).map_err(Into::into);
    };

    let fd = std::io::stdin().as_raw_fd();
    let _guard = TerminalModeGuard::new(fd)?;

    let mut termios_settings = tcgetattr(std::io::stdin())?;
    let initial_winsize = get_winsize()?;
    let pty = openpty(Some(&initial_winsize), Some(&termios_settings))?;

    cfmakeraw(&mut termios_settings);
    tcsetattr(std::io::stdin(), SetArg::TCSANOW, &termios_settings)?;

    match unsafe { nix::unistd::fork() } {
        Ok(ForkResult::Parent { .. }) => {
            close(pty.slave)?;
            parent::process(config, pty.master)
        }
        Ok(ForkResult::Child) => {
            close(pty.master)?;
            child::process(config, pty.slave)
        }
        Err(e) => {
            std::process::exit(e as i32);
        }
    }
}