mir_analyzer/parser/docblock/

mod.rs

use mir_types::{ArrayKey, Atomic, Type, Variance};
/// Docblock parser — delegates to `phpdoc_parser` for tag extraction,
/// then converts tags into mir's `ParsedDocblock` with resolved types.
use std::sync::Arc;

use indexmap::IndexMap;
use phpdoc_parser::{body_text, parse as parse_phpdoc};

// ---------------------------------------------------------------------------
// DocblockParser
// ---------------------------------------------------------------------------

pub struct DocblockParser;

impl DocblockParser {
    pub fn parse(text: &str) -> ParsedDocblock {
        let doc = parse_phpdoc(text);
        let mut result = ParsedDocblock {
            description: extract_description(text),
            ..Default::default()
        };

        for tag in &doc.tags {
            match tag.name.as_str() {
                "param-out" | "psalm-param-out" | "phpstan-param-out" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_s, name)) = parse_param_line(&body_str) {
                            record_backslash_keyword_types(
                                &mut result,
                                "param-out",
                                &ty_s,
                                tag.span.start,
                                text,
                            );
                            if let Some(msg) = validate_type_str(&ty_s, "param-out") {
                                result.invalid_annotations.push(msg);
                            } else {
                                result.out_params.push((
                                    name.trim_start_matches('$').to_string(),
                                    parse_type_string(&ty_s),
                                ));
                            }
                        }
                    }
                }
                "param" | "psalm-param" | "phpstan-param" | "phan-param" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_s, name)) = parse_param_line(&body_str) {
                            record_backslash_keyword_types(
                                &mut result,
                                "param",
                                &ty_s,
                                tag.span.start,
                                text,
                            );
                            // Check if the parsed type is valid
                            if is_inside_generics(&ty_s) {
                                // For unclosed generics, report the full body for context
                                if let Some(msg) = validate_type_str(&body_str, "param") {
                                    result.invalid_annotations.push(msg);
                                }
                            } else if let Some(msg) = validate_type_str(&ty_s, "param") {
                                // For other errors, report the parsed type
                                result.invalid_annotations.push(msg);
                            } else {
                                result
                                    .param_type_strings
                                    .push((name.trim_start_matches('$').to_string(), ty_s.clone()));
                                let bare = name.trim_start_matches('$').to_string();
                                let desc = param_description(&body_str, &ty_s, &bare);
                                if !desc.is_empty() {
                                    result.param_descriptions.push((bare.clone(), desc));
                                }
                                result.params.push((bare, parse_type_string(&ty_s)));
                            }
                        } else if let Some(msg) = validate_type_str(&body_str, "param") {
                            // If parsing failed, validate the full body to provide better error context
                            result.invalid_annotations.push(msg);
                            record_backslash_keyword_types(
                                &mut result,
                                "param",
                                &body_str,
                                tag.span.start,
                                text,
                            );
                        }
                    }
                }
                "return" | "psalm-return" | "phpstan-return" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let ty_s = extract_return_type(&body_str);
                        record_backslash_keyword_types(
                            &mut result,
                            "return",
                            &ty_s,
                            tag.span.start,
                            text,
                        );
                        if let Some(msg) = validate_type_str(&ty_s, "return") {
                            result.invalid_annotations.push(msg);
                        }
                        result.return_type = Some(parse_type_string(&ty_s));
                        let desc = description_after(&body_str, &ty_s);
                        if !desc.is_empty() {
                            result.return_description = Some(desc);
                        }
                    }
                }
                "var" | "psalm-var" | "phpstan-var" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_s, name)) = parse_var_line(&body_str) {
                            record_backslash_keyword_types(
                                &mut result,
                                "var",
                                &ty_s,
                                tag.span.start,
                                text,
                            );
                            if let Some(msg) = validate_type_str(&ty_s, "var") {
                                result.invalid_annotations.push(msg);
                            }
                            let name = name.trim_start_matches('$').to_string();
                            let ty = parse_type_string(&ty_s);
                            result.var_tags.push((Some(name.clone()), ty.clone()));
                            result.var_type = Some(ty);
                            result.var_name = Some(name);
                        } else {
                            // Spaces inside PHP types only appear within <…> generics.
                            // Stop at top-level whitespace to exclude description text that
                            // follows the type in multi-line @var bodies.
                            let ty_s = extract_type_prefix(body_str.trim());
                            record_backslash_keyword_types(
                                &mut result,
                                "var",
                                ty_s,
                                tag.span.start,
                                text,
                            );
                            if let Some(msg) = validate_type_str(ty_s, "var") {
                                result.invalid_annotations.push(msg);
                            }
                            let ty = parse_type_string(ty_s);
                            result.var_tags.push((None, ty.clone()));
                            result.var_type = Some(ty);
                        }
                    }
                }
                "throws" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let first_word = body_str.split_whitespace().next().unwrap_or("");
                        let desc = description_after(&body_str, first_word);
                        for class in first_word.split('|') {
                            if !class.is_empty() {
                                record_backslash_keyword_types(
                                    &mut result,
                                    "throws",
                                    class,
                                    tag.span.start,
                                    text,
                                );
                                result.throws.push(class.to_string());
                                if !desc.is_empty() {
                                    result
                                        .throws_descriptions
                                        .push((class.to_string(), desc.clone()));
                                }
                            }
                        }
                    }
                }
                "deprecated" => {
                    result.is_deprecated = true;
                    result.deprecated = Some(body_text(&tag.body).unwrap_or_default().to_string());
                }
                "template" | "psalm-template" | "phpstan-template" => {
                    if let Some((name, bound, default)) =
                        parse_template_line(tag.name.as_str(), body_text(&tag.body))
                    {
                        if let Some(msg) = validate_type_str(&name, "template") {
                            result.invalid_annotations.push(msg);
                        }
                        if let Some(b) = &bound {
                            record_backslash_keyword_types(
                                &mut result,
                                "template",
                                b,
                                tag.span.start,
                                text,
                            );
                            if let Some(msg) = validate_type_str(b, "template") {
                                result.invalid_annotations.push(msg);
                            }
                        }
                        result.templates.push((
                            name,
                            bound.map(|b| parse_type_string(&b)),
                            Variance::Invariant,
                            default.map(|d| parse_type_string(&d)),
                        ));
                    }
                }
                "template-covariant"
                | "psalm-template-covariant"
                | "phpstan-template-covariant" => {
                    if let Some((name, bound, default)) =
                        parse_template_line(tag.name.as_str(), body_text(&tag.body))
                    {
                        if let Some(msg) = validate_type_str(&name, "template-covariant") {
                            result.invalid_annotations.push(msg);
                        }
                        if let Some(b) = &bound {
                            record_backslash_keyword_types(
                                &mut result,
                                "template-covariant",
                                b,
                                tag.span.start,
                                text,
                            );
                            if let Some(msg) = validate_type_str(b, "template-covariant") {
                                result.invalid_annotations.push(msg);
                            }
                        }
                        result.templates.push((
                            name,
                            bound.map(|b| parse_type_string(&b)),
                            Variance::Covariant,
                            default.map(|d| parse_type_string(&d)),
                        ));
                    }
                }
                "template-contravariant"
                | "psalm-template-contravariant"
                | "phpstan-template-contravariant" => {
                    if let Some((name, bound, default)) =
                        parse_template_line(tag.name.as_str(), body_text(&tag.body))
                    {
                        if let Some(msg) = validate_type_str(&name, "template-contravariant") {
                            result.invalid_annotations.push(msg);
                        }
                        if let Some(b) = &bound {
                            record_backslash_keyword_types(
                                &mut result,
                                "template-contravariant",
                                b,
                                tag.span.start,
                                text,
                            );
                            if let Some(msg) = validate_type_str(b, "template-contravariant") {
                                result.invalid_annotations.push(msg);
                            }
                        }
                        result.templates.push((
                            name,
                            bound.map(|b| parse_type_string(&b)),
                            Variance::Contravariant,
                            default.map(|d| parse_type_string(&d)),
                        ));
                    }
                }
                "extends" | "template-extends" | "psalm-extends" | "phpstan-extends" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let trimmed = body_str.trim();
                        record_backslash_keyword_types(
                            &mut result,
                            "extends",
                            trimmed,
                            tag.span.start,
                            text,
                        );
                        if let Some(msg) = validate_type_str(trimmed, "extends") {
                            result.invalid_annotations.push(msg);
                        }
                        result.extends.push(parse_type_string(trimmed));
                    }
                }
                "implements"
                | "template-implements"
                | "psalm-implements"
                | "phpstan-implements" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let trimmed = body_str.trim();
                        record_backslash_keyword_types(
                            &mut result,
                            "implements",
                            trimmed,
                            tag.span.start,
                            text,
                        );
                        if let Some(msg) = validate_type_str(trimmed, "implements") {
                            result.invalid_annotations.push(msg);
                        }
                        result.implements.push(parse_type_string(trimmed));
                    }
                }
                "use" | "template-use" | "psalm-use" | "phpstan-use" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let trimmed = body_str.trim();
                        record_backslash_keyword_types(
                            &mut result,
                            "use",
                            trimmed,
                            tag.span.start,
                            text,
                        );
                        if let Some(msg) = validate_type_str(trimmed, "use") {
                            result.invalid_annotations.push(msg);
                        }
                        result.uses.push(parse_type_string(trimmed));
                    }
                }
                "assert" | "psalm-assert" | "phpstan-assert" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            let (ty, negated) = parse_assertion_type(&ty_str);
                            let (name, key) = split_array_key_suffix(&name);
                            result.assertions.push((name, key, ty, negated));
                        }
                    }
                }
                "if-this-is" | "psalm-if-this-is" | "phpstan-if-this-is" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let trimmed = body_str.trim();
                        if !trimmed.is_empty() {
                            result.if_this_is = Some(parse_type_string(trimmed));
                        }
                    }
                }
                "self-out" | "psalm-self-out" | "phpstan-self-out" | "this-out"
                | "psalm-this-out" | "phpstan-this-out" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let trimmed = body_str.trim();
                        if !trimmed.is_empty() {
                            result.self_out = Some(parse_self_out_type(trimmed));
                        }
                    }
                }
                "suppress" | "psalm-suppress" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        for rule in body_str.split([',', ' ']) {
                            let rule = rule.trim().to_string();
                            if !rule.is_empty() {
                                result.suppressed_issues.push(rule);
                            }
                        }
                    }
                }
                "see" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        result.see.push(body_str.to_string());
                    }
                }
                "link" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        result.see.push(body_str.to_string());
                    }
                }
                "mixin" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let base_class =
                            body_str.split('<').next().unwrap_or(&body_str).to_string();
                        result.mixins.push(base_class);
                    }
                }
                "property" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            record_backslash_keyword_types(
                                &mut result,
                                "property",
                                &ty_str,
                                tag.span.start,
                                text,
                            );
                            result.properties.push(DocProperty {
                                tag_span: tag_head_span(text, tag.span.start, tag.span.end),
                                type_hint: ty_str,
                                name: name.trim_start_matches('$').to_string(),
                                read_only: false,
                                write_only: false,
                            });
                        }
                    }
                }
                "property-read" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            record_backslash_keyword_types(
                                &mut result,
                                "property-read",
                                &ty_str,
                                tag.span.start,
                                text,
                            );
                            result.properties.push(DocProperty {
                                tag_span: tag_head_span(text, tag.span.start, tag.span.end),
                                type_hint: ty_str,
                                name: name.trim_start_matches('$').to_string(),
                                read_only: true,
                                write_only: false,
                            });
                        }
                    }
                }
                "property-write" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            record_backslash_keyword_types(
                                &mut result,
                                "property-write",
                                &ty_str,
                                tag.span.start,
                                text,
                            );
                            result.properties.push(DocProperty {
                                tag_span: tag_head_span(text, tag.span.start, tag.span.end),
                                type_hint: ty_str,
                                name: name.trim_start_matches('$').to_string(),
                                read_only: false,
                                write_only: true,
                            });
                        }
                    }
                }
                "method" | "psalm-method" => {
                    let body_str = body_text(&tag.body).unwrap_or_default().trim().to_string();
                    if let Some(err) = validate_method_body(&body_str) {
                        result.invalid_annotations.push(err);
                    } else if let Some(mut m) = parse_method_line(&body_str) {
                        m.tag_span = tag_head_span(text, tag.span.start, tag.span.end);
                        for p in &m.params {
                            record_backslash_keyword_types(
                                &mut result,
                                "method",
                                &p.type_hint,
                                tag.span.start,
                                text,
                            );
                        }
                        record_backslash_keyword_types(
                            &mut result,
                            "method",
                            &m.return_type,
                            tag.span.start,
                            text,
                        );
                        result.methods.push(m);
                    }
                }
                "psalm-type" | "phpstan-type" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((name, type_expr)) = split_type_alias_decl(&body_str) {
                            result.type_aliases.push(DocTypeAlias {
                                name: name.to_string(),
                                type_expr: type_expr.to_string(),
                            });
                        }
                    }
                }
                "psalm-import-type" | "phpstan-import-type" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some(import) = parse_import_type(&body_str) {
                            result.import_types.push(import);
                        }
                    }
                }
                "since" if result.since.is_none() => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let v = body_str.split_whitespace().next().unwrap_or("");
                        if !v.is_empty() {
                            result.since = Some(v.to_string());
                        }
                    }
                }
                "removed" if result.removed.is_none() => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let v = body_str.split_whitespace().next().unwrap_or("");
                        if !v.is_empty() {
                            result.removed = Some(v.to_string());
                        }
                    }
                }
                "internal" | "psalm-internal" => result.is_internal = true,
                "pure" | "psalm-pure" | "phpstan-pure" => result.is_pure = true,
                "impure" | "psalm-impure" | "phpstan-impure" => result.is_impure = true,
                "seal-properties" | "psalm-seal-properties" => result.seal_properties = true,
                "no-named-arguments" => result.no_named_arguments = true,
                "mutation-free" | "psalm-mutation-free" | "phpstan-mutation-free" => {
                    result.is_mutation_free = true
                }
                "external-mutation-free"
                | "psalm-external-mutation-free"
                | "phpstan-external-mutation-free" => result.is_external_mutation_free = true,
                "immutable" | "psalm-immutable" | "phpstan-immutable" => result.is_immutable = true,
                "readonly" | "psalm-readonly" | "phpstan-readonly" => result.is_readonly = true,
                "final" => result.is_final = true,
                "inheritDoc" | "inheritdoc" => result.is_inherit_doc = true,
                "api" | "psalm-api" => result.is_api = true,
                "assert-if-true" | "psalm-assert-if-true" | "phpstan-assert-if-true" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            let (ty, negated) = parse_assertion_type(&ty_str);
                            let (name, key) = split_array_key_suffix(&name);
                            result.assertions_if_true.push((name, key, ty, negated));
                        }
                    }
                }
                "assert-if-false" | "psalm-assert-if-false" | "phpstan-assert-if-false" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            let (ty, negated) = parse_assertion_type(&ty_str);
                            let (name, key) = split_array_key_suffix(&name);
                            result.assertions_if_false.push((name, key, ty, negated));
                        }
                    }
                }
                "psalm-property" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            result.properties.push(DocProperty {
                                tag_span: tag_head_span(text, tag.span.start, tag.span.end),
                                type_hint: ty_str,
                                name,
                                read_only: false,
                                write_only: false,
                            });
                        }
                    }
                }
                "psalm-property-read" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            result.properties.push(DocProperty {
                                tag_span: tag_head_span(text, tag.span.start, tag.span.end),
                                type_hint: ty_str,
                                name,
                                read_only: true,
                                write_only: false,
                            });
                        }
                    }
                }
                "psalm-property-write" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((ty_str, name)) = parse_param_line(&body_str) {
                            result.properties.push(DocProperty {
                                tag_span: tag_head_span(text, tag.span.start, tag.span.end),
                                type_hint: ty_str,
                                name,
                                read_only: false,
                                write_only: true,
                            });
                        }
                    }
                }
                "psalm-require-extends" | "phpstan-require-extends" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let cls = body_str
                            .split_whitespace()
                            .next()
                            .unwrap_or("")
                            .trim()
                            .to_string();
                        if !cls.is_empty() {
                            result.require_extends.push(cls);
                        }
                    }
                }
                "psalm-require-implements" | "phpstan-require-implements" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let cls = body_str
                            .split_whitespace()
                            .next()
                            .unwrap_or("")
                            .trim()
                            .to_string();
                        if !cls.is_empty() {
                            result.require_implements.push(cls);
                        }
                    }
                }
                "mir-check" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        if let Some((expr_part, type_part)) = body_str.split_once(" is ") {
                            // Kept verbatim (including any leading `$`) — the
                            // consumer parses this as a real PHP expression,
                            // not just a bare variable name.
                            let expr_text = expr_part.trim().to_string();
                            let type_string = type_part.trim().to_string();
                            if !expr_text.is_empty() && !type_string.is_empty() {
                                result.mir_checks.push((expr_text, type_string));
                            }
                        }
                    }
                }
                "dataProvider" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        let name = body_str
                            .trim()
                            .trim_end_matches("()")
                            .rsplit("::")
                            .next()
                            .unwrap_or("")
                            .trim();
                        if !name.is_empty() {
                            result.data_providers.push(name.to_string());
                        }
                    }
                }
                "trace" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        // Support both comma-separated and space-separated variable names
                        for part in body_str.split([',', ' ']) {
                            let var_name = part.trim().trim_start_matches('$').to_string();
                            if !var_name.is_empty() {
                                result.trace_vars.push(var_name);
                            }
                        }
                    }
                }
                "taint-sink" => {
                    if let Some(body_str) = body_text(&tag.body) {
                        // Format: `kind $param` or `kind $param1 $param2`
                        let mut tokens = body_str.split_whitespace();
                        if let Some(kind) = tokens.next() {
                            let kind = kind.to_string();
                            for param_token in tokens {
                                let param = param_token.trim_start_matches('$').to_string();
                                if !param.is_empty() {
                                    result.taint_sinks.push((param, kind.clone()));
                                }
                            }
                        }
                    }
                }
                "taint-source" => result.is_taint_source = true,
                _ => {}
            }
        }

        if text.to_ascii_lowercase().contains("{@inheritdoc}") {
            result.is_inherit_doc = true;
        }

        result
    }
}

