postrust-core 1.0.0-alpha.1

//! Table and column types.

use crate::api_request::QualifiedIdentifier;
use indexmap::IndexMap;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;

/// A database table or view.
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct Table {
    /// Schema name
    pub schema: String,
    /// Table/view name
    pub name: String,
    /// Description from comment
    pub description: Option<String>,
    /// Whether this is a view (vs a table)
    pub is_view: bool,
    /// Whether this is a partitioned table.
    ///
    /// It behaves as a table for reading and writing, but not for everything:
    /// a system column such as `xmax` cannot be read back from one.
    #[serde(default)]
    pub is_partitioned: bool,
    /// Whether INSERT is allowed
    pub insertable: bool,
    /// Whether UPDATE is allowed
    pub updatable: bool,
    /// Whether DELETE is allowed
    pub deletable: bool,
    /// Primary key column names
    pub pk_cols: Vec<String>,
    /// Unique constraints, as `(name, columns)`, the primary key among them.
    ///
    /// Named because `ON CONFLICT ON CONSTRAINT` takes a name: an upsert has
    /// to say which uniqueness it is resolving against, and a table may have
    /// several.
    #[serde(default)]
    pub unique_constraints: Vec<(String, Vec<String>)>,
    /// Columns indexed by name
    pub columns: ColumnMap,
    /// Computed columns, indexed by name.
    ///
    /// Not part of the table, and never returned by `select=*`: a function of
    /// one argument of the table's own row type, which PostgreSQL lets a query
    /// read as though it were a column. PostgREST exposes them under the name
    /// the function has, so `?select=id,always_true` and `?always_true=is.true`
    /// both work on a table that has no such column.
    #[serde(default)]
    pub computed_columns: HashMap<String, ComputedColumn>,
}

impl Column {
    /// The type a data representation would be declared on.
    ///
    /// The domain where the column has one, and the type itself otherwise.
    pub fn representation_type(&self) -> &str {
        self.domain_type.as_deref().unwrap_or(&self.nominal_type)
    }
}

impl Table {
    /// Get a column by name.
    pub fn get_column(&self, name: &str) -> Option<&Column> {
        self.columns.get(name)
    }

    /// Get a computed column by name.
    pub fn get_computed_column(&self, name: &str) -> Option<&ComputedColumn> {
        self.computed_columns.get(name)
    }

    /// Check if the table has a column.
    pub fn has_column(&self, name: &str) -> bool {
        self.columns.contains_key(name)
    }

    /// Get the qualified identifier for this table.
    pub fn qualified_identifier(&self) -> QualifiedIdentifier {
        QualifiedIdentifier::new(&self.schema, &self.name)
    }

    /// Get column names in order.
    pub fn column_names(&self) -> impl Iterator<Item = &str> {
        self.columns.keys().map(|s| s.as_str())
    }

    /// Check if this is a read-only view.
    pub fn is_readonly(&self) -> bool {
        !self.insertable && !self.updatable && !self.deletable
    }
}

/// A function of the table's row type, read as though it were a column.
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct ComputedColumn {
    /// The function to call.
    pub function: QualifiedIdentifier,
    /// What it returns.
    pub data_type: String,
    /// The comment on the function, which is the field's description.
    ///
    /// A column's description comes from a comment on the column; a computed
    /// column has no column to comment on, so it comes from the function --
    /// which is where a schema author would write it and where Hasura reads
    /// it from.
    #[serde(default)]
    pub description: Option<String>,
    /// The name of the parameter that takes the table's row.
    ///
    /// Usually the first and usually the only one, but not always: Hasura
    /// records a `table_argument` because a function may take the row second,
    /// after the session. Where it is not first the call has to be written in
    /// named notation, which is why the name is kept rather than the position.
    #[serde(default)]
    pub row_argument: Option<String>,
    /// The name of the parameter filled from the caller's session, if any.
    ///
    /// A function with one is not callable from the REST surface, which has no
    /// session document to give it, and is skipped there.
    #[serde(default)]
    pub session_argument: Option<String>,
    /// Whether the function takes anything besides the row.
    ///
    /// The REST surface has nowhere to write an argument and no session to
    /// supply, so it reads only the fields that take neither: a field it could
    /// only ever call wrongly is worse than a field that is not there.
    #[serde(default)]
    pub takes_arguments: bool,
}

