php-lsp 0.25.4

use std::sync::Arc;

use dashmap::DashMap;
use tower_lsp_server::ls_types::{Diagnostic, Uri};

use crate::analysis::semantic_diagnostics::issues_to_diagnostics_gated;
use crate::document::ast::ParsedDoc;
use crate::document::document_store::DocumentStore;
use crate::lang::config::DiagnosticsConfig;

/// Per-open-file state owned by `Backend` (Phase E4).
///
/// Previously this lived inside `DocumentStore`'s `map: DashMap<Uri, Document>`,
/// but none of these fields are salsa-shaped: `text` is the live editor buffer,
/// `version` is an async-parse gate, and `parse_diagnostics` is a publish cache.
/// Keeping them on `Backend` leaves `DocumentStore` as a pure salsa-input wrapper.
#[derive(Default, Clone)]
pub(crate) struct OpenFile {
    /// Live editor text. `Arc`-backed so readers (`OpenFiles::text`, hit on
    /// every request that needs the current buffer) clone a refcount instead
    /// of the whole buffer.
    pub(crate) text: Arc<str>,
    /// Monotonic counter bumped on every `set_open_text` / `close_open_file`;
    /// used to discard stale async parse results.
    pub(crate) version: u64,
    /// Parse-level diagnostics most recently cached for publication.
    pub(crate) parse_diagnostics: Vec<Diagnostic>,
    /// Content hash of the last `publishDiagnostics` sent for this file.
    /// Lets the dependent-republish sweep skip publishes whose content the
    /// client already displays. `None` until the first publish; the entry
    /// (and hash) drops on close, so a reopen always publishes.
    pub(crate) published_hash: Option<u64>,
    /// Most recent external-tool (PHPStan/PHPCS) findings, cached so every
    /// publish path can append them without re-running the tools.
    pub(crate) external_diagnostics: Vec<Diagnostic>,
    /// `version` this file was at when `external_diagnostics` was computed.
    /// An edit bumps `version` without touching this field, so a mismatch
    /// means the cached findings were computed against text the buffer no
    /// longer holds — [`OpenFiles::external_diagnostics`] treats that as
    /// stale and returns nothing rather than showing diagnostics against the
    /// wrong line numbers.
    pub(crate) external_diagnostics_version: Option<u64>,
}

/// Shared handle to open-file state. Cheaply cloneable — wraps an `Arc<DashMap>`
/// so it can be captured by async closures alongside `Arc<DocumentStore>`.
#[derive(Clone, Default)]
pub struct OpenFiles(Arc<DashMap<Uri, OpenFile>>);

impl OpenFiles {
    pub(crate) fn new() -> Self {
        Self::default()
    }

    pub(crate) fn set_open_text(&self, docs: &DocumentStore, uri: Uri, text: String) -> u64 {
        // Build the Arc once and hand the same allocation to both the salsa
        // mirror and the open-file entry — `mirror_text_arc` skips the extra
        // `Arc::from` copy `mirror_text` would otherwise make from `&text`.
        let text: Arc<str> = Arc::from(text);
        docs.mirror_text_arc(&uri, Arc::clone(&text));
        let mut entry = self.0.entry(uri).or_default();
        entry.version += 1;
        entry.text = text;
        entry.version
    }

    pub(crate) fn close(&self, docs: &DocumentStore, uri: &Uri) {
        self.0.remove(uri);
        docs.evict_token_cache(uri);
    }

    pub(crate) fn current_version(&self, uri: &Uri) -> Option<u64> {
        self.0.get(uri).map(|e| e.version)
    }

    pub(crate) fn text(&self, uri: &Uri) -> Option<Arc<str>> {
        self.0.get(uri).map(|e| Arc::clone(&e.text))
    }

    pub(crate) fn set_parse_diagnostics(&self, uri: &Uri, diagnostics: Vec<Diagnostic>) {
        if let Some(mut entry) = self.0.get_mut(uri) {
            entry.parse_diagnostics = diagnostics;
        }
    }