/// Record the backslash-qualified non-class (`\int`) members of a docblock
/// type spelling in `result` — the collector turns each into an
/// `InvalidDocblockType` warning anchored at the token's own source line.
fn record_backslash_keyword_types(
    result: &mut ParsedDocblock,
    tag_name: &str,
    ty: &str,
    tag_start: u32,
    text: &str,
) {
    for token in backslash_keyword_members(ty) {
        let (line, col) = token_line_col(text, tag_start, ty, &token);
        result.backslash_keyword_types.push(BackslashKeywordType {
            tag: tag_name.to_string(),
            token,
            line,
            col,
        });
    }
}

/// 0-based `(line, col)` of `token` within the docblock `text`. `token` is a
/// verbatim member of `ty`, which sits in the tag body that follows `@tag`;
/// it is located relative to `tag_start` (the tag's offset in the text).
/// Falls back to the tag start when the token cannot be found.
fn token_line_col(text: &str, tag_start: u32, ty: &str, token: &str) -> (u32, u32) {
    let start = (tag_start as usize).min(text.len());
    let mut abs = start;
    if let Some(rest) = text.get(abs..) {
        if let Some(ty_off) = rest.find(ty) {
            if let Some(ty_text) = text.get(abs + ty_off..) {
                if let Some(tok_off) = ty_text.find(token) {
                    abs = (abs + ty_off + tok_off).min(text.len());
                }
            }
        }
    }
    let mut line = 0u32;
    let mut line_start = 0usize;
    for (i, b) in text.bytes().enumerate() {
        if i >= abs {
            break;
        }
        if b == b'\n' {
            line += 1;
            line_start = i + 1;
        }
    }
    (line, (abs - line_start) as u32)
}

