Fixes for unops

1 files changed, 20 insertions, 0 deletions

diff --git a/src/formatter/utils.c b/src/formatter/utils.c
index 472829d9..37751377 100644
--- a/src/formatter/utils.c
+++ b/src/formatter/utils.c
@@ -9,6 +9,7 @@
 #include "../stdlib/optionals.h"
 #include "../stdlib/tables.h"
 #include "../stdlib/text.h"
+#include "formatter.h"
 
 const Text_t single_indent = Text("    ");
 
@@ -110,3 +111,22 @@ CONSTFUNC const char *binop_tomo_operator(ast_e tag) {
     default: return NULL;
     }
 }
+
+OptionalText_t termify_inline(ast_t *ast, Table_t comments) {
+    if (range_has_comment(ast->start, ast->end, comments)) return NONE_TEXT;
+    switch (ast->tag) {
+    case BINOP_CASES:
+    case Not:
+    case Negative: return parenthesize(format_inline_code(ast, comments), EMPTY_TEXT);
+    default: return format_inline_code(ast, comments);
+    }
+}
+
+Text_t termify(ast_t *ast, Table_t comments, Text_t indent) {
+    switch (ast->tag) {
+    case BINOP_CASES:
+    case Not:
+    case Negative: return parenthesize(format_code(ast, comments, indent), indent);
+    default: return format_inline_code(ast, comments);
+    }
+}