From ad08514f46440e53170f936f6fe2bd2195659855 Mon Sep 17 00:00:00 2001
From: 702fbtngus <702fbtngus@kaist.ac.kr>
Date: Tue, 3 Sep 2024 21:57:09 +0900
Subject: [PATCH] Function argument validation

---
 spectec/src/al/valid.ml | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/spectec/src/al/valid.ml b/spectec/src/al/valid.ml
index 36854174c3..4956a29869 100644
--- a/spectec/src/al/valid.ml
+++ b/spectec/src/al/valid.ml
@@ -562,6 +562,21 @@ let valid_algo (algo: algorithm) =
   |> print_string;
   print_endline ")";
 
+  (* TODO: Use local environment *)
+  (* Store global enviroment *)
+  let global_env = !env in
+
+  (* Add function argument to environment *)
+  (match Env.find_opt_def !env (Al_util.name_of_algo algo $ no_region) with
+  | Some (params, _, _) -> List.iter (fun param ->
+      (match param.it with
+      | DefP (id, params', typ') -> env := Env.bind_def !env id (params', typ', [])
+      | _ -> ()
+      )
+    ) params;
+  | _ -> ()
+  );
+
   init algo;
   let walker =
     { base_unit_walker with
@@ -570,7 +585,10 @@ let valid_algo (algo: algorithm) =
       walk_instr = valid_instr
     }
   in
-  walker.walk_algo walker algo
+  walker.walk_algo walker algo;
+
+  (* Reset global enviroment *)
+  env := global_env
 
 let valid (script: script) =
   List.iter valid_algo script