initial commit

rs/src/cluster1.rs

@@ -0,0 +1,4 @@
+#[cfg(test)]
+mod cluster1_test {
+    // TODO
+}

rs/src/lib.rs

@@ -0,0 +1,3 @@
+mod programs;
+mod prereqs;
+mod cluster1;

rs/src/prereqs.rs

@@ -0,0 +1,208 @@
+#[cfg(test)]
+mod prereqs_tests {
+    use solana_client::rpc_client::RpcClient;
+    use solana_program::{
+        pubkey::Pubkey,
+        system_instruction::transfer, system_program
+    };
+    use solana_sdk::{
+        message::Message,
+        signature::{Keypair, Signer, read_keypair_file},
+        transaction::Transaction
+    };
+    use std::str::FromStr;
+    use bs58;
+    use std::io::{self, BufRead};
+
+    use crate::programs::wba_prereq::{WbaPrereqProgram, CompleteArgs, UpdateArgs};
+
+    const RPC_URL: &str = "https://api.devnet.solana.com";
+
+    #[test]
+    fn keygen() {
+        // Create a new keypair
+        let keypair = Keypair::new();
+        println!("You've generated a new Solana wallet: {}", keypair.pubkey().to_string());
+        println!("");
+        println!("To save your wallet, copy and paste the following into a JSON file:");
+        println!("{:?}", keypair.to_bytes());
+    }
+
+    #[test]
+    fn base58_to_wallet() {
+        println!("Input your private key as base58:");
+        let stdin = io::stdin();
+        let base58 = stdin.lock().lines().next().unwrap().unwrap();
+        println!("Your wallet file is:");
+        let wallet = bs58::decode(base58).into_vec().unwrap();
+        println!("{:?}", wallet);
+    }
+
+    #[test]
+    fn wallet_to_base58() {
+        println!("Input your private key as a wallet file byte array:");
+        let stdin = io::stdin();
+        let wallet = stdin.lock().lines().next().unwrap().unwrap().trim_start_matches('[').trim_end_matches(']').split(',').map(|s| s.trim().parse::<u8>().unwrap()).collect::<Vec<u8>>();
+        println!("Your private key is:");
+        let base58 = bs58::encode(wallet).into_string();
+        println!("{:?}", base58);
+    }
+
+    #[test]
+    fn airdop() {
+        // Import our keypair
+        let keypair = read_keypair_file("dev-wallet.json").expect("Couldn't find wallet file");
+
+        // Connected to Solana Devnet RPC Client
+        let client = RpcClient::new(RPC_URL);
+        
+        // We're going to claim 2 devnet SOL tokens (2 billion lamports)
+        match client.request_airdrop(&keypair.pubkey(), 2_000_000_000u64) {
+            Ok(s) => {
+                println!("Success! Check out your TX here:");
+                println!("https://explorer.solana.com/tx/{}?cluster=devnet", s.to_string());
+            },
+            Err(e) => println!("Oops, something went wrong: {}", e.to_string())
+        };
+    }
+
+    #[test]
+    fn transfer_sol() {
+        // Import our keypair
+        let keypair = read_keypair_file("dev-wallet.json").expect("Couldn't find wallet file");
+
+        // Define our WBA public key
+        let to_pubkey =
+            Pubkey::from_str("GLtaTaYiTQrgz411iPJD79rsoee59HhEy18rtRdrhEUJ").unwrap();
+
+        // Create a Solana devnet connection
+        let rpc_client = RpcClient::new(RPC_URL);
+
+        // Get balance of dev wallet
+        let balance = rpc_client
+            .get_balance(&keypair.pubkey())
+            .expect("Failed to get balance");
+
+        println!("Balance: {}", balance);
+
+        // Get recent blockhash
+        let recent_blockhash = rpc_client
+        .get_latest_blockhash()
+        .expect("Failed to get recent blockhash");
+
+        // Create a test transaction to calculate fees
+        let message = Message::new_with_blockhash(
+            &[transfer(
+                &keypair.pubkey(),
+                &to_pubkey,
+                balance,
+            )], 
+            Some(&keypair.pubkey()),
+            &recent_blockhash
+        );
+
+        // Calculate exact fee rate to transfer entire SOL amount out of account minus fees
+        let fee = rpc_client
+            .get_fee_for_message(&message)
+            .expect("Failed to get fee calculator");
+        println!("fee: {}", fee);
+
+        // Deduct fee from lamports amount and create a TX with correct balance
+        let transaction = Transaction::new_signed_with_payer(
+            &[transfer(
+                &keypair.pubkey(),
+                &to_pubkey,
+                balance - fee,
+            )], 
+            Some(&keypair.pubkey()),
+            &vec![&keypair],
+            recent_blockhash
+        );
+
+        // Send the transaction
+        let signature = rpc_client
+            .send_and_confirm_transaction(&transaction)
+            .expect("Failed to send transaction");
+
+        // Print our transaction out
+        println!(
+            "Success! Check out your TX here: 
+            https://explorer.solana.com/tx/{}/?cluster=devnet",
+            signature
+        );
+    }
+
+    #[test]
+    fn enroll() {
+        // Create a Solana devnet connection
+        let rpc_client = RpcClient::new(RPC_URL);
+
+        // Let's define all our accounts
+        let signer = read_keypair_file("wba-wallet.json").expect("Couldn't find wallet file");
+        let prereq = WbaPrereqProgram::derive_program_address(&[b"prereq", signer.pubkey().to_bytes().as_ref()]);
+
+        // Define our instruction data
+        let args = CompleteArgs {
+            github: b"testaccount".to_vec()
+        };
+
+        // Get recent blockhash
+        let recent_blockhash = rpc_client
+        .get_latest_blockhash()
+        .expect("Failed to get recent blockhash");        
+
+        // Now we can invoke the "complete" function
+        let transaction = WbaPrereqProgram::complete(
+            &[&signer.pubkey(), &prereq, &system_program::id()],
+            &args,
+            Some(&signer.pubkey()),           
+            &[&signer],
+            recent_blockhash
+        );
+
+        // Send the transaction
+        let signature = rpc_client
+        .send_and_confirm_transaction(&transaction)
+        .expect("Failed to send transaction");
+
+        // Print our transaction out
+        println!("Success! Check out your TX here: https://explorer.solana.com/tx/{}/?cluster=devnet", signature);
+    }
+
+    #[test]
+    fn update() {
+        // Create a Solana devnet connection
+        let rpc_client = RpcClient::new(RPC_URL);
+
+        // Let's define all our accounts
+        let signer = read_keypair_file("wba-wallet.json").expect("Couldn't find wallet file");
+        let prereq = WbaPrereqProgram::derive_program_address(&[b"prereq", signer.pubkey().to_bytes().as_ref()]);
+
+        // Define our instruction data
+        let args = UpdateArgs {
+            github: b"testaccount".to_vec()
+        };
+
+        // Get recent blockhash
+        let recent_blockhash = rpc_client
+        .get_latest_blockhash()
+        .expect("Failed to get recent blockhash");        
+
+        // Now we can invoke the "complete" function
+        let transaction = WbaPrereqProgram::update(
+            &[&signer.pubkey(), &prereq, &system_program::id()],
+            &args,
+            Some(&signer.pubkey()),           
+            &[&signer],
+            recent_blockhash
+        );
+
+        // Send the transaction
+        let signature = rpc_client
+        .send_and_confirm_transaction(&transaction)
+        .expect("Failed to send transaction");
+
+        // Print our transaction out
+        println!("Success! Check out your TX here: https://explorer.solana.com/tx/{}/?cluster=devnet", signature);
+    }
+}

