-
Notifications
You must be signed in to change notification settings - Fork 220
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* add spl airdrop * bump to Jib 0.5.0; include priority fee option * fix Cargo toml file
- Loading branch information
1 parent
44c6c43
commit 84ee7cb
Showing
7 changed files
with
412 additions
and
119 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,109 @@ | ||
| use super::*; | ||
|
|
||
| pub struct AirdropSolArgs { | ||
| pub client: RpcClient, | ||
| pub keypair: Option<String>, | ||
| pub network: Network, | ||
| pub recipient_list: Option<String>, | ||
| pub cache_file: Option<String>, | ||
| } | ||
|
|
||
| pub async fn airdrop_sol(args: AirdropSolArgs) -> Result<()> { | ||
| let solana_opts = parse_solana_config(); | ||
| let keypair = parse_keypair(args.keypair, solana_opts); | ||
|
|
||
| let mut jib = Jib::new(vec![keypair], args.network)?; | ||
|
|
||
| let mut instructions = vec![]; | ||
|
|
||
| if args.recipient_list.is_some() && args.cache_file.is_some() { | ||
| eprintln!("Cannot provide both a recipient list and a cache file."); | ||
| std::process::exit(1); | ||
| } | ||
|
|
||
| // Get the current time as yyyy-mm-dd-hh-mm-ss | ||
| let now = chrono::Local::now(); | ||
| let timestamp = now.format("%Y-%m-%d-%H-%M-%S").to_string(); | ||
|
|
||
| let mut cache_file_name = format!("mb-cache-airdrop-{timestamp}.json"); | ||
| let successful_tx_file_name = format!("mb-successful-airdrops-{timestamp}.json"); | ||
|
|
||
| let mut airdrop_list: HashMap<String, u64> = if let Some(list_file) = args.recipient_list { | ||
| serde_json::from_reader(File::open(list_file)?)? | ||
| } else if let Some(cache_file) = args.cache_file { | ||
| cache_file_name = PathBuf::from(cache_file.clone()) | ||
| .file_name() | ||
| .unwrap() | ||
| .to_str() | ||
| .unwrap() | ||
| .to_string(); | ||
|
|
||
| let failed_txes: Vec<FailedTransaction> = serde_json::from_reader(File::open(cache_file)?)?; | ||
| failed_txes | ||
| .iter() | ||
| .flat_map(|f| f.recipients.clone()) | ||
| .collect() | ||
| } else { | ||
| eprintln!("No recipient list or cache file provided."); | ||
| std::process::exit(1); | ||
| }; | ||
|
|
||
| for (address, amount) in &airdrop_list { | ||
| let pubkey = match Pubkey::from_str(address) { | ||
| Ok(pubkey) => pubkey, | ||
| Err(_) => { | ||
| eprintln!("Invalid address: {}, skipping...", address); | ||
| continue; | ||
| } | ||
| }; | ||
|
|
||
| instructions.push(solana_sdk::system_instruction::transfer( | ||
| &jib.payer().pubkey(), | ||
| &pubkey, | ||
| *amount, | ||
| )); | ||
| } | ||
|
|
||
| jib.set_instructions(instructions); | ||
| let results = jib.hoist()?; | ||
|
|
||
| if results.iter().any(|r| r.is_failure()) { | ||
| println!("Some transactions failed. Check the {cache_file_name} cache file for details."); | ||
| } | ||
|
|
||
| let mut successes = vec![]; | ||
| let mut failures = vec![]; | ||
|
|
||
| results.iter().for_each(|r| { | ||
| if r.is_failure() { | ||
| let tx = r.transaction().unwrap(); // Transactions exist on failures. | ||
| let account_keys = tx.message().account_keys.clone(); | ||
| let transaction_accounts = account_keys.iter().map(|k| k.to_string()).collect(); | ||
|
|
||
| // All accounts except the first and last are recipients. | ||
| let recipients: HashMap<String, u64> = account_keys[1..account_keys.len() - 1] | ||
| .iter() | ||
| .map(|pubkey| pubkey.to_string()) | ||
| .map(|a| airdrop_list.remove_entry(&a).expect("Recipient not found")) | ||
| .collect(); | ||
|
|
||
| failures.push(FailedTransaction { | ||
| transaction_accounts, | ||
| recipients, | ||
| error: r.error().unwrap(), // Errors exist on failures. | ||
| }) | ||
| } else { | ||
| debug!("Transaction successful: {}", r.signature().unwrap()); // Signatures exist on successes. | ||
| successes.push(r.signature().unwrap()); // Signatures exist on successes. | ||
| } | ||
| }); | ||
|
|
||
| // Write cache file and successful transactions. | ||
| let successful_tx_file = std::fs::File::create(successful_tx_file_name)?; | ||
| serde_json::to_writer_pretty(successful_tx_file, &successes)?; | ||
|
|
||
| let cache_file = std::fs::File::create(cache_file_name)?; | ||
| serde_json::to_writer_pretty(cache_file, &failures)?; | ||
|
|
||
| Ok(()) | ||
| } |
Oops, something went wrong.