solidb 1.0.2

use crate::driver::protocol::{DriverError, Response};
use crate::driver::DriverHandler;
use crate::storage::{VectorIndexConfig, VectorMetric};

// ==================== Configuration Structs ====================

/// Configuration for geo_near query
pub struct GeoNearConfig {
    pub collection: String,
    pub field: String,
    pub latitude: f64,
    pub longitude: f64,
    pub radius: Option<f64>,
    pub limit: Option<i32>,
}

/// Configuration for creating a vector index
pub struct VectorIndexCreateConfig {
    pub collection: String,
    pub name: String,
    pub field: String,
    pub dimensions: i32,
    pub metric: Option<String>,
    pub ef_construction: Option<i32>,
    pub m: Option<i32>,
}

// ==================== Standard Index Operations ====================

pub fn handle_create_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    name: String,
    fields: Vec<String>,
    unique: bool,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => {
            // Default to Persistent index type
            let index_type = crate::storage::IndexType::Persistent;
            match coll.create_index(name, fields, index_type, unique) {
                Ok(stats) => Response::ok(serde_json::to_value(stats).unwrap_or_default()),
                Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
            }
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_delete_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    name: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => {
            // Try dropping as standard index first
            if coll.drop_index(&name).is_ok() {
                return Response::ok_empty();
            }
            // Try dropping as fulltext index
            if coll.drop_fulltext_index(&name).is_ok() {
                return Response::ok_empty();
            }
            // Try dropping as geo index
            if coll.drop_geo_index(&name).is_ok() {
                return Response::ok_empty();
            }
            // Try dropping as TTL index
            if coll.drop_ttl_index(&name).is_ok() {
                return Response::ok_empty();
            }
            Response::error(DriverError::DatabaseError(format!(
                "Index '{}' not found",
                name
            )))
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_list_indexes(
    handler: &DriverHandler,
    database: String,
    collection: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => {
            let indexes = coll.list_indexes();
            Response::ok(serde_json::to_value(indexes).unwrap_or_default())
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_rebuild_indexes(
    handler: &DriverHandler,
    database: String,
    collection: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => match coll.rebuild_all_indexes() {
            Ok(stats) => Response::ok(serde_json::to_value(stats).unwrap_or_default()),
            Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
        },
        Err(e) => Response::error(e),
    }
}

// ==================== Geo Index Operations ====================

pub fn handle_create_geo_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    name: String,
    field: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => match coll.create_geo_index(name, field) {
            Ok(_) => Response::ok_empty(),
            Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
        },
        Err(e) => Response::error(e),
    }
}

pub fn handle_list_geo_indexes(
    handler: &DriverHandler,
    database: String,
    collection: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => {
            let indexes = coll.list_geo_indexes();
            Response::ok(serde_json::to_value(indexes).unwrap_or_default())
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_delete_geo_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    name: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => match coll.drop_geo_index(&name) {
            Ok(_) => Response::ok_empty(),
            Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
        },
        Err(e) => Response::error(e),
    }
}

pub fn handle_geo_near(
    handler: &DriverHandler,
    database: String,
    config: GeoNearConfig,
) -> Response {
    match handler.get_collection(&database, &config.collection) {
        Ok(coll) => {
            let limit_val = config.limit.map(|l| l.max(0) as usize).unwrap_or(10);

            let results_opt = if let Some(r) = config.radius {
                coll.geo_within(&config.field, config.latitude, config.longitude, r)
                    .map(|mut res| {
                        if limit_val < res.len() {
                            res.truncate(limit_val);
                        }
                        res
                    })
            } else {
                coll.geo_near(&config.field, config.latitude, config.longitude, limit_val)
            };

            match results_opt {
                Some(results) => Response::ok(serde_json::json!(results)),
                None => Response::error(DriverError::DatabaseError(
                    "Geo index not found".to_string(),
                )),
            }
        }
        Err(e) => Response::error(e),
    }
}

// ==================== Vector Index Operations ====================

pub fn handle_create_vector_index(
    handler: &DriverHandler,
    database: String,
    config: VectorIndexCreateConfig,
) -> Response {
    match handler.get_collection(&database, &config.collection) {
        Ok(coll) => {
            let mut idx_config =
                VectorIndexConfig::new(config.name, config.field, config.dimensions as usize);

            if let Some(m_str) = config.metric {
                if let Ok(val) =
                    serde_json::from_value::<VectorMetric>(serde_json::Value::String(m_str))
                {
                    idx_config = idx_config.with_metric(val);
                }
            }

            if let Some(ef) = config.ef_construction {
                idx_config = idx_config.with_ef_construction(ef as usize);
            }

            if let Some(m_val) = config.m {
                idx_config = idx_config.with_m(m_val as usize);
            }

            match coll.create_vector_index(idx_config) {
                Ok(_) => Response::ok_empty(),
                Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
            }
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_list_vector_indexes(
    handler: &DriverHandler,
    database: String,
    collection: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => {
            let indexes = coll.list_vector_indexes();
            Response::ok(serde_json::to_value(indexes).unwrap_or_default())
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_delete_vector_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    name: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => match coll.drop_vector_index(&name) {
            Ok(_) => Response::ok_empty(),
            Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
        },
        Err(e) => Response::error(e),
    }
}

#[allow(clippy::too_many_arguments)]
pub fn handle_hybrid_search(
    handler: &DriverHandler,
    database: String,
    collection: String,
    vector: Vec<f32>,
    text_query: String,
    vector_index: String,
    fulltext_field: String,
    vector_weight: Option<f32>,
    text_weight: Option<f32>,
    limit: Option<u32>,
    fusion: Option<String>,
) -> Response {
    use crate::storage::{FusionMethod, HybridSearchOptions};

    let fusion = match fusion.as_deref() {
        None => FusionMethod::default(),
        Some(s) => match FusionMethod::parse(s) {
            Some(f) => f,
            None => {
                return Response::error(DriverError::InvalidCommand(format!(
                    "Invalid fusion method '{}': expected \"weighted\" or \"rrf\"",
                    s
                )))
            }
        },
    };
    let defaults = HybridSearchOptions::default();
    let opts = HybridSearchOptions {
        vector_weight: vector_weight.unwrap_or(defaults.vector_weight),
        text_weight: text_weight.unwrap_or(defaults.text_weight),
        limit: limit.map(|l| l as usize).unwrap_or(defaults.limit),
        fusion,
    };

    match handler.get_collection(&database, &collection) {
        Ok(coll) => {
            match coll.hybrid_search(&vector_index, &fulltext_field, &vector, &text_query, &opts) {
                Ok(results) => Response::ok(serde_json::json!(results)),
                Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
            }
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_vector_search(
    handler: &DriverHandler,
    database: String,
    collection: String,
    index_name: String,
    vector: Vec<f32>,
    limit: Option<i32>,
    ef_search: Option<i32>,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => {
            // TODO: Implement filter support when Collection::vector_search supports it
            match coll.vector_search(
                &index_name,
                &vector,
                limit.map(|l| l.max(0) as usize).unwrap_or(10),
                ef_search.map(|v| v.max(0) as usize),
            ) {
                Ok(results) => Response::ok(serde_json::json!(results)),
                Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
            }
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_quantize_vector_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    index_name: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => match coll.quantize_vector_index(
            &index_name,
            crate::storage::index::VectorQuantization::Scalar,
        ) {
            Ok(_) => Response::ok_empty(),
            Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
        },
        Err(e) => Response::error(e),
    }
}

pub fn handle_dequantize_vector_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    index_name: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => match coll.dequantize_vector_index(&index_name) {
            Ok(_) => Response::ok_empty(),
            Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
        },
        Err(e) => Response::error(e),
    }
}

// ==================== TTL Index Operations ====================

pub fn handle_create_ttl_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    name: String,
    field: String,
    expire_after_seconds: i64,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => match coll.create_ttl_index(name, field, expire_after_seconds as u64) {
            Ok(_) => Response::ok_empty(),
            Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
        },
        Err(e) => Response::error(e),
    }
}

pub fn handle_list_ttl_indexes(
    handler: &DriverHandler,
    database: String,
    collection: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => {
            let indexes = coll.list_ttl_indexes();
            Response::ok(serde_json::to_value(indexes).unwrap_or_default())
        }
        Err(e) => Response::error(e),
    }
}

pub fn handle_delete_ttl_index(
    handler: &DriverHandler,
    database: String,
    collection: String,
    name: String,
) -> Response {
    match handler.get_collection(&database, &collection) {
        Ok(coll) => match coll.drop_ttl_index(&name) {
            Ok(_) => Response::ok_empty(),
            Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
        },
        Err(e) => Response::error(e),
    }
}