shaperail_codegen/

diagnostics.rs

use shaperail_core::{FieldType, HttpMethod, ResourceDefinition, WASM_HOOK_PREFIX};

/// A structured diagnostic with error code, human message, fix suggestion, and example.
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize)]
pub struct Diagnostic {
    /// Stable error code (e.g., "SR001").
    pub code: &'static str,
    /// Human-readable error message.
    pub error: String,
    /// Suggested fix action.
    pub fix: String,
    /// Inline YAML example showing the correct pattern.
    pub example: String,
}

impl std::fmt::Display for Diagnostic {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "[{}] {}", self.code, self.error)
    }
}

/// Validate a parsed `ResourceDefinition` and return structured diagnostics
/// with fix suggestions. This is the "AI-friendly" counterpart to `validator::validate_resource`.
pub fn diagnose_resource(rd: &ResourceDefinition) -> Vec<Diagnostic> {
    let mut diags = Vec::new();
    let res = &rd.resource;

    if res.is_empty() {
        diags.push(Diagnostic {
            code: "SR001",
            error: "resource name must not be empty".into(),
            fix: "add a snake_case plural name to the 'resource' key".into(),
            example: "resource: users".into(),
        });
    }

    if rd.version == 0 {
        diags.push(Diagnostic {
            code: "SR002",
            error: format!("resource '{res}': version must be >= 1"),
            fix: "set version to 1 or higher".into(),
            example: "version: 1".into(),
        });
    }

    if rd.schema.is_empty() {
        diags.push(Diagnostic {
            code: "SR003",
            error: format!("resource '{res}': schema must have at least one field"),
            fix: "add at least an id field to the schema section".into(),
            example: "schema:\n  id: { type: uuid, primary: true, generated: true }".into(),
        });
    }

    let primary_count = rd.schema.values().filter(|f| f.primary).count();
    if primary_count == 0 && !rd.schema.is_empty() {
        diags.push(Diagnostic {
            code: "SR004",
            error: format!("resource '{res}': schema must have a primary key field"),
            fix: "add 'primary: true' to one field (typically 'id')".into(),
            example: "id: { type: uuid, primary: true, generated: true }".into(),
        });
    } else if primary_count > 1 {
        diags.push(Diagnostic {
            code: "SR005",
            error: format!(
                "resource '{res}': schema must have exactly one primary key, found {primary_count}"
            ),
            fix: "remove 'primary: true' from all fields except one".into(),
            example: "id: { type: uuid, primary: true, generated: true }".into(),
        });
    }

    for (name, field) in &rd.schema {
        if field.field_type == FieldType::Enum && field.values.is_none() {
            diags.push(Diagnostic {
                code: "SR010",
                error: format!("resource '{res}': field '{name}' is type enum but has no values"),
                fix: format!("add 'values: [value1, value2]' to the '{name}' field"),
                example: format!("{name}: {{ type: enum, values: [option_a, option_b] }}"),
            });
        }

        if field.field_type != FieldType::Enum && field.values.is_some() {
            diags.push(Diagnostic {
                code: "SR011",
                error: format!("resource '{res}': field '{name}' has values but is not type enum"),
                fix: format!("change the type to 'enum' or remove 'values' from '{name}'"),
                example: format!("{name}: {{ type: enum, values: [...] }}"),
            });
        }

        if field.reference.is_some() && field.field_type != FieldType::Uuid {
            diags.push(Diagnostic {
                code: "SR012",
                error: format!("resource '{res}': field '{name}' has ref but is not type uuid"),
                fix: format!("change the type of '{name}' to uuid"),
                example: format!(
                    "{name}: {{ type: uuid, ref: {}, required: true }}",
                    field.reference.as_deref().unwrap_or("resource.id")
                ),
            });
        }

        if let Some(ref reference) = field.reference {
            if !reference.contains('.') {
                diags.push(Diagnostic {
                    code: "SR013",
                    error: format!(
                        "resource '{res}': field '{name}' ref must be in 'resource.field' format, got '{reference}'"
                    ),
                    fix: "use 'resource_name.field_name' format for the ref value".into(),
                    example: format!("{name}: {{ type: uuid, ref: organizations.id }}"),
                });
            }
        }

        if field.field_type == FieldType::Array && field.items.is_none() {
            diags.push(Diagnostic {
                code: "SR014",
                error: format!("resource '{res}': field '{name}' is type array but has no items"),
                fix: format!("add 'items: <element_type>' to the '{name}' field"),
                example: format!("{name}: {{ type: array, items: string }}"),
            });
        }

        if field.format.is_some() && field.field_type != FieldType::String {
            diags.push(Diagnostic {
                code: "SR015",
                error: format!(
                    "resource '{res}': field '{name}' has format but is not type string"
                ),
                fix: format!("change the type of '{name}' to string, or remove 'format'"),
                example: format!(
                    "{name}: {{ type: string, format: {} }}",
                    field.format.as_deref().unwrap_or("email")
                ),
            });
        }

        if field.primary && !field.generated && !field.required {
            diags.push(Diagnostic {
                code: "SR016",
                error: format!(
                    "resource '{res}': primary key field '{name}' must be generated or required"
                ),
                fix: format!("add 'generated: true' or 'required: true' to '{name}'"),
                example: format!("{name}: {{ type: uuid, primary: true, generated: true }}"),
            });
        }
    }

    // Tenant key validation
    if let Some(ref tenant_key) = rd.tenant_key {
        match rd.schema.get(tenant_key) {
            Some(field) => {
                if field.field_type != FieldType::Uuid {
                    diags.push(Diagnostic {
                        code: "SR020",
                        error: format!(
                            "resource '{res}': tenant_key '{tenant_key}' must reference a uuid field, found {}",
                            field.field_type
                        ),
                        fix: format!("change the type of '{tenant_key}' to uuid"),
                        example: format!(
                            "{tenant_key}: {{ type: uuid, ref: organizations.id, required: true }}"
                        ),
                    });
                }
            }
            None => {
                diags.push(Diagnostic {
                    code: "SR021",
                    error: format!(
                        "resource '{res}': tenant_key '{tenant_key}' not found in schema"
                    ),
                    fix: format!("add a '{tenant_key}' uuid field to the schema"),
                    example: format!(
                        "{tenant_key}: {{ type: uuid, ref: organizations.id, required: true }}"
                    ),
                });
            }
        }
    }

    // Endpoint validation
    if let Some(endpoints) = &rd.endpoints {
        for (action, ep) in endpoints {
            if let Some(controller) = &ep.controller {
                if let Some(before) = &controller.before {
                    if before.is_empty() {
                        diags.push(Diagnostic {
                            code: "SR030",
                            error: format!(
                                "resource '{res}': endpoint '{action}' has an empty controller.before name"
                            ),
                            fix: "provide a function name for controller.before".into(),
                            example: "controller: { before: validate_input }".into(),
                        });
                    }
                    diagnose_controller_name(res, action, "before", before, &mut diags);
                }
                if let Some(after) = &controller.after {
                    if after.is_empty() {
                        diags.push(Diagnostic {
                            code: "SR031",
                            error: format!(
                                "resource '{res}': endpoint '{action}' has an empty controller.after name"
                            ),
                            fix: "provide a function name for controller.after".into(),
                            example: "controller: { after: enrich_response }".into(),
                        });
                    }
                    diagnose_controller_name(res, action, "after", after, &mut diags);
                }
            }

            if let Some(events) = &ep.events {
                for event in events {
                    if event.is_empty() {
                        diags.push(Diagnostic {
                            code: "SR032",
                            error: format!(
                                "resource '{res}': endpoint '{action}' has an empty event name"
                            ),
                            fix: "use 'resource.action' format for event names".into(),
                            example: format!("events: [{res}.created]"),
                        });
                    }
                }
            }

            if let Some(jobs) = &ep.jobs {
                for job in jobs {
                    if job.is_empty() {
                        diags.push(Diagnostic {
                            code: "SR033",
                            error: format!(
                                "resource '{res}': endpoint '{action}' has an empty job name"
                            ),
                            fix: "provide a snake_case job name".into(),
                            example: "jobs: [send_notification]".into(),
                        });
                    }
                }
            }

            // Input/filter/search/sort fields must exist
            for (field_kind, fields) in [
                ("input", &ep.input),
                ("filter", &ep.filters),
                ("search", &ep.search),
                ("sort", &ep.sort),
            ] {
                if let Some(fields) = fields {
                    for field_name in fields {
                        if !rd.schema.contains_key(field_name) {
                            diags.push(Diagnostic {
                                code: "SR040",
                                error: format!(
                                    "resource '{res}': endpoint '{action}' {field_kind} field '{field_name}' not found in schema"
                                ),
                                fix: format!(
                                    "add '{field_name}' to the schema, or remove it from {field_kind}"
                                ),
                                example: format!("{field_name}: {{ type: string, required: true }}"),
                            });
                        }
                    }
                }
            }

            if ep.soft_delete && !rd.schema.contains_key("updated_at") {
                diags.push(Diagnostic {
                    code: "SR041",
                    error: format!(
                        "resource '{res}': endpoint '{action}' has soft_delete but schema has no 'updated_at' field"
                    ),
                    fix: "add an 'updated_at' timestamp field to the schema".into(),
                    example: "updated_at: { type: timestamp, generated: true }".into(),
                });
            }

            if let Some(upload) = &ep.upload {
                if !matches!(
                    *ep.method(),
                    HttpMethod::Post | HttpMethod::Patch | HttpMethod::Put
                ) {
                    diags.push(Diagnostic {
                        code: "SR050",
                        error: format!(
                            "resource '{res}': endpoint '{action}' uses upload but method must be POST, PATCH, or PUT"
                        ),
                        fix: "change the method to POST, PATCH, or PUT".into(),
                        example: "method: POST".into(),
                    });
                }

                match rd.schema.get(&upload.field) {
                    Some(field) if field.field_type == FieldType::File => {}
                    Some(_) => diags.push(Diagnostic {
                        code: "SR051",
                        error: format!(
                            "resource '{res}': endpoint '{action}' upload field '{}' must be type file",
                            upload.field
                        ),
                        fix: format!("change '{}' to type file in the schema", upload.field),
                        example: format!("{}: {{ type: file, required: true }}", upload.field),
                    }),
                    None => diags.push(Diagnostic {
                        code: "SR052",
                        error: format!(
                            "resource '{res}': endpoint '{action}' upload field '{}' not found in schema",
                            upload.field
                        ),
                        fix: format!("add '{}' as a file field in the schema", upload.field),
                        example: format!("{}: {{ type: file, required: true }}", upload.field),
                    }),
                }

                if !matches!(upload.storage.as_str(), "local" | "s3" | "gcs" | "azure") {
                    diags.push(Diagnostic {
                        code: "SR053",
                        error: format!(
                            "resource '{res}': endpoint '{action}' upload storage '{}' is invalid",
                            upload.storage
                        ),
                        fix: "use one of: local, s3, gcs, azure".into(),
                        example: "upload: { field: file, storage: s3, max_size: 5mb }".into(),
                    });
                }

                if !ep
                    .input
                    .as_ref()
                    .is_some_and(|fields| fields.iter().any(|field| field == &upload.field))
                {
                    diags.push(Diagnostic {
                        code: "SR054",
                        error: format!(
                            "resource '{res}': endpoint '{action}' upload field '{}' must appear in input",
                            upload.field
                        ),
                        fix: format!("add '{}' to the input array", upload.field),
                        example: format!("input: [{}]", upload.field),
                    });
                }
            }
        }
    }

    // Relation validation
    if let Some(relations) = &rd.relations {
        for (name, rel) in relations {
            use shaperail_core::RelationType;

            if rel.relation_type == RelationType::BelongsTo && rel.key.is_none() {
                diags.push(Diagnostic {
                    code: "SR060",
                    error: format!(
                        "resource '{res}': relation '{name}' is belongs_to but has no key"
                    ),
                    fix: format!("add 'key: {res}_id' to the relation (the local FK field)"),
                    example: format!(
                        "{name}: {{ resource: {}, type: belongs_to, key: {}_id }}",
                        rel.resource, rel.resource
                    ),
                });
            }

            if matches!(
                rel.relation_type,
                RelationType::HasMany | RelationType::HasOne
            ) && rel.foreign_key.is_none()
            {
                diags.push(Diagnostic {
                    code: "SR061",
                    error: format!(
                        "resource '{res}': relation '{name}' is {} but has no foreign_key",
                        rel.relation_type
                    ),
                    fix: format!(
                        "add 'foreign_key: {res}_id' to the relation (the FK on the related table)"
                    ),
                    example: format!(
                        "{name}: {{ resource: {}, type: {}, foreign_key: {res}_id }}",
                        rel.resource, rel.relation_type
                    ),
                });
            }

            if let Some(key) = &rel.key {
                if !rd.schema.contains_key(key) {
                    diags.push(Diagnostic {
                        code: "SR062",
                        error: format!(
                            "resource '{res}': relation '{name}' key '{key}' not found in schema"
                        ),
                        fix: format!("add '{key}' as a uuid field in the schema"),
                        example: format!(
                            "{key}: {{ type: uuid, ref: {}.id, required: true }}",
                            rel.resource
                        ),
                    });
                }
            }
        }
    }

    // Index validation
    if let Some(indexes) = &rd.indexes {
        for (i, idx) in indexes.iter().enumerate() {
            if idx.fields.is_empty() {
                diags.push(Diagnostic {
                    code: "SR070",
                    error: format!("resource '{res}': index {i} has no fields"),
                    fix: "add at least one field to the index".into(),
                    example: "- { fields: [field_name] }".into(),
                });
            }
            for field_name in &idx.fields {
                if !rd.schema.contains_key(field_name) {
                    diags.push(Diagnostic {
                        code: "SR071",
                        error: format!(
                            "resource '{res}': index {i} references field '{field_name}' not in schema"
                        ),
                        fix: format!("add '{field_name}' to the schema, or remove it from the index"),
                        example: format!("{field_name}: {{ type: string, required: true }}"),
                    });
                }
            }
            if let Some(order) = &idx.order {
                if order != "asc" && order != "desc" {
                    diags.push(Diagnostic {
                        code: "SR072",
                        error: format!(
                            "resource '{res}': index {i} has invalid order '{order}', must be 'asc' or 'desc'"
                        ),
                        fix: "use 'asc' or 'desc' for the index order".into(),
                        example: "- { fields: [created_at], order: desc }".into(),
                    });
                }
            }
        }
    }

    diags
}

