remowt-endpoints 0.1.9

Nix daemon proxy
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

use std::process::Stdio;

use bifrostlink::declarative::endpoints;
use bifrostlink::Config;
use serde::{Deserialize, Serialize};
use std::result::Result;
use tokio::process::Command;

pub struct NixDaemon;

#[derive(Serialize, Deserialize, Debug, thiserror::Error)]
pub enum Error {
	#[error("nix daemon unavailable: {0}")]
	DaemonUnavailable(String),
	#[error("tunnel socket unavailable: {0}")]
	Tunnel(String),
}

#[endpoints(ns = 4)]
impl NixDaemon {
	#[endpoints(id = 2)]
	async fn serve_store(&self, store: String, socket: String) -> Result<(), Error> {
		let mut child = Command::new("nix-daemon")
			.arg("--stdio")
			.arg("--store")
			.arg(&store)
			.stdin(Stdio::piped())
			.stdout(Stdio::piped())
			.spawn()
			.map_err(|e| Error::DaemonUnavailable(e.to_string()))?;
		let tunnel = tokio::net::UnixStream::connect(&socket)
			.await
			.map_err(|e| Error::Tunnel(e.to_string()))?;
		let mut stdin = child.stdin.take().expect("piped");
		let mut stdout = child.stdout.take().expect("piped");
		tokio::spawn(async move {
			let mut tunnel = tunnel;
			let (mut tr, mut tw) = tunnel.split();
			let _ = tokio::join!(
				tokio::io::copy(&mut tr, &mut stdin),
				tokio::io::copy(&mut stdout, &mut tw),
			);
			let _ = child.wait().await;
		});
		Ok(())
	}
}