evenframe_core 0.1.5

//! Configuration builders for processing Evenframe types.

use super::{BuildConfig, EvenframeType, WorkspaceScanner};
use crate::error::Result;
use crate::{
    derive::{
        attributes::{
            parse_annotation_attributes, parse_doccom_attribute, parse_event_attributes,
            parse_format_attribute_bin, parse_index_attributes, parse_macroforge_derive_attribute,
            parse_mock_data_attribute, parse_relation_attribute, parse_rust_derives,
            parse_table_validators,
        },
        validator_parser::parse_field_validators_as_enums,
    },
    schemasync::table::TableConfig,
    schemasync::{DefineConfig, EdgeConfig, EventConfig, PermissionsConfig},
    types::{FieldType, StructConfig, StructField, TaggedUnion, Variant, VariantData},
    typesync::config::CollisionStrategy,
    validator::{StringValidator, Validator},
};
use convert_case::{Case, Casing};
use std::collections::BTreeMap;
use std::fs;
use std::path::Path;
use syn::{Fields, FieldsNamed, Item, ItemEnum, ItemStruct, parse_file};
use tracing::{debug, info, trace, warn};

/// All configurations extracted from the workspace.
pub type AllConfigs = (
    BTreeMap<String, TaggedUnion>,
    BTreeMap<String, TableConfig>,
    BTreeMap<String, StructConfig>,
);

/// Builds all configurations from the workspace using the provided build config.
///
/// Returns a tuple of (enums, tables, objects).
pub fn build_all_configs(config: &BuildConfig) -> Result<AllConfigs> {
    debug!("Starting build_all_configs");
    let mut enum_configs = BTreeMap::new();
    let mut table_configs = BTreeMap::new();
    let mut struct_configs = BTreeMap::new();

    debug!("Creating workspace scanner");
    let scanner = WorkspaceScanner::with_path(
        config.scan_path.clone(),
        config.apply_aliases.clone(),
        config.expand_macros,
    );

    let types = scanner.scan_for_evenframe_types()?;
    info!("Found {} Evenframe types", types.len());

    process_types(
        &types,
        &mut enum_configs,
        &mut table_configs,
        &mut struct_configs,
        config.collision_strategy,
    )?;

    info!(
        "First pass complete. Found {} struct configs, {} enum configs, {} table configs",
        struct_configs.len(),
        enum_configs.len(),
        table_configs.len()
    );

    resolve_relation_endpoints(&mut table_configs, &enum_configs);

    // Apply output rule plugins to enrich configs with convention-based defaults
    #[cfg(feature = "wasm-plugins")]
    {
        info!(
            "Output rule plugins configured: {}",
            config.output_rule_plugins.len()
        );
        for (name, cfg) in &config.output_rule_plugins {
            debug!("  Output rule plugin '{}' -> {}", name, cfg.path);
        }
    }
    #[cfg(feature = "wasm-plugins")]
    if !config.output_rule_plugins.is_empty() {
        apply_rule_plugins(
            config,
            &mut table_configs,
            &mut struct_configs,
            &mut enum_configs,
        );
    }

    // Apply synthetic-item plugins — these add new structs/enums/tables
    // derived from the (now finalized) scanner+rule-plugin state.
    #[cfg(feature = "wasm-plugins")]
    if !config.synthetic_item_plugins.is_empty() {
        info!(
            "Synthetic-item plugins configured: {}",
            config.synthetic_item_plugins.len()
        );
        apply_synthetic_plugins(
            config,
            &mut table_configs,
            &mut struct_configs,
            &mut enum_configs,
        )?;
        // Re-run relation endpoint resolution so any synthetic relation
        // tables get their from/to resolved from their `in`/`out` fields.
        resolve_relation_endpoints(&mut table_configs, &enum_configs);
    }

    Ok((enum_configs, table_configs, struct_configs))
}

/// Builds all configurations from the workspace using default configuration.
///
/// This loads config from `evenframe.toml` if available, otherwise uses defaults.
/// Returns a tuple of (enums, tables, objects).
pub fn build_all_configs_default() -> AllConfigs {
    debug!("Starting build_all_configs_default");
    let mut enum_configs = BTreeMap::new();
    let mut table_configs = BTreeMap::new();
    let mut struct_configs = BTreeMap::new();

    // Try to load config, fall back to defaults
    let config = match BuildConfig::from_toml() {
        Ok(cfg) => cfg,
        Err(e) => {
            warn!("Error loading configuration: {}, using defaults", e);
            BuildConfig::default()
        }
    };

    debug!("Creating workspace scanner");
    let scanner = WorkspaceScanner::with_path(
        config.scan_path.clone(),
        config.apply_aliases.clone(),
        config.expand_macros,
    );

    let types = match scanner.scan_for_evenframe_types() {
        Ok(types) => {
            info!("Found {} Evenframe types", types.len());
            types
        }
        Err(e) => {
            warn!("Error scanning workspace: {}", e);
            return (BTreeMap::new(), BTreeMap::new(), BTreeMap::new());
        }
    };

    if let Err(e) = process_types(
        &types,
        &mut enum_configs,
        &mut table_configs,
        &mut struct_configs,
        CollisionStrategy::Error,
    ) {
        warn!("Error processing types: {}", e);
        return (BTreeMap::new(), BTreeMap::new(), BTreeMap::new());
    }

    info!(
        "First pass complete. Found {} struct configs, {} enum configs, {} table configs",
        struct_configs.len(),
        enum_configs.len(),
        table_configs.len()
    );

    resolve_relation_endpoints(&mut table_configs, &enum_configs);

    (enum_configs, table_configs, struct_configs)
}

