tronic 0.5.2

A modular, async-first Rust client for the Tron blockchain.
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137

use k256::ecdsa::SigningKey;
use tronic::domain::address::TronAddress;
use tronic::domain::trx::Trx;
use std::process::Stdio;
use std::time::Duration;
use tokio::process::Command;
use tokio::runtime::Runtime;
use tronic::client::pending::AutoSigning;
use tronic::client::Client;
use tronic::provider::grpc::GrpcProvider;
use tronic::signer::LocalSigner;

#[static_init::dynamic(drop)]
pub static mut NODE: Node = Node::start("test_tron_container");

pub struct Node {
    container_name: String,
    grpc_port: u16,
    tx: tokio::sync::mpsc::Sender<(tokio::sync::oneshot::Sender<LocalSigner>, Trx)>,
    zion_addr: TronAddress,
}

impl Node {
    fn start(container_name: &'static str) -> Self {
        let grpc_port = portpicker::pick_unused_port().unwrap();

        // Create the runtime first
        let rt = Runtime::new().unwrap();

        let (tx, mut rx) = tokio::sync::mpsc::channel::<(tokio::sync::oneshot::Sender<LocalSigner>, Trx)>(1000);
        
        // Use the runtime to block on async operations
        std::thread::spawn(move || {
            rt.block_on(async {
                // Start the Tron Quickstart container
                Command::new("docker")
                    .args(&[
                        "run",
                        "-d",
                        "-it",
                        "-p",
                        &format!("{grpc_port}:50051"),
                        "-v",
                        "./tests/conf/tron.conf:/java-tron/conf/tron.conf",
                        "--rm",
                        "--name",
                        container_name,
                        "custom-tron",
                        "-jvm",
                        "{-Xmx10g -Xms10g}",
                        "-c",
                        "/java-tron/conf/tron.conf",
                        "--es",
                        "-w",
                    ])
                    .stdout(Stdio::null())
                    .stderr(Stdio::null())
                    .spawn()
                    .unwrap()
                    .wait()
                    .await
                    .unwrap();

                tokio::time::sleep(Duration::from_secs(8)).await;

                let signing_key = SigningKey::from_slice(&hex::decode("da146374a75310b9666e834ee4ad0866d6f4035967bfc76217c5a495fff9f0d0").unwrap()).unwrap();

                let provider = GrpcProvider::builder().connect(
                    format!("http://localhost:{}", grpc_port)).await.unwrap();
                
                let zion = Client::builder()
                    .provider(provider)
                    .signer(LocalSigner::from(signing_key))
                    .build();

                while let Some((tx, balance)) = rx.recv().await {
                    let signer = LocalSigner::rand();
                    let _ = zion
                        .send_trx()
                        .to(signer.address())
                        .amount(balance)
                        .build::<AutoSigning>()
                        .await
                        .unwrap()
                        .broadcast(&())
                        .await
                        .unwrap();
                    tx.send(signer).unwrap();
                }
            });
        });

        // Wait node to initialize
        std::thread::sleep(Duration::from_secs(10));

        Self {
            container_name: container_name.into(),
            grpc_port,
            tx, 
            zion_addr: LocalSigner::from(SigningKey::from_slice(&hex::decode("da146374a75310b9666e834ee4ad0866d6f4035967bfc76217c5a495fff9f0d0").unwrap()).unwrap()).address()
        }
    }

    pub fn grpc_addr(&self) -> String {
        format!("http://localhost:{}", self.grpc_port)
    }

    pub fn zion_addr(&self) -> TronAddress {
        self.zion_addr
    }

    pub async fn new_account(&self, balance: Trx) -> LocalSigner {
        let (tx, rx) = tokio::sync::oneshot::channel();
        self.tx.send((tx, balance)).await.unwrap();
        rx.await.unwrap()
    }
}

impl Drop for Node {
    fn drop(&mut self) {
        let container_name = self.container_name.clone();
        std::thread::spawn(move || {
            let rt = Runtime::new().unwrap();
            rt.block_on(async {
                Command::new("docker")
                    .args(&["stop", &container_name])
                    .stdout(Stdio::null())
                    .stderr(Stdio::null())
                    .spawn()
                    .unwrap()
                    .wait()
                    .await
                    .unwrap();
            });
        }).join().unwrap();
    }
}