use marisa_ffi::{Trie, Keyset, Agent};

fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Create a keyset and add some keys
    let mut keyset = Keyset::new()?;
    keyset.push("hello")?;
    keyset.push("world")?;
    keyset.push("rust")?;

    // Build the trie
    let trie = Trie::build(&keyset)?;

    // Lookup a key
    if let Some(id) = trie.lookup("hello") {
        println!("Found 'hello' with ID: {}", id);
    }

    // Predictive search
    let mut agent = Agent::new()?;
    if trie.predictive_search("h", &mut agent)? {
        println!("Found key: {}", agent.key()?);
    }

    Ok(())
}

impl Trie {
    /// Create a new empty trie
    pub fn new() -> Result<Self, MarisaError>;
    
    /// Build a trie from a keyset
    pub fn build(keyset: &Keyset) -> Result<Self, MarisaError>;
    
    /// Load a trie from a file
    pub fn load(&mut self, filename: &str) -> Result<(), MarisaError>;
    
    /// Save the trie to a file
    pub fn save(&self, filename: &str) -> Result<(), MarisaError>;
    
    /// Lookup a key and return its ID if found
    pub fn lookup(&self, key: &str) -> Option<u32>;
    
    /// Perform predictive search starting with the given prefix
    pub fn predictive_search(&self, prefix: &str, agent: &mut Agent) -> Result<bool, MarisaError>;
    
    /// Perform reverse lookup to get a key by its ID
    pub fn reverse_lookup(&self, id: u32, agent: &mut Agent) -> Result<(), MarisaError>;
    
    /// Perform common prefix search
    pub fn common_prefix_search(&self, prefix: &str, agent: &mut Agent) -> Result<bool, MarisaError>;
}

impl Keyset {
    /// Create a new empty keyset
    pub fn new() -> Result<Self, MarisaError>;
    
    /// Add a key to the keyset
    pub fn push(&mut self, key: &str) -> Result<(), MarisaError>;
    
    /// Add a key with a specific ID to the keyset
    pub fn push_with_id(&mut self, key: &str, id: u32) -> Result<(), MarisaError>;
    
    /// Reset the keyset
    pub fn reset(&mut self);
}

impl Agent {
    /// Create a new agent
    pub fn new() -> Result<Self, MarisaError>;
    
    /// Get the current key as a string
    pub fn key(&self) -> Result<String, MarisaError>;
    
    /// Get the current key ID
    pub fn id(&self) -> u32;
    
    /// Move to the next result
    pub fn next(&mut self) -> Result<bool, MarisaError>;
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum MarisaError {
    StateError,
    NullError,
    BoundError,
    RangeError,
    CodeError,
    ResetError,
    SizeError,
    MemoryError,
    IoError,
    FormatError,
    KeyNotFound,
    NoResults,
    Utf8Error(std::str::Utf8Error),
}