aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_hir/src/query_definitions.rs
blob: bf9ac0dfb42457dcc6c0c32b839441ca878c263b (plain)
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
use std::{
    sync::Arc,
    time::Instant,
};

use rustc_hash::FxHashMap;
use ra_syntax::{SyntaxNode, TreeArc};
use ra_db::{CrateId};

use crate::{
    SourceFileItems, SourceItemId, HirFileId,
    Function, FnScopes, Module,
    db::HirDatabase,
    nameres::{ItemMap, Resolver},
};

pub(super) fn fn_scopes(db: &impl HirDatabase, func: Function) -> Arc<FnScopes> {
    let body = db.body_hir(func);
    let res = FnScopes::new(body);
    Arc::new(res)
}

pub(super) fn file_items(db: &impl HirDatabase, file_id: HirFileId) -> Arc<SourceFileItems> {
    let source_file = db.hir_parse(file_id);
    let res = SourceFileItems::new(file_id, &source_file);
    Arc::new(res)
}

pub(super) fn file_item(
    db: &impl HirDatabase,
    source_item_id: SourceItemId,
) -> TreeArc<SyntaxNode> {
    let source_file = db.hir_parse(source_item_id.file_id);
    db.file_items(source_item_id.file_id)[source_item_id.item_id]
        .to_node(&source_file)
        .to_owned()
}

pub(super) fn item_map(db: &impl HirDatabase, crate_id: CrateId) -> Arc<ItemMap> {
    let start = Instant::now();
    let module_tree = db.module_tree(crate_id);
    let input = module_tree
        .modules()
        .map(|module_id| {
            (
                module_id,
                db.lower_module_module(Module {
                    krate: crate_id,
                    module_id,
                }),
            )
        })
        .collect::<FxHashMap<_, _>>();

    let resolver = Resolver::new(db, &input, crate_id);
    let res = resolver.resolve();
    let elapsed = start.elapsed();
    log::info!("item_map: {:?}", elapsed);
    Arc::new(res)
}