/// `self<T>`/`static<T>`/`parent<T>`/`$this<T>` written in a self-out
/// annotation (`@psalm-self-out`, `@phpstan-self-out`, `@psalm-this-out`)
/// needs its `<T>` kept intact for method-level template substitution (e.g.
/// `@psalm-self-out self<U>` on a method with its own `@template U`) —
/// unlike an ordinary `@return self<T>`, where the shared type parser
/// deliberately drops the args and reattaches the receiver's own params
/// instead (see `parse_generic`'s `self`/`static`/`parent` arms). Parsed
/// here as a `TNamedObject` sentinel whose `fqcn` is the literal keyword
/// (never a real PHP class name), which `substitute_static_atom` recognizes
/// and resolves to the actual receiver class at call time while keeping
/// `type_params` intact for the caller's later template substitution.
fn parse_self_out_type(trimmed: &str) -> Type {
    for keyword in ["self", "static", "parent", "$this"] {
        let Some(rest) = strip_ascii_ci_prefix(trimmed, keyword) else {
            continue;
        };
        let rest = rest.trim_start();
        if let Some(inner) = rest.strip_prefix('<').and_then(|s| s.strip_suffix('>')) {
            let params: Vec<Type> = split_generics(inner)
                .iter()
                .map(|p| parse_type_string(p.trim()))
                .collect();
            return Type::single(Atomic::TNamedObject {
                fqcn: mir_types::Name::from(keyword),
                type_params: mir_types::union::vec_to_type_params(params),
            });
        }
    }
    parse_type_string(trimmed)
}

