rune/hir/
ctxt.rs

Help
1use tracing::instrument_ast;
2
3use crate::alloc;
4use crate::alloc::prelude::*;
5use crate::ast::{self, Spanned};
6use crate::compile::{meta, DynLocation, Error, ItemId, Result};
7use crate::grammar::{Ignore, Node};
8use crate::hir;
9use crate::query::{GenericsParameters, Query, SecondaryBuildEntry};
10use crate::SourceId;
11
12#[derive(Default, Clone, Copy)]
13pub(super) enum Needs {
14    #[default]
15    Value,
16    Type,
17}
18
19pub(crate) struct Ctxt<'hir, 'a, 'arena> {
20    /// Arena used for allocations.
21    pub(super) arena: &'hir hir::arena::Arena,
22    pub(crate) q: Query<'a, 'arena>,
23    pub(super) source_id: SourceId,
24    pub(super) const_eval: bool,
25    pub(super) secondary_builds: Option<&'a mut Vec<SecondaryBuildEntry<'hir>>>,
26    pub(super) in_template: bool,
27    pub(super) in_path: bool,
28    pub(super) needs: Needs,
29    pub(super) scopes: hir::Scopes<'hir, 'a>,
30    pub(super) statement_buffer: Vec<hir::Stmt<'hir>>,
31    pub(super) statements: Vec<hir::Stmt<'hir>>,
32    pub(super) pattern_bindings: Vec<hir::Variable>,
33    pub(super) label: Option<ast::Label>,
34}
35
36impl<'hir, 'a, 'arena> Ctxt<'hir, 'a, 'arena> {
37    /// Construct a new context for used when constants are built separately
38    /// through the query system.
39    pub(crate) fn with_query(
40        arena: &'hir hir::arena::Arena,
41        q: Query<'a, 'arena>,
42        source_id: SourceId,
43        secondary_builds: &'a mut Vec<SecondaryBuildEntry<'hir>>,
44    ) -> alloc::Result<Self> {
45        Self::inner(arena, q, source_id, false, Some(secondary_builds))
46    }
47
48    /// Construct a new context used in a constant context where the resulting
49    /// expression is expected to be converted into a constant.
50    pub(crate) fn with_const(
51        arena: &'hir hir::arena::Arena,
52        q: Query<'a, 'arena>,
53        source_id: SourceId,
54    ) -> alloc::Result<Self> {
55        Self::inner(arena, q, source_id, true, None)
56    }
57
58    fn inner(
59        arena: &'hir hir::arena::Arena,
60        q: Query<'a, 'arena>,
61        source_id: SourceId,
62        const_eval: bool,
63        secondary_builds: Option<&'a mut Vec<SecondaryBuildEntry<'hir>>>,
64    ) -> alloc::Result<Self> {
65        let scopes = hir::Scopes::new(q.gen)?;
66
67        Ok(Self {
68            arena,
69            q,
70            source_id,
71            const_eval,
72            secondary_builds,
73            in_template: false,
74            in_path: false,
75            needs: Needs::default(),
76            scopes,
77            statement_buffer: Vec::new(),
78            statements: Vec::new(),
79            pattern_bindings: Vec::new(),
80            label: None,
81        })
82    }
83
84    #[instrument_ast(span = ast)]
85    pub(super) fn try_lookup_meta(
86        &mut self,
87        span: &dyn Spanned,
88        item: ItemId,
89        parameters: &GenericsParameters,
90    ) -> Result<Option<meta::Meta>> {
91        self.q
92            .try_lookup_meta(&DynLocation::new(self.source_id, span), item, parameters)
93    }
94
95    #[instrument_ast(span = ast)]
96    pub(super) fn lookup_meta(
97        &mut self,
98        span: &dyn Spanned,
99        item: ItemId,
100        parameters: impl AsRef<GenericsParameters>,
101    ) -> Result<meta::Meta> {
102        self.q
103            .lookup_meta(&DynLocation::new(self.source_id, span), item, parameters)
104    }
105}
106
107impl<'a> Ignore<'a> for Ctxt<'_, '_, '_> {
108    fn ignore(&mut self, _: Node<'a>) -> Result<()> {
109        Ok(())
110    }
111
112    fn error(&mut self, error: Error) -> alloc::Result<()> {
113        self.q.diagnostics.error(self.source_id, error)
114    }
115}
rune/hir/ctxt.rs

rune/hir/
ctxt.rs
