sqruff_lib/utils/functional/
context.rs

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
use sqruff_lib_core::utils::functional::segments::Segments;

use crate::core::rules::context::RuleContext;

pub struct FunctionalContext<'a> {
    context: RuleContext<'a>,
}

impl<'a> FunctionalContext<'a> {
    pub fn new(context: RuleContext<'a>) -> Self {
        FunctionalContext { context }
    }

    pub fn segment(&self) -> Segments {
        Segments::new(
            self.context.segment.clone(),
            self.context.templated_file.clone(),
        )
    }

    pub fn siblings_post(&self) -> Segments {
        Segments::from_vec(
            self.context.siblings_post(),
            self.context.templated_file.clone(),
        )
    }

    pub fn parent_stack(&self) -> Segments {
        Segments::from_vec(
            self.context.parent_stack.clone(),
            self.context.templated_file.clone(),
        )
    }
}