272 lines
9.9 KiB
Rust
272 lines
9.9 KiB
Rust
|
use crate::{
|
||
|
|
input_parsers::*,
|
||
|
|
input_validators::*,
|
||
|
|
wallet::{
|
||
|
|
check_account_for_fee, check_unique_pubkeys, log_instruction_custom_error, ProcessResult,
|
||
|
|
WalletCommand, WalletConfig, WalletError,
|
||
|
|
},
|
||
|
|
};
|
||
|
|
use clap::{App, Arg, ArgMatches, SubCommand};
|
||
|
|
use solana_client::rpc_client::RpcClient;
|
||
|
|
use solana_sdk::{
|
||
|
|
account_utils::State, message::Message, pubkey::Pubkey, signature::KeypairUtil,
|
||
|
|
system_instruction::SystemError, transaction::Transaction,
|
||
|
|
};
|
||
|
|
use solana_storage_api::storage_instruction::{self, StorageAccountType};
|
||
|
|
|
||
|
|
pub trait StorageSubCommands {
|
||
|
|
fn storage_subcommands(self) -> Self;
|
||
|
|
}
|
||
|
|
|
||
|
|
impl StorageSubCommands for App<'_, '_> {
|
||
|
|
fn storage_subcommands(self) -> Self {
|
||
|
|
self.subcommand(
|
||
|
|
SubCommand::with_name("create-replicator-storage-account")
|
||
|
|
.about("Create a replicator storage account")
|
||
|
|
.arg(
|
||
|
|
Arg::with_name("storage_account_owner")
|
||
|
|
.index(1)
|
||
|
|
.value_name("STORAGE ACCOUNT OWNER PUBKEY")
|
||
|
|
.takes_value(true)
|
||
|
|
.required(true)
|
||
|
|
.validator(is_pubkey_or_keypair),
|
||
|
|
)
|
||
|
|
.arg(
|
||
|
|
Arg::with_name("storage_account_pubkey")
|
||
|
|
.index(2)
|
||
|
|
.value_name("STORAGE ACCOUNT PUBKEY")
|
||
|
|
.takes_value(true)
|
||
|
|
.required(true)
|
||
|
|
.validator(is_pubkey_or_keypair),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
.subcommand(
|
||
|
|
SubCommand::with_name("create-validator-storage-account")
|
||
|
|
.about("Create a validator storage account")
|
||
|
|
.arg(
|
||
|
|
Arg::with_name("storage_account_owner")
|
||
|
|
.index(1)
|
||
|
|
.value_name("STORAGE ACCOUNT OWNER PUBKEY")
|
||
|
|
.takes_value(true)
|
||
|
|
.required(true)
|
||
|
|
.validator(is_pubkey_or_keypair),
|
||
|
|
)
|
||
|
|
.arg(
|
||
|
|
Arg::with_name("storage_account_pubkey")
|
||
|
|
.index(2)
|
||
|
|
.value_name("STORAGE ACCOUNT PUBKEY")
|
||
|
|
.takes_value(true)
|
||
|
|
.required(true)
|
||
|
|
.validator(is_pubkey_or_keypair),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
.subcommand(
|
||
|
|
SubCommand::with_name("claim-storage-reward")
|
||
|
|
.about("Redeem storage reward credits")
|
||
|
|
.arg(
|
||
|
|
Arg::with_name("node_account_pubkey")
|
||
|
|
.index(1)
|
||
|
|
.value_name("NODE PUBKEY")
|
||
|
|
.takes_value(true)
|
||
|
|
.required(true)
|
||
|
|
.validator(is_pubkey_or_keypair)
|
||
|
|
.help("The node account to credit the rewards to"),
|
||
|
|
)
|
||
|
|
.arg(
|
||
|
|
Arg::with_name("storage_account_pubkey")
|
||
|
|
.index(2)
|
||
|
|
.value_name("STORAGE ACCOUNT PUBKEY")
|
||
|
|
.takes_value(true)
|
||
|
|
.required(true)
|
||
|
|
.validator(is_pubkey_or_keypair)
|
||
|
|
.help("Storage account address to redeem credits for"),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
.subcommand(
|
||
|
|
SubCommand::with_name("show-storage-account")
|
||
|
|
.about("Show the contents of a storage account")
|
||
|
|
.arg(
|
||
|
|
Arg::with_name("storage_account_pubkey")
|
||
|
|
.index(1)
|
||
|
|
.value_name("STORAGE ACCOUNT PUBKEY")
|
||
|
|
.takes_value(true)
|
||
|
|
.required(true)
|
||
|
|
.validator(is_pubkey_or_keypair)
|
||
|
|
.help("Storage account pubkey"),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn parse_storage_create_replicator_account(
|
||
|
|
matches: &ArgMatches<'_>,
|
||
|
|
) -> Result<WalletCommand, WalletError> {
|
||
|
|
let account_owner = pubkey_of(matches, "storage_account_owner").unwrap();
|
||
|
|
let storage_account_pubkey = pubkey_of(matches, "storage_account_pubkey").unwrap();
|
||
|
|
Ok(WalletCommand::CreateStorageAccount {
|
||
|
|
account_owner,
|
||
|
|
storage_account_pubkey,
|
||
|
|
account_type: StorageAccountType::Replicator,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn parse_storage_create_validator_account(
|
||
|
|
matches: &ArgMatches<'_>,
|
||
|
|
) -> Result<WalletCommand, WalletError> {
|
||
|
|
let account_owner = pubkey_of(matches, "storage_account_owner").unwrap();
|
||
|
|
let storage_account_pubkey = pubkey_of(matches, "storage_account_pubkey").unwrap();
|
||
|
|
Ok(WalletCommand::CreateStorageAccount {
|
||
|
|
account_owner,
|
||
|
|
storage_account_pubkey,
|
||
|
|
account_type: StorageAccountType::Validator,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn parse_storage_claim_reward(matches: &ArgMatches<'_>) -> Result<WalletCommand, WalletError> {
|
||
|
|
let node_account_pubkey = pubkey_of(matches, "node_account_pubkey").unwrap();
|
||
|
|
let storage_account_pubkey = pubkey_of(matches, "storage_account_pubkey").unwrap();
|
||
|
|
Ok(WalletCommand::ClaimStorageReward {
|
||
|
|
node_account_pubkey,
|
||
|
|
storage_account_pubkey,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn parse_storage_get_account_command(
|
||
|
|
matches: &ArgMatches<'_>,
|
||
|
|
) -> Result<WalletCommand, WalletError> {
|
||
|
|
let storage_account_pubkey = pubkey_of(matches, "storage_account_pubkey").unwrap();
|
||
|
|
Ok(WalletCommand::ShowStorageAccount(storage_account_pubkey))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn process_create_storage_account(
|
||
|
|
rpc_client: &RpcClient,
|
||
|
|
config: &WalletConfig,
|
||
|
|
account_owner: &Pubkey,
|
||
|
|
storage_account_pubkey: &Pubkey,
|
||
|
|
account_type: StorageAccountType,
|
||
|
|
) -> ProcessResult {
|
||
|
|
check_unique_pubkeys(
|
||
|
|
(&config.keypair.pubkey(), "wallet keypair".to_string()),
|
||
|
|
(
|
||
|
|
&storage_account_pubkey,
|
||
|
|
"storage_account_pubkey".to_string(),
|
||
|
|
),
|
||
|
|
)?;
|
||
|
|
let (recent_blockhash, fee_calculator) = rpc_client.get_recent_blockhash()?;
|
||
|
|
let ixs = storage_instruction::create_storage_account(
|
||
|
|
&config.keypair.pubkey(),
|
||
|
|
&account_owner,
|
||
|
|
storage_account_pubkey,
|
||
|
|
1,
|
||
|
|
account_type,
|
||
|
|
);
|
||
|
|
let mut tx = Transaction::new_signed_instructions(&[&config.keypair], ixs, recent_blockhash);
|
||
|
|
check_account_for_fee(rpc_client, config, &fee_calculator, &tx.message)?;
|
||
|
|
let result = rpc_client.send_and_confirm_transaction(&mut tx, &[&config.keypair]);
|
||
|
|
log_instruction_custom_error::<SystemError>(result)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn process_claim_storage_reward(
|
||
|
|
rpc_client: &RpcClient,
|
||
|
|
config: &WalletConfig,
|
||
|
|
node_account_pubkey: &Pubkey,
|
||
|
|
storage_account_pubkey: &Pubkey,
|
||
|
|
) -> ProcessResult {
|
||
|
|
let (recent_blockhash, fee_calculator) = rpc_client.get_recent_blockhash()?;
|
||
|
|
|
||
|
|
let instruction =
|
||
|
|
storage_instruction::claim_reward(node_account_pubkey, storage_account_pubkey);
|
||
|
|
let signers = [&config.keypair];
|
||
|
|
let message = Message::new_with_payer(vec![instruction], Some(&signers[0].pubkey()));
|
||
|
|
|
||
|
|
let mut tx = Transaction::new(&signers, message, recent_blockhash);
|
||
|
|
check_account_for_fee(rpc_client, config, &fee_calculator, &tx.message)?;
|
||
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &signers)?;
|
||
|
|
Ok(signature_str.to_string())
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn process_show_storage_account(
|
||
|
|
rpc_client: &RpcClient,
|
||
|
|
_config: &WalletConfig,
|
||
|
|
storage_account_pubkey: &Pubkey,
|
||
|
|
) -> ProcessResult {
|
||
|
|
let account = rpc_client.get_account(storage_account_pubkey)?;
|
||
|
|
|
||
|
|
if account.owner != solana_storage_api::id() {
|
||
|
|
return Err(WalletError::RpcRequestError(
|
||
|
|
format!("{:?} is not a storage account", storage_account_pubkey).to_string(),
|
||
|
|
)
|
||
|
|
.into());
|
||
|
|
}
|
||
|
|
|
||
|
|
use solana_storage_api::storage_contract::StorageContract;
|
||
|
|
let storage_contract: StorageContract = account.state().map_err(|err| {
|
||
|
|
WalletError::RpcRequestError(
|
||
|
|
format!("Unable to deserialize storage account: {:?}", err).to_string(),
|
||
|
|
)
|
||
|
|
})?;
|
||
|
|
println!("{:#?}", storage_contract);
|
||
|
|
println!("account lamports: {}", account.lamports);
|
||
|
|
Ok("".to_string())
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
use super::*;
|
||
|
|
use crate::wallet::{app, parse_command};
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_parse_command() {
|
||
|
|
let test_commands = app("test", "desc", "version");
|
||
|
|
let pubkey = Pubkey::new_rand();
|
||
|
|
let pubkey_string = pubkey.to_string();
|
||
|
|
let storage_account_pubkey = Pubkey::new_rand();
|
||
|
|
let storage_account_string = storage_account_pubkey.to_string();
|
||
|
|
|
||
|
|
let test_create_replicator_storage_account = test_commands.clone().get_matches_from(vec![
|
||
|
|
"test",
|
||
|
|
"create-replicator-storage-account",
|
||
|
|
&pubkey_string,
|
||
|
|
&storage_account_string,
|
||
|
|
]);
|
||
|
|
assert_eq!(
|
||
|
|
parse_command(&pubkey, &test_create_replicator_storage_account).unwrap(),
|
||
|
|
WalletCommand::CreateStorageAccount {
|
||
|
|
account_owner: pubkey,
|
||
|
|
storage_account_pubkey,
|
||
|
|
account_type: StorageAccountType::Replicator,
|
||
|
|
}
|
||
|
|
);
|
||
|
|
|
||
|
|
let test_create_validator_storage_account = test_commands.clone().get_matches_from(vec![
|
||
|
|
"test",
|
||
|
|
"create-validator-storage-account",
|
||
|
|
&pubkey_string,
|
||
|
|
&storage_account_string,
|
||
|
|
]);
|
||
|
|
assert_eq!(
|
||
|
|
parse_command(&pubkey, &test_create_validator_storage_account).unwrap(),
|
||
|
|
WalletCommand::CreateStorageAccount {
|
||
|
|
account_owner: pubkey,
|
||
|
|
storage_account_pubkey,
|
||
|
|
account_type: StorageAccountType::Validator,
|
||
|
|
}
|
||
|
|
);
|
||
|
|
|
||
|
|
let test_claim_storage_reward = test_commands.clone().get_matches_from(vec![
|
||
|
|
"test",
|
||
|
|
"claim-storage-reward",
|
||
|
|
&pubkey_string,
|
||
|
|
&storage_account_string,
|
||
|
|
]);
|
||
|
|
assert_eq!(
|
||
|
|
parse_command(&pubkey, &test_claim_storage_reward).unwrap(),
|
||
|
|
WalletCommand::ClaimStorageReward {
|
||
|
|
node_account_pubkey: pubkey,
|
||
|
|
storage_account_pubkey,
|
||
|
|
}
|
||
|
|
);
|
||
|
|
}
|
||
|
|
// TODO: Add process tests
|
||
|
|
}
|