// ---------------------------------------------------------------------------
// ParsedDocblock support types
// ---------------------------------------------------------------------------

/// Byte range of a tag's first line, without trailing whitespace or the
/// docblock's closing `*/`.
fn tag_head_span(text: &str, start: impl TryInto<usize>, end: impl TryInto<usize>) -> (u32, u32) {
    let (Ok(start), Ok(end)) = (start.try_into(), end.try_into()) else {
        return (0, 0);
    };
    let end = end.min(text.len());
    let Some(raw) = text.get(start..end) else {
        return (start as u32, end as u32);
    };
    let head = raw.lines().next().unwrap_or(raw);
    let head = head.trim_end().trim_end_matches("*/").trim_end();
    (start as u32, (start + head.len()) as u32)
}

#[derive(Debug, Default, Clone)]
pub struct DocProperty {
    pub type_hint: String,
    pub name: String,     // without leading $
    pub read_only: bool,  // true for @property-read
    pub write_only: bool, // true for @property-write
    /// Byte range of the tag within the docblock text.
    pub tag_span: (u32, u32),
}

#[derive(Debug, Default, Clone)]
pub struct DocMethod {
    pub return_type: String,
    pub name: String,
    pub is_static: bool,
    pub params: Vec<DocMethodParam>,
    /// Byte range of the tag within the docblock text.
    pub tag_span: (u32, u32),
}

