language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-26 21:39:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Move TypeAnnotation::Apply to to_node

Browse source
This commit is contained in:
Joshua Warner 2024-12-19 17:36:20 -05:00
parent f169329c0e
commit 13ec99c16b
No known key found for this signature in database
GPG key ID: 89AD497003F93FDD
2 changed files with 46 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

87
crates/compiler/fmt/src/annotation.rs
View file

@ -222,8 +222,6 @@ fn fmt_ty_ann(
) {
let me = ann_lift_spaces(buf.text.bump(), me);
let self_is_multiline = me.item.is_multiline();
if !me.before.is_empty() {
buf.ensure_ends_with_newline();
fmt_comments_only(buf, me.before.iter(), NewlineAt::Bottom, indent);
@ -1214,59 +1212,43 @@ impl<'a> Nodify<'a> for TypeAnnotation<'a> {
{
match self {
TypeAnnotation::Apply(module, func, args) => {
if args.is_empty() {
return Spaces {
item: Node::TypeAnnotation(*self),
before: &[],
after: &[],
};
}
let mut new_args = Vec::with_capacity_in(args.len(), arena);
if !args.is_empty() {
for arg in args.iter().take(args.len() - 1) {
let lifted = ann_lift_spaces(arena, &arg.value);
new_args.push(Loc::at(arg.region, lower(arena, lifted)));
}
}
let after = if let Some(last) = args.last() {
let lifted = ann_lift_spaces(arena, &last.value);
if lifted.before.is_empty() {
new_args.push(Loc::at(last.region, lifted.item));
} else {
new_args.push(Loc::at(
last.region,
TypeAnnotation::SpaceBefore(arena.alloc(lifted.item), lifted.before),
));
}
lifted.after
let first = if module.is_empty() {
Node::Literal(func)
} else {
&[]
Node::Literal(arena.alloc_str(&format!("{}.{}", module, func)))
};
let item = Node::TypeAnnotation(TypeAnnotation::Apply(
module,
func,
new_args.into_bump_slice(),
));
let mut last_after: &[CommentOrNewline<'_>] = &[];
let mut rest = Vec::with_capacity_in(args.len(), arena);
if parens == Parens::InApply {
parens_around_node(
arena,
Spaces {
before: &[],
item,
after,
},
true,
)
for arg in *args {
let lifted = arg.value.to_node(arena, Parens::InApply);
let before = merge_spaces_conservative(arena, last_after, lifted.before);
last_after = lifted.after;
rest.push(Item {
before,
comma: false,
newline: false,
space: true,
node: lifted.item,
});
}
let item = Spaces {
before: &[],
item: Node::CommaSequence {
allow_blank_lines: false,
indent_rest: true,
first: arena.alloc(first),
rest: rest.into_bump_slice(),
},
after: last_after,
};
if parens == Parens::InApply && !args.is_empty() {
parens_around_node(arena, item, false)
} else {
Spaces {
before: &[],
item,
after,
}
item
}
}
TypeAnnotation::SpaceBefore(expr, spaces) => {
@ -1335,6 +1317,7 @@ impl<'a> Nodify<'a> for TypeAnnotation<'a> {
before: first_node.before,
item: Node::CommaSequence {
allow_blank_lines: false,
indent_rest: false,
first: arena.alloc(first_node.item),
rest: rest_nodes.into_bump_slice(),
},
@ -1346,7 +1329,7 @@ impl<'a> Nodify<'a> for TypeAnnotation<'a> {
|| parens == Parens::InAsPattern
|| parens == Parens::InFunctionType
{
parens_around_node(arena, inner, true)
parens_around_node(arena, item, false)
} else {
item
}
@ -1375,7 +1358,7 @@ impl<'a> Nodify<'a> for TypeAnnotation<'a> {
let item = Spaces::item(Node::Literal(text));
if *text == "implements" {
parens_around_node(arena, item)
parens_around_node(arena, item, false)
} else {
item
}

13
crates/compiler/fmt/src/node.rs
View file

@ -56,6 +56,7 @@ pub enum Node<'a> {
DelimitedSequence(Braces, &'a [(Sp<'a>, Node<'a>)], Sp<'a>),
CommaSequence {
allow_blank_lines: bool,
indent_rest: bool,
first: &'a Node<'a>,
rest: &'a [Item<'a>],
},
@ -160,6 +161,7 @@ impl<'a> Formattable for Node<'a> {
}
Node::CommaSequence {
allow_blank_lines: _,
indent_rest: _,
first,
rest,
} => first.is_multiline() || rest.iter().any(|item| item.is_multiline()),
@ -205,10 +207,16 @@ impl<'a> Formattable for Node<'a> {
}
Node::CommaSequence {
allow_blank_lines,
indent_rest,
first,
rest,
} => {
buf.indent(indent);
let inner_indent = if *indent_rest {
indent + INDENT
} else {
indent
};
first.format_with_options(buf, parens, newlines, indent);
for item in *rest {
@ -218,14 +226,15 @@ impl<'a> Formattable for Node<'a> {
if *allow_blank_lines {
fmt_spaces(buf, item.before.iter(), indent);
} else {
fmt_spaces_no_blank_lines(buf, item.before.iter(), indent);
fmt_spaces_no_blank_lines(buf, item.before.iter(), inner_indent);
}
if item.newline {
buf.ensure_ends_with_newline();
} else if item.space {
buf.ensure_ends_with_whitespace();
}
item.node.format_with_options(buf, parens, newlines, indent);
item.node
.format_with_options(buf, parens, newlines, inner_indent);
}
}
Node::Literal(text) => {