    pub(crate) fn parse_diagnostics(&self, uri: &Uri) -> Option<Vec<Diagnostic>> {
        self.0.get(uri).map(|e| e.parse_diagnostics.clone())
    }

    /// Record the content hash of a `publishDiagnostics` just sent for `uri`.
    /// No-op when the file closed mid-flight (entry gone).
    pub(crate) fn note_published(&self, uri: &Uri, hash: u64) {
        if let Some(mut entry) = self.0.get_mut(uri) {
            entry.published_hash = Some(hash);
        }
    }

    /// Hash of the last publish sent for `uri`, if it is still open.
    pub(crate) fn published_hash(&self, uri: &Uri) -> Option<u64> {
        self.0.get(uri).and_then(|e| e.published_hash)
    }

    /// Cache `diagnostics` from an external tool run against `uri` at
    /// `version`. No-op when the file closed mid-run (entry gone).
    pub(crate) fn set_external_diagnostics(
        &self,
        uri: &Uri,
        version: u64,
        diagnostics: Vec<Diagnostic>,
    ) {
        if let Some(mut entry) = self.0.get_mut(uri) {
            entry.external_diagnostics = diagnostics;
            entry.external_diagnostics_version = Some(version);
        }
    }

    /// Cached external-tool diagnostics for `uri`, or empty if none have
    /// been computed yet or the buffer has changed since they were.
    pub(crate) fn external_diagnostics(&self, uri: &Uri) -> Vec<Diagnostic> {
        self.0
            .get(uri)
            .filter(|e| e.external_diagnostics_version == Some(e.version))
            .map(|e| e.external_diagnostics.clone())
            .unwrap_or_default()
    }

    pub(crate) fn all_with_diagnostics(&self) -> Vec<(Uri, Vec<Diagnostic>, Option<i64>)> {
        self.0
            .iter()
            .map(|e| {
                (
                    e.key().clone(),
                    e.value().parse_diagnostics.clone(),
                    Some(e.value().version as i64),
                )
            })
            .collect()
    }

    pub(crate) fn urls(&self) -> Vec<Uri> {
        self.0.iter().map(|e| e.key().clone()).collect()
    }

    pub(crate) fn contains(&self, uri: &Uri) -> bool {
        self.0.contains_key(uri)
    }

    /// Open-gated parsed doc: returns `Some` only when `uri` is currently open.
    pub(crate) fn get_doc(&self, docs: &DocumentStore, uri: &Uri) -> Option<Arc<ParsedDoc>> {
        if !self.contains(uri) {
            return None;
        }
        docs.get_doc_salsa(uri)
    }

    /// Open-gated, stale-tolerant parsed doc for cursor-triggered cosmetic reads
    /// (`documentHighlight`). Never spins under write pressure — see
    /// [`DocumentStore::get_doc_snapshot_or_stale`].
    pub(crate) fn get_doc_stale(&self, docs: &DocumentStore, uri: &Uri) -> Option<Arc<ParsedDoc>> {
        if !self.contains(uri) {
            return None;
        }
        docs.get_doc_snapshot_or_stale(uri)
    }
}

