psibase 0.23.0

Library and command-line tool for interacting with psibase networks
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

use tokio::net::TcpListener;
use tokio::net::UnixStream;
//use core::net::SocketAddr;
use reqwest::Url;

pub struct AutoAbort {
    handle: tokio::task::JoinHandle<()>,
}

impl Drop for AutoAbort {
    fn drop(&mut self) {
        self.handle.abort()
    }
}

async fn make_tcp_proxy(
    path: &str,
) -> Result<(tokio::task::JoinHandle<()>, String), anyhow::Error> {
    let listener = TcpListener::bind("127.0.0.1:0").await?;
    let endpoint = listener.local_addr()?;
    let pathcpy: String = path.into();
    let task = tokio::task::spawn(async move {
        while let Ok((mut client, _)) = listener.accept().await {
            let pathcpy2 = pathcpy.clone();
            tokio::task::spawn(async move {
                if let Ok(mut upstream) = UnixStream::connect(pathcpy2).await {
                    let _ = tokio::io::copy_bidirectional(&mut client, &mut upstream).await;
                }
            });
        }
    });

    Ok((task, format!("http://{}", endpoint)))
}

pub async fn apply_proxy(
    builder: reqwest::ClientBuilder,
    proxy: &Option<Url>,
) -> Result<(reqwest::ClientBuilder, Option<AutoAbort>), anyhow::Error> {
    if let Some(url) = proxy {
        if url.scheme() == "unix" {
            let (task, endpoint) = make_tcp_proxy(url.path()).await?;
            return Ok((
                builder.proxy(reqwest::Proxy::http(endpoint)?),
                Some(AutoAbort { handle: task }),
            ));
        } else {
            return Ok((builder.proxy(reqwest::Proxy::http(url.as_str())?), None));
        }
    }
    return Ok((builder, None));
}