language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-27 13:59:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Print back tags appropriately

Browse source
This commit is contained in:
Ayaz Hafiz 2022-08-09 12:54:27 -07:00
parent bff2852639
commit e1359c3025
No known key found for this signature in database
GPG key ID: 0E2A37416A25EF58
1 changed files with 30 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

36
crates/compiler/test_derive/src/pretty_print.rs
View file

@ -19,7 +19,10 @@ pub fn pretty_print_def(c: &Ctx, d: &Def) -> String {
macro_rules! maybe_paren { macro_rules! maybe_paren {
($paren_if_above:expr, $my_prec:expr, $doc:expr) => { ($paren_if_above:expr, $my_prec:expr, $doc:expr) => {
if $my_prec > $paren_if_above { maybe_paren!($paren_if_above, $my_prec, || true, $doc)
};
($paren_if_above:expr, $my_prec:expr, $extra_cond:expr, $doc:expr) => {
if $my_prec > $paren_if_above && $extra_cond() {
$doc.parens().group() $doc.parens().group()
} else { } else {
$doc $doc
@ -47,7 +50,7 @@ fn def<'a>(c: &Ctx, f: &'a Arena<'a>, d: &'a Def) -> DocBuilder<'a, Arena<'a>> {
#[derive(PartialEq, PartialOrd)] #[derive(PartialEq, PartialOrd)]
enum EPrec { enum EPrec {
Free, Free,
CallArg, AppArg,
} }
fn expr<'a>(c: &Ctx, p: EPrec, f: &'a Arena<'a>, e: &'a Expr) -> DocBuilder<'a, Arena<'a>> { fn expr<'a>(c: &Ctx, p: EPrec, f: &'a Arena<'a>, e: &'a Expr) -> DocBuilder<'a, Arena<'a>> {
@ -137,11 +140,11 @@ fn expr<'a>(c: &Ctx, p: EPrec, f: &'a Arena<'a>, e: &'a Expr) -> DocBuilder<'a,
maybe_paren!( maybe_paren!(
Free, Free,
p, p,
expr(c, CallArg, f, &fun.value) expr(c, AppArg, f, &fun.value)
.append( .append(
f.concat(args.iter().map(|le| f.line().append(expr( f.concat(args.iter().map(|le| f.line().append(expr(
c, c,
CallArg, AppArg,
f, f,
&le.1.value &le.1.value
)))) ))))
@ -193,7 +196,7 @@ fn expr<'a>(c: &Ctx, p: EPrec, f: &'a Arena<'a>, e: &'a Expr) -> DocBuilder<'a,
EmptyRecord => f.text("{}"), EmptyRecord => f.text("{}"),
Access { Access {
loc_expr, field, .. loc_expr, field, ..
} => expr(c, CallArg, f, &loc_expr.value) } => expr(c, AppArg, f, &loc_expr.value)
.append(f.text(format!(".{}", field.as_str()))) .append(f.text(format!(".{}", field.as_str())))
.group(), .group(),
OpaqueWrapFunction(OpaqueWrapFunctionData { opaque_name, .. }) => { OpaqueWrapFunction(OpaqueWrapFunctionData { opaque_name, .. }) => {
@ -201,7 +204,28 @@ fn expr<'a>(c: &Ctx, p: EPrec, f: &'a Arena<'a>, e: &'a Expr) -> DocBuilder<'a,
} }
Accessor(_) => todo!(), Accessor(_) => todo!(),
Update { .. } => todo!(), Update { .. } => todo!(),
Tag { .. } => todo!(), Tag {
name, arguments, ..
} => maybe_paren!(
Free,
p,
|| !arguments.is_empty(),
f.text(name.0.as_str())
.append(if arguments.is_empty() {
f.nil()
} else {
f.space()
})
.append(
f.intersperse(
arguments
.iter()
.map(|(_, le)| expr(c, AppArg, f, &le.value)),
f.space(),
)
)
.group()
),
ZeroArgumentTag { .. } => todo!(), ZeroArgumentTag { .. } => todo!(),
OpaqueRef { .. } => todo!(), OpaqueRef { .. } => todo!(),
Expect { .. } => todo!(), Expect { .. } => todo!(),