flusso-kernel 0.17.0

use serde::{Deserialize, Serialize};

use std::collections::BTreeMap;

use super::MappingType;
use crate::common::GenericValue;

/// The declared type of a leaf field — the single vocabulary that bridges a
/// Postgres column type and an OpenSearch mapping type.
///
/// A self-describing schema names one of these per scalar field, so the document
/// shape (and the index mapping) is known without ever touching the database. A
/// variant pins both ends: which Postgres types satisfy it (for validation, when
/// a database is reachable) and which OpenSearch [`MappingType`] it emits. That
/// is what lets the two disagree on purpose — an `Enum` is stored as text in
/// Postgres but must be a `keyword` in OpenSearch.
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum FlussoType {
    /// Analyzed natural-language full text — descriptions, bios, comments. The
    /// default analyzed type; carries the `flusso_text` analyzer (plain tokenize
    /// + accent/case fold). PG `text` / `varchar` → OS `text`.
    Text,
    /// Identifier-like short text — names, SKUs, codes, statuses. Analyzed with
    /// the `flusso_code` analyzer, which splits on punctuation/case so `C-01234`
    /// is found by `c01234` or `01234`. PG `text` / `varchar` → OS `text`.
    Identifier,
    /// Exact, aggregatable token. PG `text` / `varchar` → OS `keyword`.
    Keyword,
    /// A closed set of string values — stored as text in Postgres (a `varchar`
    /// or a PG `enum`), indexed exactly as a `keyword`. An optional declared
    /// variant order (on the field's [`Column`](super::Column), not here) makes
    /// it sort by that order rather than alphabetically.
    Enum,
    /// PG `uuid` → OS `keyword`.
    Uuid,
    /// PG `boolean` → OS `boolean`.
    Boolean,
    /// PG `smallint` / `int2` → OS `short`.
    Short,
    /// PG `integer` / `int4` → OS `integer`.
    Integer,
    /// PG `bigint` / `int8` → OS `long`.
    Long,
    /// PG `real` / `float4` → OS `float`.
    Float,
    /// PG `double precision` / `float8` → OS `double`.
    Double,
    /// PG `numeric` / `decimal` / `money` → OS `double` (lossy but searchable;
    /// declare [`Custom`](Self::Custom) `scaled_float` when exactness matters).
    Decimal,
    /// PG `date` → OS `date`.
    Date,
    /// PG `timestamp` / `timestamptz` / `time` → OS `date`.
    Timestamp,
    /// PG `bytea` → OS `binary`.
    Binary,
    /// PG `json` / `jsonb` → OS `object`.
    Json,
    /// A dynamic-key object: a `json`/`jsonb` column whose keys are
    /// runtime-determined but whose values all share one leaf type (e.g.
    /// translations `{"en": …, "it": …}`). Maps to OS `object` with
    /// `dynamic: true`, so runtime keys are accepted without enumerating them
    /// in the schema; every key is mapped like a plain field of `values`, the
    /// leaf type of every value.
    Map { values: Box<FlussoType> },
    /// A geographic point → OS `geo_point`.
    ///
    /// The document is assembled server-side as JSON, so the source column must
    /// already hold a value OpenSearch accepts as a `geo_point` and that carries
    /// through JSON verbatim: `json`/`jsonb` shaped as `{"lat": …, "lon": …}` or
    /// `[lon, lat]`, or `text` as `"lat,lon"`. A PostGIS `geometry` or PG-native
    /// `point` is **not** accepted — it would serialize as WKB / `(x,y)`; expose
    /// a generated `jsonb`/`text` column in that shape instead.
    GeoPoint,
    /// An escape hatch: an explicit OpenSearch type with the Postgres types it
    /// accepts, for anything the named variants don't cover (`geo_shape`,
    /// `scaled_float`, …).
    Custom {
        postgres: Vec<String>,
        opensearch: String,
    },
}

impl FlussoType {
    /// The mapping options a field of this type carries unless it sets them
    /// itself: `identifier` is analyzed `text` with the `flusso_code` analyzer.
    /// Empty for every other type. A plain column and a `map`'s keys both start
    /// from these, so a key of a type is mapped like a field of it.
    pub fn default_options(&self) -> BTreeMap<String, GenericValue> {
        match self {
            FlussoType::Identifier => BTreeMap::from([(
                "analyzer".to_owned(),
                GenericValue::String("flusso_code".to_owned()),
            )]),
            _ => BTreeMap::new(),
        }
    }