#[derive(Debug, Default, Clone)]
pub struct DocMethodParam {
    pub name: String,
    pub type_hint: String,
    pub is_variadic: bool,
    pub is_byref: bool,
    pub is_optional: bool,
}

#[derive(Debug, Default, Clone)]
pub struct DocTypeAlias {
    pub name: String,
    pub type_expr: String,
}

/// A docblock type token written with a leading `\` (the fully-qualified
/// class qualifier) on a non-class type keyword (`\int`, `?\string`, …).
/// `line`/`col` are 0-based positions within the docblock text, so the
/// collector can anchor the warning at the token's own source line.
#[derive(Debug, Clone)]
pub struct BackslashKeywordType {
    /// The annotation tag carrying the type (`param`, `return`, `var`, …).
    pub tag: String,
    /// The offending type token, verbatim with its leading backslash.
    pub token: String,
    /// Line of the token within the docblock text.
    pub line: u32,
    /// Column of the token within its docblock-text line.
    pub col: u32,
}

#[derive(Debug, Default, Clone)]
pub struct DocImportType {
    /// The name exported by the source class (the original alias name).
    pub original: String,
    /// The local name to use in this class (`as LocalAlias`); defaults to `original`.
    pub local: String,
    /// The FQCN of the class to import the type from.
    pub from_class: String,
}