/// Build the full diagnostic bundle for an already-open file.
///
/// Reuses cached parse diagnostics from `OpenFiles` (set by the file's own
/// debounced parse) and recomputes the rest. `semantic_issues` is
/// salsa-cached; for files unaffected by the triggering change it's a cache
/// hit.
///
/// Before the initial workspace scan finishes (`!docs.is_index_ready()`),
/// symbol lookups against the workspace-wide table only see the subset of
/// files indexed so far, so issues that depend on such a lookup (e.g.
/// `UndefinedClass` for a global class declared in a file not yet reached
/// by the scan) can misreport a real declaration as missing — see issue
/// #242. Those specific kinds are held back until the index is ready (the
/// caller republishes once it is, see `republish_after_index_ready`); issues
/// checked purely against the file's own contents are unaffected and always
/// reported immediately.
pub(crate) fn compute_open_file_diagnostics(
    docs: &DocumentStore,
    open_files: &OpenFiles,
    uri: &Uri,
    diag_cfg: &DiagnosticsConfig,
    is_laravel: bool,
) -> Vec<Diagnostic> {
    let mut out = open_files.parse_diagnostics(uri).unwrap_or_default();
    if let Some(issues) = docs.get_semantic_issues_salsa(uri) {
        out.extend(issues_to_diagnostics_gated(
            &issues,
            uri,
            diag_cfg,
            docs.is_index_ready(),
        ));
    }
    out.extend(crate::laravel::unguarded_model_diagnostics(
        docs, uri, is_laravel,
    ));
    out.extend(open_files.external_diagnostics(uri));
    out
}

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

    /// `OpenFiles::text` must hand out the same `Arc` allocation on repeat
    /// reads (a refcount bump), not a fresh clone of the buffer — the whole
    /// point of storing `Arc<str>` instead of `String` in `OpenFile`.
    #[test]
    fn text_reuses_the_same_arc_across_reads() {
        let docs = DocumentStore::new();
        let open_files = OpenFiles::new();
        let uri = ("file:///test.php").parse::<Uri>().unwrap();
        open_files.set_open_text(&docs, uri.clone(), "<?php".to_owned());

        let first = open_files.text(&uri).unwrap();
        let second = open_files.text(&uri).unwrap();
        assert!(
            Arc::ptr_eq(&first, &second),
            "repeated reads must share one allocation, not clone the buffer per call"
        );
    }

    /// Issue #242: before the workspace index is ready, a reference to a
    /// class the store hasn't seen yet is indistinguishable from a genuine
    /// typo — `UndefinedClass` must be held back either way. Once
    /// `mark_index_ready` flips, the same (still-unresolvable) reference must
    /// be reported, so this isn't just permanently swallowed.
    #[test]
    fn undefined_class_suppressed_until_index_ready() {
        let docs = DocumentStore::new();
        let open_files = OpenFiles::new();
        let uri = ("file:///app.php").parse::<Uri>().unwrap();
        open_files.set_open_text(
            &docs,
            uri.clone(),
            "<?php\n\nnew TrulyMissingClass();\n".to_owned(),
        );
        let diag_cfg = DiagnosticsConfig::default();

        let before = compute_open_file_diagnostics(&docs, &open_files, &uri, &diag_cfg, false);
        assert!(
            before.is_empty(),
            "UndefinedClass must be suppressed before the index is ready, got: {before:?}"
        );

        docs.mark_index_ready();
        let after = compute_open_file_diagnostics(&docs, &open_files, &uri, &diag_cfg, false);
        assert!(
            after
                .iter()
                .any(|d| d.message.contains("TrulyMissingClass")),
            "UndefinedClass must be reported once the index is ready, got: {after:?}"
        );
    }

    /// A diagnostic checked purely against the file's own contents (a
    /// same-function return-type mismatch) must never wait on the workspace
    /// index — only symbol lookups against the workspace-wide table are
    /// index-ready-gated (see `issue_needs_workspace_index`).
    #[test]
    fn local_only_issue_not_suppressed_before_index_ready() {
        let docs = DocumentStore::new();
        let open_files = OpenFiles::new();
        let uri = ("file:///app.php").parse::<Uri>().unwrap();
        open_files.set_open_text(
            &docs,
            uri.clone(),
            "<?php\nfunction f(): string { return 42; }\n".to_owned(),
        );
        let diag_cfg = DiagnosticsConfig::default();

        let before = compute_open_file_diagnostics(&docs, &open_files, &uri, &diag_cfg, false);
        assert!(
            !before.is_empty(),
            "a same-file return-type mismatch must not wait on the workspace index"
        );
    }
}