sql_forge_macro/

lib.rs

// =============================================================================
// Imports
// =============================================================================

use proc_macro::TokenStream;
use proc_macro2::{Delimiter, Group, Span, TokenStream as TokenStream2, TokenTree};
use quote::{format_ident, quote};
use std::collections::{HashMap, HashSet};
use std::fmt::Write;
use std::fs;
use std::path::Path;
use syn::parse::{Parse, ParseStream};
use syn::parse_quote;
use syn::spanned::Spanned;
use syn::{
    Expr, ExprBlock, ExprGroup, ExprLit, ExprParen, Fields, Ident, ItemStruct, Lit, LitStr, Pat,
    Stmt, Token, Type,
};

// =============================================================================
// Input types
// =============================================================================

mod kw {
    syn::custom_keyword!(scalar);
}

/// A `:name = expr` parameter binding.
#[derive(Clone)]
struct ParamAssign {
    name: Ident,
    expr: Expr,
}

impl Parse for ParamAssign {
    fn parse(input: ParseStream<'_>) -> syn::Result<Self> {
        input.parse::<Token![:]>()?;
        let name: Ident = input.parse()?;
        input.parse::<Token![=]>()?;
        let expr: Expr = input.parse()?;
        Ok(Self { name, expr })
    }
}

/// A single section value: SQL string + optional local parameter bindings.
#[derive(Clone)]
struct SectionFragment {
    sql: String,
    span: Span,
    params: ParamsSource,
}

/// One arm in a `match { ... }` inside a section assignment.
#[derive(Clone)]
struct SectionMatchArm {
    pat: Pat,
    guard: Option<Expr>,
    value: SectionValue,
}

/// The right-hand side of a section `#name = ...` assignment.
#[derive(Clone)]
enum SectionValue {
    /// A plain string (or `(string, params)` tuple).
    Single(SectionFragment),
    /// A tuple of values, one per section when using `#(a, b) = ...`.
    Grouped(Vec<SectionValue>),
    /// A `match expr { arm => ..., arm => ... }` expression.
    Match {
        expr: Expr,
        arms: Vec<SectionMatchArm>,
    },
}

/// A full `#name = value` or `#(a, b) = value` section assignment.
#[derive(Clone)]
struct SectionAssign {
    names: Vec<Ident>,
    value: SectionValue,
}

impl Parse for SectionAssign {
    fn parse(input: ParseStream<'_>) -> syn::Result<Self> {
        input.parse::<Token![#]>()?;

        // Parse `#(a, b)` for grouped sections, or `#name` for single.
        let names = if input.peek(syn::token::Paren) {
            let content;
            syn::parenthesized!(content in input);
            let mut out = Vec::new();
            while !content.is_empty() {
                out.push(content.parse::<Ident>()?);
                if content.is_empty() {
                    break;
                }
                content.parse::<Token![,]>()?;
            }
            if out.is_empty() {
                return Err(input.error("sql_forge!: grouped section key list cannot be empty"));
            }
            out
        } else {
            vec![input.parse::<Ident>()?]
        };

        input.parse::<Token![=]>()?;
        let value = parse_section_value(input, names.len())?;
        Ok(Self { names, value })
    }
}

/// The fully-parsed macro invocation.
struct SqlForgeInput {
    db: Option<Type>,
    result: ResultSpec,
    force_scalar: bool,
    sql: SqlTemplate,
    params: ParamsSource,
    sections: Vec<SectionAssign>,
    batch: Option<Expr>,
}

/// One entry in a result map `(>key = Model, ...)`.
#[derive(Clone)]
struct ResultAssign {
    name: Ident,
    model: Type,
    force_scalar: bool,
}

#[derive(Clone)]
enum ResultSpec {
    /// Execute-only (no model), e.g. `sql_forge!("SQL", ...)`
    None,
    /// e.g. `sql_forge!(User, ...)`
    Single(Box<Type>),
    /// e.g. `sql_forge!((>a = X, >b = Y), ...)`
    Group(Vec<ResultAssign>),
}

#[derive(Clone)]
enum ParamsSource {
    None,
    /// `( :name = expr, ... )`
    Map(Vec<ParamAssign>),
    /// A struct expression whose fields are matched to `:name` placeholders.
    Struct(Box<Expr>),
}

/// The SQL template. Only string literals are supported.
enum SqlTemplate {
    Literal(LitStr),
}

impl SqlTemplate {
    fn span(&self) -> Span {
        match self {
            Self::Literal(lit) => lit.span(),
        }
    }

