Implement function calls

1 files changed, 21 insertions, 0 deletions

diff --git a/generate.go b/generate.go
index 317e11c..3e21a0c 100644
--- a/generate.go
+++ b/generate.go
@@ -378,3 +378,24 @@ func (n *numberExpr) generate(w io.Writer) (string, error) {
 
 	return fmt.Sprintf("%d", num), nil
 }
+
+func (c *callExpr) generate(w io.Writer) (string, error) {
+	ret := allocReg()
+
+	fmt.Fprintf(w, "%s =w call $%s(", ret, c.funcName)
+
+	delim := ""
+	for arg := c.args; arg != nil; arg = arg.next {
+		reg, err := arg.value.generate(w)
+		if err != nil {
+			return "", err
+		}
+
+		fmt.Fprintf(w, "w %s%s", reg, delim)
+		delim = ","
+	}
+
+	fmt.Fprintf(w, ")\n")
+
+	return ret, nil
+}