/// Resolves `from`/`to` on relation tables by inspecting `in`/`out` field types.
///
/// For each table with a relation config that has empty `from`/`to`, this looks at
/// the `in` and `out` fields' `RecordLink<T>` types and resolves `T` to table names.
/// If `T` is a persistable_union enum, all variant table names are collected.
fn resolve_relation_endpoints(
    table_configs: &mut BTreeMap<String, TableConfig>,
    enum_configs: &BTreeMap<String, TaggedUnion>,
) {
    // Snapshot table names to avoid borrow conflicts
    let known_tables: std::collections::BTreeSet<String> = table_configs.keys().cloned().collect();

    for table_config in table_configs.values_mut() {
        let Some(relation) = table_config.relation.as_mut() else {
            continue;
        };

        // Default edge_name from table_name if empty
        if relation.edge_name.is_empty() {
            relation.edge_name = table_config.table_name.clone();
        }

        // Resolve from/to from in/out field types
        if relation.from.is_empty()
            && let Some(tables) = resolve_field_to_tables(
                &table_config.struct_config,
                "in",
                enum_configs,
                &known_tables,
            )
        {
            debug!(
                "Auto-resolved relation.from for '{}': {:?}",
                table_config.table_name, tables
            );
            relation.from = tables;
        }
        if relation.to.is_empty()
            && let Some(tables) = resolve_field_to_tables(
                &table_config.struct_config,
                "out",
                enum_configs,
                &known_tables,
            )
        {
            debug!(
                "Auto-resolved relation.to for '{}': {:?}",
                table_config.table_name, tables
            );
            relation.to = tables;
        }
    }
}

/// Resolves a relation field (`in` or `out`) to the table names it references.
fn resolve_field_to_tables(
    struct_config: &crate::types::StructConfig,
    field_name: &str,
    enum_configs: &BTreeMap<String, TaggedUnion>,
    known_tables: &std::collections::BTreeSet<String>,
) -> Option<Vec<String>> {
    let field = struct_config
        .fields
        .iter()
        .find(|f| f.field_name == field_name)?;

    // Extract the inner type name from RecordLink<T>
    let inner_type_name = match &field.field_type {
        FieldType::RecordLink(inner) => match inner.as_ref() {
            FieldType::Other(name) => name.clone(),
            _ => return None,
        },
        _ => return None,
    };

    // Try direct table match
    let snake = inner_type_name.to_case(Case::Snake);
    if known_tables.contains(&snake) {
        return Some(vec![snake]);
    }

    // Try enum variant resolution
    if let Some(tagged) = enum_configs.get(&inner_type_name) {
        let mut tables = Vec::new();
        for variant in &tagged.variants {
            if let Some(data) = &variant.data {
                let struct_name = match data {
                    VariantData::InlineStruct(s) => &s.struct_name,
                    VariantData::DataStructureRef(FieldType::Other(name)) => name,
                    _ => continue,
                };
                let t = struct_name.to_case(Case::Snake);
                if known_tables.contains(&t) {
                    tables.push(t);
                }
            }
        }
        if !tables.is_empty() {
            return Some(tables);
        }
    }

    None
}

/// Recursively collects all `Item::Struct` and `Item::Enum` from nested `mod` blocks.
/// For non-expanded files (no `Item::Mod`), this returns the same items unchanged.
fn collect_items_flat(items: Vec<Item>) -> Vec<Item> {
    let mut result = Vec::new();
    for item in items {
        match item {
            Item::Mod(ref m) => {
                if let Some((_, ref mod_items)) = m.content {
                    result.extend(collect_items_flat(mod_items.clone()));
                }
            }
            item @ (Item::Struct(_) | Item::Enum(_)) => result.push(item),
            _ => {}
        }
    }
    result
}