    /// Parse the SQL into a sequence of textual segments and `{#section}` slots.
    fn into_segments(self) -> Result<Vec<Segment>, String> {
        match self {
            Self::Literal(lit) => parse_literal_segments(&lit.value()),
        }
    }
}

fn parse_sql_template(input: ParseStream<'_>) -> syn::Result<SqlTemplate> {
    if input.peek(LitStr) {
        Ok(SqlTemplate::Literal(input.parse::<LitStr>()?))
    } else {
        Err(input.error("sql_forge!: SQL template must be a string literal"))
    }
}

/// One piece of the parsed SQL template.
#[derive(Clone)]
enum Segment {
    /// Plain SQL text (may contain `:param` placeholders).
    Text(String),
    /// A `{#section_name}` placeholder.
    Section { name: String },
    /// A `{( ... )}` batch value template (repeated per batch item).
    Batch { parts: Vec<TextPart> },
}

/// A fragment of SQL text after splitting on `:param`.
#[derive(Clone)]
enum TextPart {
    /// Literal SQL text.
    Lit(String),
    /// A `:param` placeholder.
    Param { name: String, is_list: bool },
}

// =============================================================================
// Parsing helpers
// =============================================================================

/// Used by `detect_parenthesized_map_kind` to identify what a `(...)` argument is.
enum MapKind {
    Results,
    Params,
    Sections,
}

// =============================================================================
// Section value parsing (string fragments, match, grouped tuples)
// =============================================================================

fn detect_parenthesized_map_kind(input: ParseStream<'_>) -> syn::Result<Option<MapKind>> {
    let fork = input.fork();
    let content;
    syn::parenthesized!(content in fork);

    if content.is_empty() {
        return Err(input.error("sql_forge!: map argument cannot be empty"));
    }

    if content.peek(Token![>]) {
        Ok(Some(MapKind::Results))
    } else if content.peek(Token![:]) {
        Ok(Some(MapKind::Params))
    } else if content.peek(Token![#]) {
        Ok(Some(MapKind::Sections))
    } else {
        Ok(None)
    }
}

impl Parse for ResultAssign {
    fn parse(input: ParseStream<'_>) -> syn::Result<Self> {
        input.parse::<Token![>]>()?;
        let name: Ident = input.parse()?;
        input.parse::<Token![=]>()?;
        let (force_scalar, model) = if input.peek(kw::scalar) {
            input.parse::<kw::scalar>()?;
            (true, input.parse::<Type>()?)
        } else {
            (false, input.parse::<Type>()?)
        };
        Ok(Self {
            name,
            model,
            force_scalar,
        })
    }
}

fn parse_result_map(input: ParseStream<'_>) -> syn::Result<Vec<ResultAssign>> {
    let content;
    syn::parenthesized!(content in input);

    let mut results = Vec::new();
    while !content.is_empty() {
        results.push(content.parse::<ResultAssign>()?);
        if content.is_empty() {
            break;
        }
        content.parse::<Token![,]>()?;
    }

    if results.is_empty() {
        return Err(input.error("sql_forge!: result map cannot be empty"));
    }

    Ok(results)
}

fn parse_param_map(input: ParseStream<'_>) -> syn::Result<Vec<ParamAssign>> {
    let content;
    syn::parenthesized!(content in input);

    let mut params = Vec::new();
    while !content.is_empty() {
        params.push(content.parse::<ParamAssign>()?);
        if content.is_empty() {
            break;
        }
        content.parse::<Token![,]>()?;
    }

    Ok(params)
}

fn parse_section_map(input: ParseStream<'_>) -> syn::Result<Vec<SectionAssign>> {
    let content;
    syn::parenthesized!(content in input);

    let mut sections = Vec::new();
    while !content.is_empty() {
        sections.push(content.parse::<SectionAssign>()?);
        if content.is_empty() {
            break;
        }
        content.parse::<Token![,]>()?;
    }

    Ok(sections)
}

fn parse_params_source_expr(
    input: ParseStream<'_>,
    allow_sections: bool,
) -> syn::Result<ParamsSource> {
    if input.peek(syn::token::Paren) {
        match detect_parenthesized_map_kind(input)? {
            Some(MapKind::Results) => Err(input
                .error("sql_forge!: result maps are only allowed as the macro result argument")),
            Some(MapKind::Params) => Ok(ParamsSource::Map(parse_param_map(input)?)),
            Some(MapKind::Sections) if allow_sections => {
                Err(input.error("sql_forge!: section maps are not allowed here"))
            }
            Some(MapKind::Sections) => Err(input.error(
                "sql_forge!: use :name = expr for section-local parameters, not #name = expr",
            )),
            None => Ok(ParamsSource::Struct(Box::new(input.parse::<Expr>()?))),
        }
    } else {
        Ok(ParamsSource::Struct(Box::new(input.parse::<Expr>()?)))
    }
}

fn parse_section_fragment(input: ParseStream<'_>) -> syn::Result<SectionFragment> {
    if input.peek(syn::token::Paren) {
        let fork = input.fork();
        let content;
        syn::parenthesized!(content in fork);

        if let Ok(first_expr) = content.parse::<Expr>() {
            if extract_lit_str(&first_expr).is_some() && content.parse::<Token![,]>().is_ok() {
                let _ = parse_params_source_expr(&content, false)?;
                if content.peek(Token![,]) {
                    content.parse::<Token![,]>()?;
                }
                if content.is_empty() {
                    let content;
                    syn::parenthesized!(content in input);
                    let first_expr: Expr = content.parse()?;
                    let sql = extract_lit_str(&first_expr).ok_or_else(|| {
                        input.error("sql_forge!: section tuple must start with a string literal")
                    })?;
                    let span = first_expr.span();
                    content.parse::<Token![,]>()?;
                    let params = parse_params_source_expr(&content, false)?;
                    if content.peek(Token![,]) {
                        content.parse::<Token![,]>()?;
                    }
                    if !content.is_empty() {
                        return Err(content.error(
                            "sql_forge!: unexpected tokens after section-local parameter source",
                        ));
                    }
                    return Ok(SectionFragment { sql, span, params });
                }
            }
        }
    }

    let expr: Expr = input.parse()?;
    let sql = extract_lit_str(&expr).ok_or_else(|| {
        input
            .error("sql_forge!: section values must be string literals or (string literal, params)")
    })?;
    Ok(SectionFragment {
        sql,
        span: expr.span(),
        params: ParamsSource::None,
    })
}

fn parse_section_value(input: ParseStream<'_>, width: usize) -> syn::Result<SectionValue> {
    if input.peek(Token![match]) {
        input.parse::<Token![match]>()?;
        let expr: Expr = input.call(Expr::parse_without_eager_brace)?;
        let content;
        syn::braced!(content in input);
        let mut arms = Vec::new();
        while !content.is_empty() {
            let pat = content.call(Pat::parse_multi_with_leading_vert)?;
            let guard = if content.peek(Token![if]) {
                content.parse::<Token![if]>()?;
                Some(content.parse::<Expr>()?)
            } else {
                None
            };
            content.parse::<Token![=>]>()?;
            let value = parse_section_value(&content, width)?;
            if content.peek(Token![,]) {
                content.parse::<Token![,]>()?;
            }
            arms.push(SectionMatchArm { pat, guard, value });
        }
        return Ok(SectionValue::Match { expr, arms });
    }

    if width == 1 {
        return Ok(SectionValue::Single(parse_section_fragment(input)?));
    }

    let content;
    syn::parenthesized!(content in input);
    let mut items = Vec::new();
    while !content.is_empty() {
        items.push(parse_section_value(&content, 1)?);
        if content.is_empty() {
            break;
        }
        content.parse::<Token![,]>()?;
    }

    if items.len() != width {
        return Err(input.error(format!(
            "sql_forge!: grouped section value must provide exactly {} items",
            width,
        )));
    }

    Ok(SectionValue::Grouped(items))
}

// =============================================================================
// Top-level macro input parsing (SqlForgeInput::parse)
// =============================================================================

impl Parse for SqlForgeInput {
    fn parse(input: ParseStream<'_>) -> syn::Result<Self> {
        let (db, result, force_scalar, sql) = if input.peek(LitStr) {
            let sql = parse_sql_template(input)?;
            (None, ResultSpec::None, false, sql)
        } else if input.peek(kw::scalar) {
            input.parse::<kw::scalar>()?;
            let model: Type = input.parse()?;
            input.parse::<Token![,]>()?;
            let sql = parse_sql_template(input)?;
            (None, ResultSpec::Single(Box::new(model)), true, sql)
        } else if input.peek(syn::token::Paren) {
            let result_map_kind = detect_parenthesized_map_kind(input)?;
            match result_map_kind {
                Some(MapKind::Results) => {
                    let result = ResultSpec::Group(parse_result_map(input)?);
                    input.parse::<Token![,]>()?;
                    let sql = parse_sql_template(input)?;
                    (None, result, false, sql)
                }
                _ => {
                    return Err(input.error(
                        "sql_forge!: expected a result map like (>name = Model, ...) or a model type",
                    ));
                }
            }
        } else {
            let first_ty: Type = input.parse()?;
            input.parse::<Token![,]>()?;

            if input.peek(LitStr) {
                let model = first_ty;
                let sql = parse_sql_template(input)?;
                (None, ResultSpec::Single(Box::new(model)), false, sql)
            } else if input.peek(kw::scalar) {
                input.parse::<kw::scalar>()?;
                let model: Type = input.parse()?;
                input.parse::<Token![,]>()?;
                let sql = parse_sql_template(input)?;
                (
                    Some(first_ty),
                    ResultSpec::Single(Box::new(model)),
                    true,
                    sql,
                )
            } else if input.peek(syn::token::Paren)
                && matches!(
                    detect_parenthesized_map_kind(input)?,
                    Some(MapKind::Results)
                )
            {
                let result = ResultSpec::Group(parse_result_map(input)?);
                input.parse::<Token![,]>()?;
                let sql = parse_sql_template(input)?;
                (Some(first_ty), result, false, sql)
            } else {
                let db = Some(first_ty);
                let model: Type = input.parse()?;
                input.parse::<Token![,]>()?;
                let sql = parse_sql_template(input)?;
                (db, ResultSpec::Single(Box::new(model)), false, sql)
            }
        };

        let mut batch = None;
        let mut params = ParamsSource::None;
        let mut sections = Vec::new();
        let mut seen_params = false;
        let mut seen_sections = false;

        if input.parse::<Token![,]>().is_ok() {
            while !input.is_empty() {
                if input.peek(Token![..]) {
                    if batch.is_some() {
                        return Err(
                            input.error("sql_forge!: only one batch source argument is allowed")
                        );
                    }
                    input.parse::<Token![..]>()?;
                    batch = Some(input.parse::<Expr>()?);
                } else if input.peek(syn::token::Paren) {
                    match detect_parenthesized_map_kind(input)? {
                        Some(MapKind::Results) => {
                            return Err(input.error(
                                "sql_forge!: result maps are only allowed as the macro result argument",
                            ));
                        }
                        Some(MapKind::Params) => {
                            if seen_params {
                                return Err(
                                    input.error("sql_forge!: only one parameter source is allowed")
                                );
                            }
                            params = ParamsSource::Map(parse_param_map(input)?);
                            seen_params = true;
                        }
                        Some(MapKind::Sections) => {
                            if seen_sections {
                                return Err(
                                    input.error("sql_forge!: duplicate section map argument")
                                );
                            }
                            sections = parse_section_map(input)?;
                            seen_sections = true;
                        }
                        None => {
                            if seen_params {
                                return Err(
                                    input.error("sql_forge!: only one parameter source is allowed")
                                );
                            }
                            params = ParamsSource::Struct(Box::new(input.parse::<Expr>()?));
                            seen_params = true;
                        }
                    }
                } else {
                    if seen_params {
                        return Err(input.error("sql_forge!: only one parameter source is allowed"));
                    }
                    params = ParamsSource::Struct(Box::new(input.parse::<Expr>()?));
                    seen_params = true;
                }

                if input.parse::<Token![,]>().is_ok() {
                    continue;
                }
                break;
            }
        }

        if !input.is_empty() {
            return Err(input.error("sql_forge!: unexpected tokens in macro invocation"));
        }

        Ok(Self {
            db,
            result,
            force_scalar,
            sql,
            params,
            sections,
            batch,
        })
    }
}

// =============================================================================
// Database type resolution
// =============================================================================

fn resolve_db_from_env() -> Result<Type, String> {
    if let Ok(val) = std::env::var("SQL_FORGE_DB_TYPE") {
        return syn::parse_str::<Type>(&val).map_err(|err| {
            format!(
                "sql_forge!: invalid DB type `{}` in SQL_FORGE_DB_TYPE env var: {}",
                val, err
            )
        });
    }

    let manifest_dir = match std::env::var("CARGO_MANIFEST_DIR") {
        Ok(d) => d,
        Err(_) => {
            return Err(
                "sql_forge!: pass DB as first macro argument, set SQL_FORGE_DB_TYPE, \
                 or configure [package.metadata.sql_forge] in Cargo.toml"
                    .to_string(),
            );
        }
    };
    let manifest_path = Path::new(&manifest_dir).join("Cargo.toml");

    let cargo_toml = fs::read_to_string(&manifest_path).map_err(|err| {
        format!(
            "sql_forge!: failed to read {}: {}",
            manifest_path.display(),
            err
        )
    })?;

    let value: toml::Value = toml::from_str(&cargo_toml)
        .map_err(|err| format!("sql_forge!: failed to parse Cargo.toml: {}", err))?;

    let db_str = value
        .get("package")
        .and_then(|v| v.get("metadata"))
        .and_then(|v| v.get("sql_forge"))
        .and_then(|v| v.get("db"))
        .and_then(|v| v.as_str())
        .ok_or({
            "sql_forge!: missing [package.metadata.sql_forge] db = \"...\" in Cargo.toml, \
             SQL_FORGE_DB_TYPE env var, or DB as first macro argument"
        })?;

    syn::parse_str::<Type>(db_str).map_err(|err| {
        format!(
            "sql_forge!: invalid DB type `{}` in Cargo.toml metadata: {}",
            db_str, err
        )
    })
}

fn uses_dollar_params(db: &Type) -> bool {
    let Type::Path(type_path) = db else {
        return false;
    };
    type_path
        .path
        .segments
        .last()
        .is_some_and(|s| s.ident == "Postgres")
}

fn is_builtin_scalar_type(ty: &Type) -> bool {
    let Type::Path(type_path) = ty else {
        return false;
    };

    if type_path.qself.is_some()
        || type_path.path.leading_colon.is_some()
        || type_path.path.segments.len() != 1
    {
        return false;
    }

    let ident = &type_path.path.segments[0].ident;
    ident == "i8"
        || ident == "i16"
        || ident == "i32"
        || ident == "i64"
        || ident == "isize"
        || ident == "u8"
        || ident == "u16"
        || ident == "u32"
        || ident == "u64"
        || ident == "usize"
        || ident == "f32"
        || ident == "f64"
        || ident == "bool"
        || ident == "String"
}

fn scalar_output_type(model: &Type) -> Option<&Type> {
    if is_builtin_scalar_type(model) {
        return Some(model);
    }
    None
}

fn push_text_segment(out: &mut Vec<Segment>, text: String) {
    if text.is_empty() {
        return;
    }
    match out.last_mut() {
        Some(Segment::Text(existing)) => existing.push_str(&text),
        _ => out.push(Segment::Text(text)),
    }
}

fn parse_literal_segments(sql: &str) -> Result<Vec<Segment>, String> {
    let mut out = Vec::new();
    let mut text = String::new();
    let mut chars = sql.chars().peekable();

    while let Some(ch) = chars.next() {
        if ch != '{' {
            text.push(ch);
            continue;
        }

        if chars.peek() == Some(&'(') {
            push_text_segment(&mut out, std::mem::take(&mut text));

            let mut paren_depth = 0u32;
            let mut content = String::new();
            let mut found_close = false;
            for ch in chars.by_ref() {
                if ch == '{' {
                    return Err(
                        "sql_forge!: nested braces not allowed inside batch section".to_string()
                    );
                }
                if ch == '}' {
                    if paren_depth != 0 {
                        return Err(
                            "sql_forge!: batch section {( ... )} has unbalanced parentheses"
                                .to_string(),
                        );
                    }
                    found_close = true;
                    break;
                }
                if ch == '(' {
                    paren_depth += 1;
                } else if ch == ')' {
                    if paren_depth == 0 {
                        return Err(
                            "sql_forge!: batch section {( ... )} has unbalanced parentheses"
                                .to_string(),
                        );
                    }
                    paren_depth -= 1;
                }
                content.push(ch);
            }
            if !found_close {
                return Err("sql_forge!: batch section {( ... )} without closing }".to_string());
            }
            let parts = parse_text_parts(&content);
            for part in &parts {
                if let TextPart::Param { is_list: true, .. } = part {
                    return Err(
                        "sql_forge!: list parameters (:name[]) are not allowed inside {( ... )} \
                         batch sections; use plain parameters (:name) instead"
                            .to_string(),
                    );
                }
            }
            out.push(Segment::Batch { parts });
            continue;
        }

        if chars.peek() != Some(&'#') {
            text.push(ch);
            continue;
        }

        chars.next();
        push_text_segment(&mut out, std::mem::take(&mut text));

        let mut name = String::new();
        loop {
            let Some(next) = chars.next() else {
                return Err("sql_forge!: section placeholder without closing }".to_string());
            };
            if next == '}' {
                break;
            }
            name.push(next);
        }

        if name.is_empty() {
            return Err("sql_forge!: empty section placeholder name".to_string());
        }

        out.push(Segment::Section { name });
    }

    push_text_segment(&mut out, text);
    Ok(out)
}

// =============================================================================
// SQL template parsing: {#sections} and :param placeholders
// =============================================================================

fn is_ident_start(ch: char) -> bool {
    ch == '_' || ch.is_ascii_alphabetic()
}

fn is_ident_continue(ch: char) -> bool {
    is_ident_start(ch) || ch.is_ascii_digit()
}

fn sanitize_backticked_alias_ident(content: &str) -> String {
    let mut split_at = content.len();
    for (idx, ch) in content.char_indices() {
        if ch == '!' || ch == '?' || ch == ':' {
            split_at = idx;
            break;
        }
    }

    if split_at == content.len() {
        return content.to_string();
    }

    let base = content[..split_at].trim_end();
    if base.is_empty() {
        content.to_string()
    } else {
        base.to_string()
    }
}

fn sanitize_runtime_sql_text(text: &str) -> String {
    let mut out = String::with_capacity(text.len());
    let mut chars = text.chars().peekable();

    while let Some(ch) = chars.next() {
        if ch != '`' {
            out.push(ch);
            continue;
        }

        let mut content = String::new();
        let mut closed = false;

        for next in chars.by_ref() {
            if next == '`' {
                closed = true;
                break;
            }
            content.push(next);
        }

        if closed {
            out.push('`');
            out.push_str(&sanitize_backticked_alias_ident(&content));
            out.push('`');
        } else {
            out.push('`');
            out.push_str(&content);
            break;
        }
    }

    out
}

fn parse_text_parts(text: &str) -> Vec<TextPart> {
    let mut parts = Vec::new();
    let mut last = 0usize;
    let mut iter = text.char_indices().peekable();

    while let Some((idx, ch)) = iter.next() {
        if ch != ':' {
            continue;
        }

        let Some(&(next_idx, next_ch)) = iter.peek() else {
            continue;
        };

        if !is_ident_start(next_ch) {
            continue;
        }

        if text[..idx].ends_with(':') {
            continue;
        }

        if last < idx {
            parts.push(TextPart::Lit(text[last..idx].to_string()));
        }

        iter.next();

        let mut name = String::new();
        name.push(next_ch);
        let mut end = next_idx + next_ch.len_utf8();

        while let Some(&(j, c)) = iter.peek() {
            if is_ident_continue(c) {
                name.push(c);
                end = j + c.len_utf8();
                iter.next();
            } else {
                break;
            }
        }

        let mut is_list = false;
        if text[end..].starts_with("[]") {
            is_list = true;
            end += 2;
        }

        parts.push(TextPart::Param { name, is_list });
        last = end;
    }

    if last < text.len() {
        parts.push(TextPart::Lit(text[last..].to_string()));
    }

    parts
}

fn render_validator_text(
    text: &str,
    use_dollar_params: bool,
    param_offset: &mut usize,
    list_count: usize,
) -> (String, Vec<(String, bool)>) {
    let mut out_sql = String::new();
    let mut occurrences = Vec::new();

    for part in parse_text_parts(text) {
        match part {
            TextPart::Lit(lit) => out_sql.push_str(&lit),
            TextPart::Param { name, is_list } => {
                if is_list && list_count > 1 {
                    let slots: Vec<String> = if use_dollar_params {
                        (0..list_count)
                            .map(|i| format!("${}", *param_offset + i + 1))
                            .collect()
                    } else {
                        (0..list_count).map(|_| "?".to_string()).collect()
                    };
                    if use_dollar_params {
                        *param_offset += list_count;
                    }
                    out_sql.push_str(&slots.join(", "));
                } else if use_dollar_params {
                    *param_offset += 1;
                    write!(out_sql, "${}", *param_offset).unwrap();
                } else {
                    out_sql.push('?');
                }
                occurrences.push((name, is_list));
            }
        }
    }

    (out_sql, occurrences)
}

fn strip_expr(expr: &Expr) -> &Expr {
    match expr {
        Expr::Paren(ExprParen { expr, .. }) => strip_expr(expr),
        Expr::Group(ExprGroup { expr, .. }) => strip_expr(expr),
        Expr::Block(ExprBlock { block, .. }) => {
            if block.stmts.len() != 1 {
                return expr;
            }
            match &block.stmts[0] {
                Stmt::Expr(inner, None) => strip_expr(inner),
                _ => expr,
            }
        }
        _ => expr,
    }
}

fn extract_lit_str(expr: &Expr) -> Option<String> {
    match strip_expr(expr) {
        Expr::Lit(ExprLit {
            lit: Lit::Str(lit), ..
        }) => Some(lit.value()),
        _ => None,
    }
}

// =============================================================================
// Preprocessing: {>key} compile-time result flags
// =============================================================================

fn result_flag_ident(name: &str) -> syn::Ident {
    format_ident!("__enhanced_result_flag_{}", name)
}

/// Replaces `{>key}` tokens inside braced groups with `__enhanced_result_flag_key`
/// identifiers. This is a preprocessing step so that the rest of the parser sees
/// plain identifiers instead of braced groups it does not understand.
fn preprocess_result_key_placeholders(input: TokenStream2) -> TokenStream2 {
    fn walk(stream: TokenStream2) -> TokenStream2 {
        let mut out = TokenStream2::new();
        let iter = stream.into_iter().peekable();

        for token in iter {
            match token {
                TokenTree::Group(group) => {
                    if group.delimiter() == Delimiter::Brace {
                        let mut inner = group.stream().into_iter();
                        let first = inner.next();
                        let second = inner.next();
                        let third = inner.next();

                        if let (
                            Some(TokenTree::Punct(p)),
                            Some(TokenTree::Ident(name_ident)),
                            None,
                        ) = (first, second, third)
                        {
                            if p.as_char() == '>' {
                                let ident = result_flag_ident(&name_ident.to_string());
                                out.extend(std::iter::once(TokenTree::Ident(ident)));
                                continue;
                            }
                        }
                    }

                    let new_inner = walk(group.stream());
                    let mut new_group = Group::new(group.delimiter(), new_inner);
                    new_group.set_span(group.span());
                    out.extend(std::iter::once(TokenTree::Group(new_group)));
                }
                other => out.extend(std::iter::once(other)),
            }
        }

        out
    }

    walk(input)
}

fn build_result_flag_bindings(keys: &[String], active_key: Option<&str>) -> Vec<TokenStream2> {
    keys.iter()
        .map(|key| {
            let ident = result_flag_ident(key);
            let enabled = Some(key.as_str()) == active_key;
            quote! { let #ident: bool = #enabled; }
        })
        .collect()
}

fn transpose_section_case_matrix(
    case_matrix: Vec<Vec<SectionFragment>>,
    width: usize,
) -> Result<Vec<Vec<SectionFragment>>, String> {
    let mut per_section: Vec<Vec<SectionFragment>> = (0..width).map(|_| Vec::new()).collect();

    for row in case_matrix {
        if row.len() != width {
            return Err(
                "sql_forge!: grouped sections must return one item per section".to_string(),
            );
        }
        for (section_idx, fragment) in row.into_iter().enumerate() {
            per_section[section_idx].push(fragment);
        }
    }

    Ok(per_section)
}

fn collect_section_case_matrix(
    value: SectionValue,
    width: usize,
    active_key: Option<&str>,
) -> Result<Vec<Vec<SectionFragment>>, String> {
    match value {
        SectionValue::Single(fragment) => {
            if width != 1 {
                return Err(
                    "sql_forge!: grouped sections must return one item per section".to_string(),
                );
            }
            Ok(vec![vec![fragment]])
        }
        SectionValue::Grouped(values) => {
            if values.len() != width {
                return Err(
                    "sql_forge!: grouped sections must return one item per section".to_string(),
                );
            }

            let mut variants_by_section = Vec::<Vec<SectionFragment>>::with_capacity(width);
            let mut nmax = 1usize;

            for value in values {
                let item_matrix = collect_section_case_matrix(value, 1, active_key)?;
                let mut item_variants = Vec::<SectionFragment>::with_capacity(item_matrix.len());
                for mut row in item_matrix {
                    let fragment = row.pop().ok_or_else(|| {
                        "sql_forge!: grouped sections must return one item per section".to_string()
                    })?;
                    if !row.is_empty() {
                        return Err(
                            "sql_forge!: grouped sections must return one item per section"
                                .to_string(),
                        );
                    }
                    item_variants.push(fragment);
                }
                if item_variants.is_empty() {
                    return Err("sql_forge!: section match must have at least one arm".to_string());
                }
                nmax = nmax.max(item_variants.len());
                variants_by_section.push(item_variants);
            }

            let mut case_matrix = Vec::<Vec<SectionFragment>>::with_capacity(nmax);
            for case_idx in 0..nmax {
                let mut row = Vec::<SectionFragment>::with_capacity(width);
                for variants in &variants_by_section {
                    row.push(variants[case_idx % variants.len()].clone());
                }
                case_matrix.push(row);
            }

            Ok(case_matrix)
        }
        SectionValue::Match { expr, arms } => {
            let mut case_matrix = Vec::<Vec<SectionFragment>>::new();

            if let Some(key) = expr_result_flag_key(&expr) {
                let target = active_key == Some(key.as_str());
                for arm in arms {
                    if arm.guard.is_none() {
                        if let Some(false) = pattern_matches_bool(&arm.pat, target) {
                            continue;
                        }
                    }
                    let mut arm_cases = collect_section_case_matrix(arm.value, width, active_key)?;
                    wrap_section_case_matrix_for_match_arm(
                        &mut arm_cases,
                        &expr,
                        &arm.pat,
                        arm.guard.as_ref(),
                    );
                    case_matrix.extend(arm_cases);
                }
            } else {
                for arm in arms {
                    let mut arm_cases = collect_section_case_matrix(arm.value, width, active_key)?;
                    wrap_section_case_matrix_for_match_arm(
                        &mut arm_cases,
                        &expr,
                        &arm.pat,
                        arm.guard.as_ref(),
                    );
                    case_matrix.extend(arm_cases);
                }
            }

            if case_matrix.is_empty() {
                return Err("sql_forge!: section match must have at least one arm".to_string());
            }

            Ok(case_matrix)
        }
    }
}

/// This function rewrites a section-local validator expression so it stays inside the same
/// match arm scope that originally introduced it
fn wrap_expr_for_match_arm(expr: Expr, match_expr: &Expr, pat: &Pat, guard: Option<&Expr>) -> Expr {
    let match_expr = match_expr.clone();
    let pat = pat.clone();
    let pattern_binds_values = match &pat {
        Pat::Ident(_) => true,
        Pat::Or(pat_or) => pat_or
            .cases
            .iter()
            .any(|case| matches!(case, Pat::Ident(_))),
        Pat::Paren(pat_paren) => matches!(pat_paren.pat.as_ref(), Pat::Ident(_)),
        Pat::Reference(pat_reference) => matches!(pat_reference.pat.as_ref(), Pat::Ident(_)),
        Pat::Slice(pat_slice) => pat_slice
            .elems
            .iter()
            .any(|elem| matches!(elem, Pat::Ident(_))),
        Pat::Struct(pat_struct) => pat_struct
            .fields
            .iter()
            .any(|field| matches!(*field.pat, Pat::Ident(_))),
        Pat::Tuple(pat_tuple) => pat_tuple
            .elems
            .iter()
            .any(|elem| matches!(elem, Pat::Ident(_))),
        Pat::TupleStruct(pat_tuple_struct) => pat_tuple_struct
            .elems
            .iter()
            .any(|elem| matches!(elem, Pat::Ident(_))),
        Pat::Type(pat_type) => matches!(pat_type.pat.as_ref(), Pat::Ident(_)),
        _ => false,
    };

    if pattern_binds_values {
        let pat_refs: Vec<TokenStream2> = pat_var_idents(&pat)
            .into_iter()
            .map(|ident| quote! { let _ = &#ident; })
            .collect();
        if let Some(guard) = guard.cloned() {
            parse_quote! {
                match &(#match_expr) {
                    #pat if #guard => { #( #pat_refs )* #expr },
                    _ => unreachable!("sql_forge!: validator arm mismatch"),
                }
            }
        } else {
            parse_quote! {
                match &(#match_expr) {
                    #pat => { #( #pat_refs )* #expr },
                    _ => unreachable!("sql_forge!: validator arm mismatch"),
                }
            }
        }
    } else if let Some(guard) = guard.cloned() {
        parse_quote! {
            match &(#match_expr) {
                #pat if #guard => { &(#expr) },
                _ => unreachable!("sql_forge!: validator arm mismatch"),
            }
        }
    } else {
        parse_quote! {
            match &(#match_expr) {
                #pat => { &(#expr) },
                _ => unreachable!("sql_forge!: validator arm mismatch"),
            }
        }
    }
}

fn wrap_params_source_for_match_arm(
    params: &mut ParamsSource,
    match_expr: &Expr,
    pat: &Pat,
    guard: Option<&Expr>,
) {
    match params {
        ParamsSource::None => {}
        ParamsSource::Map(entries) => {
            for entry in entries {
                entry.expr = wrap_expr_for_match_arm(entry.expr.clone(), match_expr, pat, guard);
            }
        }
        ParamsSource::Struct(expr) => {
            **expr = wrap_expr_for_match_arm((**expr).clone(), match_expr, pat, guard);
        }
    }
}

fn wrap_section_case_matrix_for_match_arm(
    case_matrix: &mut [Vec<SectionFragment>],
    match_expr: &Expr,
    pat: &Pat,
    guard: Option<&Expr>,
) {
    for row in case_matrix {
        for fragment in row {
            wrap_params_source_for_match_arm(&mut fragment.params, match_expr, pat, guard);
        }
    }
}

// Returns Vec<Vec<SectionFragment>> indexed [section_idx][case_idx].
// =============================================================================
// Section variant collection
// =============================================================================

/// Collects all possible `SectionFragment` values per section index. Each
/// returned `Vec<Vec<SectionFragment>>` is indexed `[section_idx][case_idx]`,
/// listing every variant that section can take across all match arms.
/// Used for full validation (cycling strategy over all variants).
fn collect_section_variants(
    value: SectionValue,
    width: usize,
) -> Result<Vec<Vec<SectionFragment>>, String> {
    transpose_section_case_matrix(collect_section_case_matrix(value, width, None)?, width)
}

fn expr_result_flag_key(expr: &Expr) -> Option<String> {
    match strip_expr(expr) {
        Expr::Path(path) if path.qself.is_none() && path.path.segments.len() == 1 => {
            let name = path.path.segments[0].ident.to_string();
            name.strip_prefix("__enhanced_result_flag_")
                .map(|v| v.to_string())
        }
        _ => None,
    }
}

fn pattern_matches_bool(pat: &Pat, value: bool) -> Option<bool> {
    match pat {
        Pat::Lit(expr_lit) => match &expr_lit.lit {
            Lit::Bool(lit_bool) => Some(lit_bool.value == value),
            _ => None,
        },
        Pat::Wild(_) => Some(true),
        _ => None,
    }
}

/// Like `collect_section_variants`, but filters `match` arms by the active
/// result key when the match expression is a `{>key}` flag. When building the
/// query for a specific key, only the matching arm (true/false) is included;
/// arms with guards or non-flag expressions include all variants as usual.
fn collect_section_variants_for_result(
    value: SectionValue,
    width: usize,
    active_key: Option<&str>,
) -> Result<Vec<Vec<SectionFragment>>, String> {
    transpose_section_case_matrix(
        collect_section_case_matrix(value, width, active_key)?,
        width,
    )
}

// =============================================================================
// Parameter binding generation
// =============================================================================

/// Generates `let` bindings for all parameters used in the SQL or section
/// fragments. Returns a map of param-name → local ident for later reference,
/// and a list of `let` statements.
fn build_param_bindings(
    params: &ParamsSource,
    used_param_names: &[String],
    prefix: &str,
    for_validator: bool,
    enforce_usage_check: bool,
) -> Result<(HashMap<String, syn::Ident>, Vec<TokenStream2>), TokenStream> {
    let mut declared_params = HashMap::<String, syn::Ident>::new();
    let mut bindings = Vec::<TokenStream2>::new();

    match params {
        ParamsSource::None => {}
        ParamsSource::Map(entries) => {
            for entry in entries {
                let key = entry.name.to_string();
                if declared_params.contains_key(&key) {
                    return Err(syn::Error::new(
                        entry.name.span(),
                        "sql_forge!: duplicated parameter mapping",
                    )
                    .to_compile_error()
                    .into());
                }
                if enforce_usage_check && !used_param_names.iter().any(|n| n == &key) {
                    return Err(syn::Error::new(
                        entry.name.span(),
                        format!(
                            "sql_forge!: parameter :{} is unused in the SQL template",
                            key,
                        ),
                    )
                    .to_compile_error()
                    .into());
                }
                let local_ident = format_ident!("__enhanced_{}_{}", prefix, key);
                let expr = &entry.expr;
                if for_validator {
                    bindings.push(quote! {
                        let #local_ident = &(#expr);
                    });
                } else {
                    bindings.push(quote! {
                        let #local_ident = #expr;
                    });
                }
                declared_params.insert(key, local_ident);
            }
        }
        ParamsSource::Struct(expr) => {
            let source_ident = format_ident!("__enhanced_source_{}", prefix);
            bindings.push(quote! {
                let #source_ident = &(#expr);
            });
            for name in used_param_names {
                let local_ident = format_ident!("__enhanced_{}_{}", prefix, name);
                let field_ident = format_ident!("{}", name);
                if for_validator {
                    bindings.push(quote! {
                        let #local_ident = &#source_ident.#field_ident;
                    });
                } else {
                    bindings.push(quote! {
                        let #local_ident = #source_ident.#field_ident;
                    });
                }
                declared_params.insert(name.to_string(), local_ident);
            }
        }
    }

    Ok((declared_params, bindings))
}

struct ValidatorRenderContext<'a> {
    local_params: &'a HashMap<String, syn::Ident>,
    top_level_params: &'a HashMap<String, syn::Ident>,
    allow_top_level_fallback: bool,
    use_dollar_params: bool,
    sql_span: Span,
    list_count: usize,
}

/// Builds the placeholders SQL string and argument list for the compile-time
/// validator (sqlx::query_as! / query_scalar!). Each `:param` in the SQL is
/// replaced by `?` (MySQL/SQLite) or `$1`/`$2`/... (PostgreSQL), and the
/// corresponding value expression is collected into the args list.
fn render_validator_args(
    sql: &str,
    param_offset: &mut usize,
    arg_index: &mut usize,
    context: &ValidatorRenderContext<'_>,
) -> Result<(String, Vec<TokenStream2>, Vec<TokenStream2>), TokenStream> {
    let (rendered_sql, occurrences) = render_validator_text(
        sql,
        context.use_dollar_params,
        param_offset,
        context.list_count,
    );

    let mut setup = Vec::<TokenStream2>::new();
    let mut args = Vec::<TokenStream2>::new();

    for (name, is_list) in occurrences {
        let local_ident = if context.allow_top_level_fallback {
            context
                .local_params
                .get(&name)
                .or_else(|| context.top_level_params.get(&name))
        } else {
            context.local_params.get(&name)
        };

        let Some(local_ident) = local_ident else {
            return Err(syn::Error::new(
                context.sql_span,
                format!("sql_forge!: parameter :{} has no mapping", name),
            )
            .to_compile_error()
            .into());
        };

        if is_list {
            for _ in 0..context.list_count {
                let value_ident = format_ident!("__enhanced_validator_arg_{}", *arg_index);
                *arg_index += 1;
                if context.use_dollar_params {
                    setup.push(quote! {
                        let #value_ident = sql_forge::sql_forge_validator_value(
                            (#local_ident)
                                .as_slice()
                                .first()
                                .expect("sql_forge!: list parameters used in validation must have at least one representative element")
                        );
                    });
                } else {
                    setup.push(quote! {
                        let #value_ident = (#local_ident)
                            .as_slice()
                            .first()
                            .expect("sql_forge!: list parameters used in validation must have at least one representative element");
                    });
                }
                args.push(quote! { #value_ident });
            }
        } else {
            let value_ident = format_ident!("__enhanced_validator_arg_{}", *arg_index);
            *arg_index += 1;
            if context.use_dollar_params {
                setup.push(quote! {
                    let #value_ident = sql_forge::sql_forge_validator_value(#local_ident);
                });
            } else {
                setup.push(quote! {
                    let #value_ident = #local_ident;
                });
            }
            args.push(quote! { #value_ident });
        }
    }

    Ok((rendered_sql, setup, args))
}

// =============================================================================
// Runtime code generation (QueryBuilder-based)
// =============================================================================

/// Generates the `push()` / `push_bind()` calls for a single section fragment
/// at runtime using `sqlx::QueryBuilder`.
fn render_runtime_fragment(
    fragment: &SectionFragment,
    local_params: &HashMap<String, syn::Ident>,
) -> Result<TokenStream2, TokenStream> {
    let mut steps = Vec::<TokenStream2>::new();

    for part in parse_text_parts(&fragment.sql) {
        match part {
            TextPart::Lit(lit) => {
                let lit_str = LitStr::new(&lit, fragment.span);
                steps.push(quote! { __builder.push(#lit_str); });
            }
            TextPart::Param { name, is_list } => {
                let Some(local_ident) = local_params.get(&name) else {
                    return Err(syn::Error::new(
                        fragment.span,
                        format!("sql_forge!: parameter :{} has no mapping", name),
                    )
                    .to_compile_error()
                    .into());
                };

                if is_list {
                    steps.push(quote! {
                        let __enhanced_values = #local_ident;
                        let mut __separated = __builder.separated(", ");
                        for __value in __enhanced_values {
                            __separated.push_bind(__value);
                        }
                    });
                } else {
                    steps.push(quote! {
                        __builder.push_bind(#local_ident);
                    });
                }
            }
        }
    }

    Ok(quote! { #( #steps )* })
}

fn is_pat_binding(ident: &Ident) -> bool {
    let name = ident.to_string();
    !name.is_empty()
        && name
            .chars()
            .next()
            .is_some_and(|c| c.is_ascii_lowercase() || c == '_')
}

fn pat_var_idents(pat: &Pat) -> Vec<Ident> {
    let mut names = Vec::new();
    fn walk(p: &Pat, names: &mut Vec<Ident>) {
        match p {
            Pat::Ident(pi) if is_pat_binding(&pi.ident) => names.push(pi.ident.clone()),
            Pat::Tuple(pt) => pt.elems.iter().for_each(|e| walk(e, names)),
            Pat::Struct(ps) => ps.fields.iter().for_each(|f| walk(&f.pat, names)),
            Pat::TupleStruct(pts) => pts.elems.iter().for_each(|e| walk(e, names)),
            Pat::Or(po) => po.cases.iter().for_each(|c| walk(c, names)),
            Pat::Paren(pp) => walk(&pp.pat, names),
            Pat::Reference(pr) => walk(&pr.pat, names),
            Pat::Slice(psl) => psl.elems.iter().for_each(|e| walk(e, names)),
            Pat::Type(pt) => walk(&pt.pat, names),
            _ => {}
        }
    }
    walk(pat, &mut names);
    names
}

fn section_value_refers_to(value: &SectionValue, name: &str) -> bool {
    match value {
        SectionValue::Single(f) => {
            if collect_used_param_names_in_sql(&f.sql)
                .iter()
                .any(|n| n == name)
            {
                return true;
            }
            if let ParamsSource::Map(entries) = &f.params {
                for e in entries {
                    let expr = &e.expr;
                    let expr_str = quote! { #expr }.to_string();
                    if expr_str.trim() == name {
                        return true;
                    }
                }
            }
            false
        }
        SectionValue::Grouped(vals) => vals.iter().any(|v| section_value_refers_to(v, name)),
        SectionValue::Match { arms, .. } => arms.iter().any(|arm| {
            let pat_vars: HashSet<_> = pat_var_idents(&arm.pat)
                .into_iter()
                .map(|i| i.to_string())
                .collect();
            if pat_vars.contains(name) {
                false
            } else {
                section_value_refers_to(&arm.value, name)
            }
        }),
    }
}

fn build_section_runtime_action(
    value: &SectionValue,
    section_idx: usize,
    prefix: &str,
) -> Result<TokenStream2, TokenStream> {
    match value {
        SectionValue::Single(fragment) => {
            let used_param_names = collect_used_param_names_in_sql(&fragment.sql);
            let (local_params, bindings) =
                build_param_bindings(&fragment.params, &used_param_names, prefix, false, true)?;
            let body = render_runtime_fragment(fragment, &local_params)?;
            Ok(quote! {{ #( #bindings )* #body }})
        }
        SectionValue::Grouped(fragments) => build_section_runtime_action(
            &fragments[section_idx],
            0,
            &format!("{}_grouped_{}", prefix, section_idx),
        ),
        SectionValue::Match { expr, arms } => {
            let arm_tokens: Result<Vec<TokenStream2>, TokenStream> = arms
                .iter()
                .enumerate()
                .map(|(arm_idx, arm)| {
                    let pat = &arm.pat;
                    let guard_tokens = arm.guard.as_ref().map(|guard| quote! { if #guard });
                    let body = build_section_runtime_action(
                        &arm.value,
                        section_idx,
                        &format!("{}_{}", prefix, arm_idx),
                    )?;
                    let noop_refs: Vec<TokenStream2> = pat_var_idents(pat)
                        .into_iter()
                        .filter(|ident| section_value_refers_to(&arm.value, &ident.to_string()))
                        .map(|ident| quote! { ::core::hint::black_box(&#ident); })
                        .collect();
                    Ok::<TokenStream2, TokenStream>(quote! {
                        #pat #guard_tokens => {
                            #( #noop_refs )*
                            #body
                        }
                    })
                })
                .collect();
            let arm_tokens = arm_tokens?;
            Ok(quote! {
                match #expr {
                    #( #arm_tokens ),*
                }
            })
        }
    }
}

fn collect_used_param_names(segments: &[Segment]) -> Vec<String> {
    let mut names = Vec::new();
    let mut seen = HashSet::<String>::new();

    for segment in segments {
        match segment {
            Segment::Text(text) => {
                for name in collect_used_param_names_in_sql(text) {
                    if seen.insert(name.clone()) {
                        names.push(name);
                    }
                }
            }
            Segment::Batch { parts } => {
                for part in parts {
                    if let TextPart::Param { name, .. } = part {
                        if seen.insert(name.clone()) {
                            names.push(name.clone());
                        }
                    }
                }
            }
            _ => {}
        }
    }

    names
}

fn collect_used_param_names_in_sql(sql: &str) -> Vec<String> {
    let mut names = Vec::new();
    let mut seen = HashSet::<String>::new();
    for part in parse_text_parts(sql) {
        if let TextPart::Param { name, .. } = part {
            if seen.insert(name.to_string()) {
                names.push(name);
            }
        }
    }
    names
}

/// Builds a parameterized SQL query with compile-time type-checking and a
/// runtime [`sqlx::QueryBuilder`] for dynamic SQL.
///
/// Combines `sqlx::query_as!` / `sqlx::query_scalar!` validation (never called
/// at runtime) with `QueryBuilder::push_bind` for safe value binding.
///
/// # Syntax
///
/// ```text
/// sql_forge!(
///     [DB,]        // optional: sqlx::MySql | sqlx::Postgres | sqlx::Sqlite
///     [Model,]     // optional result spec
///     SQL,         // string literal
///     [params,]    // optional: ( :name = expr, ... )  or  struct_expr
///     [(sections),]// optional: ( #name = ..., ... )
///     [..batch]    // optional: batch source expression used by {( ... )}
/// )
/// ```
///
/// `Model` has three forms:
/// - omitted: execute-only query; only `.execute(...)` is available
/// - `Type` or `scalar Type`: a single result query
/// - `( >key1 = TypeA, >key2 = scalar TypeB )`: a grouped multi-result query
///
/// The trailing parameter source, section map, and batch source are optional.
/// The batch source may appear alongside the others as a single `..expr` argument.
///
/// The DB type may be omitted when `SQL_FORGE_DB_TYPE` is set (e.g.
/// `SQL_FORGE_DB_TYPE=sqlx::MySql`) or when
/// `[package.metadata.sql_forge] db = "..."` is set in `Cargo.toml`.
/// The env var takes priority over Cargo.toml metadata.
///
/// # Parameters
///
/// Named parameters are written `:name` in the SQL. At runtime each occurrence
/// is replaced by a `push_bind` call; at compile time it becomes a
/// database-specific placeholder: `?` for MySQL and SQLite, and `$1`, `$2`, ...
/// for Postgres.
///
/// **Inline map** – bind individual expressions:
/// ```rust,ignore
/// sql_forge!(User, "SELECT ... WHERE id <= :max_id", ( :max_id = filter.max_id ))
/// ```
///
/// **Struct source** – field names are matched to `:name` placeholders automatically:
/// ```rust,ignore
/// sql_forge!(User, "SELECT ... WHERE id <= :max_id LIMIT :limit", filter)
/// ```
///
/// # Sections (`{#name}`)
///
/// Sections are runtime SQL slots; each section's variants are validated at
/// compile time via `query_as!` / `query_scalar!`, though not every combination
/// of variants across sections is checked. The section map is a second parenthesised
/// argument starting with `#`:
///
/// ```rust,ignore
/// sql_forge!(
///     User,
///     "SELECT * FROM users {#join_org}",
///     (
///         #join_org = match include_org {
///             true  => " JOIN organisations o ON o.id = users.org_id ",
///             false => "",
///         }
///     )
/// )
/// ```
///
/// A section arm can also carry local parameters as a tuple `("sql", params)`:
///
/// ```rust,ignore
/// sql_forge!(
///     User,
///     "SELECT * FROM users {#filter}",
///     (
///         #filter = (
///             " WHERE id <= :max_id AND status = :status ",
///             ( :max_id = max_id, :status = "active" ),
///         )
///     )
/// )
/// ```
///
/// Multiple placeholders driven by one `match` use `#(a, b)` with each arm
/// returning a tuple of the same width:
///
/// ```rust,ignore
/// sql_forge!(
///     User,
///     "SELECT * FROM users {#join_org} {#filter_org}",
///     (
///         #(join_org, filter_org) = match include_org {
///             true  => (
///                 " JOIN organisations o ON o.id = users.org_id ",
///                 (
///                     " AND o.active = :active ",
///                     ( :active = true ),
///                 ),
///             ),
///             false => ("", ""),
///         }
///     )
/// )
/// ```
///
/// Grouped section items may themselves use nested `match` expressions. Those
/// nested matches use smart cycling within the arm rather than a cartesian
/// product. For example, if one grouped arm returns a fixed first item plus two
/// nested binary matches for the second and third items, that arm contributes
/// two aligned variants `(0, 0)` and `(1, 1)`, not four `(0, 0)`, `(0, 1)`,
/// `(1, 0)`, `(1, 1)` combinations.
///
/// # `IN (...)` with list parameters
///
/// Wrap the placeholder in parentheses to expand a `Vec` into multiple bound
/// values:
///
/// ```rust,ignore
/// sql_forge!(User, "SELECT * FROM users WHERE id IN (:ids[])", ( :ids = ids ))
/// ```
///
/// **Empty lists** are not rewritten; `IN ()` is a database syntax error.
/// Guard against empty inputs explicitly, e.g. with a dynamic section:
///
/// ```rust,ignore
/// sql_forge!(
///     User,
///     "SELECT id, name FROM users WHERE {#filter}",
///     (
///         #filter = match ids.is_empty() {
///             true  => "1 = 0",
///             false => ("id IN (:ids[])", ( :ids = ids )),
///         }
///     )
/// )
/// ```
///
/// # Batch inserts (`{( ... )}`)
///
/// A batch section `{( ... )}` repeats its content for each item in an iterable
/// source passed as `..expr`. Inside the batch, `:name` refers to a field on the
/// current item. List parameters (`:name[]`) are **not** allowed inside batch
/// sections.
///
/// ## Struct batch
///
/// ```rust,ignore
/// struct BatchItem { name: String, price: i64 }
///
/// let items = vec![
///     BatchItem { name: "A".into(), price: 100 },
///     BatchItem { name: "B".into(), price: 200 },
/// ];
///
/// sql_forge!(
///     "INSERT INTO products (name, price, stock, category)
///      VALUES {(:name, :price, 10, 'Batch')}",
///     ..items
/// )
/// .execute(&pool)
/// .await?;
/// ```
///
/// For compile-time checking, the validator expands the batch to 3 fake copies
/// (`(?, ?, 10, 'Batch'), (?, ?, 10, 'Batch'), (?, ?, 10, 'Batch')`).
/// At runtime the iterable drives the actual number of rows.
///
/// # Scalar output
///
/// When `Model` is a primitive (`i32`, `i64`, `String`, etc.) the macro uses
/// `query_scalar!` for validation and `build_query_scalar` for execution.
///
/// # Multiple results
///
/// A result map produces a `SqlForgeQueryGroup` with one query per key.
/// Each key can be a struct or a primitive (used as a scalar):
///
/// ```rust,ignore
/// sql_forge!(
///     (
///         >count = i64,
///         >items = Item,
///     ),
///     "SELECT {#fields} FROM items WHERE category_id = :cat",
///     ( :cat = category_id ),
///     (
///         #fields = match {>count} {           // {>key} is true when building
///             true  => "COUNT(*) AS total",    // the query for that model/result
///             false => "id, name, price",      // key and false otherwise
///         }
///     )
/// )
/// ```
///
/// The generated struct has one field per key (`group.count`, `group.items`),
/// each implementing `SqlForgeQuery<T, Db = DB>` and usable with any SQLx
/// executor method (`fetch_one`, `fetch_all`, etc.).
///
/// # Execute-only (no model)
///
/// When the model type is omitted, the macro produces a value implementing
/// `SqlForgeQueryExecute`. Only `.execute(executor)`
/// is available and there is no return type to deserialize into. This is useful
/// for `INSERT`, `UPDATE`, `DELETE`, and other DML statements.
///
/// ```rust,ignore
/// sql_forge!(
///     "UPDATE products SET stock = stock + 1 WHERE id = :id",
///     ( :id = 42i64 ),
/// )
/// .execute(&pool)
/// .await?;
/// ```
///
/// Sections and struct parameter sources work the same way as in model-backed queries:
///
/// ```rust,ignore
/// sql_forge!(
///     "UPDATE products SET price = :new_price {#filter}",
///     ( #filter = "WHERE category = :cat", ( :cat = "Electronics" ) ),
/// )
/// .execute(&pool)
/// .await?;
/// ```
#[proc_macro]
#[allow(clippy::too_many_lines)]
pub fn sql_forge(input: TokenStream) -> TokenStream {
    // ---- Phase 1: Parse the macro input into structured data ----
    let preprocessed = preprocess_result_key_placeholders(TokenStream2::from(input));
    let SqlForgeInput {
        db,
        result,
        force_scalar,
        sql,
        params,
        sections,
        batch,
    } = match syn::parse2::<SqlForgeInput>(preprocessed) {
        Ok(v) => v,
        Err(err) => return err.to_compile_error().into(),
    };

    // ---- Phase 2: Resolve database type (from macro arg or Cargo.toml) ----
    let db = match db {
        Some(db) => db,
        None => match resolve_db_from_env() {
            Ok(db) => db,
            Err(msg) => {
                return syn::Error::new(Span::call_site(), msg)
                    .to_compile_error()
                    .into();
            }
        },
    };

    let use_dollar_params = uses_dollar_params(&db);
    let is_sqlite = if let syn::Type::Path(type_path) = &db {
        type_path
            .path
            .segments
            .last()
            .is_some_and(|s| s.ident == "Sqlite")
    } else {
        false
    };
    let list_count: usize = if is_sqlite { 1 } else { 3 };

    // ---- Phase 3: Build result case definitions ----
    // Each result case is (optional_key, model_type, optional_scalar_type).
    // Scalar type is set for primitives and `scalar`-marked types.
    let result_cases: Vec<(Option<String>, Option<Type>, Option<Type>)> = match result {
        ResultSpec::None => {
            vec![(None, None, None)]
        }
        ResultSpec::Single(ref model) => {
            let model_ty = (**model).clone();
            let scalar = if force_scalar {
                Some(model_ty.clone())
            } else {
                scalar_output_type(model.as_ref()).cloned()
            };
            vec![(None, Some(model_ty), scalar)]
        }
        ResultSpec::Group(ref cases) => {
            if force_scalar {
                return syn::Error::new(
                    Span::call_site(),
                    "sql_forge!: scalar mode is not supported for grouped result maps",
                )
                .to_compile_error()
                .into();
            }

            let mut out = Vec::new();
            let mut seen = HashSet::new();
            for case in cases {
                let key = case.name.to_string();
                if !seen.insert(key.clone()) {
                    return syn::Error::new(
                        case.name.span(),
                        "sql_forge!: duplicated key in result map",
                    )
                    .to_compile_error()
                    .into();
                }

                let model = case.model.clone();
                let scalar = if case.force_scalar {
                    Some(model.clone())
                } else {
                    scalar_output_type(&case.model).cloned()
                };
                out.push((Some(key), Some(model), scalar));
            }
            out
        }
    };
    let group_result_keys: Vec<String> = result_cases
        .iter()
        .filter_map(|(key, _, _)| key.as_ref().cloned())
        .collect();
    let is_grouped_result = !group_result_keys.is_empty();
    let sql_span = sql.span();

    // ---- Phase 4: Parse SQL into segments (text + {#section} slots) ----
    let segments = match sql.into_segments() {
        Ok(segments) => segments,
        Err(msg) => {
            return syn::Error::new(sql_span, msg).to_compile_error().into();
        }
    };

    let has_batch_segment = segments.iter().any(|s| matches!(s, Segment::Batch { .. }));
    match (&batch, has_batch_segment) {
        (None, true) => {
            return syn::Error::new(
                sql_span,
                "sql_forge!: SQL contains {( ... )} batch section but no batch source argument (..expr) \
                 was provided"
            )
            .to_compile_error()
            .into();
        }
        (Some(_), false) => {
            return syn::Error::new(
                sql_span,
                "sql_forge!: batch source argument (..expr) provided but SQL has no {( ... )} \
                 batch section",
            )
            .to_compile_error()
            .into();
        }
        _ => {}
    }

    let used_param_names = collect_used_param_names(&segments);

    // Batch-only params come from batch items, not the top-level params map.
    // They must be excluded from the usage check so that a param like :category
    // that appears only inside {( ... )} is flagged as unused when given in the
    // params map, as it would never be read from there at runtime.
    let batch_param_names: std::collections::HashSet<String> = segments
        .iter()
        .filter_map(|s| {
            if let Segment::Batch { parts } = s {
                Some(parts.iter().filter_map(|p| {
                    if let TextPart::Param { name, .. } = p {
                        Some(name.clone())
                    } else {
                        None
                    }
                }))
            } else {
                None
            }
        })
        .flatten()
        .collect();
    let top_level_used_names: Vec<String> = used_param_names
        .iter()
        .filter(|n| !batch_param_names.contains(*n))
        .cloned()
        .collect();

    // ---- Phase 5: Build parameter bindings for the top-level params ----
    let (declared_params, validator_param_bindings) =
        match build_param_bindings(&params, &top_level_used_names, "top_level", true, true) {
            Ok(v) => v,
            Err(err) => return err,
        };

    let mut runtime_section_actions = HashMap::<String, TokenStream2>::new();

    // ---- Phase 6: Process sections: build runtime actions and collect validation variants ----
    for assign in &sections {
        let SectionAssign { names, value } = assign;

        // Build runtime actions first, while `value` is still available by reference.
        let mut named_actions: Vec<(String, TokenStream2)> = Vec::new();
        for (section_idx, name_ident) in names.iter().enumerate() {
            let name = name_ident.to_string();
            if runtime_section_actions.contains_key(&name) {
                return syn::Error::new(
                    name_ident.span(),
                    "sql_forge!: duplicated section mapping",
                )
                .to_compile_error()
                .into();
            }
            let action = match build_section_runtime_action(
                value,
                section_idx,
                &format!("section_{}", name),
            ) {
                Ok(action) => action,
                Err(err) => return err,
            };
            named_actions.push((name, action));
        }

        // Consume `value` here so invalid grouped/nested section structures fail early.
        if let Err(msg) = collect_section_variants(value.clone(), names.len()) {
            return syn::Error::new(names[0].span(), msg)
                .to_compile_error()
                .into();
        }

        for (name, action) in named_actions {
            runtime_section_actions.insert(name, action);
        }
    }

    let sql_section_names: std::collections::HashSet<&str> = segments
        .iter()
        .filter_map(|seg| {
            if let Segment::Section { name } = seg {
                Some(name.as_str())
            } else {
                None
            }
        })
        .collect();
    for name in runtime_section_actions.keys() {
        if !sql_section_names.contains(name.as_str()) {
            return syn::Error::new(
                sql_span,
                format!(
                    "sql_forge!: section `#{}` is declared in the section map but `{{#{}}}` never appears in the SQL",
                    name, name,
                ),
            )
            .to_compile_error()
            .into();
        }
    }

    // ---- Phase 8: For each result case, generate validator + runtime tokens ----
    let mut generated_query_defs = Vec::<TokenStream2>::new();
    let mut generated_query_values = Vec::<TokenStream2>::new();
    let mut group_field_defs = Vec::<TokenStream2>::new();
    let mut group_method_defs = Vec::<TokenStream2>::new();
    let mut group_field_idents = Vec::<syn::Ident>::new();
    let mut group_field_tys = Vec::<TokenStream2>::new();
    let mut group_trait_impls = Vec::<TokenStream2>::new();

    let mut grouped_validator_invocations = Vec::<TokenStream2>::new();

    for (result_key, model_opt, scalar_model_ty) in result_cases.iter() {
        let suffix = result_key.as_deref().unwrap_or("single");
        let query_ident = format_ident!("__SqlForgeQuery_{}", suffix);
        let query_value_ident = format_ident!("__sql_forge_value_{}", suffix);

        let flag_bindings = build_result_flag_bindings(&group_result_keys, result_key.as_deref());

        let mut section_variants_for_validation = HashMap::<String, Vec<SectionFragment>>::new();
        for assign in &sections {
            let SectionAssign { names, value } = assign;
            let variants_by_section = match collect_section_variants_for_result(
                value.clone(),
                names.len(),
                result_key.as_deref(),
            ) {
                Ok(v) => v,
                Err(msg) => {
                    return syn::Error::new(names[0].span(), msg)
                        .to_compile_error()
                        .into();
                }
            };

            for (name_ident, section_cases) in names.iter().zip(variants_by_section) {
                section_variants_for_validation.insert(name_ident.to_string(), section_cases);
            }
        }

        let mut nmax = 1usize;
        for segment in &segments {
            if let Segment::Section { name } = segment {
                if let Some(variants) = section_variants_for_validation.get(name) {
                    if variants.is_empty() {
                        return syn::Error::new(
                            sql_span,
                            format!("sql_forge!: section {{#{}}} has no possible variants", name),
                        )
                        .to_compile_error()
                        .into();
                    }
                    nmax = nmax.max(variants.len());
                } else {
                    return syn::Error::new(
                        sql_span,
                        format!("sql_forge!: section {{#{}}} has no mapping", name),
                    )
                    .to_compile_error()
                    .into();
                }
            }
        }

        let mut validator_cases = Vec::<(LitStr, Vec<TokenStream2>, Vec<TokenStream2>)>::new();
        for case_idx in 0..nmax {
            let mut sql_case = String::new();
            let mut case_setup = Vec::<TokenStream2>::new();
            let mut case_args = Vec::<TokenStream2>::new();
            let mut param_offset = 0usize;
            let mut arg_index = 0usize;
            let empty_params = HashMap::<String, syn::Ident>::new();
            let root_validator_context = ValidatorRenderContext {
                local_params: &empty_params,
                top_level_params: &declared_params,
                allow_top_level_fallback: true,
                use_dollar_params,
                sql_span,
                list_count,
            };

            for segment in &segments {
                match segment {
                    Segment::Text(text) => {
                        let (chunk_sql, chunk_setup, chunk_args) = match render_validator_args(
                            text,
                            &mut param_offset,
                            &mut arg_index,
                            &root_validator_context,
                        ) {
                            Ok(value) => value,
                            Err(err) => return err,
                        };
                        sql_case.push_str(&chunk_sql);
                        case_setup.extend(chunk_setup);
                        case_args.extend(chunk_args);
                    }
                    Segment::Section { name } => {
                        let Some(variants) = section_variants_for_validation.get(name) else {
                            return syn::Error::new(
                                sql_span,
                                format!("sql_forge!: section {{#{}}} has no mapping", name),
                            )
                            .to_compile_error()
                            .into();
                        };

                        let fragment = &variants[case_idx % variants.len()];
                        let used_param_names = collect_used_param_names_in_sql(&fragment.sql);
                        let (local_params, bindings) = match build_param_bindings(
                            &fragment.params,
                            &used_param_names,
                            &format!("section_case_{}_{}_{}", suffix, case_idx, name),
                            true,
                            true,
                        ) {
                            Ok(value) => value,
                            Err(err) => return err,
                        };
                        let section_validator_context = ValidatorRenderContext {
                            local_params: &local_params,
                            top_level_params: &declared_params,
                            allow_top_level_fallback: false,
                            use_dollar_params,
                            sql_span: fragment.span,
                            list_count,
                        };
                        let (chunk_sql, chunk_setup, chunk_args) = match render_validator_args(
                            &fragment.sql,
                            &mut param_offset,
                            &mut arg_index,
                            &section_validator_context,
                        ) {
                            Ok(value) => value,
                            Err(err) => return err,
                        };
                        sql_case.push_str(&chunk_sql);
                        case_setup.extend(bindings);
                        case_setup.extend(chunk_setup);
                        case_args.extend(chunk_args);
                    }
                    Segment::Batch { parts } => {
                        let mut first = true;
                        for _ in 0..list_count {
                            let sep = if first { "" } else { ", " };
                            first = false;
                            sql_case.push_str(sep);
                            for tp in parts {
                                match tp {
                                    TextPart::Lit(lit) => sql_case.push_str(lit),
                                    TextPart::Param { name, .. } => {
                                        if let Some(batch_expr) = &batch {
                                            let field_ident = format_ident!("{}", name);
                                            if use_dollar_params {
                                                param_offset += 1;
                                                write!(sql_case, "${}", param_offset).unwrap();
                                            } else {
                                                sql_case.push('?');
                                            }
                                            case_args.push(quote! { #batch_expr[0].#field_ident });
                                        } else if use_dollar_params {
                                            param_offset += 1;
                                            write!(sql_case, "${}", param_offset).unwrap();
                                        } else {
                                            sql_case.push('?');
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }

            validator_cases.push((LitStr::new(&sql_case, sql_span), case_setup, case_args));
        }

        let mut validator_invocations = Vec::<TokenStream2>::new();
        for (sql_lit, case_setup, args) in &validator_cases {
            if model_opt.is_none() {
                if args.is_empty() {
                    validator_invocations.push(quote! {
                        {
                            #( #case_setup )*
                            let _ = sqlx::query_scalar!(
                                #sql_lit,
                            );
                        }
                    });
                } else {
                    validator_invocations.push(quote! {
                        {
                            #( #case_setup )*
                            let _ = sqlx::query_scalar!(
                                #sql_lit,
                                #( #args ),*
                            );
                        }
                    });
                }
            } else if let Some(scalar_ty) = scalar_model_ty {
                if args.is_empty() {
                    validator_invocations.push(quote! {
                        {
                            #( #case_setup )*
                            let _ = sqlx::query_scalar!(
                                #sql_lit,
                            );
                        }
                    });
                } else {
                    validator_invocations.push(quote! {
                        {
                            #( #case_setup )*
                            let _ = sqlx::query_scalar!(
                                #sql_lit,
                                #( #args ),*
                            );
                        }
                    });
                }
                let _ = scalar_ty;
            } else if args.is_empty() {
                validator_invocations.push(quote! {
                    {
                        #( #case_setup )*
                        let _ = sqlx::query_as!(
                            __EnhancedModel,
                            #sql_lit,
                        );
                    }
                });
            } else {
                validator_invocations.push(quote! {
                    {
                        #( #case_setup )*
                        let _ = sqlx::query_as!(
                            __EnhancedModel,
                            #sql_lit,
                            #( #args ),*
                        );
                    }
                });
            }
        }

        let model_alias = if let Some(model) = model_opt {
            if scalar_model_ty.is_none() {
                quote! { type __EnhancedModel = #model; }
            } else {
                quote! {}
            }
        } else {
            quote! {}
        };
        grouped_validator_invocations.push(quote! {
            {
                #( #flag_bindings )*
                #model_alias
                #( #validator_invocations )*
            }
        });

        let (runtime_declared_params, runtime_param_bindings) =
            match build_param_bindings(&params, &used_param_names, "runtime", false, false) {
                Ok(v) => v,
                Err(err) => return err,
            };

        let mut runtime_steps = Vec::<TokenStream2>::new();
        for (seg_idx, segment) in segments.iter().enumerate() {
            match segment {
                Segment::Text(text) => {
                    for part in parse_text_parts(text) {
                        match part {
                            TextPart::Lit(lit) => {
                                let lit = sanitize_runtime_sql_text(&lit);
                                let lit_str = LitStr::new(&lit, sql_span);
                                runtime_steps.push(quote! {
                                    __builder.push(#lit_str);
                                });
                            }
                            TextPart::Param { name, is_list } => {
                                let Some(local_ident) = runtime_declared_params.get(&name) else {
                                    return syn::Error::new(
                                        sql_span,
                                        format!("sql_forge!: parameter :{} has no mapping", name),
                                    )
                                    .to_compile_error()
                                    .into();
                                };

                                if is_list {
                                    runtime_steps.push(quote! {
                                        let __enhanced_values = #local_ident;
                                        let mut __separated = __builder.separated(", ");
                                        for __value in __enhanced_values {
                                            __separated.push_bind(__value);
                                        }
                                    });
                                } else {
                                    runtime_steps.push(quote! {
                                        __builder.push_bind(#local_ident);
                                    });
                                }
                            }
                        }
                    }
                }
                Segment::Section { name } => {
                    let Some(section_action) = runtime_section_actions.get(name) else {
                        let _ = seg_idx;
                        return syn::Error::new(
                            sql_span,
                            format!("sql_forge!: section {{#{}}} has no mapping", name),
                        )
                        .to_compile_error()
                        .into();
                    };
                    runtime_steps.push(quote! {
                        #section_action
                    });
                }
                Segment::Batch { parts } => {
                    if let Some(batch_expr) = &batch {
                        let mut body = Vec::<TokenStream2>::new();
                        for part in parts {
                            match part {
                                TextPart::Lit(lit) => {
                                    let lit_str = LitStr::new(lit, sql_span);
                                    body.push(quote! {
                                        __builder.push(#lit_str);
                                    });
                                }
                                TextPart::Param { name, .. } => {
                                    let field_ident = format_ident!("{}", name);
                                    body.push(quote! {
                                        __builder.push_bind(__item.#field_ident);
                                    });
                                }
                            }
                        }
                        runtime_steps.push(quote! {
                            {
                                let mut __first = true;
                                for __item in #batch_expr {
                                    if !__first {
                                        __builder.push(", ");
                                    }
                                    __first = false;
                                    #( #body )*
                                }
                            }
                        });
                    }
                }
            }
        }

        let exec_methods = if model_opt.is_none() {
            quote! {
                async fn execute<'e, E>(mut self, executor: E) -> Result<<#db as sqlx::Database>::QueryResult, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner.build().execute(executor).await
                }
            }
        } else if let Some(scalar_ty) = scalar_model_ty {
            quote! {
                async fn fetch_all<'e, E>(mut self, executor: E) -> Result<Vec<#scalar_ty>, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner
                        .build_query_scalar::<#scalar_ty>()
                        .fetch_all(executor)
                        .await
                }

                async fn fetch_one<'e, E>(mut self, executor: E) -> Result<#scalar_ty, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner
                        .build_query_scalar::<#scalar_ty>()
                        .fetch_one(executor)
                        .await
                }

                async fn fetch_optional<'e, E>(mut self, executor: E) -> Result<Option<#scalar_ty>, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner
                        .build_query_scalar::<#scalar_ty>()
                        .fetch_optional(executor)
                        .await
                }

                async fn execute<'e, E>(mut self, executor: E) -> Result<<#db as sqlx::Database>::QueryResult, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner.build().execute(executor).await
                }
            }
        } else {
            let model = model_opt.as_ref().unwrap();
            quote! {
                async fn fetch_all<'e, E>(mut self, executor: E) -> Result<Vec<#model>, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner.build_query_as::<#model>().fetch_all(executor).await
                }

                async fn fetch_one<'e, E>(mut self, executor: E) -> Result<#model, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner.build_query_as::<#model>().fetch_one(executor).await
                }

                async fn fetch_optional<'e, E>(mut self, executor: E) -> Result<Option<#model>, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner
                        .build_query_as::<#model>()
                        .fetch_optional(executor)
                        .await
                }

                async fn execute<'e, E>(mut self, executor: E) -> Result<<#db as sqlx::Database>::QueryResult, sqlx::Error>
                where
                    E: sqlx::Executor<'e, Database = #db>,
                {
                    self.inner.build().execute(executor).await
                }
            }
        };

        let final_type: TokenStream2 = if let Some(model) = model_opt {
            if let Some(scalar_ty) = scalar_model_ty {
                quote! { #scalar_ty }
            } else {
                quote! { #model }
            }
        } else {
            quote! {}
        };
        let trait_impl = if model_opt.is_none() {
            quote! {
                impl<'args> sql_forge::SqlForgeQueryExecute
                    for #query_ident<'args>
                {
                    type Db = #db;

                    fn execute<'e, E>(self, executor: E) -> impl std::future::Future<Output = Result<<#db as sqlx::Database>::QueryResult, sqlx::Error>> + Send + 'e
                    where
                        Self: Sized + 'e,
                        E: sqlx::Executor<'e, Database = #db> + Send + 'e,
                        #db: 'e,
                    {
                        #query_ident::execute(self, executor)
                    }
                }
            }
        } else {
            quote! {
                impl<'args> sql_forge::SqlForgeQuery<#final_type>
                    for #query_ident<'args>
                {
                    type Db = #db;

                    fn fetch_all<'e, E>(self, executor: E) -> impl std::future::Future<Output = Result<Vec<#final_type>, sqlx::Error>> + Send + 'e
                    where
                        Self: Sized + 'e,
                        E: sqlx::Executor<'e, Database = #db> + Send + 'e,
                        #db: 'e,
                    {
                        #query_ident::fetch_all(self, executor)
                    }

                    fn fetch_one<'e, E>(self, executor: E) -> impl std::future::Future<Output = Result<#final_type, sqlx::Error>> + Send + 'e
                    where
                        Self: Sized + 'e,
                        E: sqlx::Executor<'e, Database = #db> + Send + 'e,
                        #db: 'e,
                    {
                        #query_ident::fetch_one(self, executor)
                    }

                    fn fetch_optional<'e, E>(self, executor: E) -> impl std::future::Future<Output = Result<Option<#final_type>, sqlx::Error>> + Send + 'e
                    where
                        Self: Sized + 'e,
                        E: sqlx::Executor<'e, Database = #db> + Send + 'e,
                        #db: 'e,
                    {
                        #query_ident::fetch_optional(self, executor)
                    }

                    fn execute<'e, E>(self, executor: E) -> impl std::future::Future<Output = Result<<#db as sqlx::Database>::QueryResult, sqlx::Error>> + Send + 'e
                    where
                        Self: Sized + 'e,
                        E: sqlx::Executor<'e, Database = #db> + Send + 'e,
                        #db: 'e,
                    {
                        #query_ident::execute(self, executor)
                    }
                }
            }
        };

        generated_query_defs.push(quote! {
            struct #query_ident<'args> {
                inner: sqlx::QueryBuilder<'args, #db>,
            }

            impl<'args> #query_ident<'args> {
                #exec_methods
            }

            #trait_impl
        });

        generated_query_values.push(quote! {
            #( #runtime_param_bindings )*
            #( #flag_bindings )*
            let mut __builder: sqlx::QueryBuilder<#db> = sqlx::QueryBuilder::new("");
            #( #runtime_steps )*
            let #query_value_ident = #query_ident { inner: __builder };
        });

        if let Some(key) = result_key {
            let method_ident = format_ident!("{}", key);
            group_field_defs.push(quote! {
                #method_ident: #query_ident<'args>
            });
            group_field_tys.push(quote! { #query_ident<'args> });
            group_method_defs.push(quote! {
                pub fn #method_ident(self) -> #query_ident<'args> {
                    self.#method_ident
                }
            });

            let key_ty_ident = format_ident!("__SqlForgeQueryGroupKey_{}", key);
            group_trait_impls.push(quote! {
                struct #key_ty_ident;

                impl<'args> sql_forge::SqlForgeQueryGroupGet<#key_ty_ident, #final_type> for __SqlForgeQueryGroup<'args> {
                    type Query = #query_ident<'args>;

                    fn get(self, _: #key_ty_ident) -> Self::Query {
                        self.#method_ident
                    }
                }
            });
            group_field_idents.push(method_ident);
        }
    }

    // ---- Phase 8: Emit the final token stream ----
    let validator_tokens = quote! {
        let _sql_forge_validator = || {
            #( #validator_param_bindings )*
            #( #grouped_validator_invocations )*
        };
    };

    if !is_grouped_result {
        let single_query_value_ident = format_ident!("__sql_forge_value_single");
        return quote! {
            {
                #validator_tokens
                #( #generated_query_defs )*
                #( #generated_query_values )*
                #single_query_value_ident
            }
        }
        .into();
    }

    let group_field_inits: Vec<TokenStream2> = result_cases
        .iter()
        .filter_map(|(key, _, _)| key.as_ref())
        .map(|key| {
            let method_ident = format_ident!("{}", key);
            let query_value_ident = format_ident!("__sql_forge_value_{}", key);
            quote! { #method_ident: #query_value_ident }
        })
        .collect();

    quote! {
        {
            #validator_tokens

            #( #generated_query_defs )*
            #( #generated_query_values )*

            struct __SqlForgeQueryGroup<'args> {
                #( #group_field_defs, )*
            }

            impl<'args> __SqlForgeQueryGroup<'args> {
                #( #group_method_defs )*

                pub fn into_parts(self) -> ( #( #group_field_tys ),* ) {
                    ( #( self.#group_field_idents ),* )
                }
            }

            impl<'args> sql_forge::SqlForgeQueryGroup for __SqlForgeQueryGroup<'args> {
                type Db = #db;
            }

            #( #group_trait_impls )*

            __SqlForgeQueryGroup {
                #( #group_field_inits, )*
            }
        }
    }
    .into()
}

/// Expands to the database type from the `SQL_FORGE_DB_TYPE` environment variable,
/// falling back to `[package.metadata.sql_forge]` in `Cargo.toml`.
///
/// ```rust,ignore
/// use sql_forge::db_type;
///
/// pub type AppDb = db_type!();
/// // expands to the type set via SQL_FORGE_DB_TYPE or Cargo.toml metadata
/// ```
///
/// Priority:
/// 1. `SQL_FORGE_DB_TYPE` env var (e.g. `sqlx::MySql`, `sqlx::Postgres`)
/// 2. `[package.metadata.sql_forge] db = "..."` in `Cargo.toml`
#[proc_macro]
pub fn db_type(input: TokenStream) -> TokenStream {
    if !input.is_empty() {
        return syn::Error::new(Span::call_site(), "db_type!() takes no arguments")
            .to_compile_error()
            .into();
    }

    match resolve_db_from_env() {
        Ok(db) => quote! { #db }.into(),
        Err(msg) => syn::Error::new(Span::call_site(), msg)
            .to_compile_error()
            .into(),
    }
}

/// Marks a single-value tuple struct as a transparent wrapper for use with
/// `sql_forge!` parameters.
///
/// Expands to `#[derive(sqlx::Type)]` + `#[sqlx(transparent)]` (needed for
/// all database backends so the type implements `sqlx::Encode` + `sqlx::Type`)
/// and additionally implements `SqlForgeValidatorValue<InnerType>`, which is
/// **required for PostgreSQL** to pass compile-time parameter validation in
/// `query_as!`. MySQL and SQLite do not use the trait.
///
/// ```rust,ignore
/// #[derive(Debug, PartialEq, Eq)]
/// #[sql_forge_transparent]
/// struct UserId(pub i64);
/// ```
#[proc_macro_attribute]
pub fn sql_forge_transparent(_attr: TokenStream, item: TokenStream) -> TokenStream {
    let input: ItemStruct = match syn::parse(item) {
        Ok(v) => v,
        Err(err) => return err.to_compile_error().into(),
    };

    let struct_name = &input.ident;
    let inner_type = match &input.fields {
        Fields::Unnamed(fields) if fields.unnamed.len() == 1 => &fields.unnamed.first().unwrap().ty,
        _ => {
            return syn::Error::new(
                input.span(),
                "#[sql_forge_transparent] expects a tuple struct with exactly one field",
            )
            .to_compile_error()
            .into();
        }
    };

    let attrs = input.attrs;
    let generics = &input.generics;
    let vis = &input.vis;
    let struct_token = input.struct_token;
    let semi_token = input.semi_token;
    let fields = &input.fields;

    let expanded = quote! {
        #( #attrs )*
        #[derive(sqlx::Type)]
        #[sqlx(transparent)]
        #vis #struct_token #struct_name #generics #fields #semi_token

        impl #generics sql_forge::SqlForgeValidatorValue<#inner_type> for #struct_name #generics {
            fn sql_forge_validator_value(&self) -> #inner_type {
                self.0.clone()
            }
        }
    };

    expanded.into()
}