Adds an incredibly simple equality operator

1 files changed, 15 insertions, 0 deletions

diff --git a/subex/arithmetic.go b/subex/arithmetic.go
index 9e5e530..4cbc9db 100644
--- a/subex/arithmetic.go
+++ b/subex/arithmetic.go
@@ -156,3 +156,18 @@ func notValues(values walk.ValueList) (notted walk.ValueList, err error) {
 	}
 	return notted, nil
 }
+
+// Returns true if all values are equal, false if not
+func equalValues(values walk.ValueList) (walk.ValueList, error) {
+	if len(values) == 0 {
+		return walk.ValueList{walk.BoolScalar(true)}, nil
+	}
+	first := values[0]
+	for _, value := range values[1:] {
+		// TODO: Refine the equality check
+		if value != first {
+			return walk.ValueList{walk.BoolScalar(false)}, nil
+		}
+	}
+	return walk.ValueList{walk.BoolScalar(true)}, nil
+}