sqlx-firebirdsql 0.1.0

use super::FirebirdColumn;
use crate::column::ColumnIndex;
use crate::error::Error;
use crate::{Firebird, FirebirdArguments, FirebirdTypeInfo};
use either::Either;
use sqlx_core::sql_str::SqlStr;
use std::collections::HashMap;
use std::sync::Arc;

pub(crate) use sqlx_core::statement::*;

#[derive(Debug, Clone)]
pub struct FirebirdStatement {
    pub(crate) sql: SqlStr,
    pub(crate) metadata: FirebirdStatementMetadata,
}

#[derive(Debug, Default, Clone)]
pub(crate) struct FirebirdStatementMetadata {
    pub(crate) columns: Arc<Vec<FirebirdColumn>>,
    pub(crate) column_names: Arc<HashMap<String, usize>>,
    pub(crate) parameters: usize,
    #[cfg_attr(not(feature = "offline"), allow(dead_code))]
    pub(crate) nullable: Vec<Option<bool>>,
}

impl Statement for FirebirdStatement {
    type Database = Firebird;

    fn into_sql(self) -> SqlStr {
        self.sql
    }

    fn sql(&self) -> &SqlStr {
        &self.sql
    }

    fn parameters(&self) -> Option<Either<&[FirebirdTypeInfo], usize>> {
        Some(Either::Right(self.metadata.parameters))
    }

    fn columns(&self) -> &[FirebirdColumn] {
        &self.metadata.columns
    }

    sqlx_core::impl_statement_query!(FirebirdArguments);
}

impl ColumnIndex<FirebirdStatement> for &'_ str {
    fn index(&self, statement: &FirebirdStatement) -> Result<usize, Error> {
        statement
            .metadata
            .column_names
            .get(*self)
            .ok_or_else(|| Error::ColumnNotFound((*self).into()))
            .copied()
    }
}