/// Processes found Evenframe types into configurations.
fn process_types(
    types: &[EvenframeType],
    enum_configs: &mut BTreeMap<String, TaggedUnion>,
    table_configs: &mut BTreeMap<String, TableConfig>,
    struct_configs: &mut BTreeMap<String, StructConfig>,
    collision_strategy: CollisionStrategy,
) -> Result<()> {
    debug!("Grouping types by file");
    let mut types_by_file: BTreeMap<String, Vec<_>> = BTreeMap::new();
    for evenframe_type in types {
        types_by_file
            .entry(evenframe_type.file_path.clone())
            .or_default()
            .push(evenframe_type);
    }
    debug!("Grouped into {} files", types_by_file.len());

    // Track which file each type name was first defined in (for collision messages)
    let mut struct_origins: BTreeMap<String, String> = BTreeMap::new();
    let mut enum_origins: BTreeMap<String, String> = BTreeMap::new();
    // Track renames so we can update field references after all types are collected
    let mut renames: BTreeMap<String, String> = BTreeMap::new();

    debug!("Starting first pass: parsing structs and enums");
    for (file_path, file_types) in &types_by_file {
        trace!("Processing file: {}", file_path);
        let content = match fs::read_to_string(file_path) {
            Ok(content) => content,
            Err(e) => {
                warn!("Error reading file {}: {}", file_path, e);
                continue;
            }
        };

        let syntax = match parse_file(&content) {
            Ok(syntax) => syntax,
            Err(e) => {
                warn!("Error parsing file {}: {}", file_path, e);
                continue;
            }
        };
        trace!("Parsed {} items from {}", syntax.items.len(), file_path);

        for item in collect_items_flat(syntax.items) {
            match item {
                Item::Struct(item_struct) => {
                    if let Some(evenframe_type) =
                        file_types.iter().find(|&t| item_struct.ident == t.name)
                    {
                        debug!("Found Evenframe struct: {:?}", item_struct.ident);
                        if let Some(mut struct_config) = parse_struct_config(&item_struct) {
                            struct_config.pipeline = evenframe_type.pipeline;
                            // Check for name collision
                            if let Some(existing_file) =
                                struct_origins.get(&struct_config.struct_name)
                            {
                                match collision_strategy {
                                    CollisionStrategy::Error => {
                                        return Err(crate::error::EvenframeError::Config(format!(
                                            "Type name collision: '{}' is defined in both '{}' and '{}'. \
                                             Rename one of them, or set collision_strategy = \"auto_rename\" \
                                             in [typesync] config.",
                                            struct_config.struct_name, existing_file, file_path
                                        )));
                                    }
                                    CollisionStrategy::AutoRename => {
                                        let stem = Path::new(file_path)
                                            .file_stem()
                                            .and_then(|s| s.to_str())
                                            .unwrap_or("unknown");
                                        let prefix = stem.to_case(Case::Pascal);
                                        let old_name = struct_config.struct_name.clone();
                                        let new_name = format!("{}{}", prefix, old_name);
                                        warn!(
                                            "Type '{}' in '{}' renamed to '{}' to avoid collision with '{}'",
                                            old_name, file_path, new_name, existing_file
                                        );
                                        struct_config.struct_name = new_name.clone();
                                        renames.insert(old_name, new_name);
                                    }
                                }
                            }

                            struct_origins
                                .insert(struct_config.struct_name.clone(), file_path.clone());
                            trace!(
                                "Inserting struct config {:?}: {:#?}",
                                &struct_config.struct_name, &struct_config
                            );
                            struct_configs
                                .insert(struct_config.struct_name.clone(), struct_config.clone());

                            if evenframe_type.has_id_field {
                                let table_name = struct_config.struct_name.to_case(Case::Snake);
                                debug!(
                                    "Building table config for: {} (snake_case: {})",
                                    struct_config.struct_name, &table_name
                                );

                                let mock_generation_config =
                                    parse_mock_data_attribute(&item_struct.attrs).ok().flatten();

                                let events = match parse_event_attributes(&item_struct.attrs) {
                                    Ok(events) => events,
                                    Err(e) => {
                                        warn!(
                                            error = %e,
                                            struct_name = %struct_config.struct_name,
                                            "Failed to parse event attributes, skipping"
                                        );
                                        Vec::new()
                                    }
                                };

                                let known_field_names: std::collections::BTreeSet<String> =
                                    struct_config
                                        .fields
                                        .iter()
                                        .map(|f| {
                                            f.field_name
                                                .trim_start_matches("r#")
                                                .to_string()
                                        })
                                        .collect();
                                let indexes = parse_index_attributes(
                                    &item_struct.attrs,
                                    &known_field_names,
                                )
                                .map_err(|e| {
                                    crate::error::EvenframeError::Config(format!(
                                        "Failed to parse #[index(...)] on struct '{}' in '{}': {}",
                                        struct_config.struct_name, file_path, e
                                    ))
                                })?;

                                let table_config = TableConfig {
                                    table_name: table_name.clone(),
                                    struct_config: struct_config.clone(),
                                    relation: parse_relation_attribute(&item_struct.attrs)
                                        .ok()
                                        .flatten(),
                                    permissions: PermissionsConfig::parse(&item_struct.attrs)
                                        .ok()
                                        .flatten(),
                                    mock_generation_config,
                                    events: events
                                        .into_iter()
                                        .map(|statement| EventConfig { statement })
                                        .collect(),
                                    indexes,
                                };
                                trace!(
                                    "Inserting table config {:?}: {:#?}",
                                    &table_config.table_name, &struct_config
                                );
                                table_configs.insert(table_name, table_config);
                            }
                        }
                    }
                }
                Item::Enum(item_enum) => {
                    if let Some(evenframe_type) =
                        file_types.iter().find(|&t| item_enum.ident == t.name)
                    {
                        debug!("Found Evenframe enum: {}", item_enum.ident);
                        if let Some(mut tagged_union) = parse_enum_config(&item_enum) {
                            tagged_union.pipeline = evenframe_type.pipeline;
                            // Check for name collision
                            if let Some(existing_file) = enum_origins.get(&tagged_union.enum_name) {
                                match collision_strategy {
                                    CollisionStrategy::Error => {
                                        return Err(crate::error::EvenframeError::Config(format!(
                                            "Type name collision: '{}' is defined in both '{}' and '{}'. \
                                             Rename one of them, or set collision_strategy = \"auto_rename\" \
                                             in [typesync] config.",
                                            tagged_union.enum_name, existing_file, file_path
                                        )));
                                    }
                                    CollisionStrategy::AutoRename => {
                                        let stem = Path::new(file_path)
                                            .file_stem()
                                            .and_then(|s| s.to_str())
                                            .unwrap_or("unknown");
                                        let prefix = stem.to_case(Case::Pascal);
                                        let old_name = tagged_union.enum_name.clone();
                                        let new_name = format!("{}{}", prefix, old_name);
                                        warn!(
                                            "Type '{}' in '{}' renamed to '{}' to avoid collision with '{}'",
                                            old_name, file_path, new_name, existing_file
                                        );
                                        tagged_union.enum_name = new_name.clone();
                                        renames.insert(old_name, new_name);
                                    }
                                }
                            }

                            enum_origins.insert(tagged_union.enum_name.clone(), file_path.clone());
                            trace!(
                                "Inserting enum config {:?}: {:#?}",
                                &tagged_union.enum_name, &tagged_union
                            );
                            enum_configs
                                .insert(tagged_union.enum_name.clone(), tagged_union.clone());

                            for variant in &tagged_union.variants {
                                if let Some(VariantData::InlineStruct(ref enum_struct)) =
                                    variant.data
                                {
                                    struct_configs.insert(
                                        enum_struct.struct_name.clone(),
                                        enum_struct.clone(),
                                    );
                                }
                            }
                        }
                    }
                }
                _ => {}
            }
        }
    }

    // Propagate renames through field type references
    if !renames.is_empty() {
        debug!(
            "Propagating {} type renames through field references",
            renames.len()
        );
        for struct_config in struct_configs.values_mut() {
            for field in &mut struct_config.fields {
                rename_field_type(&mut field.field_type, &renames);
            }
        }
        for table_config in table_configs.values_mut() {
            for field in &mut table_config.struct_config.fields {
                rename_field_type(&mut field.field_type, &renames);
            }
        }
    }

    Ok(())
}