rs/src/programs/mod.rs

@@ -0,0 +1,2 @@
+// Programs
+pub mod wba_prereq;

rs/src/programs/wba_prereq.rs

@@ -0,0 +1,80 @@
+use solana_idlgen::idlgen;
+idlgen!({
+    "version": "0.1.0",
+    "name": "wba_prereq",
+    "instructions": [
+        {
+        "name": "complete",
+        "accounts": [
+            {
+            "name": "signer",
+            "isMut": true,
+            "isSigner": true
+            },
+            {
+            "name": "prereq",
+            "isMut": true,
+            "isSigner": false
+            },
+            {
+            "name": "systemProgram",
+            "isMut": false,
+            "isSigner": false
+            }
+        ],
+        "args": [
+            {
+            "name": "github",
+            "type": "bytes"
+            }
+        ]
+        },
+        {
+        "name": "update",
+        "accounts": [
+            {
+            "name": "signer",
+            "isMut": true,
+            "isSigner": true
+            },
+            {
+            "name": "prereq",
+            "isMut": true,
+            "isSigner": false
+            },
+            {
+            "name": "systemProgram",
+            "isMut": false,
+            "isSigner": false
+            }
+        ],
+        "args": [
+            {
+            "name": "github",
+            "type": "bytes"
+            }
+        ]
+        }
+    ],
+    "accounts": [
+        {
+        "name": "PrereqAccount",
+        "type": {
+            "kind": "struct",
+            "fields": [
+            {
+                "name": "github",
+                "type": "bytes"
+            },
+            {
+                "name": "key",
+                "type": "publicKey"
+            }
+            ]
+        }
+        }
+    ],
+    "metadata": {
+        "address": "HC2oqz2p6DEWfrahenqdq2moUcga9c9biqRBcdK3XKU1"
+    }
+});