es_entity/

macros.rs

#[macro_export]
macro_rules! idempotency_guard {
    ($events:expr, $( $pattern:pat $(if $guard:expr)? ),+ $(,)?) => {
        for event in $events {
            match event {
                $(
                    $pattern $(if $guard)? => return $crate::FromIdempotentIgnored::from_ignored(),
                )+
                _ => {}
            }
        }
    };
    ($events:expr, $( $pattern:pat $(if $guard:expr)? ),+,
     => $break_pattern:pat $(if $break_guard:expr)?) => {
        for event in $events {
            match event {
                $($pattern $(if $guard)? => return $crate::FromIdempotentIgnored::from_ignored(),)+
                $break_pattern $(if $break_guard)? => break,
                _ => {}
            }
        }
    };
}

/// Execute an event-sourced query with automatic entity reconstruction.
///
/// This macro supports the following optional parameters:
/// - `entity_ty`: Override the entity type (useful when table name is customized)
/// - `id_ty`: Override the ID type (defaults to {Entity}Id)
/// - Prefix literal: Table prefix to ignore when deriving names
/// - `executor`: The database executor
/// - `sql`: The SQL query string
/// - Additional arguments for the SQL query
///
/// # Examples
/// ```ignore
/// // Basic usage
/// es_query!(executor, "SELECT id FROM users WHERE id = $1", id)
///
/// // With custom entity and ID types
/// es_query!(
///     entity_ty = MyEntity,
///     id_ty = MyEntityId,
///     executor,
///     "SELECT id FROM custom_table WHERE id = $1",
///     id
/// )
/// ```
#[macro_export]
macro_rules! es_query {
    // With entity_ty and id_ty
    (entity_ty = $entity_ty:ident, id_ty = $id_ty:ident, $prefix:literal, $db:expr, $query:expr) => ({
        $crate::expand_es_query!(
            entity_ty = $entity_ty,
            id_ty = $id_ty,
            ignore_prefix = $prefix,
            executor = $db,
            sql = $query
        )
    });
    (entity_ty = $entity_ty:ident, id_ty = $id_ty:ident, $prefix:literal, $db:expr, $query:expr, $($args:tt)*) => ({
        $crate::expand_es_query!(
            entity_ty = $entity_ty,
            id_ty = $id_ty,
            ignore_prefix = $prefix,
            executor = $db,
            sql = $query,
            args = [$($args)*]
        )
    });
    (entity_ty = $entity_ty:ident, id_ty = $id_ty:ident, $db:expr, $query:expr) => ({
        $crate::expand_es_query!(
            entity_ty = $entity_ty,
            id_ty = $id_ty,
            executor = $db,
            sql = $query
        )
    });
    (entity_ty = $entity_ty:ident, id_ty = $id_ty:ident, $db:expr, $query:expr, $($args:tt)*) => ({
        $crate::expand_es_query!(
            entity_ty = $entity_ty,
            id_ty = $id_ty,
            executor = $db,
            sql = $query,
            args = [$($args)*]
        )
    });
    // With only entity_ty
    (entity_ty = $entity_ty:ident, $prefix:literal, $db:expr, $query:expr) => ({
        $crate::expand_es_query!(
            entity_ty = $entity_ty,
            ignore_prefix = $prefix,
            executor = $db,
            sql = $query
        )
    });
    (entity_ty = $entity_ty:ident, $prefix:literal, $db:expr, $query:expr, $($args:tt)*) => ({
        $crate::expand_es_query!(
            entity_ty = $entity_ty,
            ignore_prefix = $prefix,
            executor = $db,
            sql = $query,
            args = [$($args)*]
        )
    });
    (entity_ty = $entity_ty:ident, $db:expr, $query:expr) => ({
        $crate::expand_es_query!(
            entity_ty = $entity_ty,
            executor = $db,
            sql = $query
        )
    });
    (entity_ty = $entity_ty:ident, $db:expr, $query:expr, $($args:tt)*) => ({
        $crate::expand_es_query!(
            entity_ty = $entity_ty,
            executor = $db,
            sql = $query,
            args = [$($args)*]
        )
    });
    // With only id_ty (existing patterns)
    (id_ty = $id_ty:ident, $prefix:literal, $db:expr, $query:expr) => ({
        $crate::expand_es_query!(
            id_ty = $id_ty,
            ignore_prefix = $prefix,
            executor = $db,
            sql = $query
        )
    });
    (id_ty = $id_ty:ident, $prefix:literal, $db:expr, $query:expr, $($args:tt)*) => ({
        $crate::expand_es_query!(
            id_ty = $id_ty,
            ignore_prefix = $prefix,
            executor = $db,
            sql = $query,
            args = [$($args)*]
        )
    });
    (id_ty = $id_ty:ident, $db:expr, $query:expr) => ({
        $crate::expand_es_query!(
            id_ty = $id_ty,
            executor = $db,
            sql = $query
        )
    });
    (id_ty = $id_ty:ident, $db:expr, $query:expr, $($args:tt)*) => ({
        $crate::expand_es_query!(
            id_ty = $id_ty,
            executor = $db,
            sql = $query,
            args = [$($args)*]
        )
    });
    // Without entity_ty or id_ty (existing patterns)
    ($prefix:literal, $db:expr, $query:expr) => ({
        $crate::expand_es_query!(
            ignore_prefix = $prefix,
            executor = $db,
            sql = $query
        )
    });
    ($prefix:literal, $db:expr, $query:expr, $($args:tt)*) => ({
        $crate::expand_es_query!(
            ignore_prefix = $prefix,
            executor = $db,
            sql = $query,
            args = [$($args)*]
        )
    });
    ($db:expr, $query:expr) => ({
        $crate::expand_es_query!(
            executor = $db,
            sql = $query
        )
    });
    ($db:expr, $query:expr, $($args:tt)*) => ({
        $crate::expand_es_query!(
            executor = $db,
            sql = $query,
            args = [$($args)*]
        )
    });
}