/// Recursively updates type references that were renamed due to collisions.
fn rename_field_type(field_type: &mut FieldType, renames: &BTreeMap<String, String>) {
    match field_type {
        FieldType::Other(name) => {
            if let Some(new_name) = renames.get(name.as_str()) {
                *name = new_name.clone();
            }
        }
        FieldType::Option(inner) | FieldType::Vec(inner) | FieldType::RecordLink(inner) => {
            rename_field_type(inner, renames);
        }
        FieldType::HashMap(k, v) | FieldType::BTreeMap(k, v) => {
            rename_field_type(k, renames);
            rename_field_type(v, renames);
        }
        FieldType::Tuple(items) => {
            for item in items {
                rename_field_type(item, renames);
            }
        }
        FieldType::Struct(fields) => {
            for (_, ft) in fields {
                rename_field_type(ft, renames);
            }
        }
        _ => {}
    }
}

fn parse_struct_config(item_struct: &ItemStruct) -> Option<StructConfig> {
    let struct_name = item_struct.ident.to_string();
    trace!("Parsing struct config for: {}", struct_name);
    let mut fields = Vec::new();

    if let Fields::Named(ref fields_named) = item_struct.fields {
        debug!(
            "Processing {} fields for struct {}",
            fields_named.named.len(),
            struct_name
        );
        fields = process_struct_fields(fields_named);
    }

    let table_validators = parse_table_validators(&item_struct.attrs)
        .ok()
        .unwrap_or_default();

    let doccom = parse_doccom_attribute(&item_struct.attrs).ok().flatten();
    let macroforge_derives = parse_macroforge_derive_attribute(&item_struct.attrs)
        .ok()
        .unwrap_or_default();
    let annotations = parse_annotation_attributes(&item_struct.attrs)
        .ok()
        .unwrap_or_default();
    let rust_derives = parse_rust_derives(&item_struct.attrs);
    let raw_attributes = collect_raw_attributes(&item_struct.attrs);

    Some(StructConfig {
        struct_name,
        fields,
        validators: table_validators
            .into_iter()
            .map(|v| Validator::StringValidator(StringValidator::StringEmbedded(v)))
            .collect(),
        doccom,
        macroforge_derives,
        annotations,
        pipeline: crate::types::Pipeline::default(),
        rust_derives,
        output_override: None,
        raw_attributes,
    })
}

fn parse_enum_config(item_enum: &ItemEnum) -> Option<TaggedUnion> {
    let enum_name = item_enum.ident.to_string();
    trace!("Parsing enum config for: {}", enum_name);
    let mut variants = Vec::new();

    let enum_doccom = parse_doccom_attribute(&item_enum.attrs).ok().flatten();
    let enum_macroforge_derives = parse_macroforge_derive_attribute(&item_enum.attrs)
        .ok()
        .unwrap_or_default();
    let enum_annotations = parse_annotation_attributes(&item_enum.attrs)
        .ok()
        .unwrap_or_default();
    let representation =
        crate::derive::attributes::parse_serde_enum_representation(&item_enum.attrs)
            .ok()
            .unwrap_or_default();
    let enum_rust_derives = parse_rust_derives(&item_enum.attrs);

    for variant in &item_enum.variants {
        let variant_name = variant.ident.to_string();
        trace!("Processing variant: {} in enum {}", variant_name, enum_name);

        let variant_doccom = parse_doccom_attribute(&variant.attrs).ok().flatten();
        let variant_annotations = parse_annotation_attributes(&variant.attrs)
            .ok()
            .unwrap_or_default();
        let variant_macroforge_derives = parse_macroforge_derive_attribute(&variant.attrs)
            .ok()
            .unwrap_or_default();

        let data = match &variant.fields {
            Fields::Unit => None,
            Fields::Unnamed(fields) => {
                if fields.unnamed.is_empty() {
                    None
                } else if fields.unnamed.len() == 1 {
                    let field = &fields.unnamed[0];
                    let field_type = FieldType::parse_syn_ty(&field.ty);
                    Some(VariantData::DataStructureRef(field_type))
                } else {
                    let field_types: Vec<_> = fields
                        .unnamed
                        .iter()
                        .map(|f| FieldType::parse_syn_ty(&f.ty))
                        .collect();
                    Some(VariantData::DataStructureRef(FieldType::Tuple(field_types)))
                }
            }
            Fields::Named(fields_named) => {
                debug!(
                    "Processing {} fields for enum struct {}",
                    fields_named.named.len(),
                    variant_name
                );
                let struct_fields = process_struct_fields(fields_named);

                Some(VariantData::InlineStruct(StructConfig {
                    struct_name: variant_name.clone(),
                    fields: struct_fields,
                    macroforge_derives: variant_macroforge_derives,
                    ..Default::default()
                }))
            }
        };

        let variant_raw_attributes = collect_raw_attributes(&variant.attrs);
        let is_default_variant = variant
            .attrs
            .iter()
            .any(|a| a.path().is_ident("default"));

        variants.push(Variant {
            name: variant_name,
            data,
            doccom: variant_doccom,
            annotations: variant_annotations,
            output_override: None,
            raw_attributes: variant_raw_attributes,
            is_default: is_default_variant,
        });
    }

    let enum_raw_attributes = collect_raw_attributes(&item_enum.attrs);

    Some(TaggedUnion {
        enum_name,
        variants,
        representation,
        doccom: enum_doccom,
        macroforge_derives: enum_macroforge_derives,
        annotations: enum_annotations,
        pipeline: crate::types::Pipeline::default(),
        rust_derives: enum_rust_derives,
        output_override: None,
        raw_attributes: enum_raw_attributes,
    })
}

