cleanup

program/src/open.rs

@@ -0,0 +1,196 @@
+use ore_api::prelude::*;
+use solana_program::program_pack::Pack;
+use spl_token_2022::instruction::AuthorityType;
+use steel::*;
+
+/// Opens a new block.
+pub fn process_open(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
+    // Parse data.
+    let args = Open::try_from_bytes(data)?;
+    let id = u64::from_le_bytes(args.id);
+
+    // Load accounts.
+    let clock = Clock::get()?;
+    let [signer_info, block_info, collateral_info, commitment_info, market_info, mint_base_info, mint_quote_info, vault_base_info, vault_quote_info, system_program, token_program, associated_token_program, rent_sysvar] =
+        accounts
+    else {
+        return Err(ProgramError::NotEnoughAccountKeys);
+    };
+    signer_info.is_signer()?;
+    block_info
+        .is_empty()?
+        .is_writable()?
+        .has_seeds(&[BLOCK, &id.to_le_bytes()], &ore_api::ID)?;
+    market_info
+        .is_empty()?
+        .is_writable()?
+        .has_seeds(&[MARKET, &id.to_le_bytes()], &ore_api::ID)?;
+    mint_base_info
+        .is_empty()?
+        .is_writable()?
+        .has_seeds(&[MINT, &id.to_le_bytes()], &ore_api::ID)?;
+    mint_quote_info.has_address(&MINT_ADDRESS)?.as_mint()?;
+    system_program.is_program(&system_program::ID)?;
+    token_program.is_program(&spl_token::ID)?;
+    associated_token_program.is_program(&spl_associated_token_account::ID)?;
+    rent_sysvar.is_sysvar(&sysvar::rent::ID)?;
+
+    // Error out if start slot is within the current period.
+    let start_slot = id * 1500;
+    let current_period_start = (clock.slot / 1500) * 1500;
+    let current_period_end = current_period_start + 1500;
+    if start_slot < current_period_end {
+        return Err(ProgramError::InvalidArgument);
+    }
+
+    // Initialize config.
+    create_program_account::<Block>(
+        block_info,
+        system_program,
+        signer_info,
+        &ore_api::ID,
+        &[BLOCK, &id.to_le_bytes()],
+    )?;
+    let block = block_info.as_account_mut::<Block>(&ore_api::ID)?;
+    block.id = id;
+    block.min_difficulty = MIN_DIFFICULTY;
+    block.reward_rate = REWARD_RATE;
+    block.slot_hash = [0; 32];
+    block.start_slot = start_slot;
+    block.total_hashes = 0;
+    block.winning_hashes = 0;
+
+    // Initialize market.
+    create_program_account::<Market>(
+        market_info,
+        system_program,
+        signer_info,
+        &ore_api::ID,
+        &[MARKET, &id.to_le_bytes()],
+    )?;
+    let market = market_info.as_account_mut::<Market>(&ore_api::ID)?;
+    market.base = TokenParams {
+        mint: *mint_base_info.key,
+        balance: HASH_TOKEN_SUPPLY,
+        balance_virtual: 0,
+    };
+    market.quote = TokenParams {
+        mint: *mint_quote_info.key,
+        balance: 0,
+        balance_virtual: VIRTUAL_LIQUIDITY,
+    };
+    market.fee = FeeParams {
+        rate: FEE_RATE_BPS,
+        uncollected: 0,
+        cumulative: 0,
+    };
+    market.snapshot = Snapshot {
+        enabled: 1,
+        base_balance: 0,
+        quote_balance: 0,
+        slot: 0,
+    };
+    market.id = id;
+
+    // Initialize hash token mint.
+    let mint_bump = mint_pda(block.id).1;
+    allocate_account_with_bump(
+        mint_base_info,
+        system_program,
+        signer_info,
+        spl_token::state::Mint::LEN,
+        &spl_token::ID,
+        &[MINT, &id.to_le_bytes()],
+        mint_bump,
+    )?;
+    initialize_mint_signed_with_bump(
+        mint_base_info,
+        block_info,
+        None,
+        token_program,
+        rent_sysvar,
+        0,
+        &[MINT, &id.to_le_bytes()],
+        mint_bump,
+    )?;
+
+    // TODO Initialize hash token metadata.
+
+    // Initialize collateral and commitment token accounts.
+    if collateral_info.data_is_empty() {
+        create_associated_token_account(
+            signer_info,
+            block_info,
+            collateral_info,
+            mint_quote_info,
+            system_program,
+            token_program,
+            associated_token_program,
+        )?;
+    } else {
+        collateral_info.as_associated_token_account(block_info.key, mint_quote_info.key)?;
+    }
+    if commitment_info.data_is_empty() {
+        create_associated_token_account(
+            signer_info,
+            block_info,
+            commitment_info,
+            mint_base_info,
+            system_program,
+            token_program,
+            associated_token_program,
+        )?;
+    } else {
+        commitment_info.as_associated_token_account(block_info.key, mint_base_info.key)?;
+    }
+
+    // Initialize vault token accounts.
+    if vault_base_info.data_is_empty() {
+        create_associated_token_account(
+            signer_info,
+            market_info,
+            vault_base_info,
+            mint_base_info,
+            system_program,
+            token_program,
+            associated_token_program,
+        )?;
+    } else {
+        vault_base_info.as_associated_token_account(market_info.key, mint_base_info.key)?;
+    }
+    if vault_quote_info.data_is_empty() {
+        create_associated_token_account(
+            signer_info,
+            market_info,
+            vault_quote_info,
+            mint_quote_info,
+            system_program,
+            token_program,
+            associated_token_program,
+        )?;
+    } else {
+        vault_quote_info.as_associated_token_account(market_info.key, mint_quote_info.key)?;
+    }
+
+    // Mint hash tokens to market.
+    mint_to_signed(
+        mint_base_info,
+        vault_base_info,
+        block_info,
+        token_program,
+        HASH_TOKEN_SUPPLY,
+        &[BLOCK, &id.to_le_bytes()],
+    )?;
+
+    // Burn mint authority.
+    set_authority_signed(
+        mint_base_info,
+        block_info,
+        None,
+        AuthorityType::MintTokens,
+        token_program,
+        &[BLOCK, &id.to_le_bytes()],
+    )?;
+
+    Ok(())
+}