grit_pattern_matcher/pattern/
add.rs

1use super::{
2    patterns::{Matcher, Pattern, PatternName},
3    resolved_pattern::ResolvedPattern,
4    state::State,
5};
6use crate::{
7    constant::Constant,
8    context::{ExecContext, QueryContext},
9};
10use grit_util::{error::GritResult, AnalysisLogs};
11
12#[derive(Debug, Clone)]
13pub struct Add<Q: QueryContext> {
14    pub lhs: Pattern<Q>,
15    pub rhs: Pattern<Q>,
16}
17
18impl<Q: QueryContext> Add<Q> {
19    pub fn new(lhs: Pattern<Q>, rhs: Pattern<Q>) -> Self {
20        Self { lhs, rhs }
21    }
22
23    pub fn call<'a>(
24        &'a self,
25        state: &mut State<'a, Q>,
26        context: &'a Q::ExecContext<'a>,
27        logs: &mut AnalysisLogs,
28    ) -> GritResult<Q::ResolvedPattern<'a>> {
29        let res = self.evaluate(state, context, logs)?;
30        Ok(Q::ResolvedPattern::from_constant(Constant::Float(res)))
31    }
32
33    fn evaluate<'a>(
34        &'a self,
35        state: &mut State<'a, Q>,
36        context: &'a Q::ExecContext<'a>,
37        logs: &mut AnalysisLogs,
38    ) -> GritResult<f64> {
39        let lhs = self.lhs.float(state, context, logs)?;
40        let rhs = self.rhs.float(state, context, logs)?;
41        let res = lhs + rhs;
42        Ok(res)
43    }
44}
45
46impl<Q: QueryContext> PatternName for Add<Q> {
47    fn name(&self) -> &'static str {
48        "ADD"
49    }
50}
51
52impl<Q: QueryContext> Matcher<Q> for Add<Q> {
53    fn execute<'a>(
54        &'a self,
55        binding: &Q::ResolvedPattern<'a>,
56        state: &mut State<'a, Q>,
57        context: &'a Q::ExecContext<'a>,
58        logs: &mut AnalysisLogs,
59    ) -> GritResult<bool> {
60        let binding_text = binding.text(&state.files, context.language())?;
61        let binding_int = binding_text.parse::<f64>()?;
62        let target = self.evaluate(state, context, logs)?;
63        Ok(binding_int == target)
64    }
65}
grit_pattern_matcher/pattern/add.rs

grit_pattern_matcher/pattern/
add.rs
