typedb-driver 3.10.0

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

use std::fmt;

use chrono::{DateTime, NaiveDate, NaiveDateTime};

pub use self::{
    instance::{Attribute, Entity, Relation},
    type_::{AttributeType, EntityType, RelationType, RoleType},
    value::{Value, ValueType},
};
use crate::{
    IID,
    concept::value::{Decimal, Duration, Struct, TimeZone},
};

pub mod instance;
pub mod type_;
pub mod value;

/// The fundamental TypeQL object.
#[derive(Clone, PartialEq)]
pub enum Concept {
    EntityType(EntityType),
    RelationType(RelationType),
    RoleType(RoleType),
    AttributeType(AttributeType),

    Entity(Entity),
    Relation(Relation),
    Attribute(Attribute),

    Value(Value),
}

#[derive(Clone, PartialEq)]
pub enum ConceptCategory {
    EntityType,
    RelationType,
    RoleType,
    AttributeType,
    Entity,
    Relation,
    Attribute,
    Value,
}

impl ConceptCategory {
    pub const fn name(&self) -> &'static str {
        match self {
            ConceptCategory::EntityType => "EntityType",
            ConceptCategory::RelationType => "RelationType",
            ConceptCategory::RoleType => "RoleType",
            ConceptCategory::AttributeType => "AttributeType",
            ConceptCategory::Entity => "Entity",
            ConceptCategory::Relation => "Relation",
            ConceptCategory::Attribute => "Attribute",
            ConceptCategory::Value => "Value",
        }
    }
}

impl fmt::Display for ConceptCategory {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        fmt::Debug::fmt(self, f)
    }
}

impl fmt::Debug for ConceptCategory {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.name())
    }
}

impl Concept {
    pub(crate) const UNKNOWN_LABEL: &'static str = "unknown";

    /// Retrieves the unique id (IID) of this Concept.
    /// If this is an Entity or Relation Instance, returns the IID of the instance.
    /// Otherwise, returns None.
    pub fn try_get_iid(&self) -> Option<&IID> {
        match self {
            Self::Entity(entity) => Some(&entity.iid),
            Self::Relation(relation) => Some(&relation.iid),
            _ => None,
        }
    }

    /// Retrieves the label of this Concept.
    /// If this is an Instance, returns the label of the type of this instance ("unknown" if type fetching is disabled).
    /// If this is a Value, returns the label of the value type of the value.
    /// If this is a Type, returns the label of the type.
    pub fn get_label(&self) -> &str {
        self.try_get_label().unwrap_or(Self::UNKNOWN_LABEL)
    }

    /// Retrieves the optional label of the concept.
    /// If this is an Instance, returns the label of the type of this instance (None if type fetching is disabled).
    /// If this is a Value, returns the label of the value type of the value.
    /// If this is a Type, returns the label of the type.
    pub fn try_get_label(&self) -> Option<&str> {
        match self {
            Self::EntityType(entity_type) => Some(entity_type.label()),
            Self::RelationType(relation_type) => Some(relation_type.label()),
            Self::AttributeType(attribute_type) => Some(attribute_type.label()),
            Self::RoleType(role_type) => Some(role_type.label()),
            Self::Entity(entity) => entity.type_().map(|type_| type_.label()),
            Self::Relation(relation) => relation.type_().map(|type_| type_.label()),
            Self::Attribute(attribute) => attribute.type_().map(|type_| type_.label()),
            Self::Value(value) => Some(value.get_type_name()),
        }
    }

    /// Retrieves the label of the value type of the concept, if it exists.
    /// If this is an Attribute Instance, returns the label of the value of this instance.
    /// If this is a Value, returns the label of the value.
    /// If this is an Attribute Type, returns the label of the value type that the schema permits for the attribute type, if one is defined.
    /// Otherwise, returns None.
    pub fn try_get_value_label(&self) -> Option<&str> {
        match self {
            Self::AttributeType(attribute_type) => attribute_type.value_type().map(|value_type| value_type.name()),
            Self::Attribute(attribute) => Some(attribute.value.get_type_name()),
            Self::Value(value) => Some(value.get_type_name()),
            _ => None,
        }
    }

    /// Retrieves the value type enum of the concept, if it exists.
    /// If this is an Attribute Instance, returns the value type of the value of this instance.
    /// If this is a Value, returns the value type of the value.
    /// If this is an Attribute Type, returns value type that the schema permits for the attribute type, if one is defined.
    /// Otherwise, returns None.
    pub fn try_get_value_type(&self) -> Option<ValueType> {
        match self {
            Self::AttributeType(attribute_type) => attribute_type.value_type().cloned(),
            Self::Attribute(attribute) => Some(attribute.value.get_type()),
            Self::Value(value) => Some(value.get_type()),
            _ => None,
        }
    }

    /// Retrieves the value of this Concept, if it exists.
    /// If this is an Attribute Instance, returns the value of this instance.
    /// If this a Value, returns the value.
    /// Otherwise, returns empty.
    pub fn try_get_value(&self) -> Option<&Value> {
        match self {
            Self::Attribute(attribute) => Some(&attribute.value),
            Self::Value(value) => Some(value),
            _ => None,
        }
    }

