php-lsp 0.5.0

A PHP Language Server Protocol implementation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

//! `method_returns` salsa query — derives the per-file map of
//! `class_name -> method_name -> return_class_name`. Depends on `parsed_doc`.
//!
//! This is the sole cache for method-return inference since Phase E3 removed
//! the `OnceLock<MethodReturnsMap>` from `ParsedDoc`. Production call sites
//! (inlay_hints, type_definition, hover, completion) fetch the memoized
//! `Arc<MethodReturnsMap>` via `DocumentStore::get_method_returns_salsa` /
//! `other_docs_with_returns` and pass it into the `TypeMap` constructors.

use std::sync::Arc;

use salsa::Database;

use crate::ast::MethodReturnsMap;
use crate::db::input::SourceFile;
use crate::db::parse::parsed_doc;
use crate::type_map::build_method_returns;

#[derive(Clone)]
pub struct MethodReturnsArc(pub Arc<MethodReturnsMap>);

impl MethodReturnsArc {
    pub fn get(&self) -> &MethodReturnsMap {
        &self.0
    }
}

// SAFETY: identical contract to `ParsedArc::maybe_update`.
crate::impl_arc_update!(MethodReturnsArc);

#[salsa::tracked(no_eq)]
pub fn method_returns(db: &dyn Database, file: SourceFile) -> MethodReturnsArc {
    let doc = parsed_doc(db, file);
    MethodReturnsArc(Arc::new(build_method_returns(doc.get())))
}

#[cfg(test)]
mod tests {
    use std::sync::Arc;

    use super::*;
    use crate::db::analysis::AnalysisHost;
    use crate::db::input::{FileId, SourceFile};
    use salsa::Setter;

    #[test]
    fn method_returns_captures_factory_return() {
        let host = AnalysisHost::new();
        let file = SourceFile::new(
            host.db(),
            FileId(0),
            Arc::<str>::from("file:///t.php"),
            Arc::<str>::from(
                "<?php\nclass Foo {\n    public function make(): Bar { return new Bar(); }\n}\nclass Bar {}",
            ),
            None,
        );
        let m = method_returns(host.db(), file);
        let foo = m.get().get("Foo").expect("class Foo in map");
        assert_eq!(foo.get("make").map(String::as_str), Some("Bar"));
    }

    #[test]
    fn method_returns_reruns_after_edit() {
        let mut host = AnalysisHost::new();
        let file = SourceFile::new(
            host.db(),
            FileId(1),
            Arc::<str>::from("file:///t.php"),
            Arc::<str>::from(
                "<?php\nclass F {\n    public function m(): A { return new A(); }\n}\nclass A {}\nclass B {}",
            ),
            None,
        );
        let a1 = method_returns(host.db(), file);
        let first = Arc::as_ptr(&a1.0);

        file.set_text(host.db_mut()).to(Arc::<str>::from(
            "<?php\nclass F {\n    public function m(): B { return new B(); }\n}\nclass A {}\nclass B {}",
        ));
        let a2 = method_returns(host.db(), file);
        assert_ne!(
            first,
            Arc::as_ptr(&a2.0),
            "editing the source should produce a new map"
        );
        assert_eq!(
            a2.get()
                .get("F")
                .and_then(|m| m.get("m"))
                .map(String::as_str),
            Some("B")
        );
    }
}