This commit is contained in:
Hardhat Chad
2025-06-06 16:57:23 -07:00
parent 09ff016280
commit 4bd55ae1a8
6 changed files with 231 additions and 122 deletions

View File

@@ -20,6 +20,7 @@ default = []
[dependencies]
bincode.workspace = true
mpl-token-metadata.workspace = true
ore-api.workspace = true
ore-delegate-api.workspace = true
solana-program.workspace = true
spl-token.workspace = true

View File

@@ -1,69 +1,77 @@
use ore_api::prelude::*;
use ore_delegate_api::prelude::*;
use steel::*;
/// Deposits hash tokens for cranking.
pub fn process_deposit(accounts: &[AccountInfo<'_>], _data: &[u8]) -> ProgramResult {
pub fn process_deposit(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
// Load data.
let args = Deposit::try_from_bytes(data)?;
let amount = u64::from_le_bytes(args.amount);
// Load accounts.
// let clock = Clock::get()?;
// let [signer_info, block_info, market_info, mint_base_info, mint_quote_info, recipient_info, treasury_info, vault_base_info, vault_quote_info, system_program, token_program] =
// accounts
// else {
// return Err(ProgramError::NotEnoughAccountKeys);
// };
// signer_info.is_signer()?;
// let block = block_info
// .as_account_mut::<Block>(&ore_api::ID)?
// .assert_mut(|b| clock.slot >= b.start_slot + 1500)?;
// let market = market_info
// .as_account_mut::<Market>(&ore_api::ID)?
// .assert_mut(|m| m.id == block.id)?;
// mint_base_info.has_address(&market.base.mint)?.as_mint()?;
// mint_quote_info.has_address(&market.quote.mint)?.as_mint()?;
// let vault_base =
// vault_base_info.as_associated_token_account(market_info.key, mint_base_info.key)?;
// let vault_quote =
// vault_quote_info.as_associated_token_account(market_info.key, mint_quote_info.key)?;
// system_program.is_program(&system_program::ID)?;
// token_program.is_program(&spl_token::ID)?;
let [signer_info, block_info, delegate_info, escrow_info, market_info, mint_info, sender_info, system_program, token_program, associated_token_program] =
accounts
else {
return Err(ProgramError::NotEnoughAccountKeys);
};
signer_info.is_signer()?;
let block = block_info.as_account_mut::<Block>(&ore_api::ID)?;
let market = market_info
.as_account_mut::<Market>(&ore_api::ID)?
.assert_mut(|m| m.id == block.id)?;
mint_info.has_address(&market.base.mint)?.as_mint()?;
sender_info
.is_writable()?
.as_associated_token_account(signer_info.key, &market.base.mint)?
.assert_mut(|t| t.amount() >= amount)?;
system_program.is_program(&system_program::ID)?;
token_program.is_program(&spl_token::ID)?;
associated_token_program.is_program(&spl_associated_token_account::ID)?;
// // Payout block reward.
// if block.best_miner != Pubkey::default() {
// recipient_info.as_associated_token_account(&block.best_miner, &MINT_ADDRESS)?;
// mint_to_signed(
// mint_quote_info,
// recipient_info,
// treasury_info,
// token_program,
// block.reward,
// &[TREASURY],
// )?;
// }
// Initialize delegate.
let delegate = if delegate_info.data_is_empty() {
create_program_account::<Delegate>(
delegate_info,
system_program,
signer_info,
&ore_delegate_api::ID,
&[
DELEGATE,
&signer_info.key.to_bytes(),
&block.id.to_le_bytes(),
],
)?;
let delegate = delegate_info.as_account_mut::<Delegate>(&ore_delegate_api::ID)?;
delegate.authority = *signer_info.key;
delegate.block_id = block.id;
delegate.fee = 0; // TODO: Set fee.
delegate
} else {
delegate_info.as_account_mut::<Delegate>(&ore_delegate_api::ID)?
};
// // Burn base liquidity.
// burn_signed(
// vault_base_info,
// mint_base_info,
// market_info,
// token_program,
// vault_base.amount(),
// &[MARKET, &market.id.to_le_bytes()],
// )?;
// Initialize escrow.
if escrow_info.data_is_empty() {
create_associated_token_account(
signer_info,
delegate_info,
escrow_info,
mint_info,
system_program,
token_program,
associated_token_program,
)?;
} else {
escrow_info
.is_writable()?
.as_associated_token_account(delegate_info.key, &mint_info.key)?;
}
// // Burn quote liquidity.
// burn_signed(
// vault_quote_info,
// mint_quote_info,
// market_info,
// token_program,
// vault_quote.amount(),
// &[MARKET, &market.id.to_le_bytes()],
// )?;
// Update delegate.
delegate.balance += amount;
// // Close block.
// block_info.close(signer_info)?;
// // Close market.
// market_info.close(signer_info)?;
// Transfer tokens.
transfer(signer_info, sender_info, escrow_info, token_program, amount)?;
Ok(())
}

View File

@@ -1,79 +1,61 @@
use ore_delegate_api::prelude::*;
use solana_program::keccak;
use steel::*;
/// Withdraws hash tokens from the delegate.
pub fn process_withdraw(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
// Parse data.
// Load data.
let args = Withdraw::try_from_bytes(data)?;
let amount = u64::from_le_bytes(args.amount);
// Load accounts.
// let clock = Clock::get()?;
// let [signer_info, block_info, market_info, miner_info, mint_info, sender_info, system_program, token_program] =
// accounts
// else {
// return Err(ProgramError::NotEnoughAccountKeys);
// };
// signer_info.is_signer()?;
// let block = block_info
// .as_account_mut::<Block>(&ore_api::ID)?
// .assert_mut(|b| clock.slot >= b.start_slot)?
// .assert_mut(|b| clock.slot < b.start_slot + 1500)?;
// let market = market_info
// .as_account::<Market>(&ore_api::ID)?
// .assert(|m| m.id == block.id)?;
// mint_info.has_address(&market.base.mint)?.as_mint()?;
// sender_info
// .is_writable()?
// .as_associated_token_account(signer_info.key, &mint_info.key)?
// .assert(|t| t.amount() >= amount)?;
// system_program.is_program(&system_program::ID)?;
// token_program.is_program(&spl_token::ID)?;
let [signer_info, delegate_info, escrow_info, mint_info, recipient_info, system_program, token_program] =
accounts
else {
return Err(ProgramError::NotEnoughAccountKeys);
};
signer_info.is_signer()?;
let delegate = delegate_info
.as_account_mut::<Delegate>(&ore_delegate_api::ID)?
.assert_mut(|d| d.authority == *signer_info.key)?;
let escrow = escrow_info.as_associated_token_account(delegate_info.key, &mint_info.key)?;
recipient_info
.is_writable()?
.as_associated_token_account(signer_info.key, &mint_info.key)?;
system_program.is_program(&system_program::ID)?;
token_program.is_program(&spl_token::ID)?;
// // Load miner account.
// let miner = if miner_info.data_is_empty() {
// create_program_account::<Miner>(
// miner_info,
// system_program,
// signer_info,
// &ore_api::ID,
// &[MINER, &signer_info.key.to_bytes()],
// )?;
// let miner = miner_info.as_account_mut::<Miner>(&ore_api::ID)?;
// miner.authority = *signer_info.key;
// miner.block_id = 0;
// miner.hash = [0; 32];
// miner.total_hashes = 0;
// miner.total_rewards = 0;
// miner
// } else {
// miner_info
// .as_account_mut::<Miner>(&ore_api::ID)?
// .assert_mut(|m| m.authority == *signer_info.key)?
// };
// Update delegate.
delegate.balance -= escrow.amount().min(amount);
// // Reset miner hash if mining new block.
// if miner.block_id != block.id {
// miner.block_id = block.id;
// miner.hash =
// keccak::hashv(&[block.slot_hash.as_ref(), miner.authority.as_ref()]).to_bytes();
// }
// Transfer tokens.
transfer_signed(
signer_info,
escrow_info,
recipient_info,
token_program,
amount,
&[
DELEGATE,
&delegate.authority.to_bytes(),
&delegate.block_id.to_le_bytes(),
],
)?;
// // Mine.
// for _ in 0..amount {
// miner.hash = keccak::hashv(&[miner.hash.as_ref()]).to_bytes();
// if miner.hash < block.best_hash {
// block.best_hash = miner.hash;
// block.best_miner = miner.authority;
// }
// }
// // Update miner stats.
// miner.total_hashes += amount;
// // Burn hash tokens.
// burn(sender_info, mint_info, signer_info, token_program, amount)?;
// Close accounts if empty.
if delegate.balance == 0 {
delegate_info.close(signer_info)?;
close_token_account_signed(
escrow_info,
signer_info,
delegate_info,
token_program,
&[
DELEGATE,
&delegate.authority.to_bytes(),
&delegate.block_id.to_le_bytes(),
],
)?;
}
Ok(())
}