    /// Retrieves the boolean value of this Concept, if it exists.
    /// If this is a boolean-valued Attribute Instance, returns the boolean value of this instance.
    /// If this a boolean-valued Value, returns the boolean value.
    /// Otherwise, returns None.
    pub fn try_get_boolean(&self) -> Option<bool> {
        self.try_get_value().and_then(|value| value.get_boolean())
    }

    /// Retrieves the integer value of this Concept, if it exists.
    /// If this is an integer-valued Attribute Instance, returns the integer value of this instance.
    /// If this an integer-valued Value, returns the integer value.
    /// Otherwise, returns None.
    pub fn try_get_integer(&self) -> Option<i64> {
        self.try_get_value().and_then(|value| value.get_integer())
    }

    /// Retrieves the double value of this Concept, if it exists.
    /// If this is a double-valued Attribute Instance, returns the double value of this instance.
    /// If this a double-valued Value, returns the double value.
    /// Otherwise, returns None.
    pub fn try_get_double(&self) -> Option<f64> {
        self.try_get_value().and_then(|value| value.get_double())
    }

    /// Retrieves the fixed-decimal value of this Concept, if it exists.
    /// If this is a fixed-decimal valued Attribute Instance, returns the fixed-decimal value of this instance.
    /// If this a fixed-decimal valued Value, returns the fixed-decimal value.
    /// Otherwise, returns None.
    pub fn try_get_decimal(&self) -> Option<Decimal> {
        self.try_get_value().and_then(|value| value.get_decimal())
    }

    /// Retrieves the string value of this Concept, if it exists.
    /// If this is a string-valued Attribute Instance, returns the string value of this instance.
    /// If this a string-valued Value, returns the string value.
    /// Otherwise, returns None.
    pub fn try_get_string(&self) -> Option<&str> {
        self.try_get_value().and_then(|value| value.get_string())
    }

    /// Retrieves the date value of this Concept, if it exists.
    /// If this is a date-valued Attribute Instance, returns the date value of this instance.
    /// If this a date-valued Value, returns the date value.
    /// Otherwise, returns None.
    pub fn try_get_date(&self) -> Option<NaiveDate> {
        self.try_get_value().and_then(|value| value.get_date())
    }

    /// Retrieves the datetime value of this Concept, if it exists.
    /// If this is a datetime-valued Attribute Instance, returns the datetime value of this instance.
    /// If this a datetime-valued Value, returns the datetime value.
    /// Otherwise, returns None.
    pub fn try_get_datetime(&self) -> Option<NaiveDateTime> {
        self.try_get_value().and_then(|value| value.get_datetime())
    }

    /// Retrieves the timezoned-datetime value of this Concept, if it exists.
    /// If this is a timezoned-datetime valued Attribute Instance, returns the timezoned-datetime value of this instance.
    /// If this a timezoned-datetime valued Value, returns the timezoned-datetime value.
    /// Otherwise, returns None.
    pub fn try_get_datetime_tz(&self) -> Option<DateTime<TimeZone>> {
        self.try_get_value().and_then(|value| value.get_datetime_tz())
    }

    /// Retrieves the duration value of this Concept, if it exists.
    /// If this is a duration-valued Attribute Instance, returns the duration value of this instance.
    /// If this a duration-valued Value, returns the duration value.
    /// Otherwise, returns None.
    pub fn try_get_duration(&self) -> Option<Duration> {
        self.try_get_value().and_then(|value| value.get_duration())
    }

    /// Retrieves the struct value of this Concept, if it exists.
    /// If this is a struct-valued Attribute Instance, returns the struct value of this instance.
    /// If this a struct-valued Value, returns the struct value.
    /// Otherwise, returns None.
    pub fn try_get_struct(&self) -> Option<&Struct> {
        self.try_get_value().and_then(|value| value.get_struct())
    }

    /// Retrieves the category of this Concept.
    pub fn get_category(&self) -> ConceptCategory {
        match self {
            Self::EntityType(_) => ConceptCategory::EntityType,
            Self::RelationType(_) => ConceptCategory::RelationType,
            Self::RoleType(_) => ConceptCategory::RoleType,
            Self::AttributeType(_) => ConceptCategory::AttributeType,
            Self::Entity(_) => ConceptCategory::Entity,
            Self::Relation(_) => ConceptCategory::Relation,
            Self::Attribute(_) => ConceptCategory::Attribute,
            Self::Value(_) => ConceptCategory::Value,
        }
    }

    /// Check if this Concept represents a Type from the schema of the database.
    /// These are exactly: Entity Types, Relation Types, Role Types, and Attribute Types
    ///
    /// Equivalent to:
    /// ```
    /// concept.is_entity_type() || concept.is_relation_type() || concept.is_role_type() || concept.is_attribute_type()
    /// ```
    pub fn is_type(&self) -> bool {
        matches!(self, Self::EntityType(_) | Self::RelationType(_) | Self::RoleType(_) | Self::AttributeType(_))
    }