// ---------------------------------------------------------------------------
// ParsedDocblock
// ---------------------------------------------------------------------------

#[derive(Debug, Default, Clone)]
pub struct ParsedDocblock {
    /// `@param Type $name`
    pub params: Vec<(String, Type)>,
    /// Raw `@param` type spellings keyed by parameter name. Kept so later
    /// analysis can distinguish ambiguous pseudo-type aliases like
    /// `integer`/`boolean` from a same-named class that should take
    /// precedence once the full symbol table is available.
    pub param_type_strings: Vec<(String, String)>,
    /// `@param-out Type $name` / `@psalm-param-out Type $name` — the type written
    /// back to the caller's by-ref argument after the call.
    pub out_params: Vec<(String, Type)>,
    /// `@return Type`
    pub return_type: Option<Type>,
    /// `@var Type` or `@var Type $name` — type and optional variable name (last tag wins)
    pub var_type: Option<Type>,
    /// Optional variable name from `@var Type $name`
    pub var_name: Option<String>,
    /// Every `@var` tag in docblock order, for statement-level annotations.
    pub var_tags: Vec<(Option<String>, Type)>,
    /// `@template T` / `@template T of Bound` / `@template-covariant T` / `@template-contravariant T`
    /// The last element is the optional `@template T = Default` default type.
    pub templates: Vec<(String, Option<Type>, Variance, Option<Type>)>,
    /// `@extends ClassName<T>` — a class has at most one entry (its single
    /// parent); an interface may have several, one per base interface named
    /// in its native `extends A, B` clause.
    pub extends: Vec<Type>,
    /// `@implements InterfaceName<T>`
    pub implements: Vec<Type>,
    /// `@use TraitName<T>` — explicit type argument(s) for a `use`d trait's
    /// own `@template`, mirroring `@implements` for interfaces.
    pub uses: Vec<Type>,
    /// `@throws ClassName`
    pub throws: Vec<String>,
    /// `(parameter name, prose)` for each `@param` that has a description.
    pub param_descriptions: Vec<(String, String)>,
    /// Prose after the type of `@return`.
    pub return_description: Option<String>,
    /// `(class as written, prose)` for each `@throws` that has a description.
    pub throws_descriptions: Vec<(String, String)>,
    /// `@psalm-assert Type $var` — the `bool` is true for the `!Type` negated
    /// form; the key path is non-empty when the target is a specific
    /// (possibly nested) array key of `$var` (`@psalm-assert Type
    /// $var['a']['b']`) rather than the whole variable.
    pub assertions: Vec<(String, Vec<ArrayKey>, Type, bool)>,
    /// `@psalm-assert-if-true Type $var`
    pub assertions_if_true: Vec<(String, Vec<ArrayKey>, Type, bool)>,
    /// `@psalm-assert-if-false Type $var`
    pub assertions_if_false: Vec<(String, Vec<ArrayKey>, Type, bool)>,
    /// `@psalm-suppress IssueName`
    pub suppressed_issues: Vec<String>,
    pub is_deprecated: bool,
    pub is_internal: bool,
    pub is_pure: bool,
    /// An explicit `@impure`/`@psalm-impure`/`@phpstan-impure` declaration.
    /// This opts an implementation out of an inherited purity contract.
    pub is_impure: bool,
    pub is_mutation_free: bool,
    pub is_external_mutation_free: bool,
    pub no_named_arguments: bool,
    pub is_immutable: bool,
    pub is_readonly: bool,
    pub is_api: bool,
    /// `@final` — class should be treated as final even without the PHP `final` keyword.
    pub is_final: bool,
    /// `@inheritDoc` or `{@inheritDoc}` was present — documentation should be
    /// inherited from the nearest ancestor that has a real docblock.
    pub is_inherit_doc: bool,
    /// Free text before first `@` tag — used for hover display
    pub description: String,
    /// `@deprecated message` — Some(message) or Some("") if no message
    pub deprecated: Option<String>,
    /// `@see ClassName` / `@link URL`
    pub see: Vec<String>,
    /// `@mixin ClassName`
    pub mixins: Vec<String>,
    /// `@property`, `@property-read`, `@property-write`
    pub properties: Vec<DocProperty>,
    /// `@method [static] ReturnType name([params])`
    pub methods: Vec<DocMethod>,
    /// `@psalm-type Alias = TypeExpr` / `@phpstan-type Alias = TypeExpr`
    pub type_aliases: Vec<DocTypeAlias>,
    /// `@psalm-import-type Alias from SourceClass` / `@phpstan-import-type ...`
    pub import_types: Vec<DocImportType>,
    /// `@psalm-require-extends ClassName` / `@phpstan-require-extends ClassName`
    pub require_extends: Vec<String>,
    /// `@psalm-require-implements InterfaceName` / `@phpstan-require-implements InterfaceName`
    pub require_implements: Vec<String>,
    /// `@since X.Y` — first PHP version this symbol exists in.
    pub since: Option<String>,
    /// `@removed X.Y` — first PHP version this symbol no longer exists in.
    pub removed: Option<String>,
    /// Malformed type annotations detected during parsing.
    pub invalid_annotations: Vec<String>,
    /// Docblock type spellings that use a leading `\` (the fully-qualified
    /// class qualifier) on a non-class type keyword (`\int`, `?\string`, …);
    /// the collector emits an `InvalidDocblockType` warning for each, anchored
    /// at the token's own line.
    pub backslash_keyword_types: Vec<BackslashKeywordType>,
    /// `@mir-check EXPR is TYPE` — (expr_text, type_string). `expr_text` is
    /// kept verbatim (e.g. `$h->status`, `self::$prop`, `$arr['key']`) and
    /// parsed as a real PHP expression by the consumer, not just a bare
    /// variable name.
    pub mir_checks: Vec<(String, String)>,
    /// `@trace $var1, $var2` or `@trace $var1 $var2` — variable names to trace
    pub trace_vars: Vec<String>,
    /// `@taint-sink <kind> $param` — (param_name_without_dollar, sink_kind_string)
    pub taint_sinks: Vec<(String, String)>,
    /// `@taint-source` — this function/method's return value is tainted.
    pub is_taint_source: bool,
    /// `@seal-properties` / `@psalm-seal-properties` — disallows undeclared property access.
    pub seal_properties: bool,
    /// `@if-this-is Type` / `@psalm-if-this-is Type` — the method may only be
    /// called when `$this` satisfies this type. Stored as the raw parsed type;
    /// class names are resolved later by the collector.
    pub if_this_is: Option<Type>,
    /// `@psalm-self-out Type` / `@phpstan-self-out Type` — the receiver's type
    /// after this call returns. Stored as the raw parsed type; class names
    /// (and `self`/`static`) are resolved later by the collector.
    pub self_out: Option<Type>,
    /// `@dataProvider methodName` (PHPUnit) — name of the method that supplies
    /// this test's data, invoked by PHPUnit via reflection rather than a call.
    pub data_providers: Vec<String>,
}

