use crate::bot::{Context, Command};
use async_trait::async_trait;
use reqwest::Client;
use serde_json::Value;

#[derive(Default)]
pub struct Waifu {
    http: Client
}

#[async_trait]
impl Command for Waifu {
    async fn execute(&mut self, msg: Context<'_>) -> anyhow::Result<String> {
        let category = msg.content.unwrap_or("waifu");
        let request = self.http.get(format!("https://api.waifu.pics/sfw/{}", category)).build()?;
        let response = self.http.execute(request)
            .await?
            .text()
            .await?;
        let response = response.trim();
        let value: Value = serde_json::from_str(response)?;
        let url = value["url"]
            .as_str()
            .unwrap_or("Invalid API Response.")
            .to_string();
        Ok(url)
    }
}