/// A table column.
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct Column {
    /// Column name
    pub name: String,
    /// Description from comment
    pub description: Option<String>,
    /// Whether NULL is allowed
    pub nullable: bool,
    /// PostgreSQL data type
    pub data_type: String,
    /// Base type (for domains)
    pub nominal_type: String,
    /// The domain this column is declared over, where it is declared over one.
    ///
    /// `information_schema` reports the type *underneath* a domain in
    /// `data_type` and `udt_name`, so nothing else says that a value is a
    /// `color` rather than an integer -- and a data representation is declared
    /// on the domain.
    #[serde(default)]
    pub domain_type: Option<String>,
    /// Maximum length (for varchar, etc.)
    pub max_len: Option<i32>,
    /// Default value expression
    pub default: Option<String>,
    /// Enum values (for enum types)
    pub enum_values: Vec<String>,
    /// Whether this is part of the primary key
    pub is_pk: bool,
    /// Column position (1-based)
    pub position: i32,
    /// Whether PostgreSQL generates this column's value and refuses any other.
    ///
    /// `GENERATED ALWAYS AS IDENTITY` and `GENERATED ALWAYS AS (...) STORED`
    /// both mean the same thing to a writer: naming the column at all is an
    /// error, not merely unnecessary. A `BY DEFAULT` identity is not this --
    /// it has a default and accepts a value -- which is why the flag is about
    /// what a write may say rather than about how a value is produced.
    #[serde(default)]
    pub always_generated: bool,
}

impl Column {
    /// Check if this column has a default value.
    pub fn has_default(&self) -> bool {
        self.default.is_some()
    }

    /// Check if this is an auto-generated column.
    pub fn is_auto(&self) -> bool {
        self.default
            .as_ref()
            .map(|d| d.contains("nextval(") || d.contains("gen_random_uuid()"))
            .unwrap_or(false)
    }

    /// Check if this is a JSON/JSONB column.
    pub fn is_json(&self) -> bool {
        self.data_type == "json" || self.data_type == "jsonb"
    }

    /// Check if this is an array type.
    pub fn is_array(&self) -> bool {
        self.data_type.starts_with('_') || self.data_type.ends_with("[]")
    }

    /// Check if this is a range type.
    pub fn is_range(&self) -> bool {
        self.data_type.ends_with("range")
    }
}

/// Map of column name to column.
pub type ColumnMap = IndexMap<String, Column>;

/// Map of qualified identifier to table.
pub type TablesMap = HashMap<QualifiedIdentifier, Table>;

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_table_qualified_identifier() {
        let table = Table {
            schema: "public".into(),
            name: "users".into(),
            description: None,
            is_view: false,
            insertable: true,
            updatable: true,
            deletable: true,
            pk_cols: vec!["id".into()],
            unique_constraints: Vec::new(),
            columns: IndexMap::new(),
            computed_columns: Default::default(),
            is_partitioned: false,
        };

        let qi = table.qualified_identifier();
        assert_eq!(qi.schema, "public");
        assert_eq!(qi.name, "users");
    }

    #[test]
    fn test_column_is_auto() {
        let col1 = Column {
            name: "id".into(),
            description: None,
            nullable: false,
            data_type: "integer".into(),
            nominal_type: "integer".into(),
            max_len: None,
            default: Some("nextval('users_id_seq'::regclass)".into()),
            enum_values: vec![],
            is_pk: true,
            position: 1,
            always_generated: false,
            domain_type: None,
        };
        assert!(col1.is_auto());

        let col2 = Column {
            name: "uuid".into(),
            description: None,
            nullable: false,
            data_type: "uuid".into(),
            nominal_type: "uuid".into(),
            max_len: None,
            default: Some("gen_random_uuid()".into()),
            enum_values: vec![],
            is_pk: false,
            position: 2,
            always_generated: false,
            domain_type: None,
        };
        assert!(col2.is_auto());

        let col3 = Column {
            name: "name".into(),
            description: None,
            nullable: false,
            data_type: "text".into(),
            nominal_type: "text".into(),
            max_len: None,
            default: None,
            enum_values: vec![],
            is_pk: false,
            position: 3,
            always_generated: false,
            domain_type: None,
        };
        assert!(!col3.is_auto());
    }
}