From 78a29c41098db5e5f8291e0345a3cd443c52b329 Mon Sep 17 00:00:00 2001
From: Mel <einebeere@gmail.com>
Date: Mon, 30 May 2022 02:02:53 +0000
Subject: Specify arg count on VM Functions

---
 pkg/lang/vm/value/data.go | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'pkg/lang/vm/value/data.go')

diff --git a/pkg/lang/vm/value/data.go b/pkg/lang/vm/value/data.go
index c0e5e9a..13716fd 100644
--- a/pkg/lang/vm/value/data.go
+++ b/pkg/lang/vm/value/data.go
@@ -116,6 +116,7 @@ type NullData struct{}
 
 type FunctionData struct {
 	pc     int
+	args   uint
 	env    mem.Ptr
 	native NativeFunc
 }
@@ -126,6 +127,10 @@ func (f FunctionData) Pc() int {
 	return f.pc
 }
 
+func (f FunctionData) Args() uint {
+	return f.args
+}
+
 func (f FunctionData) Env() mem.Ptr {
 	return f.env
 }
@@ -135,14 +140,14 @@ func (f FunctionData) Native() NativeFunc {
 }
 
 func (f FunctionData) WithEnv(env mem.Ptr) FunctionData {
-	return FunctionData{pc: f.pc, env: env, native: f.native}
+	return FunctionData{pc: f.pc, args: f.args, env: env, native: f.native}
 }
 
 func (f FunctionData) String(_ mem.Mem) (string, error) {
 	if f.native != nil {
-		return "<fn native>", nil
+		return fmt.Sprintf("<fn(%d) native>", f.args), nil
 	} else {
-		return fmt.Sprintf("<fn %d>", f.pc), nil
+		return fmt.Sprintf("<fn(%d) %d>", f.args, f.pc), nil
 	}
 }
 }
-- 
cgit 1.4.1