sentinel-driver 2.0.0

use crate::error::{Error, Result};
use crate::types::FromSql;
use crate::types::Oid;

// ── Primitive types ──────────────────────────────────

impl FromSql for bool {
    fn oid() -> Oid {
        Oid::BOOL
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        if buf.len() != 1 {
            return Err(Error::Decode(format!(
                "bool: expected 1 byte, got {}",
                buf.len()
            )));
        }
        Ok(buf[0] != 0)
    }
}

impl FromSql for i16 {
    fn oid() -> Oid {
        Oid::INT2
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let arr: [u8; 2] = buf
            .try_into()
            .map_err(|_| Error::Decode(format!("int2: expected 2 bytes, got {}", buf.len())))?;
        Ok(i16::from_be_bytes(arr))
    }
}

impl FromSql for i32 {
    fn oid() -> Oid {
        Oid::INT4
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let arr: [u8; 4] = buf
            .try_into()
            .map_err(|_| Error::Decode(format!("int4: expected 4 bytes, got {}", buf.len())))?;
        Ok(i32::from_be_bytes(arr))
    }
}

impl FromSql for i64 {
    fn oid() -> Oid {
        Oid::INT8
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let arr: [u8; 8] = buf
            .try_into()
            .map_err(|_| Error::Decode(format!("int8: expected 8 bytes, got {}", buf.len())))?;
        Ok(i64::from_be_bytes(arr))
    }
}

impl FromSql for f32 {
    fn oid() -> Oid {
        Oid::FLOAT4
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let arr: [u8; 4] = buf
            .try_into()
            .map_err(|_| Error::Decode(format!("float4: expected 4 bytes, got {}", buf.len())))?;
        Ok(f32::from_be_bytes(arr))
    }
}

impl FromSql for f64 {
    fn oid() -> Oid {
        Oid::FLOAT8
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let arr: [u8; 8] = buf
            .try_into()
            .map_err(|_| Error::Decode(format!("float8: expected 8 bytes, got {}", buf.len())))?;
        Ok(f64::from_be_bytes(arr))
    }
}

// ── String types ─────────────────────────────────────

impl FromSql for String {
    fn oid() -> Oid {
        Oid::TEXT
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        String::from_utf8(buf.to_vec())
            .map_err(|e| Error::Decode(format!("text: invalid UTF-8: {e}")))
    }
}

// ── Byte types ───────────────────────────────────────

impl FromSql for Vec<u8> {
    fn oid() -> Oid {
        Oid::BYTEA
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        Ok(buf.to_vec())
    }
}

// ── chrono types ─────────────────────────────────────

/// PG epoch offset in microseconds from Unix epoch.
const PG_EPOCH_OFFSET_US: i64 = 946_684_800_000_000;

impl FromSql for chrono::NaiveDateTime {
    fn oid() -> Oid {
        Oid::TIMESTAMP
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let us_from_pg_epoch = i64::from_sql(buf)?;
        if us_from_pg_epoch == i64::MAX {
            return Ok(chrono::NaiveDateTime::MAX);
        }
        if us_from_pg_epoch == i64::MIN {
            return Ok(chrono::NaiveDateTime::MIN);
        }
        let us_from_unix_epoch = us_from_pg_epoch + PG_EPOCH_OFFSET_US;
        let secs = us_from_unix_epoch.div_euclid(1_000_000);
        let nsecs = (us_from_unix_epoch.rem_euclid(1_000_000) * 1000) as u32;
        chrono::DateTime::from_timestamp(secs, nsecs)
            .map(|dt| dt.naive_utc())
            .ok_or_else(|| Error::Decode(format!("timestamp out of range: {us_from_pg_epoch}")))
    }
}

impl FromSql for chrono::DateTime<chrono::Utc> {
    fn oid() -> Oid {
        Oid::TIMESTAMPTZ
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let us_from_pg_epoch = i64::from_sql(buf)?;
        if us_from_pg_epoch == i64::MAX {
            return Ok(chrono::NaiveDateTime::MAX.and_utc());
        }
        if us_from_pg_epoch == i64::MIN {
            return Ok(chrono::NaiveDateTime::MIN.and_utc());
        }
        let us_from_unix_epoch = us_from_pg_epoch + PG_EPOCH_OFFSET_US;
        let secs = us_from_unix_epoch.div_euclid(1_000_000);
        let nsecs = (us_from_unix_epoch.rem_euclid(1_000_000) * 1000) as u32;
        chrono::DateTime::from_timestamp(secs, nsecs)
            .ok_or_else(|| Error::Decode(format!("timestamptz out of range: {us_from_pg_epoch}")))
    }
}

