iroh-pigeons 0.2.0

carrier pigeons for your SSH connections
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
108
109
110
111
112
113
114
115
116
117
118
119

use std::{
    path::PathBuf,
    process::{Command, Stdio},
};

use crate::{Service, ServiceParams};

#[cfg(target_os = "linux")]
#[derive(Debug, Clone)]
pub(crate) struct LinuxService;

#[cfg(target_os = "linux")]
impl Service for LinuxService {
    async fn install(service_params: ServiceParams) -> anyhow::Result<()> {
        let path = Self::init_install_script(service_params)?;
        tracing::debug!("running install script: {}", path.display());

        let status = Command::new("sh")
            .arg(&path)
            .stdin(Stdio::inherit())
            .stdout(Stdio::inherit())
            .stderr(Stdio::inherit())
            .status()?;

        if !status.success() {
            anyhow::bail!("install script failed with exit code: {}", status);
        }

        Ok(())
    }

    async fn info() -> anyhow::Result<()> {
        anyhow::bail!("service info is not yet supported")
    }

    async fn uninstall() -> anyhow::Result<()> {
        let path = Self::init_uninstall_script()?;
        tracing::debug!("running uninstall script: {}", path.display());

        let status = Command::new("sh")
            .arg(&path)
            .stdin(Stdio::inherit())
            .stdout(Stdio::inherit())
            .stderr(Stdio::inherit())
            .status()?;

        if !status.success() {
            anyhow::bail!("uninstall script failed with exit code: {}", status);
        }

        Ok(())
    }

    async fn restart() -> anyhow::Result<()> {
        let status = Command::new("systemctl")
            .args(["restart", "pigeons.service"])
            .stdin(Stdio::inherit())
            .stdout(Stdio::inherit())
            .stderr(Stdio::inherit())
            .status()?;

        if !status.success() {
            anyhow::bail!("systemctl restart failed with exit code: {}", status);
        }

        Ok(())
    }
}

#[cfg(target_os = "linux")]
impl LinuxService {
    const INSTALL_SH_BYTES: &str = include_str!("../../service/install_linux.sh");
    const UNINSTALL_SH_BYTES: &str = include_str!("../../service/uninstall_linux.sh");

    fn init_install_script(service_params: ServiceParams) -> anyhow::Result<PathBuf> {
        use std::io::Write as _;

        let mut relay_args = String::new();
        for url in &service_params.relay_url {
            relay_args.push_str(&format!(" --relay-url {url}"));
        }

        let mut temp_sh = tempfile::Builder::new()
            .prefix("pigeons_install-")
            .suffix(".sh")
            .tempfile_in("/tmp")?;
        temp_sh.write_all(
            Self::INSTALL_SH_BYTES
                .replace("[SSHPORT]", &service_params.ssh_port.to_string())
                .replace("[RELAYARGS]", &relay_args)
                .replace(
                    "[BINARYPATH]",
                    service_params
                        .binary_path
                        .to_str()
                        .ok_or_else(|| anyhow::anyhow!("binary path is not valid UTF-8"))?,
                )
                .as_bytes(),
        )?;
        let sh_path = temp_sh.path().to_path_buf();
        temp_sh.keep()?;

        Ok(sh_path)
    }

    fn init_uninstall_script() -> anyhow::Result<PathBuf> {
        use std::io::Write as _;

        let mut temp_sh = tempfile::Builder::new()
            .prefix("pigeons_uninstall-")
            .suffix(".sh")
            .tempfile_in("/tmp")?;
        temp_sh.write_all(Self::UNINSTALL_SH_BYTES.as_bytes())?;
        let sh_path = temp_sh.path().to_path_buf();
        temp_sh.keep()?;

        Ok(sh_path)
    }
}