#[macro_export]
macro_rules! from_es_entity_error {
    ($name:ident) => {
        impl $name {
            pub fn was_not_found(&self) -> bool {
                matches!(self, $name::EsEntityError($crate::EsEntityError::NotFound))
            }
            pub fn was_concurrent_modification(&self) -> bool {
                matches!(
                    self,
                    $name::EsEntityError($crate::EsEntityError::ConcurrentModification)
                )
            }
        }
        impl From<$crate::EsEntityError> for $name {
            fn from(e: $crate::EsEntityError) -> Self {
                $name::EsEntityError(e)
            }
        }
    };
}

// Helper macro for common entity_id implementations (internal use only)
#[doc(hidden)]
#[macro_export]
macro_rules! __entity_id_common_impls {
    ($name:ident) => {
        impl $name {
            #[allow(clippy::new_without_default)]
            pub fn new() -> Self {
                $crate::prelude::uuid::Uuid::new_v4().into()
            }
        }

        impl From<$crate::prelude::uuid::Uuid> for $name {
            fn from(uuid: $crate::prelude::uuid::Uuid) -> Self {
                Self(uuid)
            }
        }

        impl From<$name> for $crate::prelude::uuid::Uuid {
            fn from(id: $name) -> Self {
                id.0
            }
        }

        impl From<&$name> for $crate::prelude::uuid::Uuid {
            fn from(id: &$name) -> Self {
                id.0
            }
        }

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

        impl std::str::FromStr for $name {
            type Err = $crate::prelude::uuid::Error;

            fn from_str(s: &str) -> Result<Self, Self::Err> {
                Ok(Self($crate::prelude::uuid::Uuid::parse_str(s)?))
            }
        }
    };
}

// Helper macro for GraphQL-specific entity_id implementations (internal use only)
#[doc(hidden)]
#[macro_export]
macro_rules! __entity_id_graphql_impls {
    ($name:ident) => {
        impl From<$crate::graphql::UUID> for $name {
            fn from(id: $crate::graphql::UUID) -> Self {
                $name($crate::prelude::uuid::Uuid::from(&id))
            }
        }

        impl From<&$crate::graphql::UUID> for $name {
            fn from(id: &$crate::graphql::UUID) -> Self {
                $name($crate::prelude::uuid::Uuid::from(id))
            }
        }
    };
}

