Garmelon/tada
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make code generic over allocator

Browse source 
This way, I can use the same code with any allocator I want. More
importantly, I can also use the nice DocBuilder-exclusive helper
functions!
This commit is contained in:
Joscha 2022-11-20 21:41:05 +01:00
parent 1a3772e6f7
commit 2ba56f0c92
4 changed files with 45 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/pretty/program.rs
View file

@ -1,29 +1,31 @@
use pretty::RcDoc;
use pretty::{DocAllocator, DocBuilder, Pretty};
use crate::ast::Program;
impl Program {
pub fn to_doc(&self) -> RcDoc {
impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for Program {
fn pretty(self, allocator: &'a D) -> DocBuilder<'a, D> {
match self {
Program::Expr {
s0,
expr,
s1,
span: _,
} => expr.to_doc(),
} => expr.pretty(allocator),
Program::Module {
s0,
s1,
elems,
s2,
span: _,
} => RcDoc::text("module")
.append(RcDoc::line())
.append(RcDoc::line())
.append(elems.to_doc(
|e| RcDoc::text("<elem>"),
|(s0, s1)| RcDoc::text(",").append(RcDoc::line()),
|s| RcDoc::text(","),
} => allocator
.text("module")
.append(allocator.line())
.append(allocator.line())
.append(elems.pretty(
allocator,
|a, e| a.text("<elem>"),
|a, (s0, s1)| a.text(",").append(a.line()),
|a, s| a.text(","),
)),
}
}