Improve self printing (#367)

Akuli · Mar 26, 2023 · 7233ba9 · 7233ba9
1 parent e838dd6
commit 7233ba9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/self_hosted/ast.jou b/self_hosted/ast.jou
@@ -620,7 +620,10 @@ class AstSignature:
         for i = 0; i < self->nargs; i++:
             if i != 0:
                 printf(", ")
-            self->args[i].print(NULL)
+            if strcmp(self->args[i].name, "self") == 0:
+                printf("self")
+            else:
+                self->args[i].print(NULL)
 
         if self->takes_varargs:
             if self->nargs != 0:

diff --git a/src/print.c b/src/print.c
@@ -172,8 +172,12 @@ static void print_ast_function_signature(const AstSignature *sig)
     printf("%s(", sig->name);
     for (const AstNameTypeValue *ntv = sig->args.ptr; ntv < End(sig->args); ntv++) {
         if (ntv > sig->args.ptr) printf(", ");
-        printf("%s: ", ntv->name);
-        print_ast_type(&ntv->type);
+        if (!strcmp(ntv->name, "self"))
+            printf("self");
+        else{
+            printf("%s: ", ntv->name);
+            print_ast_type(&ntv->type);
+        }
         assert(!ntv->value);
     }
     if (sig->takes_varargs) {