entity-derive-impl 0.3.0

// SPDX-FileCopyrightText: 2025-2026 RAprogramm <andrey.rozanov.vl@gmail.com>
// SPDX-License-Identifier: MIT

//! Migration generation for entity-derive.
//!
//! Generates `MIGRATION_UP` and `MIGRATION_DOWN` constants containing
//! SQL DDL statements for creating/dropping tables.
//!
//! # Features
//!
//! - Full type mapping (Rust → PostgreSQL)
//! - Column constraints (UNIQUE, CHECK, DEFAULT)
//! - Indexes (btree, hash, gin, gist, brin)
//! - Foreign keys with ON DELETE actions
//! - Composite indexes
//!
//! # Usage
//!
//! ```rust,ignore
//! #[derive(Entity)]
//! #[entity(table = "users", migrations)]
//! pub struct User {
//!     #[id]
//!     pub id: Uuid,
//!
//!     #[column(unique, index)]
//!     pub email: String,
//! }
//!
//! // Apply migration:
//! sqlx::query(User::MIGRATION_UP).execute(&pool).await?;
//! ```

mod postgres;
pub mod types;

use proc_macro2::TokenStream;

use super::parse::{DatabaseDialect, EntityDef};

/// Generate migration constants based on entity configuration.
///
/// Returns empty `TokenStream` if migrations are not enabled.
pub fn generate(entity: &EntityDef) -> TokenStream {
    if !entity.migrations {
        return TokenStream::new();
    }

    match entity.dialect {
        DatabaseDialect::Postgres => postgres::generate(entity),
        DatabaseDialect::ClickHouse => TokenStream::new(), // TODO: future
        DatabaseDialect::MongoDB => TokenStream::new()     // N/A for document DB
    }
}

#[cfg(test)]
mod tests {
    use syn::DeriveInput;

    use super::*;

    fn parse_entity(tokens: proc_macro2::TokenStream) -> EntityDef {
        let input: DeriveInput = syn::parse_quote!(#tokens);
        EntityDef::from_derive_input(&input).unwrap()
    }

    #[test]
    fn generate_returns_empty_when_migrations_disabled() {
        let entity = parse_entity(quote::quote! {
            #[entity(table = "users")]
            pub struct User {
                #[id]
                pub id: uuid::Uuid,
            }
        });
        let result = generate(&entity);
        assert!(result.is_empty());
    }

    #[test]
    fn generate_returns_tokens_when_migrations_enabled() {
        let entity = parse_entity(quote::quote! {
            #[entity(table = "users", migrations)]
            pub struct User {
                #[id]
                pub id: uuid::Uuid,
            }
        });
        let result = generate(&entity);
        assert!(!result.is_empty());
    }

    #[test]
    fn generate_returns_empty_for_clickhouse() {
        let entity = parse_entity(quote::quote! {
            #[entity(table = "users", dialect = "clickhouse", migrations)]
            pub struct User {
                #[id]
                pub id: uuid::Uuid,
            }
        });
        let result = generate(&entity);
        assert!(result.is_empty());
    }

    #[test]
    fn generate_returns_empty_for_mongodb() {
        let entity = parse_entity(quote::quote! {
            #[entity(table = "users", dialect = "mongodb", migrations)]
            pub struct User {
                #[id]
                pub id: uuid::Uuid,
            }
        });
        let result = generate(&entity);
        assert!(result.is_empty());
    }
}