fn process_struct_fields(fields_named: &FieldsNamed) -> Vec<StructField> {
    let mut struct_fields = Vec::new();
    for field in &fields_named.named {
        let field_name = field
            .ident
            .as_ref()
            .expect("Something went wrong getting the field name")
            .to_string();
        let field_name = field_name.trim_start_matches("r#").to_string();

        let field_type = FieldType::parse_syn_ty(&field.ty);

        let edge_config = EdgeConfig::parse(field).ok().flatten();
        let define_config = DefineConfig::parse(field).ok().flatten();
        let format = parse_format_attribute_bin(&field.attrs).ok().flatten();
        let validators = parse_field_validators_as_enums(&field.attrs);
        let doccom = parse_doccom_attribute(&field.attrs).ok().flatten();
        let annotations = parse_annotation_attributes(&field.attrs)
            .ok()
            .unwrap_or_default();

        let field_raw_attributes = collect_raw_attributes(&field.attrs);

        struct_fields.push(StructField {
            field_name,
            field_type,
            edge_config,
            define_config,
            format,
            validators,
            always_regenerate: false,
            doccom,
            annotations,
            unique: false,
            mock_plugin: None,
            output_override: None,
            raw_attributes: field_raw_attributes,
        });
    }
    struct_fields
}

/// Attributes that evenframe's own attribute parsers handle. Everything
/// else ends up in `StructConfig::raw_attributes` so plugins can see it.
const KNOWN_ATTRS: &[&str] = &[
    "annotation",
    "apply",
    "define_field_statement",
    "derive",
    "doc",
    "doccom",
    "edge",
    "event",
    "fetch",
    "format",
    "macroforge_derive",
    "mock_data",
    "mockmake",
    "permissions",
    "relation",
    "serde",
    "subquery",
    "unique",
    "validators",
    // These are handled by proc-macros but aren't plugin-relevant metadata.
    "cfg",
    "cfg_attr",
    "allow",
    "warn",
    "deny",
    "forbid",
    "must_use",
    "non_exhaustive",
    "repr",
    "automatically_derived",
];

/// Collects every attribute that evenframe doesn't natively handle into
/// a map of `attr_name → [raw_body, …]`. The "body" is the stringified
/// token stream inside the parens (or empty string for path-only attrs).
fn collect_raw_attributes(attrs: &[syn::Attribute]) -> BTreeMap<String, Vec<String>> {
    use quote::ToTokens;

    let mut out: BTreeMap<String, Vec<String>> = BTreeMap::new();
    for attr in attrs {
        let Some(ident) = attr.path().get_ident() else {
            continue;
        };
        let name = ident.to_string();
        if KNOWN_ATTRS.contains(&name.as_str()) {
            continue;
        }
        let body = match &attr.meta {
            syn::Meta::Path(_) => String::new(),
            syn::Meta::List(list) => list.tokens.to_token_stream().to_string(),
            syn::Meta::NameValue(nv) => nv.value.to_token_stream().to_string(),
        };
        out.entry(name).or_default().push(body);
    }
    out
}

/// Merges tables and objects into a single struct config map.
pub fn merge_tables_and_objects(
    tables: &BTreeMap<String, TableConfig>,
    objects: &BTreeMap<String, StructConfig>,
) -> BTreeMap<String, StructConfig> {
    debug!(
        "Merging {} tables and {} objects",
        tables.len(),
        objects.len()
    );
    let mut struct_configs = objects.clone();

    // Tables are present in `objects` under their PascalCase `struct_name`
    // (inserted by `process_types` during the scan) and carry whatever
    // output_override was set by the rule-plugin struct loop. We also want
    // the table's authoritative struct_config (with the output_override set
    // by the rule-plugin table loop) accessible under the snake_case key.
    //
    // Drop the PascalCase duplicate before inserting the table entry so
    // downstream consumers that dedup by PascalCase `struct_name` (e.g.
    // `generate_macroforge_for_types` via its `seen_structs` set) see
    // exactly one entry per table — the authoritative one from the table
    // loop — rather than racing HashMap iteration order.
    for (name, table_config) in tables {
        trace!("Merging table config for: {}", name);
        struct_configs.remove(&table_config.struct_config.struct_name);
        struct_configs.insert(name.clone(), table_config.struct_config.clone());
    }

    debug!(
        "Merge complete. Total struct configs: {}",
        struct_configs.len()
    );
    struct_configs
}