fn diagnose_controller_name(
    res: &str,
    action: &str,
    phase: &str,
    name: &str,
    diags: &mut Vec<Diagnostic>,
) {
    if let Some(wasm_path) = name.strip_prefix(WASM_HOOK_PREFIX) {
        if wasm_path.is_empty() {
            diags.push(Diagnostic {
                code: "SR035",
                error: format!(
                    "resource '{res}': endpoint '{action}' controller.{phase} has 'wasm:' prefix but no path"
                ),
                fix: "provide a .wasm file path after the 'wasm:' prefix".into(),
                example: format!("controller: {{ {phase}: \"wasm:./plugins/validator.wasm\" }}"),
            });
        } else if !wasm_path.ends_with(".wasm") {
            diags.push(Diagnostic {
                code: "SR036",
                error: format!(
                    "resource '{res}': endpoint '{action}' controller.{phase} WASM path must end with '.wasm', got '{wasm_path}'"
                ),
                fix: "ensure the WASM plugin path ends with '.wasm'".into(),
                example: format!("controller: {{ {phase}: \"wasm:./plugins/validator.wasm\" }}"),
            });
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::parser::parse_resource;

    #[test]
    fn valid_resource_produces_no_diagnostics() {
        let yaml = include_str!("../../resources/users.yaml");
        let rd = parse_resource(yaml).unwrap();
        let diags = diagnose_resource(&rd);
        assert!(diags.is_empty(), "Expected no diagnostics, got: {diags:?}");
    }

    #[test]
    fn enum_without_values_has_fix_suggestion() {
        let yaml = r#"
resource: items
version: 1
schema:
  id: { type: uuid, primary: true, generated: true }
  status: { type: enum, required: true }
"#;
        let rd = parse_resource(yaml).unwrap();
        let diags = diagnose_resource(&rd);
        let d = diags.iter().find(|d| d.code == "SR010").unwrap();
        assert!(d.fix.contains("values"));
        assert!(d.example.contains("values:"));
    }

    #[test]
    fn missing_primary_key_has_fix_suggestion() {
        let yaml = r#"
resource: items
version: 1
schema:
  name: { type: string, required: true }
"#;
        let rd = parse_resource(yaml).unwrap();
        let diags = diagnose_resource(&rd);
        let d = diags.iter().find(|d| d.code == "SR004").unwrap();
        assert!(d.fix.contains("primary: true"));
    }

    #[test]
    fn diagnostics_serialize_to_json() {
        let d = Diagnostic {
            code: "SR010",
            error: "field 'role' is type enum but has no values".into(),
            fix: "add values".into(),
            example: "role: { type: enum, values: [a, b] }".into(),
        };
        let json = serde_json::to_string(&d).unwrap();
        assert!(json.contains("SR010"));
        assert!(json.contains("fix"));
    }
}