ES6: Implement WeakSet

src/collection.js

 // Copyright 2012 the V8 project authors. All rights reserved.
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are
 // met:
 //
 //     * Redistributions of source code must retain the above copyright
 //       notice, this list of conditions and the following disclaimer.
 //     * Redistributions in binary form must reproduce the above
 //       copyright notice, this list of conditions and the following
 //       disclaimer in the documentation and/or other materials provided
@@ skipping 307 matching lines @@
   InstallFunctions($WeakMap.prototype, DONT_ENUM, $Array(
     "get", WeakMapGet,
     "set", WeakMapSet,
     "has", WeakMapHas,
     "delete", WeakMapDelete,
     "clear", WeakMapClear
   ));
 }
 
 SetUpWeakMap();
+
+
+// -------------------------------------------------------------------
+// Harmony WeakSet
+
+var weaksetDataSymbol = %CreateSymbol(void 0);
+
+
+function WeakSet() {
+  if (%_IsConstructCall()) {
+    this[weaksetDataSymbol] = new $WeakMap();
+  } else {
+    return new $WeakSet();
+  }
+}
+
+
+var $WeakSet = WeakSet;
+
+
+function WeakSetAdd(value) {
+  var weaksetData = this[weaksetDataSymbol];
+  if (!weaksetData) {
+    throw %MakeTypeError('incompatible_method_receiver',
+                         ['WeakSet.prototype.add', this]);
+  }
+  if (!(IS_SPEC_OBJECT(value) || IS_SYMBOL(value))) {
+    throw %MakeTypeError('invalid_weakset_value', [this, value]);
+  }
+  return %WeakMapSet(weaksetData, value, true);
+}
+
+
+function WeakSetHas(value) {
+  var weaksetData = this[weaksetDataSymbol];
+  if (!weaksetData) {
+    throw %MakeTypeError('incompatible_method_receiver',
+                         ['WeakSet.prototype.has', this]);
+  }
+  if (!(IS_SPEC_OBJECT(value) || IS_SYMBOL(value))) {
+    throw %MakeTypeError('invalid_weakset_value', [this, value]);
+  }
+  return %WeakMapHas(weaksetData, value);
+}
+
+
+function WeakSetDelete(value) {
+  var weaksetData = this[weaksetDataSymbol];
+  if (!weaksetData) {
+    throw MakeTypeError('incompatible_method_receiver',
+                        ['WeakSet.prototype.delete', this]);
+  }
+  if (!(IS_SPEC_OBJECT(value) || IS_SYMBOL(value))) {
+    throw %MakeTypeError('invalid_weakset_value', [this, value]);
+  }
+  return %WeakMapDelete(weaksetData, value);
+}
+
+
+function WeakSetClear() {
+  if (!this[weaksetDataSymbol]) {
+    throw %MakeTypeError('incompatible_method_receiver',
+                         ['WeakSet.prototype.clear', this]);
+  }
+
+  this[weaksetDataSymbol] = new $WeakMap();
+}
+
+
+// -------------------------------------------------------------------
+
+function SetUpWeakSet() {
+  %CheckIsBootstrapping();
+
+  %SetProperty(global, 'WeakSet', $WeakSet, DONT_ENUM);
+
+  %FunctionSetInstanceClassName($WeakSet, 'WeakSet');
+
+  %FunctionSetPrototype($WeakSet, new $Object());
+  %SetProperty($WeakSet.prototype, 'constructor', $WeakSet, DONT_ENUM);
+
+  // Set up the non-enumerable functions on the WeakSet prototype object.
+  InstallFunctions($WeakSet.prototype, DONT_ENUM, $Array(
+    'add', WeakSetAdd,
+    'has', WeakSetHas,
+    'delete', WeakSetDelete,
+    'clear', WeakSetClear
+  ));
+}
+
+SetUpWeakSet();