impl FromSql for chrono::NaiveDate {
    fn oid() -> Oid {
        Oid::DATE
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let days = i32::from_sql(buf)?;
        if days == i32::MAX {
            return Ok(chrono::NaiveDate::MAX);
        }
        if days == i32::MIN {
            return Ok(chrono::NaiveDate::MIN);
        }
        #[allow(clippy::expect_used)]
        let epoch = chrono::NaiveDate::from_ymd_opt(2000, 1, 1).expect("PG epoch is valid");
        epoch
            .checked_add_signed(chrono::Duration::days(days as i64))
            .ok_or_else(|| Error::Decode(format!("date out of range: {days} days from epoch")))
    }
}

impl FromSql for chrono::NaiveTime {
    fn oid() -> Oid {
        Oid::TIME
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let us = i64::from_sql(buf)?;
        let secs = (us / 1_000_000) as u32;
        let micro = (us % 1_000_000) as u32;
        chrono::NaiveTime::from_num_seconds_from_midnight_opt(secs, micro * 1000)
            .ok_or_else(|| Error::Decode(format!("time out of range: {us} microseconds")))
    }
}

// ── UUID ─────────────────────────────────────────────

impl FromSql for uuid::Uuid {
    fn oid() -> Oid {
        Oid::UUID
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        let arr: [u8; 16] = buf
            .try_into()
            .map_err(|_| Error::Decode(format!("uuid: expected 16 bytes, got {}", buf.len())))?;
        Ok(uuid::Uuid::from_bytes(arr))
    }
}

// ── Array types ─────────────────────────────────────

/// Decode a PostgreSQL 1-D binary array into `Vec<T>`.
/// Read a big-endian i32 from a byte slice at the given offset.
fn read_i32(buf: &[u8], offset: usize) -> i32 {
    i32::from_be_bytes([
        buf[offset],
        buf[offset + 1],
        buf[offset + 2],
        buf[offset + 3],
    ])
}

/// Read a big-endian u32 from a byte slice at the given offset.
fn read_u32(buf: &[u8], offset: usize) -> u32 {
    u32::from_be_bytes([
        buf[offset],
        buf[offset + 1],
        buf[offset + 2],
        buf[offset + 3],
    ])
}

/// Walk a one-dimensional PG array, calling `decode_elem` for each slot.
///
/// `decode_elem` receives `Some(&[u8])` for non-NULL element bodies and
/// `None` for SQL NULL (signalled by a per-element length of `-1` in the
/// PG binary array wire format). The closure decides what to do in each
/// case — non-nullable callers raise an error on `None`; nullable callers
/// map it to a sentinel value such as `Option::None`.
fn decode_array_inner<T, F>(
    buf: &[u8],
    expected_elem_oid: Oid,
    mut decode_elem: F,
) -> Result<Vec<T>>
where
    F: FnMut(Option<&[u8]>) -> Result<T>,
{
    if buf.len() < 12 {
        return Err(Error::Decode("array: header too short".into()));
    }

    let ndim = read_i32(buf, 0);
    // has_null at buf[4..8] is informational; the per-element length
    // sentinel is the authoritative source for NULL.
    let elem_oid = read_u32(buf, 8);

    if ndim == 0 {
        return Ok(Vec::new());
    }

    if ndim != 1 {
        return Err(Error::Decode(format!(
            "array: multi-dimensional arrays not supported (ndim={ndim})"
        )));
    }

    if elem_oid != expected_elem_oid.0 {
        return Err(Error::Decode(format!(
            "array: expected element OID {}, got {elem_oid}",
            expected_elem_oid.0
        )));
    }

    if buf.len() < 20 {
        return Err(Error::Decode("array: dimension header too short".into()));
    }

    let dim_len = read_i32(buf, 12) as usize;
    // dim_lbound at buf[16..20] — skip, not needed for decoding

    let mut offset = 20;
    let mut result = Vec::with_capacity(dim_len);

    for _ in 0..dim_len {
        if offset + 4 > buf.len() {
            return Err(Error::Decode("array: unexpected end of data".into()));
        }

        let elem_len = read_i32(buf, offset);
        offset += 4;

        if elem_len < 0 {
            result.push(decode_elem(None)?);
        } else {
            let elem_len = elem_len as usize;
            if offset + elem_len > buf.len() {
                return Err(Error::Decode("array: element data truncated".into()));
            }
            result.push(decode_elem(Some(&buf[offset..offset + elem_len]))?);
            offset += elem_len;
        }
    }

    Ok(result)
}

fn decode_array<T: FromSql>(buf: &[u8], expected_elem_oid: Oid) -> Result<Vec<T>> {
    decode_array_inner(buf, expected_elem_oid, |opt| match opt {
        Some(bytes) => T::from_sql(bytes),
        None => Err(Error::Decode(
            "array: NULL elements not supported (use Vec<Option<T>>)".into(),
        )),
    })
}

fn decode_array_nullable<T: FromSql>(buf: &[u8], expected_elem_oid: Oid) -> Result<Vec<Option<T>>> {
    decode_array_inner(buf, expected_elem_oid, |opt| match opt {
        Some(bytes) => T::from_sql(bytes).map(Some),
        None => Ok(None),
    })
}

