orchid/src/pipeline/project_tree/prefix.rs

use super::collect_ops::ExportedOpsCache;
use crate::ast::{Constant, Rule};
use crate::interner::{Interner, Tok};
use crate::representations::sourcefile::{FileEntry, Member, ModuleBlock};
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())
    .cloned()
    .chain(mod_stack.iter().rev_vec_clone().into_iter())
    .collect::<Vec<_>>();
  match data {
    Member::Module(ModuleBlock { name, body }) => {
      let new_stack = mod_stack.push(name.clone());
      let new_body = entv_rec(new_stack, body, path, ops_cache, i);
      Member::Module(ModuleBlock { 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| {
      let (mem, wrapper): (Member, fn(Member) -> FileEntry) = match fe {
        FileEntry::Exported(mem) => (mem, FileEntry::Exported),
        FileEntry::Internal(mem) => (mem, FileEntry::Internal),
        // XXX should [FileEntry::Export] be prefixed?
        any => return any,
      };
      wrapper(member_rec(mod_stack.clone(), mem, path, ops_cache, i))
    })
    .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)
}