type_state_builder/

lib.rs

//! Type-State Builder Pattern Implementation
//!
//! A derive macro that generates compile-time safe builders using the type-state pattern.
//! It prevents incomplete object construction by making missing required fields a
//! compile-time error rather than a runtime failure.
//!
//! For complete documentation, installation instructions, and guides, see the
//! [README](https://github.com/welf/type-state-builder#readme).
//!
//! # Design Philosophy
//!
//! This crate was designed with AI-assisted development in mind. Two principles
//! guided its design:
//!
//! ## Compiler-Enforced Correctness
//!
//! In AI-assisted development, code generation happens rapidly. By encoding field
//! requirements in the type system, errors are caught immediately by the compiler
//! rather than manifesting as runtime failures. If the code compiles, the builder
//! is correctly configured.
//!
//! ## Actionable Error Messages
//!
//! Instead of using generic type parameters to track field states (which produce
//! cryptic error messages), this crate generates named structs for each state:
//!
//! ```text
//! UserBuilder_HasName_MissingEmail    // Name set, email missing
//! UserBuilder_HasName_HasEmail        // Both set - build() available
//! ```
//!
//! When an error occurs, the type name immediately indicates which fields are missing.
//! This is particularly valuable for AI assistants that can parse and act on the error.
//!
//! ## Trade-offs
//!
//! This approach generates more structs than type-parameter-based solutions, slightly
//! increasing compile time. However, the improved error message clarity is worth this
//! cost. There is no runtime cost due to Rust's zero-cost abstractions.
//!
//! # Compatibility
//!
//! - **no_std**: Fully compatible. Generated code uses only `core` types.
//! - **MSRV**: Rust 1.70.0 or later.
//!
//! # Overview
//!
//! The macro automatically selects between two builder patterns:
//!
//! - **Type-State Builder**: For structs with required fields, providing compile-time
//!   safety that prevents calling `build()` until all required fields are set
//! - **Regular Builder**: For structs with only optional fields, providing a simple
//!   builder with immediate `build()` availability
//!
//! # Quick Start
//!
//! Add the derive macro to your struct and mark required fields:
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! #[builder(impl_into)]   // Setters arguments are now `impl Into<FieldType>`
//! struct User {
//!     #[builder(required)]
//!     name: String,
//!
//!     #[builder(required)]
//!     email: String,
//!
//!     #[builder(default = Some(30))] // Default value for age
//!     age: Option<u32>,
//!
//!     #[builder(setter_prefix = "is_")] // The setter is now named `is_active`
//!     active: bool, // Will use Default::default() if not set
//! }
//!
//! // Usage - this enforces that name and email are set
//! let user = User::builder()
//!     .name("Alice")              // `impl_into` allows to pass any type that can be converted into String
//!     .email("alice@example.com") // `impl_into` allows to pass any type that can be converted into String
//!     // age is not set and defaults to Some(30)
//!     // active is not set and defaults to false (Default::default())
//!     .build(); // The `build()` method is available since all required fields are set

