string_eqeq

wren-lang · Mar 10, 2023 · d896dc9 · d896dc9
1 parent 3f268ff
commit d896dc9
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/vm/wren_core.c b/src/vm/wren_core.c
@@ -1168,6 +1168,21 @@ DEF_PRIMITIVE(string_startsWith)
   RETURN_BOOL(memcmp(string->value, search->value, search->length) == 0);
 }
 
+static bool string_eqeq(Value lhs, Value rhs) {
+  if (!IS_STRING(rhs)) return false;
+  return wrenStringEquals(AS_STRING(lhs), AS_STRING(rhs));
+}
+
+DEF_PRIMITIVE(string_eqeq)
+{
+  RETURN_BOOL(string_eqeq(args[0], args[1]));
+}
+
+DEF_PRIMITIVE(string_bangeq)
+{
+  RETURN_BOOL(!string_eqeq(args[0], args[1]));
+}
+
 DEF_PRIMITIVE(string_plus)
 {
   if (!validateString(vm, args[1], "Right operand")) return false;
@@ -1568,6 +1583,8 @@ void wrenInitializeCore(WrenVM* vm)
   vm->stringClass = AS_CLASS(wrenFindVariable(vm, coreModule, "String"));
   PRIMITIVE(vm->stringClass->obj.classObj, "fromCodePoint(_)", string_fromCodePoint);
   PRIMITIVE(vm->stringClass->obj.classObj, "fromByte(_)", string_fromByte);
+  PRIMITIVE(vm->stringClass, "==(_)", string_eqeq);
+  PRIMITIVE(vm->stringClass, "!=(_)", string_bangeq);
   PRIMITIVE(vm->stringClass, "+(_)", string_plus);
   PRIMITIVE(vm->stringClass, "[_]", string_subscript);
   PRIMITIVE(vm->stringClass, "byteAt_(_)", string_byteAt);