ai-registry 0.1.0

use ai_registry_api::{
    consts::{AGENT, FLAG_ACTIVE},
    event::ActiveStateChangedEvent,
    instruction::SetActiveState,
    state::Agent,
};
use steel::*;

/// Process a set active state instruction
pub fn process_set_active_state(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
    let args = SetActiveState::try_from_bytes(data)?;

    // Load accounts (signer first, following Solana best practices)
    let [controller_info, agent_info] = accounts else {
        return Err(ProgramError::NotEnoughAccountKeys);
    };

    // Validate accounts
    agent_info.is_writable()?.has_seeds(
        &[
            AGENT,
            agent_info
                .as_account::<Agent>(&ai_registry_api::ID)?
                .creator
                .as_ref(),
        ],
        &ai_registry_api::ID,
    )?;
    controller_info.is_signer()?;

    // Load agent and verify controller
    let agent = agent_info.as_account::<Agent>(&ai_registry_api::ID)?;
    if controller_info.key != &agent.controller {
        return Err(ai_registry_api::error::AgentError::UnauthorizedController.into());
    }

    // Update active state
    let clock = Clock::get()?;
    let agent = agent_info.as_account_mut::<Agent>(&ai_registry_api::ID)?;
    if args.is_active == 1 {
        agent.state_flags |= FLAG_ACTIVE;
    } else {
        agent.state_flags &= !FLAG_ACTIVE;
    }
    agent.updated_at = clock.unix_timestamp;

    // Emit event
    ActiveStateChangedEvent {
        creator: agent.creator,
        agent: *agent_info.key,
        is_active: args.is_active,
        timestamp: clock.unix_timestamp,
        _padding: [0; 7],
    }
    .log();

    Ok(())
}