julefmt: minor improvements

julelang · Apr 1, 2024 · 101fbe9 · 101fbe9
1 parent a6cba7e
commit 101fbe9
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/src/format.jule b/src/format.jule
@@ -1466,9 +1466,24 @@ impl ExprFormatter {
  }
 
  fn unsafexpr(&self, mut u: &ast::UnsafeExpr) {
- self.write("unsafe { ")
+ self.write("unsafe {")
+ let line = u.expr.token.row != u.expr.end.row
+ if line {
+ self.fmt.add_indent()
+ self.write("\n")
+ self.write(self.fmt.indent)
+ } else {
+ self.write(" ")
+ }
  self.format(u.expr)
- self.write(" }")
+ if line {
+ self.fmt.done_indent()
+ self.write("\n")
+ self.write(self.fmt.indent)
+ } else {
+ self.write(" ")
+ }
+ self.write("}")
  }
 
  fn coexpr(&self, mut c: &ast::CoExpr) {