/// Implement `FromSql` for both `Vec<T>` and `Vec<Option<T>>` for a
/// specific element type.
///
/// - `Vec<T>` errors on any SQL NULL element with a message that points
///   the caller at `Vec<Option<T>>`.
/// - `Vec<Option<T>>` maps SQL NULL to `Option::None`.
///
/// Both impls advertise the same array OID; PostgreSQL does not carry a
/// distinct "nullable element" array type — nullability is per-element on
/// the wire.
macro_rules! impl_array_from_sql {
    ($elem_ty:ty, $array_oid:expr, $elem_oid:expr) => {
        impl FromSql for Vec<$elem_ty> {
            fn oid() -> Oid {
                $array_oid
            }

            fn from_sql(buf: &[u8]) -> Result<Self> {
                decode_array::<$elem_ty>(buf, $elem_oid)
            }
        }

        impl FromSql for Vec<Option<$elem_ty>> {
            fn oid() -> Oid {
                $array_oid
            }

            fn from_sql(buf: &[u8]) -> Result<Self> {
                decode_array_nullable::<$elem_ty>(buf, $elem_oid)
            }
        }
    };
}

impl_array_from_sql!(bool, Oid::BOOL_ARRAY, Oid::BOOL);
impl_array_from_sql!(i16, Oid::INT2_ARRAY, Oid::INT2);
impl_array_from_sql!(i32, Oid::INT4_ARRAY, Oid::INT4);
impl_array_from_sql!(i64, Oid::INT8_ARRAY, Oid::INT8);
impl_array_from_sql!(f32, Oid::FLOAT4_ARRAY, Oid::FLOAT4);
impl_array_from_sql!(f64, Oid::FLOAT8_ARRAY, Oid::FLOAT8);
impl_array_from_sql!(String, Oid::TEXT_ARRAY, Oid::TEXT);
impl_array_from_sql!(uuid::Uuid, Oid::UUID_ARRAY, Oid::UUID);
impl_array_from_sql!(
    crate::types::interval::PgInterval,
    Oid::INTERVAL_ARRAY,
    Oid::INTERVAL
);
impl_array_from_sql!(crate::types::money::PgMoney, Oid::MONEY_ARRAY, Oid::MONEY);
impl_array_from_sql!(crate::types::xml::PgXml, Oid::XML_ARRAY, Oid::XML);
impl_array_from_sql!(crate::types::lsn::PgLsn, Oid::PG_LSN_ARRAY, Oid::PG_LSN);
impl_array_from_sql!(crate::types::network::PgInet, Oid::INET_ARRAY, Oid::INET);
impl_array_from_sql!(crate::types::network::PgCidr, Oid::CIDR_ARRAY, Oid::CIDR);
impl_array_from_sql!(
    crate::types::network::PgMacAddr,
    Oid::MACADDR_ARRAY,
    Oid::MACADDR
);
#[cfg(feature = "with-rust-decimal")]
impl_array_from_sql!(rust_decimal::Decimal, Oid::NUMERIC_ARRAY, Oid::NUMERIC);
impl_array_from_sql!(crate::types::bit::PgBit, Oid::VARBIT_ARRAY, Oid::VARBIT);
impl_array_from_sql!(chrono::NaiveDateTime, Oid::TIMESTAMP_ARRAY, Oid::TIMESTAMP);
impl_array_from_sql!(
    chrono::DateTime<chrono::Utc>,
    Oid::TIMESTAMPTZ_ARRAY,
    Oid::TIMESTAMPTZ
);
impl_array_from_sql!(chrono::NaiveDate, Oid::DATE_ARRAY, Oid::DATE);
impl_array_from_sql!(chrono::NaiveTime, Oid::TIME_ARRAY, Oid::TIME);
impl_array_from_sql!(
    crate::types::geometric::PgPoint,
    Oid::POINT_ARRAY,
    Oid::POINT
);
impl_array_from_sql!(
    crate::types::geometric::PgCircle,
    Oid::CIRCLE_ARRAY,
    Oid::CIRCLE
);

impl_array_from_sql!(
    crate::types::timetz::PgTimeTz,
    Oid::TIMETZ_ARRAY,
    Oid::TIMETZ
);
impl_array_from_sql!(
    crate::types::network::PgMacAddr8,
    Oid::MACADDR8_ARRAY,
    Oid::MACADDR8
);

impl FromSql for Vec<Vec<u8>> {
    fn oid() -> Oid {
        Oid::BYTEA_ARRAY
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        decode_array::<Vec<u8>>(buf, Oid::BYTEA)
    }
}

impl FromSql for Vec<Option<Vec<u8>>> {
    fn oid() -> Oid {
        Oid::BYTEA_ARRAY
    }

    fn from_sql(buf: &[u8]) -> Result<Self> {
        decode_array_nullable::<Vec<u8>>(buf, Oid::BYTEA)
    }
}