solana-trader 0.1.0

Implemented most of the commonly used practical transaction-related functions on the Solana network.
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

use bs58;
use solana_sdk::{signature::Signer, signer::keypair::Keypair};

use crate::wallet::tool::private_key_base58_to_bytes;

#[derive(Debug, Clone)]
pub struct Wallet {
    pub public_key: String,
    pub private_key_32: String,
    pub private_key_64: String,
}

impl Wallet {
    /// create new wallet
    /// # Returns
    /// wallet object
    /// # Example
    /// ```rust
    /// let w: Wallet = Wallet::create_new_wallet();
    /// ```
    pub fn create_new_wallet() -> Wallet {
        let keypair = Keypair::new();
        let public_key = keypair.pubkey();
        let secret_key_32_bytes = keypair.secret_bytes();
        let secret_key_64_bytes = keypair.to_bytes();
        let secret_key_32 = bs58::encode(&secret_key_32_bytes).into_string();
        let secret_key_64 = bs58::encode(&secret_key_64_bytes).into_string();
        Wallet {
            public_key: public_key.to_string(),
            private_key_32: secret_key_32,
            private_key_64: secret_key_64,
        }
    }
    /// restore a wallet from a 64 bytes private key.
    /// # private_key
    /// * 64 bytes private key
    /// ```rust
    /// let w = Wallet::from_private_key_64("64 bytes private");
    /// ```
    pub fn from_private_key_64(private_key: &str) -> Wallet {
        let k = Keypair::from_base58_string(private_key);
        let private_key_32 = bs58::encode(k.secret_bytes()).into_string();
        Wallet {
            public_key: k.pubkey().to_string(),
            private_key_32: private_key_32,
            private_key_64: private_key.to_string(),
        }
    }
    /// restore a wallet from a 32 bytes private key.
    /// # private_key
    /// * 64 bytes private key
    /// ```rust
    /// let w = Wallet::from_private_key_64("32 bytes private");
    /// ```
    pub fn from_private_key_32(private_key: &str) -> Wallet {
        match private_key_base58_to_bytes(private_key) {
            Ok(v) => {
                if v.len() == 32 {
                    let mut buf = [0u8; 32];
                    buf.copy_from_slice(&v[..]);
                    let k = Keypair::new_from_array(buf);
                    let w = Wallet {
                        public_key: k.pubkey().to_string(),
                        private_key_32: private_key.to_string(),
                        private_key_64: bs58::encode(k.to_bytes()).into_string(),
                    };
                    return w;
                } else {
                    return Wallet::default();
                }
            }
            Err(_) => {
                return Wallet::default();
            }
        }
    }
}

pub mod tool {
    pub fn private_key_base58_to_bytes(private_key: &str) -> Result<Vec<u8>, String> {
        match bs58::decode(private_key).into_vec() {
            Ok(v) => return Ok(v),
            Err(_) => return Err("base58 decode error".to_string()),
        }
    }
}

impl Default for Wallet {
    fn default() -> Self {
        Self {
            public_key: "".to_string(),
            private_key_32: "".to_string(),
            private_key_64: "".to_string(),
        }
    }
}