/// Filter configs to only types that participate in the typesync pipeline.
pub fn filter_for_typesync(
    enums: BTreeMap<String, TaggedUnion>,
    tables: BTreeMap<String, TableConfig>,
    objects: BTreeMap<String, StructConfig>,
) -> (
    BTreeMap<String, TaggedUnion>,
    BTreeMap<String, TableConfig>,
    BTreeMap<String, StructConfig>,
) {
    (
        enums
            .into_iter()
            .filter(|(_, v)| v.pipeline.includes_typesync())
            .collect(),
        tables
            .into_iter()
            .filter(|(_, v)| v.struct_config.pipeline.includes_typesync())
            .collect(),
        objects
            .into_iter()
            .filter(|(_, v)| v.pipeline.includes_typesync())
            .collect(),
    )
}

/// Filter configs to only types that participate in the schemasync pipeline.
pub fn filter_for_schemasync(
    enums: BTreeMap<String, TaggedUnion>,
    tables: BTreeMap<String, TableConfig>,
    objects: BTreeMap<String, StructConfig>,
) -> (
    BTreeMap<String, TaggedUnion>,
    BTreeMap<String, TableConfig>,
    BTreeMap<String, StructConfig>,
) {
    (
        enums
            .into_iter()
            .filter(|(_, v)| v.pipeline.includes_schemasync())
            .collect(),
        tables
            .into_iter()
            .filter(|(_, v)| v.struct_config.pipeline.includes_schemasync())
            .collect(),
        objects
            .into_iter()
            .filter(|(_, v)| v.pipeline.includes_schemasync())
            .collect(),
    )
}

// ============================================================
// Rule plugin application
// ============================================================

/// Apply output rule plugins to set output overrides on configs.
///
/// For each table/struct/enum, calls the plugin and sets the `output_override`
/// field directly. The typesync and schemasync generators check this field
/// before computing output — if set, they use the override string directly.
#[cfg(feature = "wasm-plugins")]
#[cfg(feature = "wasm-plugins")]
fn apply_rule_plugins(
    config: &BuildConfig,
    table_configs: &mut BTreeMap<String, TableConfig>,
    struct_configs: &mut BTreeMap<String, StructConfig>,
    enum_configs: &mut BTreeMap<String, TaggedUnion>,
) {
    use crate::typesync::plugin::OutputRulePluginManager;

    let mut pm = match OutputRulePluginManager::new(&config.output_rule_plugins, &config.scan_path)
    {
        Ok(pm) => pm,
        Err(e) => {
            warn!("Failed to load output rule plugins: {}", e);
            return;
        }
    };

    // Process table configs (handler structs)
    let table_names: Vec<String> = table_configs.keys().cloned().collect();
    info!(
        "Applying output rule plugins to {} tables",
        table_names.len()
    );
    for table_name in &table_names {
        let tc = &table_configs[table_name];
        let input = build_plugin_input_struct(&tc.struct_config, Some(tc));

        for plugin_name in pm.plugin_names().to_vec() {
            match pm.transform_type(&plugin_name, &input) {
                Ok(plugin_output) => {
                    if plugin_output.error.is_some() {
                        continue;
                    }
                    let tc = table_configs.get_mut(table_name).unwrap();
                    let to = &plugin_output.type_override;
                    if !to.macroforge_derives.is_empty() || !to.annotations.is_empty() {
                        let mut ov = tc.struct_config.clone();
                        ov.output_override = None;
                        if !to.macroforge_derives.is_empty() {
                            ov.macroforge_derives = to.macroforge_derives.clone();
                        }
                        if !to.annotations.is_empty() {
                            for ann in &to.annotations {
                                if !ov.annotations.contains(ann) {
                                    ov.annotations.push(ann.clone());
                                }
                            }
                        }
                        tc.struct_config.output_override = Some(Box::new(ov));
                    }
                    if let Some(ref perms) = to.permissions {
                        tc.permissions = Some(crate::schemasync::PermissionsConfig {
                            all_permissions: None,
                            select_permissions: Some(perms.select.clone()),
                            create_permissions: Some(perms.create.clone()),
                            update_permissions: Some(perms.update.clone()),
                            delete_permissions: Some(perms.delete.clone()),
                        });
                    }
                    for event in &to.events {
                        tc.events.push(crate::schemasync::EventConfig {
                            statement: event.statement.clone(),
                        });
                    }
                    for (field_name, field_override) in &plugin_output.field_overrides {
                        if let Some(field) = tc
                            .struct_config
                            .fields
                            .iter_mut()
                            .find(|f| &f.field_name == field_name)
                            && !field_override.annotations.is_empty()
                        {
                            let mut ov = field.clone();
                            ov.output_override = None;
                            ov.annotations = field_override.annotations.clone();
                            field.output_override = Some(Box::new(ov));
                        }
                    }
                }
                Err(e) => {
                    warn!(
                        "Output rule plugin '{}' failed for table '{}': {}",
                        plugin_name, table_name, e
                    );
                }
            }
        }
    }

    // Process non-table struct configs
    let struct_names: Vec<String> = struct_configs.keys().cloned().collect();
    info!(
        "Applying output rule plugins to {} structs",
        struct_names.len()
    );
    for struct_name in &struct_names {
        let sc = &struct_configs[struct_name];
        let input = build_plugin_input_struct(sc, None);

        for plugin_name in pm.plugin_names().to_vec() {
            match pm.transform_type(&plugin_name, &input) {
                Ok(plugin_output) => {
                    if plugin_output.error.is_some() {
                        continue;
                    }
                    let sc = struct_configs.get_mut(struct_name).unwrap();
                    let to = &plugin_output.type_override;
                    if !to.macroforge_derives.is_empty() || !to.annotations.is_empty() {
                        let mut ov = sc.clone();
                        ov.output_override = None;
                        if !to.macroforge_derives.is_empty() {
                            ov.macroforge_derives = to.macroforge_derives.clone();
                        }
                        if !to.annotations.is_empty() {
                            for ann in &to.annotations {
                                if !ov.annotations.contains(ann) {
                                    ov.annotations.push(ann.clone());
                                }
                            }
                        }
                        sc.output_override = Some(Box::new(ov));
                    }
                    for (field_name, field_override) in &plugin_output.field_overrides {
                        if let Some(field) =
                            sc.fields.iter_mut().find(|f| &f.field_name == field_name)
                            && !field_override.annotations.is_empty()
                        {
                            let mut ov = field.clone();
                            ov.output_override = None;
                            ov.annotations = field_override.annotations.clone();
                            field.output_override = Some(Box::new(ov));
                        }
                    }
                }
                Err(e) => {
                    warn!(
                        "Output rule plugin '{}' failed for struct '{}': {}",
                        plugin_name, struct_name, e
                    );
                }
            }
        }
    }

    // Process enum configs
    let enum_names: Vec<String> = enum_configs.keys().cloned().collect();
    info!("Applying output rule plugins to {} enums", enum_names.len());
    for enum_name in &enum_names {
        let ec = &enum_configs[enum_name];
        let input = build_plugin_input_enum(ec);

        for plugin_name in pm.plugin_names().to_vec() {
            match pm.transform_type(&plugin_name, &input) {
                Ok(plugin_output) => {
                    if plugin_output.error.is_some() {
                        continue;
                    }
                    let ec = enum_configs.get_mut(enum_name).unwrap();
                    let to = &plugin_output.type_override;
                    if !to.macroforge_derives.is_empty() || !to.annotations.is_empty() {
                        let mut ov = ec.clone();
                        ov.output_override = None;
                        if !to.macroforge_derives.is_empty() {
                            ov.macroforge_derives = to.macroforge_derives.clone();
                        }
                        if !to.annotations.is_empty() {
                            for ann in &to.annotations {
                                if !ov.annotations.contains(ann) {
                                    ov.annotations.push(ann.clone());
                                }
                            }
                        }
                        ec.output_override = Some(Box::new(ov));
                    }
                    // Apply per-variant annotations: the plugin's field_overrides
                    // map variant names to annotations, mirroring how struct
                    // fields are handled above.
                    for (variant_name, field_override) in &plugin_output.field_overrides {
                        if let Some(variant) =
                            ec.variants.iter_mut().find(|v| &v.name == variant_name)
                            && !field_override.annotations.is_empty()
                        {
                            for ann in &field_override.annotations {
                                if !variant.annotations.contains(ann) {
                                    variant.annotations.push(ann.clone());
                                }
                            }
                        }
                    }
                }
                Err(e) => {
                    warn!(
                        "Output rule plugin '{}' failed for enum '{}': {}",
                        plugin_name, enum_name, e
                    );
                }
            }
        }
    }

    info!("Output rule plugins applied to all configs");
}

