forked from Orchid/orchid
73 lines
1.9 KiB
Rust
73 lines
1.9 KiB
Rust
use super::collect_ops::ExportedOpsCache;
|
|
use crate::ast::{Constant, Rule};
|
|
use crate::interner::{Interner, Tok};
|
|
use crate::representations::sourcefile::{FileEntry, Member, Namespace};
|
|
use crate::utils::Substack;
|
|
|
|
fn member_rec(
|
|
// level
|
|
mod_stack: Substack<Tok<String>>,
|
|
// object
|
|
data: Member,
|
|
// context
|
|
path: &[Tok<String>],
|
|
ops_cache: &ExportedOpsCache,
|
|
i: &Interner,
|
|
) -> Member {
|
|
let prefix = (path.iter())
|
|
.copied()
|
|
.chain(mod_stack.iter().rev_vec_clone().into_iter())
|
|
.collect::<Vec<_>>();
|
|
match data {
|
|
Member::Namespace(Namespace { name, body }) => {
|
|
let new_body = entv_rec(mod_stack.push(name), body, path, ops_cache, i);
|
|
Member::Namespace(Namespace { name, body: new_body })
|
|
},
|
|
Member::Constant(constant) => Member::Constant(Constant {
|
|
name: constant.name,
|
|
value: constant.value.prefix(&prefix, &|_| false),
|
|
}),
|
|
Member::Rule(rule) => Member::Rule(Rule {
|
|
prio: rule.prio,
|
|
pattern: (rule.pattern.into_iter())
|
|
.map(|e| e.prefix(&prefix, &|_| false))
|
|
.collect(),
|
|
template: (rule.template.into_iter())
|
|
.map(|e| e.prefix(&prefix, &|_| false))
|
|
.collect(),
|
|
}),
|
|
}
|
|
}
|
|
|
|
fn entv_rec(
|
|
// level
|
|
mod_stack: Substack<Tok<String>>,
|
|
// object
|
|
data: Vec<FileEntry>,
|
|
// context
|
|
path: &[Tok<String>],
|
|
ops_cache: &ExportedOpsCache,
|
|
i: &Interner,
|
|
) -> Vec<FileEntry> {
|
|
data
|
|
.into_iter()
|
|
.map(|fe| match fe {
|
|
FileEntry::Exported(mem) =>
|
|
FileEntry::Exported(member_rec(mod_stack, mem, path, ops_cache, i)),
|
|
FileEntry::Internal(mem) =>
|
|
FileEntry::Internal(member_rec(mod_stack, mem, path, ops_cache, i)),
|
|
// XXX should [FileEntry::Export] be prefixed?
|
|
any => any,
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
pub fn prefix(
|
|
data: Vec<FileEntry>,
|
|
path: &[Tok<String>],
|
|
ops_cache: &ExportedOpsCache,
|
|
i: &Interner,
|
|
) -> Vec<FileEntry> {
|
|
entv_rec(Substack::Bottom, data, path, ops_cache, i)
|
|
}
|