// Helper macro for additional conversions (internal use only)
#[doc(hidden)]
#[macro_export]
macro_rules! __entity_id_conversions {
    ($($from:ty => $to:ty),* $(,)?) => {
        $(
            impl From<$from> for $to {
                fn from(id: $from) -> Self {
                    <$to>::from($crate::prelude::uuid::Uuid::from(id))
                }
            }
            impl From<$to> for $from {
                fn from(id: $to) -> Self {
                    <$from>::from($crate::prelude::uuid::Uuid::from(id))
                }
            }
        )*
    };
}

#[cfg(all(feature = "graphql", feature = "json-schema"))]
#[macro_export]
macro_rules! entity_id {
    // Match identifiers without conversions
    ($($name:ident),+ $(,)?) => {
        $crate::entity_id! { $($name),+ ; }
    };
    ($($name:ident),+ $(,)? ; $($from:ty => $to:ty),* $(,)?) => {
        $(
            #[derive(
                $crate::prelude::sqlx::Type,
                Debug,
                Clone,
                Copy,
                PartialEq,
                Eq,
                PartialOrd,
                Ord,
                Hash,
                $crate::prelude::serde::Deserialize,
                $crate::prelude::serde::Serialize,
                $crate::prelude::schemars::JsonSchema,
            )]
            #[serde(transparent)]
            #[sqlx(transparent)]
            pub struct $name($crate::prelude::uuid::Uuid);
            $crate::__entity_id_common_impls!($name);
            $crate::__entity_id_graphql_impls!($name);
        )+
        $crate::__entity_id_conversions!($($from => $to),*);
    };
}

#[cfg(all(feature = "graphql", not(feature = "json-schema")))]
#[macro_export]
macro_rules! entity_id {
    // Match identifiers without conversions
    ($($name:ident),+ $(,)?) => {
        $crate::entity_id! { $($name),+ ; }
    };
    ($($name:ident),+ $(,)? ; $($from:ty => $to:ty),* $(,)?) => {
        $(
            #[derive(
                $crate::prelude::sqlx::Type,
                Debug,
                Clone,
                Copy,
                PartialEq,
                Eq,
                PartialOrd,
                Ord,
                Hash,
                $crate::prelude::serde::Deserialize,
                $crate::prelude::serde::Serialize,
            )]
            #[serde(transparent)]
            #[sqlx(transparent)]
            pub struct $name($crate::prelude::uuid::Uuid);
            $crate::__entity_id_common_impls!($name);
            $crate::__entity_id_graphql_impls!($name);
        )+
        $crate::__entity_id_conversions!($($from => $to),*);
    };
}

#[cfg(all(feature = "json-schema", not(feature = "graphql")))]
#[macro_export]
macro_rules! entity_id {
    // Match identifiers without conversions
    ($($name:ident),+ $(,)?) => {
        $crate::entity_id! { $($name),+ ; }
    };
    ($($name:ident),+ $(,)? ; $($from:ty => $to:ty),* $(,)?) => {
        $(
            #[derive(
                $crate::prelude::sqlx::Type,
                Debug,
                Clone,
                Copy,
                PartialEq,
                Eq,
                PartialOrd,
                Ord,
                Hash,
                $crate::prelude::serde::Deserialize,
                $crate::prelude::serde::Serialize,
                $crate::prelude::schemars::JsonSchema,
            )]
            #[serde(transparent)]
            #[sqlx(transparent)]
            pub struct $name($crate::prelude::uuid::Uuid);
            $crate::__entity_id_common_impls!($name);
        )+
        $crate::__entity_id_conversions!($($from => $to),*);
    };
}

#[cfg(all(not(feature = "json-schema"), not(feature = "graphql")))]
#[macro_export]
macro_rules! entity_id {
    // Match identifiers without conversions
    ($($name:ident),+ $(,)?) => {
        $crate::entity_id! { $($name),+ ; }
    };
    ($($name:ident),+ $(,)? ; $($from:ty => $to:ty),* $(,)?) => {
        $(
            #[derive(
                $crate::prelude::sqlx::Type,
                Debug,
                Clone,
                Copy,
                PartialEq,
                Eq,
                PartialOrd,
                Ord,
                Hash,
                $crate::prelude::serde::Deserialize,
                $crate::prelude::serde::Serialize,
            )]
            #[serde(transparent)]
            #[sqlx(transparent)]
            pub struct $name($crate::prelude::uuid::Uuid);
            $crate::__entity_id_common_impls!($name);
        )+
        $crate::__entity_id_conversions!($($from => $to),*);
    };
}