    /// The OpenSearch [`MappingType`] this declared type maps to.
    pub fn opensearch(&self) -> MappingType {
        match self {
            FlussoType::Text | FlussoType::Identifier => MappingType::Text,
            FlussoType::Keyword | FlussoType::Enum | FlussoType::Uuid => MappingType::Keyword,
            FlussoType::Boolean => MappingType::Boolean,
            FlussoType::Short => MappingType::Short,
            FlussoType::Integer => MappingType::Integer,
            FlussoType::Long => MappingType::Long,
            FlussoType::Float => MappingType::Float,
            FlussoType::Double | FlussoType::Decimal => MappingType::Double,
            FlussoType::Date | FlussoType::Timestamp => MappingType::Date,
            FlussoType::Binary => MappingType::Other("binary".to_owned()),
            FlussoType::Json | FlussoType::Map { .. } => MappingType::Object,
            FlussoType::GeoPoint => MappingType::Other("geo_point".to_owned()),
            FlussoType::Custom { opensearch, .. } => MappingType::from_name(opensearch),
        }
    }

    /// Whether `sql_type` (a Postgres type name as `format_type` reports it, e.g.
    /// `bigint`, `character varying(255)`, `numeric(10,2)`, `integer[]`) is a
    /// Postgres type this declared type accepts. Used to validate a declared
    /// schema against a live database; with no database, it is never consulted.
    pub fn accepts_pg(&self, sql_type: &str) -> bool {
        let base = normalize_pg_type(sql_type);
        match self {
            FlussoType::Text | FlussoType::Identifier | FlussoType::Keyword | FlussoType::Enum => {
                matches!(
                    base.as_str(),
                    "text"
                        | "character varying"
                        | "varchar"
                        | "character"
                        | "char"
                        | "bpchar"
                        | "citext"
                        | "name"
                )
            }
            FlussoType::Uuid => base == "uuid",
            FlussoType::Boolean => matches!(base.as_str(), "boolean" | "bool"),
            FlussoType::Short => matches!(base.as_str(), "smallint" | "int2" | "smallserial"),
            FlussoType::Integer => matches!(base.as_str(), "integer" | "int" | "int4" | "serial"),
            FlussoType::Long => matches!(base.as_str(), "bigint" | "int8" | "bigserial"),
            FlussoType::Float => matches!(base.as_str(), "real" | "float4"),
            FlussoType::Double => matches!(base.as_str(), "double precision" | "float8"),
            FlussoType::Decimal => matches!(base.as_str(), "numeric" | "decimal" | "money"),
            FlussoType::Date => base == "date",
            FlussoType::Timestamp => matches!(
                base.as_str(),
                "timestamp with time zone"
                    | "timestamp without time zone"
                    | "timestamp"
                    | "timestamptz"
                    | "time with time zone"
                    | "time without time zone"
                    | "time"
                    | "timetz"
            ),
            FlussoType::Binary => base == "bytea",
            FlussoType::Json | FlussoType::Map { .. } => {
                matches!(base.as_str(), "json" | "jsonb")
            }
            // Geo data must reach OpenSearch as JSON `{lat,lon}` / `[lon,lat]` or
            // a `"lat,lon"` string — i.e. it lives in a json/jsonb or text-like
            // column. PostGIS `geometry` / PG `point` are intentionally rejected.
            FlussoType::GeoPoint => matches!(
                base.as_str(),
                "json"
                    | "jsonb"
                    | "text"
                    | "character varying"
                    | "varchar"
                    | "character"
                    | "char"
                    | "bpchar"
                    | "citext"
                    | "name"
            ),
            FlussoType::Custom { postgres, .. } => {
                postgres.iter().any(|t| normalize_pg_type(t) == base)
            }
        }
    }
}

/// Normalize a `format_type` Postgres type name to its bare base name: drop an
/// array `[]` suffix (OpenSearch fields are natively multi-valued) and any
/// `(precision)` / `(length)` modifier, then lowercase. Mirrors the
/// normalization the Postgres source uses when reading a column's type.
fn normalize_pg_type(sql_type: &str) -> String {
    let base = sql_type.trim().trim_end_matches("[]").trim();
    let base = base.split('(').next().unwrap_or(base).trim();
    base.to_ascii_lowercase()
}

#[cfg(test)]
mod tests;