Agent Marketplace

use hydraclaw_sdk::prelude::*;

#[agent]
pub struct CustomAgent {
    config: AgentConfig,
    strategy: Box<dyn Strategy>,
}

impl CustomAgent {
    pub fn new(config: AgentConfig) -> Self {
        Self {
            config,
            strategy: Box::new(MyCustomStrategy::new()),
        }
    }
}

#[async_trait]
impl Agent for CustomAgent {
    async fn on_new_token(&self, token: Token) -> Result<Action> {
        // Your custom logic here
        let analysis = self.strategy.analyze(&token).await?;

        if analysis.should_buy {
            Ok(Action::Buy {
                token: token.address,
                amount: self.config.max_position,
                slippage: self.config.slippage,
            })
        } else {
            Ok(Action::Skip)
        }
    }

    async fn on_price_change(&self, token: Token, change: f64) -> Result<Action> {
        // React to price movements
        if change > self.config.take_profit {
            Ok(Action::Sell { token: token.address, percentage: 100 })
        } else if change < -self.config.stop_loss {
            Ok(Action::Sell { token: token.address, percentage: 100 })
        } else {
            Ok(Action::Hold)
        }
    }
}