impl ParsedDocblock {
    /// Returns the type for a given parameter name (strips leading `$`).
    ///
    /// Uses the **last** match so that `@psalm-param` / `@phpstan-param` (which
    /// php-rs-parser maps to the same `Param` variant as `@param`) overrides a
    /// preceding plain `@param` annotation.
    pub fn get_param_type(&self, name: &str) -> Option<&Type> {
        let name = name.trim_start_matches('$');
        self.params
            .iter()
            .rfind(|(n, _)| n.trim_start_matches('$') == name)
            .map(|(_, ty)| ty)
    }

    /// Returns the raw `@param` type spelling for a given parameter name.
    /// Uses the **last** match, mirroring [`Self::get_param_type`].
    pub fn get_param_type_string(&self, name: &str) -> Option<&str> {
        let name = name.trim_start_matches('$');
        self.param_type_strings
            .iter()
            .rfind(|(n, _)| n.trim_start_matches('$') == name)
            .map(|(_, ty)| ty.as_str())
    }

    /// Returns the `@param-out` / `@psalm-param-out` type for a given parameter
    /// name, if declared. Uses the **last** match.
    pub fn get_out_param_type(&self, name: &str) -> Option<&Type> {
        let name = name.trim_start_matches('$');
        self.out_params
            .iter()
            .rfind(|(n, _)| n.trim_start_matches('$') == name)
            .map(|(_, ty)| ty)
    }
}

// ---------------------------------------------------------------------------
// Type string parser
// ---------------------------------------------------------------------------

#[cfg(test)]
mod tests;
/// Parse a PHPDoc type expression string into a `Type`.
/// Handles: `string`, `int|null`, `array<string>`, `list<int>`,
/// `ClassName`, `?string` (nullable), `string[]` (array shorthand).
mod types;
mod validate;

pub(crate) use types::SelfIntConstantsGuard;
use types::*;
use validate::*;

pub(crate) use types::is_docblock_type_keyword;
pub(crate) use types::parse_type_string;