diff options
Diffstat (limited to 'src/formatter')
| -rw-r--r-- | src/formatter/formatter.c | 4 | ||||
| -rw-r--r-- | src/formatter/utils.c | 2 |
2 files changed, 6 insertions, 0 deletions
diff --git a/src/formatter/formatter.c b/src/formatter/formatter.c index 12f08810..2d220493 100644 --- a/src/formatter/formatter.c +++ b/src/formatter/formatter.c @@ -128,6 +128,8 @@ OptionalText_t format_inline_code(ast_t *ast, Table_t comments) { Text_t message = fmt_inline(assert->message, comments); return Texts("assert ", expr, ", ", message); } + /*inline*/ case Defer: + return Texts("defer ", fmt_inline(Match(ast, Defer)->body, comments)); /*inline*/ case Lambda: { DeclareMatch(lambda, ast, Lambda); Text_t code = Texts("func(", format_inline_args(lambda->args, comments)); @@ -573,6 +575,8 @@ Text_t format_code(ast_t *ast, Table_t comments, Text_t indent) { DeclareMatch(extend, ast, Extend); return Texts("lang ", Text$from_str(extend->name), format_namespace(extend->body, comments, indent)); } + /*multiline*/ case Defer: + return Texts("defer ", format_namespace(Match(ast, Defer)->body, comments, indent)); /*multiline*/ case List: { if (inlined_fits) return inlined; ast_list_t *items = Match(ast, List)->items; diff --git a/src/formatter/utils.c b/src/formatter/utils.c index 0f638a57..9cd0227d 100644 --- a/src/formatter/utils.c +++ b/src/formatter/utils.c @@ -64,6 +64,7 @@ CONSTFUNC int suggested_blank_lines(ast_t *first, ast_t *second) { case While: case For: case Block: + case Defer: case ConvertDef: case FunctionDef: case Lambda: @@ -97,6 +98,7 @@ CONSTFUNC int suggested_blank_lines(ast_t *first, ast_t *second) { case While: case For: case Block: + case Defer: case ConvertDef: case FunctionDef: case Lambda: |