ethexe-node-wrapper 2.0.0-pre.1

Node process wrapper utilities for ethexe
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

// Copyright (C) Gear Technologies Inc.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

use crate::Error;
use ethexe_common::Address;
use ethexe_rpc_client::DevClient;
use jsonrpsee::{
    http_client::HttpClient,
    ws_client::{WsClient, WsClientBuilder},
};
use std::{net::SocketAddrV4, process::Child};

/// The Vara.eth CLI instance. Will close the instance when dropped.
///
/// Can be constructed only from [spawn_immediate](super::node::VaraEth::spawn_immediate)
/// or [spawn_ready](super::node::VaraEth::spawn_ready).
#[derive(Debug)]
pub struct VaraEthInstance {
    /// The Vara.eth rpc address.
    pub(crate) rpc_addr: SocketAddrV4,
    /// The Vara.eth anvil rpc address.
    pub(crate) eth_rpc_addr: SocketAddrV4,
    /// The child process of instance
    pub(crate) child: Child,
}

impl VaraEthInstance {
    /// Fetches the Ethereum Router address.
    pub async fn router_address(&self) -> Result<Address, Error> {
        self.http_client()?
            .router_address()
            .await
            .map_err(Error::QueryRouterAddress)
    }

    /// Returns the websocket client.
    pub async fn ws_client(&self) -> Result<WsClient, Error> {
        WsClientBuilder::new()
            .build(self.ws_endpoint())
            .await
            .map_err(Error::BuildWsClient)
    }

    /// Returns the HTTP client.
    pub fn http_client(&self) -> Result<HttpClient, Error> {
        HttpClient::builder()
            .build(self.http_endpoint())
            .map_err(Error::BuildHttpClient)
    }

    /// Returns the Websocket endpoint of Vara.eth rpc.
    pub fn ws_endpoint(&self) -> String {
        format!("ws://{}", self.rpc_addr)
    }

    /// Returns the HTTP endpoint of Vara.eth rpc.
    pub fn http_endpoint(&self) -> String {
        format!("http://{}", self.rpc_addr)
    }

    /// Returns the Websocket endpoint Vara.eth node connected to.
    pub fn ethereum_ws_endpoint(&self) -> String {
        format!("ws://{}", self.eth_rpc_addr)
    }

    /// Returns the HTTP endpoint Vara.eth node connected to.
    pub fn ethereum_http_endpoint(&self) -> String {
        format!("http://{}", self.eth_rpc_addr)
    }
}

impl Drop for VaraEthInstance {
    fn drop(&mut self) {
        #[cfg(unix)]
        {
            // Here is hack for unix systems - kill all processes in group to force anvil drop.
            let pid = self.child.id() as i32;
            unsafe { libc::kill(-pid, libc::SIGTERM) };
        }

        #[cfg(not(unix))]
        let _ = self.child.kill();
    }
}