    /// Check if this Concept represents an Entity Type from the schema of the database
    pub fn is_entity_type(&self) -> bool {
        matches!(self.get_category(), ConceptCategory::EntityType)
    }

    /// Check if this Concept represents a Relation Type from the schema of the database
    pub fn is_relation_type(&self) -> bool {
        matches!(self.get_category(), ConceptCategory::RelationType)
    }

    /// Check if this Concept represents a Role Type from the schema of the database
    pub fn is_role_type(&self) -> bool {
        matches!(self.get_category(), ConceptCategory::RoleType)
    }

    /// Check if this Concept represents an Attribute Type from the schema of the database
    pub fn is_attribute_type(&self) -> bool {
        matches!(self.get_category(), ConceptCategory::AttributeType)
    }

    /// Check if this Concept represents a stored database instance from the database.
    /// These are exactly: Entity, Relation, and Attribute
    ///
    /// Equivalent to:
    /// ```
    /// concept.is_entity() || concept.is_relation() ||  concept.is_attribute()
    /// ```
    pub fn is_instance(&self) -> bool {
        matches!(self, Self::Entity(_) | Self::Relation(_) | Self::Attribute(_))
    }

    /// Check if this Concept represents an Entity instance from the database
    pub fn is_entity(&self) -> bool {
        matches!(self.get_category(), ConceptCategory::Entity)
    }

    /// Check if this Concept represents an Relation instance from the database
    pub fn is_relation(&self) -> bool {
        matches!(self.get_category(), ConceptCategory::Relation)
    }

    /// Check if this Concept represents an Attribute instance from the database
    pub fn is_attribute(&self) -> bool {
        matches!(self.get_category(), ConceptCategory::Attribute)
    }

    /// Check if this Concept represents a Value returned by the database
    pub fn is_value(&self) -> bool {
        matches!(self.get_category(), ConceptCategory::Value)
    }

    /// Check if this Concept holds a boolean as an AttributeType, an Attribute, or a Value
    pub fn is_boolean(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::Boolean))
    }

    /// Check if this Concept holds an integer as an AttributeType, an Attribute, or a Value
    pub fn is_integer(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::Integer))
    }

    /// Check if this Concept holds a fixed-decimal as an AttributeType, an Attribute, or a Value
    pub fn is_decimal(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::Decimal))
    }

    /// Check if this Concept holds a double as an AttributeType, an Attribute, or a Value
    pub fn is_double(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::Double))
    }

    /// Check if this Concept holds a string as an AttributeType, an Attribute, or a Value
    pub fn is_string(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::String))
    }

    /// Check if this Concept holds a date as an AttributeType, an Attribute, or a Value
    pub fn is_date(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::Date))
    }

    /// Check if this Concept holds a datetime as an AttributeType, an Attribute, or a Value
    pub fn is_datetime(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::Datetime))
    }

    /// Check if this Concept holds a timezoned-datetime as an AttributeType, an Attribute, or a Value
    pub fn is_datetime_tz(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::DatetimeTZ))
    }

    /// Check if this Concept holds a duration as an AttributeType, an Attribute, or a Value
    pub fn is_duration(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::Duration))
    }

    /// Check if this Concept holds a struct as an AttributeType, an Attribute, or a Value
    pub fn is_struct(&self) -> bool {
        matches!(self.try_get_value_type(), Some(ValueType::Struct(_)))
    }
}

impl fmt::Display for Concept {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        fmt::Debug::fmt(self, f)
    }
}

impl fmt::Debug for Concept {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        if self.is_type() {
            write!(f, "{}({})", self.get_category(), self.get_label())
        } else {
            write!(f, "{}({}", self.get_category(), self.get_label())?;
            if self.try_get_iid().is_some() {
                write!(f, ": {}", self.try_get_iid().unwrap())?;
                if self.try_get_value().is_some() {
                    write!(f, ", {}", self.try_get_value().unwrap())?;
                }
            } else if self.try_get_value().is_some() {
                write!(f, ": {}", self.try_get_value().unwrap())?;
            } else {
                // shouldn't be reachable?
            }
            write!(f, ")")
        }
    }
}

/// Kind represents the base of a defined type to describe its capabilities.
/// For example, "define entity person;" defines a type "person" of a kind "entity".
#[repr(C)]
#[derive(Copy, Clone, Hash, PartialEq, Eq)]
pub enum Kind {
    Entity,
    Attribute,
    Relation,
    Role,
}

impl Kind {
    pub const fn name(&self) -> &'static str {
        match self {
            Kind::Entity => "entity",
            Kind::Attribute => "attribute",
            Kind::Relation => "relation",
            Kind::Role => "relation:role",
        }
    }
}

impl fmt::Debug for Kind {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "Kind[{}]", self.name())
    }
}

impl fmt::Display for Kind {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.name())
    }
}