builtins_set.go

package evaluator

import "github.com/Abathargh/harlock/internal/object"

func setBuiltinAdd(this object.Object, args ...object.Object) object.Object {
	setThis := this.(*object.Set)

	hashable, isHashable := args[0].(object.Hashable)
	if !isHashable {
		return newTypeError("the passed type is not hashable")
	}

	key := hashable.HashKey()
	setThis.Elements[key] = args[0]
	return nil
}

func setBuiltinRemove(this object.Object, args ...object.Object) object.Object {
	setThis := this.(*object.Set)

	hashable, isHashable := args[0].(object.Hashable)
	if !isHashable {
		return newTypeError("the passed type is not hashable")
	}

	key := hashable.HashKey()
	delete(setThis.Elements, key)
	return nil
}