aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_hir/src/diagnostics.rs
blob: 562f3fe5c48fec1064aa8c10bd3f204f77718162 (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
//! FIXME: write short doc here
pub use hir_def::diagnostics::UnresolvedModule;
pub use hir_expand::diagnostics::{AstDiagnostic, Diagnostic, DiagnosticSink};
pub use hir_ty::diagnostics::{MissingFields, MissingMatchArms, MissingOkInTailExpr, NoSuchField};

use std::sync::Arc;

use crate::code_model::Function;
use crate::db::HirDatabase;
use crate::has_source::HasSource;
use hir_ty::{
    diagnostics::{MissingUnsafe, UnnecessaryUnsafe},
    expr::unsafe_expressions,
    InferenceResult,
};
use ra_syntax::AstPtr;

pub struct UnsafeValidator<'a, 'b: 'a> {
    func: &'a Function,
    infer: Arc<InferenceResult>,
    sink: &'a mut DiagnosticSink<'b>,
}

impl<'a, 'b> UnsafeValidator<'a, 'b> {
    pub fn new(
        func: &'a Function,
        infer: Arc<InferenceResult>,
        sink: &'a mut DiagnosticSink<'b>,
    ) -> UnsafeValidator<'a, 'b> {
        UnsafeValidator { func, infer, sink }
    }

    pub fn validate_body(&mut self, db: &dyn HirDatabase) {
        let def = self.func.id.into();
        let unsafe_expressions = unsafe_expressions(db, self.infer.as_ref(), def);
        let func_data = db.function_data(self.func.id);
        let unnecessary = func_data.is_unsafe && unsafe_expressions.len() == 0;
        let missing = !func_data.is_unsafe && unsafe_expressions.len() > 0;
        if !(unnecessary || missing) {
            return;
        }

        let in_file = self.func.source(db);
        let file = in_file.file_id;
        let fn_def = AstPtr::new(&in_file.value);
        let fn_name = func_data.name.clone().into();

        if unnecessary {
            self.sink.push(UnnecessaryUnsafe { file, fn_def, fn_name })
        } else {
            self.sink.push(MissingUnsafe { file, fn_def, fn_name })
        }
    }
}