From cea1a5f5bd55d9b76a0b8647c62dfc1a465818ec Mon Sep 17 00:00:00 2001
From: SimonFJ20 <simonfromjakobsen@gmail.com>
Date: Fri, 7 Feb 2025 12:56:33 +0100
Subject: [PATCH] compiler: remember call arg

---
 slige/compiler/check/checker.ts      | 5 ++++-
 slige/compiler/common/diagnostics.ts | 5 ++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/slige/compiler/check/checker.ts b/slige/compiler/check/checker.ts
index a46ba3e..3edb111 100644
--- a/slige/compiler/check/checker.ts
+++ b/slige/compiler/check/checker.ts
@@ -218,7 +218,10 @@ export class Checker {
         const _args = kind.args.map((arg, i) =>
             this.checkExpr(arg, paramTys[i])
         );
-        return fnTy.kind.returnTy;
+
+        const ty = fnTy.kind.returnTy;
+        this.exprTys.set(expr.id, ty);
+        return ty;
     }
 
     private tyTy(ty: ast.Ty): Ty {
diff --git a/slige/compiler/common/diagnostics.ts b/slige/compiler/common/diagnostics.ts
index 7f5d06a..8f0fc18 100644
--- a/slige/compiler/common/diagnostics.ts
+++ b/slige/compiler/common/diagnostics.ts
@@ -54,9 +54,8 @@ export function prettyPrintReport(ctx: Ctx, rep: Report) {
     const { severity, msg } = rep;
     const origin = rep.origin ? `\x1b[1m${rep.origin}:\x1b[0m ` : "";
     console.error(
-        `${origin}${
-            severityColor(severity)
-        }${severity}:\x1b[0m \x1b[37m${msg}\x1b[0m`,
+        `${origin}${severityColor(severity)}${severity}:\x1b[0m %c${msg}`,
+        "color: white; font-weight: bold;",
     );
     if (!rep.file) {
         return;