#[cfg(feature = "wasm-plugins")]
fn build_plugin_input_struct(
    sc: &StructConfig,
    table: Option<&crate::schemasync::table::TableConfig>,
) -> crate::typesync::plugin_types::OutputRulePluginInput {
    use crate::typesync::plugin_types::OutputRulePluginInput;

    let pipeline = format!("{:?}", sc.pipeline);
    match table {
        Some(tc) => OutputRulePluginInput::Table {
            pipeline,
            generator: String::new(),
            struct_config: sc.clone(),
            table_config: Box::new(tc.clone()),
        },
        None => OutputRulePluginInput::Struct {
            pipeline,
            generator: String::new(),
            config: sc.clone(),
        },
    }
}

#[cfg(feature = "wasm-plugins")]
fn build_plugin_input_enum(
    ec: &TaggedUnion,
) -> crate::typesync::plugin_types::OutputRulePluginInput {
    use crate::typesync::plugin_types::OutputRulePluginInput;

    OutputRulePluginInput::Enum {
        pipeline: format!("{:?}", ec.pipeline),
        generator: String::new(),
        config: ec.clone(),
    }
}

// ============================================================
// Synthetic-item plugin application
// ============================================================

/// Call each configured synthetic-item plugin in sequence, merging the items
/// it emits into the running config maps. Each plugin sees the output of
/// all previous plugins, matching the sequential composition of
/// `apply_rule_plugins`.
///
/// Collisions between plugin output and existing items honor the configured
/// [`CollisionStrategy`]:
///
/// - `Error` → abort the whole build with a collision message.
/// - `AutoRename` → prefix the synthetic item's name with the plugin name,
///   Pascal-cased.
#[cfg(feature = "wasm-plugins")]
fn apply_synthetic_plugins(
    config: &BuildConfig,
    table_configs: &mut BTreeMap<String, TableConfig>,
    struct_configs: &mut BTreeMap<String, StructConfig>,
    enum_configs: &mut BTreeMap<String, TaggedUnion>,
) -> Result<()> {
    use crate::typesync::synthetic_plugin::SyntheticItemPluginManager;

    let mut pm = SyntheticItemPluginManager::new(&config.synthetic_item_plugins, &config.scan_path)
        .map_err(|e| {
            crate::error::EvenframeError::Plugin(format!(
                "Failed to load synthetic-item plugins: {}",
                e
            ))
        })?;

    for plugin_name in pm.plugin_names().to_vec() {
        let input = build_synthetic_input(table_configs, struct_configs, enum_configs);

        let output = pm.generate_items(&plugin_name, &input).map_err(|e| {
            crate::error::EvenframeError::Plugin(format!(
                "Synthetic-item plugin '{}' failed: {}",
                plugin_name, e
            ))
        })?;

        if let Some(err) = output.error {
            return Err(crate::error::EvenframeError::Plugin(format!(
                "Synthetic-item plugin '{}' reported error: {}",
                plugin_name, err
            )));
        }

        info!(
            "Synthetic-item plugin '{}' produced: {} structs, {} enums, {} tables",
            plugin_name,
            output.new_structs.len(),
            output.new_enums.len(),
            output.new_tables.len()
        );

        merge_synthetic_output(
            &plugin_name,
            output,
            table_configs,
            struct_configs,
            enum_configs,
            config.collision_strategy,
        )?;
    }

    Ok(())
}