//!
//! assert_eq!(user.name, "Alice".to_string());
//! assert_eq!(user.email, "alice@example.com".to_string());
//! assert_eq!(user.age, Some(30));
//! assert_eq!(user.active, false);
//! ```
//!
//! # Supported Attributes
//!
//! ## Struct-level Attributes
//!
//! - `#[builder(build_method = "method_name")]` - Custom build method name
//! - `#[builder(setter_prefix = "prefix_")]` - Prefix for all setter method names
//! - `#[builder(impl_into)]` - Generate setters with `impl Into<FieldType>` parameters
//!
//! ## Field-level Attributes
//!
//! - `#[builder(required)]` - Mark field as required
//! - `#[builder(setter_name = "name")]` - Custom setter method name
//! - `#[builder(setter_prefix = "prefix_")]` - Custom prefix for setter method name
//! - `#[builder(default = expression)]` - Custom default value
//! - `#[builder(skip_setter)]` - Don't generate setter (requires default)
//! - `#[builder(impl_into)]` - Generate setter with `impl Into<FieldType>` parameter
//! - `#[builder(impl_into = false)]` - Override struct-level `impl_into` for this field
//! - `#[builder(converter = |param: InputType| -> FieldType { expression })` - Custom conversion logic for setter input
//!
//! # Advanced Examples
//!
//! ## Custom Defaults and Setter Names
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! #[builder(build_method = "create")]
//! struct Document {
//!     #[builder(required)]
//!     title: String,
//!
//!     #[builder(required, setter_name = "set_content")]
//!     content: String,
//!
//!     #[builder(default = 42)]
//!     page_count: u32,
//!
//!     #[builder(default = String::from("draft"), skip_setter)]
//!     status: String,
//! }
//!
//! let doc = Document::builder()
//!     .title("My Document".to_string())
//!     .set_content("Document content here".to_string())
//!     .create(); // Custom build method name
//! ```
//!
//! ## Generic Types and Lifetimes
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct Container<T: Clone>
//! where
//!     T: Send
//! {
//!     #[builder(required)]
//!     value: T,
//!
//!     #[builder(required)]
//!     name: String,
//!
//!     tags: Vec<String>,
//! }
//!
//! let container = Container::builder()
//!     .value(42)
//!     .name("test".to_string())
//!     .tags(vec!["tag1".to_string()])
//!     .build();
//! ```
//!
//! ## Setter Prefix Examples
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! // Struct-level setter prefix applies to all fields
//! #[derive(TypeStateBuilder)]
//! #[builder(setter_prefix = "with_")]
//! struct Config {
//!     #[builder(required)]
//!     host: String,
//!
//!     #[builder(required)]
//!     port: u16,
//!
//!     timeout: Option<u32>,
//! }
//!
//! let config = Config::builder()
//!     .with_host("localhost".to_string())
//!     .with_port(8080)
//!     .with_timeout(Some(30))
//!     .build();
//! ```
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! // Field-level setter prefix overrides struct-level prefix
//! #[derive(TypeStateBuilder)]
//! #[builder(setter_prefix = "with_")]
//! struct Database {
//!     #[builder(required)]
//!     connection_string: String,
//!
//!     #[builder(required, setter_prefix = "set_")]
//!     credentials: String,
//!
//!     #[builder(setter_name = "timeout_seconds")]
//!     timeout: Option<u32>,
//! }
//!
//! let db = Database::builder()
//!     .with_connection_string("postgresql://...".to_string())
//!     .set_credentials("user:pass".to_string())
//!     .with_timeout_seconds(Some(60))
//!     .build();
//! ```
//!
//! ## Ergonomic Conversions with `impl_into`
//!
//! The `impl_into` attribute generates setter methods that accept `impl Into<FieldType>`
//! parameters, allowing for more ergonomic API usage by automatically converting
//! compatible types.
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! // Struct-level impl_into applies to all setters
//! #[derive(TypeStateBuilder)]
//! #[builder(impl_into)]
//! struct ApiClient {
//!     #[builder(required)]
//!     base_url: String,
//!
//!     #[builder(required)]
//!     api_key: String,
//!
//!     timeout: Option<u32>,
//!     user_agent: String, // Uses Default::default()
//! }
//!
//! // Can now use &str directly instead of String::from() or .to_string()
//! let client = ApiClient::builder()
//!     .base_url("https://api.example.com")    // &str -> String
//!     .api_key("secret-key")                   // &str -> String
//!     .timeout(Some(30))
//!     .user_agent("MyApp/1.0")                 // &str -> String
//!     .build();
//! ```
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! // Field-level control with precedence rules
//! #[derive(TypeStateBuilder)]
//! #[builder(impl_into)]  // Default for all fields
//! struct Document {
//!     #[builder(required)]
//!     title: String,  // Inherits impl_into = true
//!
//!     #[builder(required, impl_into = false)]
//!     content: String,  // Override: requires String directly
//!
//!     #[builder(impl_into = true)]
//!     category: Option<String>,  // Explicit impl_into = true
//!
//!     #[builder(impl_into = false)]
//!     tags: Vec<String>,  // Override: requires Vec<String> directly
//! }
//!
//! let doc = Document::builder()
//!     .title("My Document")                // &str -> String (inherited)
//!     .content("Content".to_string())      // Must use String (override)
//!     .category(Some("tech".to_string()))  // impl Into for Option<String>
//!     .tags(vec!["rust".to_string()])      // Must use Vec<String> (override)
//!     .build();
//! ```
//!
//! **Note**: `impl_into` is incompatible with `skip_setter` since skipped fields
//! don't have setter methods generated.
//!
//! ### Complete `impl_into` Example
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//! use std::path::PathBuf;
//!
//! // Demonstrate both struct-level and field-level impl_into usage
//! #[derive(TypeStateBuilder)]
//! #[builder(impl_into)]  // Struct-level default: enable for all fields
//! struct CompleteExample {
//!     #[builder(required)]
//!     name: String,                        // Inherits: accepts impl Into<String>
//!
//!     #[builder(required, impl_into = false)]
//!     id: String,                          // Override: requires String directly
//!
//!     #[builder(impl_into = true)]
//!     description: Option<String>,         // Explicit: accepts impl Into<String>
//!
//!     #[builder(default = PathBuf::from("/tmp"))]
//!     work_dir: PathBuf,                   // Inherits: accepts impl Into<PathBuf>
//!
//!     #[builder(impl_into = false, default = Vec::new())]
//!     tags: Vec<String>,                   // Override: requires Vec<String>
//! }
//!
//! // Usage demonstrating the different setter behaviors
//! let example = CompleteExample::builder()
//!     .name("Alice")                       // &str -> String (inherited impl_into)
//!     .id("user_123".to_string())          // Must use String (override = false)
//!     .description(Some("Engineer".to_string()))  // Option<String> required
//!     .work_dir("/home/alice")             // &str -> PathBuf (inherited impl_into)
//!     .tags(vec!["rust".to_string()])      // Must use Vec<String> (override = false)
//!     .build();
//!
//! assert_eq!(example.name, "Alice");
//! assert_eq!(example.id, "user_123");
//! assert_eq!(example.description, Some("Engineer".to_string()));
//! assert_eq!(example.work_dir, PathBuf::from("/home/alice"));
//! assert_eq!(example.tags, vec!["rust"]);
//! ```
//!
//! ## Custom Conversions with `converter`
//!
//! The `converter` attribute allows you to specify custom conversion logic for field setters,
//! enabling more powerful transformations than `impl_into` can provide. Use closure syntax
//! to define the conversion function with explicit parameter types.
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct User {
//!     #[builder(required)]
//!     name: String,
//!
//!     // Normalize email to lowercase and trim whitespace
//!     #[builder(required, converter = |email: &str| email.trim().to_lowercase())]
//!     email: String,
//!
//!     // Parse comma-separated tags into Vec<String>
//!     #[builder(converter = |tags: &str| tags.split(',').map(|s| s.trim().to_string()).collect())]
//!     interests: Vec<String>,
//!
//!     // Parse age from string with fallback
//!     #[builder(converter = |age: &str| age.parse().unwrap_or(0))]
//!     age: u32,
//! }
//!
//! let user = User::builder()
//!     .name("Alice".to_string())
//!     .email("  ALICE@EXAMPLE.COM  ")  // Will be normalized to "alice@example.com"
//!     .interests("rust, programming, web")  // Parsed to Vec<String>
//!     .age("25")  // Parsed from string to u32
//!     .build();
//!
//! assert_eq!(user.email, "alice@example.com");
//! assert_eq!(user.interests, vec!["rust", "programming", "web"]);
//! assert_eq!(user.age, 25);
//! ```
//!
//! ### Complex Converter Examples
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//! use std::collections::HashMap;
//!
//! #[derive(TypeStateBuilder)]
//! struct Config {
//!     // Convert environment-style boolean strings
//!     #[builder(converter = |enabled: &str| {
//!         matches!(enabled.to_lowercase().as_str(), "true" | "1" | "yes" | "on")
//!     })]
//!     debug_enabled: bool,
//!
//!     // Parse key=value pairs into HashMap
//!     #[builder(converter = |pairs: &str| {
//!         pairs.split(',')
//!              .filter_map(|pair| {
//!                  let mut split = pair.split('=');
//!                  Some((split.next()?.trim().to_string(),
//!                       split.next()?.trim().to_string()))
//!              })
//!              .collect()
//!     })]
//!     env_vars: HashMap<String, String>,
//!
//!     // Transform slice to owned Vec
//!     #[builder(converter = |hosts: &[&str]| {
//!         hosts.iter().map(|s| s.to_string()).collect()
//!     })]
//!     allowed_hosts: Vec<String>,
//! }
//!
//! let config = Config::builder()
//!     .debug_enabled("true")
//!     .env_vars("LOG_LEVEL=debug,PORT=8080")
//!     .allowed_hosts(&["localhost", "127.0.0.1"])
//!     .build();
//!
//! assert_eq!(config.debug_enabled, true);
//! assert_eq!(config.env_vars.get("LOG_LEVEL"), Some(&"debug".to_string()));
//! assert_eq!(config.allowed_hosts, vec!["localhost", "127.0.0.1"]);
//! ```
//!
//! ### Converter with Generics
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct Container<T: Clone + Default> {
//!     #[builder(required, impl_into)]
//!     name: String,
//!
//!     // Converter works with generic fields too
//!     #[builder(converter = |items: &[T]| items.into_iter().map(|item| item.clone()).collect())]
//!     data: Vec<T>,
//! }
//!
//! // Usage with concrete type
//! let container = Container::builder()
//!     .name("numbers")        // Convert &str to String thanks to `impl_into`
//!     .data(&[1, 2, 3, 4, 5]) // Convert &[T] to Vec<T> thanks to `converter`
//!     .build();               // Available only when all required fields are set
//!
//! assert_eq!(container.data, vec![1, 2, 3, 4, 5]);
//! ```
//!
//! ### Converter vs impl_into Comparison
//!
//! | Feature | `impl_into` | `converter` |
//! |---------|-------------|-------------|
//! | **Type conversions** | Only `Into` trait | Any custom logic |
//! | **Parsing strings** | Limited | Full support |
//! | **Data validation** | No | Custom validation |
//! | **Complex transformations** | No | Full support |
//! | **Multiple input formats** | Into only | Any input type |
//! | **Performance** | Zero-cost | Depends on logic |
//! | **Syntax** | Attribute flag | Closure expression |
//!
//! **When to use `converter`:**
//! - Ergonomic setter generation for `Option<String>` fields
//! - Parsing structured data from strings
//! - Normalizing or validating input data
//! - Complex data transformations
//! - Converting between incompatible types
//! - Custom business logic in setters
//!
//! **Note**: `converter` is incompatible with `skip_setter` and `impl_into` since
//! they represent different approaches to setter generation.
//!
//! ## Optional-Only Structs (Regular Builder)
//!
//! ```
//! use type_state_builder::TypeStateBuilder;
//!
//! // No required fields = regular builder pattern
//! #[derive(TypeStateBuilder)]
//! struct Settings {
//!     debug: bool,
//!     max_connections: Option<u32>,
//!
//!     #[builder(default = "default.log".to_string())]
//!     log_file: String,
//!
//!     #[builder(skip_setter, default = 42)]
//!     magic_number: i32,
//! }
//!
//! // Can call build() immediately since no required fields
//! let settings = Settings::builder()
//!     .debug(true)
//!     .max_connections(Some(100))
//!     .build();
//! ```
//!
//! # Error Prevention
//!
//! The macro prevents common mistakes at compile time:
//!
//! ```compile_fail
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct User {
//!     #[builder(required)]
//!     name: String,
//! }
//!
//! let user = User::builder().build(); // ERROR: required field not set
//! ```
//!
//! ```compile_fail
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct BadConfig {
//!     #[builder(required, default = test)]  // ERROR: Invalid combination
//!     name: String,
//! }
//! ```
//!
//! # Invalid Attribute Combinations
//!
//! Several attribute combinations are invalid and will produce compile-time errors
//! with helpful error messages:
//!
//! ## skip_setter + setter_name
//!
//! You can't name a setter method that doesn't exist:
//!
//! ```compile_fail
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct InvalidSetterName {
//!     #[builder(skip_setter, setter_name = "custom_name")]  // ERROR: Conflicting attributes
//!     field: String,
//! }
//! ```
//!
//! ## skip_setter + setter_prefix
//!
//! You can't apply prefixes to non-existent setter methods:
//!
//! ```compile_fail
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct InvalidSetterPrefix {
//!     #[builder(skip_setter, setter_prefix = "with_")]  // ERROR: Conflicting attributes
//!     field: String,
//! }
//! ```
//!
//! ## skip_setter + impl_into
//!
//! Skipped setters can't have parameter conversion rules:
//!
//! ```compile_fail
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct InvalidImplInto {
//!     #[builder(skip_setter, impl_into = true)]  // ERROR: Conflicting attributes
//!     field: String,
//! }
//! ```
//!
//! ## skip_setter + converter
//!
//! Skipped setters can't have custom conversion logic:
//!
//! ```compile_fail
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct InvalidConverter {
//!     #[builder(skip_setter, converter = |x: String| x)]  // ERROR: Conflicting attributes
//!     field: String,
//! }
//! ```
//!
//! ## converter + impl_into
//!
//! Custom converters and `impl_into` are different approaches to parameter handling:
//!
//! ```compile_fail
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct InvalidConverterImplInto {
//!     #[builder(converter = |x: String| x, impl_into = true)]  // ERROR: Conflicting attributes
//!     field: String,
//! }
//! ```
//!
//! ## skip_setter + required
//!
//! Required fields must have setter methods:
//!
//! ```compile_fail
//! use type_state_builder::TypeStateBuilder;
//!
//! #[derive(TypeStateBuilder)]
//! struct InvalidRequired {
//!     #[builder(skip_setter, required)]  // ERROR: Conflicting attributes
//!     field: String,
//! }
//! ```
//!
//! # Module Organization
//!
//! The crate is organized into several modules that handle different aspects
//! of the builder generation process. Most users will only interact with the
//! [`TypeStateBuilder`] derive macro.
//!
//! For complete documentation, examples, and guides, see the
//! [README](https://github.com/welf/type-state-builder#readme).

use proc_macro::TokenStream;
use syn::{parse_macro_input, DeriveInput};

// Internal modules - not exported due to proc-macro restrictions
mod analysis;
mod attributes;
mod generation;
mod utils;
mod validation;

/// Derives a type-safe builder for a struct with compile-time validation of required fields.
///
/// This macro automatically generates an appropriate builder pattern based on the struct
/// configuration:
/// - **Type-State Builder** for structs with required fields
/// - **Regular Builder** for structs with only optional fields
///
/// # Basic Usage
///
/// ```
/// use type_state_builder::TypeStateBuilder;
///
/// #[derive(TypeStateBuilder)]
/// struct Person {
///     #[builder(required)]
///     name: String,
///     age: Option<u32>,
/// }
///
/// let person = Person::builder()
///     .name("Alice".to_string())
///     .build();
/// ```
///
/// # Attribute Reference
///
/// ## Struct Attributes
///
/// - `#[builder(build_method = "name")]` - Custom build method name (default: "build")
/// - `#[builder(setter_prefix = "prefix_")]` - Prefix for all setter method names
///
/// ## Field Attributes
///
/// - `#[builder(required)]` - Field must be set before build() (creates type-state builder)
/// - `#[builder(setter_name = "name")]` - Custom setter method name
/// - `#[builder(setter_prefix = "prefix_")]` - Custom prefix for this field's setter (overrides struct-level)
/// - `#[builder(default = expr)]` - Custom default value (must be valid Rust expression)
/// - `#[builder(skip_setter)]` - Don't generate setter method (requires default value)
///
/// # Generated Methods
///
/// The macro generates:
/// - `YourStruct::builder()` - Creates a new builder instance
/// - `.field_name(value)` - Setter methods for each field (unless skipped)
/// - `.build()` - Constructs the final instance (or custom name from `build_method`)
///
/// # Compile-Time Safety
///
/// The type-state builder (used when there are required fields) prevents:
/// - Calling `build()` before setting required fields
/// - Setting the same required field multiple times
/// - Invalid attribute combinations
///
/// # Error Messages
///
/// The macro provides clear error messages for common mistakes:
/// - Missing required fields at build time
/// - Invalid attribute combinations
/// - Generic parameter mismatches
/// - Syntax errors in default values
///
/// # Examples
///
/// ```
/// use type_state_builder::TypeStateBuilder;
///
/// #[derive(TypeStateBuilder)]
/// struct User {
///     #[builder(required)]
///     name: String,
///     age: Option<u32>,
/// }
///
/// let user = User::builder()
///     .name("Alice".to_string())
///     .build();
/// ```
#[proc_macro_derive(TypeStateBuilder, attributes(builder))]
pub fn derive_type_state_builder(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);

    match generate_builder_implementation(&input) {
        Ok(tokens) => tokens.into(),
        Err(error) => error.to_compile_error().into(),
    }
}

/// Generates the complete builder implementation for a struct.
///
/// This is the main implementation function that coordinates the analysis,
/// validation, and generation process.
///
/// # Arguments
///
/// * `input` - The parsed derive input from the struct definition
///
/// # Returns
///
/// A `syn::Result<proc_macro2::TokenStream>` containing the complete builder
/// implementation or detailed error information.
///
/// # Process
///
/// 1. **Analysis** - Parse and analyze the struct definition
/// 2. **Validation** - Ensure the configuration is valid
/// 3. **Generation** - Create the appropriate builder code
/// 4. **Assembly** - Combine all components into the final output
///
/// # Error Handling
///
/// Errors are returned with detailed information about:
/// - What went wrong during analysis or generation
/// - How to fix configuration issues
/// - Examples of correct usage
fn generate_builder_implementation(input: &DeriveInput) -> syn::Result<proc_macro2::TokenStream> {
    // Step 1: Analyze the struct definition
    let analysis = analysis::analyze_struct(input)?;

    // Step 2: Validate the analysis for builder generation
    analysis.validate_for_generation()?;

    // Step 3: Generate the appropriate builder implementation
    generation::generate_builder(&analysis)
}

// Internal types for testing - not exported due to proc-macro restrictions

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

    #[test]
    fn test_derive_macro_with_required_fields() {
        let input: DeriveInput = parse_quote! {
            struct Example {
                #[builder(required)]
                name: String,
                age: Option<u32>,
            }
        };

        let result = generate_builder_implementation(&input);
        assert!(result.is_ok());

        let tokens = result.unwrap();
        let code = tokens.to_string();

        // Should contain builder method
        assert!(code.contains("builder"));
        // Should contain setter for required field
        assert!(code.contains("name"));
        // Should contain build method
        assert!(code.contains("build"));
    }

    #[test]
    fn test_derive_macro_with_optional_only() {
        let input: DeriveInput = parse_quote! {
            struct Example {
                name: Option<String>,
                age: u32,
            }
        };

        let result = generate_builder_implementation(&input);
        assert!(result.is_ok());

        let tokens = result.unwrap();
        let code = tokens.to_string();

        // Should generate regular builder for optional-only struct
        assert!(code.contains("builder"));
        assert!(code.contains("build"));
    }

    #[test]
    fn test_derive_macro_with_custom_attributes() {
        let input: DeriveInput = parse_quote! {
            #[builder(build_method = "create")]
            struct Example {
                #[builder(required, setter_name = "set_name")]
                name: String,

                #[builder(default = 42)]
                count: i32,
            }
        };

        let result = generate_builder_implementation(&input);
        assert!(result.is_ok());

        let tokens = result.unwrap();
        let code = tokens.to_string();

        // Should respect custom build method name
        assert!(code.contains("create"));
        // Should respect custom setter name
        assert!(code.contains("set_name"));
    }

    #[test]
    fn test_derive_macro_with_generics() {
        let input: DeriveInput = parse_quote! {
            struct Example<T: Clone> {
                #[builder(required)]
                value: T,
                name: String,
            }
        };

        let result = generate_builder_implementation(&input);
        assert!(result.is_ok());

        let tokens = result.unwrap();
        let code = tokens.to_string();

        // Should handle generics properly
        assert!(code.contains("<"));
        assert!(code.contains("T"));
    }

    #[test]
    fn test_derive_macro_with_undeclared_generic() {
        // This test verifies that undeclared generics are allowed through our validation
        // and the Rust compiler will catch the error later
        let input: DeriveInput = parse_quote! {
            struct Example {
                value: T,  // T is not declared - Rust compiler will catch this
            }
        };

        let result = generate_builder_implementation(&input);
        // Should succeed in our validation (Rust compiler will catch undeclared generic later)
        assert!(result.is_ok());

        let code = result.unwrap().to_string();
        // Should generate code that includes the undeclared generic
        assert!(code.contains("value"));
    }

    #[test]
    fn test_unsupported_input_types() {
        // Test enum - should fail
        let enum_input: DeriveInput = parse_quote! {
            enum Example {
                A, B
            }
        };
        let result = generate_builder_implementation(&enum_input);
        assert!(result.is_err());

        // Test tuple struct - should fail
        let tuple_input: DeriveInput = parse_quote! {
            struct Example(String, i32);
        };
        let result = generate_builder_implementation(&tuple_input);
        assert!(result.is_err());

        // Test unit struct - should fail
        let unit_input: DeriveInput = parse_quote! {
            struct Example;
        };
        let result = generate_builder_implementation(&unit_input);
        assert!(result.is_err());
    }
}