🎨 update

Gwion · Mar 11, 2024 · 8cb7034 · 8cb7034
1 parent 16503e3
commit 8cb7034
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 11 deletions.
diff --git a/include/gwfmt.h b/include/gwfmt.h
@@ -55,3 +55,4 @@ ANN void gwfmt_type_def(Gwfmt *a, Type_Def b);
 ANN void gwfmt_prim_def(Gwfmt *a, Prim_Def b);
 ANN void gwfmt_ast(Gwfmt *a, Ast b);
 ANN void gwfmt_type_decl(Gwfmt *a, const Type_Decl *b);
+ANN void gwfmt_variable(Gwfmt *a, const Variable *b);
diff --git a/include/gwfmt_internal.h b/include/gwfmt_internal.h
@@ -15,7 +15,7 @@ ANN static void gwfmt_prim_interp(Gwfmt *a, Exp* *b);
 ANN static void gwfmt_prim_char(Gwfmt *a, m_str *b);
 ANN static void gwfmt_prim_nil(Gwfmt *a, void *b);
 ANN static void gwfmt_prim(Gwfmt *a, Exp_Primary *b);
-ANN static void gwfmt_var_decl(Gwfmt *a, Var_Decl *b);
+ANN static void gwfmt_var_decl(Gwfmt *a, const Var_Decl *b);
 ANN static void gwfmt_exp_decl(Gwfmt *a, Exp_Decl *b);
 ANN static void gwfmt_exp_binary(Gwfmt *a, Exp_Binary *b);
 ANN static void gwfmt_exp_unary(Gwfmt *a, Exp_Unary *b);

diff --git a/src/lint.c b/src/lint.c
@@ -593,7 +593,7 @@ ANN static void gwfmt_prim(Gwfmt *a, Exp_Primary *b) {
   gwfmt_prim_func[b->prim_type](a, &b->d);
 }
 
-ANN static void gwfmt_var_decl(Gwfmt *a, Var_Decl *b) {
+ANN static void gwfmt_var_decl(Gwfmt *a, const Var_Decl *b) {
   if (b->tag.sym) COLOR(a, "{W+}", s_name(b->tag.sym));
 }
 
@@ -1162,14 +1162,17 @@ ANN static void gwfmt_stmt(Gwfmt *a, Stmt* b) {
   if (!skip_indent) gwfmt_nl(a);
 }
 
+ANN void gwfmt_variable(Gwfmt *a, const Variable *b) {
+  if (b->td) {
+    gwfmt_type_decl(a, b->td);
+    if (b->vd.tag.sym) gwfmt_space(a);
+   }
+  gwfmt_var_decl(a, &b->vd);
+}
 ANN /*static */void gwfmt_arg_list(Gwfmt *a, Arg_List b, const bool locale) {
   for(uint32_t i = locale; i < b->len; i++) {
     Arg *arg = mp_vector_at(b, Arg, i);
-    if (arg->var.td) {
-      gwfmt_type_decl(a, arg->var.td);
-      if (arg->var.vd.tag.sym) gwfmt_space(a);
-    }
-    gwfmt_var_decl(a, &arg->var.vd);
+    gwfmt_variable(a, &arg->var);
     if (arg->exp) {
       gwfmt_space(a);
       gwfmt(a, ":");
@@ -1187,9 +1190,7 @@ ANN static void gwfmt_variable_list(Gwfmt *a, Variable_List b) {
   for(uint32_t i = 0; i < b->len; i++) {
     Variable *um = mp_vector_at(b, Variable, i);
     gwfmt_indent(a);
-    gwfmt_type_decl(a, um->td);
-    gwfmt_space(a);
-    gwfmt_var_decl(a, &um->vd);
+    gwfmt_variable(a, um);
     gwfmt_sc(a);
     gwfmt_nl(a);
   }
@@ -1248,8 +1249,8 @@ ANN void gwfmt_func_def(Gwfmt *a, Func_Def b) {
     gwfmt_nl(a);
     return;
   }
-  gwfmt_space(a);
   if (!GET_FLAG(b->base, abstract) && b->d.code) {
+    gwfmt_space(a);
     gwfmt_lbrace(a);
     gwfmt_nl(a);
     INDENT(a, gwfmt_stmt_list(a, b->d.code));