/// Builds the full-context snapshot handed to each synthetic-item plugin.
///
/// Plugins receive the actual `StructConfig` / `TaggedUnion` / `TableConfig`
/// maps (not lightweight summaries) so they can make system-wide decisions:
/// walk relations, copy field types verbatim into new structs, inspect
/// derives, etc. The clone is cheap (all three types derive `Clone`) and
/// simpler than threading references through the JSON boundary.
#[cfg(feature = "wasm-plugins")]
fn build_synthetic_input(
    table_configs: &BTreeMap<String, TableConfig>,
    struct_configs: &BTreeMap<String, StructConfig>,
    enum_configs: &BTreeMap<String, TaggedUnion>,
) -> crate::typesync::synthetic_plugin_types::SyntheticPluginInput {
    use crate::typesync::synthetic_plugin_types::SyntheticPluginInput;

    SyntheticPluginInput {
        structs: struct_configs.clone(),
        enums: enum_configs.clone(),
        tables: table_configs.clone(),
    }
}

#[cfg(feature = "wasm-plugins")]
fn merge_synthetic_output(
    plugin_name: &str,
    output: crate::typesync::synthetic_plugin_types::SyntheticPluginOutput,
    table_configs: &mut BTreeMap<String, TableConfig>,
    struct_configs: &mut BTreeMap<String, StructConfig>,
    enum_configs: &mut BTreeMap<String, TaggedUnion>,
    collision_strategy: CollisionStrategy,
) -> Result<()> {
    let prefix = plugin_name.to_case(Case::Pascal);

    for mut sc in output.new_structs {
        let original = sc.struct_name.clone();
        if struct_configs.contains_key(&sc.struct_name) {
            match collision_strategy {
                CollisionStrategy::Error => {
                    return Err(crate::error::EvenframeError::Config(format!(
                        "Synthetic-item plugin '{}' produced struct '{}' which collides with an \
                         existing type. Rename it inside the plugin, or set \
                         collision_strategy = \"auto_rename\" in [typesync] config.",
                        plugin_name, sc.struct_name
                    )));
                }
                CollisionStrategy::AutoRename => {
                    sc.struct_name = format!("{}{}", prefix, sc.struct_name);
                    warn!(
                        "Synthetic struct '{}' from plugin '{}' renamed to '{}' to avoid collision",
                        original, plugin_name, sc.struct_name
                    );
                }
            }
        }
        debug!(
            "Synthetic plugin '{}' added struct '{}'",
            plugin_name, sc.struct_name
        );
        struct_configs.insert(sc.struct_name.clone(), sc);
    }

    for mut ec in output.new_enums {
        let original = ec.enum_name.clone();
        if enum_configs.contains_key(&ec.enum_name) {
            match collision_strategy {
                CollisionStrategy::Error => {
                    return Err(crate::error::EvenframeError::Config(format!(
                        "Synthetic-item plugin '{}' produced enum '{}' which collides with an \
                         existing type. Rename it inside the plugin, or set \
                         collision_strategy = \"auto_rename\" in [typesync] config.",
                        plugin_name, ec.enum_name
                    )));
                }
                CollisionStrategy::AutoRename => {
                    ec.enum_name = format!("{}{}", prefix, ec.enum_name);
                    warn!(
                        "Synthetic enum '{}' from plugin '{}' renamed to '{}' to avoid collision",
                        original, plugin_name, ec.enum_name
                    );
                }
            }
        }
        debug!(
            "Synthetic plugin '{}' added enum '{}'",
            plugin_name, ec.enum_name
        );
        enum_configs.insert(ec.enum_name.clone(), ec);
    }

    for mut tc in output.new_tables {
        let original = tc.table_name.clone();
        if table_configs.contains_key(&tc.table_name) {
            match collision_strategy {
                CollisionStrategy::Error => {
                    return Err(crate::error::EvenframeError::Config(format!(
                        "Synthetic-item plugin '{}' produced table '{}' which collides with an \
                         existing table. Rename it inside the plugin, or set \
                         collision_strategy = \"auto_rename\" in [typesync] config.",
                        plugin_name, tc.table_name
                    )));
                }
                CollisionStrategy::AutoRename => {
                    let new_table = format!("{}_{}", prefix.to_case(Case::Snake), tc.table_name);
                    tc.table_name = new_table;
                    tc.struct_config.struct_name =
                        format!("{}{}", prefix, tc.struct_config.struct_name);
                    warn!(
                        "Synthetic table '{}' from plugin '{}' renamed to '{}' to avoid collision",
                        original, plugin_name, tc.table_name
                    );
                }
            }
        }
        debug!(
            "Synthetic plugin '{}' added table '{}'",
            plugin_name, tc.table_name
        );
        table_configs.insert(tc.table_name.clone(), tc);
    }

    Ok(())
}