Support value iterators in WebIDL interfaces.

4 files changed, 98 insertions, 0 deletions

diff --git a/components/script/dom/bindings/codegen/CodegenRust.py b/components/script/dom/bindings/codegen/CodegenRust.py
index 56a20677691..bd41a214740 100644
--- a/components/script/dom/bindings/codegen/CodegenRust.py
+++ b/components/script/dom/bindings/codegen/CodegenRust.py
@@ -1519,6 +1519,46 @@ class MethodDefiner(PropertyDefiner):
                                  "length": 0,
                                  "condition": "Condition::Satisfied"})
 
+        # Generate the keys/values/entries aliases for value iterables.
+        maplikeOrSetlikeOrIterable = descriptor.interface.maplikeOrSetlikeOrIterable
+        if (not static and not unforgeable and
+            (maplikeOrSetlikeOrIterable and
+             maplikeOrSetlikeOrIterable.isIterable() and
+             maplikeOrSetlikeOrIterable.isValueIterator())):
+            # Add our keys/values/entries/forEach
+            self.regular.append({
+                "name": "keys",
+                "methodInfo": False,
+                "selfHostedName": "ArrayKeys",
+                "length": 0,
+                "condition": PropertyDefiner.getControllingCondition(m,
+                                                                     descriptor)
+            })
+            self.regular.append({
+                "name": "values",
+                "methodInfo": False,
+                "selfHostedName": "ArrayValues",
+                "length": 0,
+                "condition": PropertyDefiner.getControllingCondition(m,
+                                                                     descriptor)
+            })
+            self.regular.append({
+                "name": "entries",
+                "methodInfo": False,
+                "selfHostedName": "ArrayEntries",
+                "length": 0,
+                "condition": PropertyDefiner.getControllingCondition(m,
+                                                                     descriptor)
+            })
+            self.regular.append({
+                "name": "forEach",
+                "methodInfo": False,
+                "selfHostedName": "ArrayForEach",
+                "length": 0,
+                "condition": PropertyDefiner.getControllingCondition(m,
+                                                                     descriptor)
+            })
+
         isUnforgeableInterface = bool(descriptor.interface.getExtendedAttribute("Unforgeable"))
         if not static and unforgeable == isUnforgeableInterface:
             stringifier = descriptor.operations['Stringifier']
diff --git a/components/script/dom/mod.rs b/components/script/dom/mod.rs
index 432246d814f..1cf74390257 100644
--- a/components/script/dom/mod.rs
+++ b/components/script/dom/mod.rs
@@ -384,6 +384,7 @@ pub mod storageevent;
 pub mod stylesheet;
 pub mod stylesheetlist;
 pub mod testbinding;
+pub mod testbindingiterable;
 pub mod testbindingproxy;
 pub mod text;
 pub mod textdecoder;
diff --git a/components/script/dom/testbindingiterable.rs b/components/script/dom/testbindingiterable.rs
new file mode 100644
index 00000000000..1e462a98531
--- /dev/null
+++ b/components/script/dom/testbindingiterable.rs
@@ -0,0 +1,43 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+// check-tidy: no specs after this line
+
+use dom::bindings::cell::DOMRefCell;
+use dom::bindings::codegen::Bindings::TestBindingIterableBinding::{self, TestBindingIterableMethods};
+use dom::bindings::error::Fallible;
+use dom::bindings::global::GlobalRef;
+use dom::bindings::js::Root;
+use dom::bindings::reflector::{Reflector, reflect_dom_object};
+use dom::bindings::str::DOMString;
+
+#[dom_struct]
+pub struct TestBindingIterable {
+    reflector: Reflector,
+    vals: DOMRefCell<Vec<DOMString>>,
+}
+
+impl TestBindingIterable {
+    fn new(global: GlobalRef) -> Root<TestBindingIterable> {
+        reflect_dom_object(box TestBindingIterable {
+            reflector: Reflector::new(),
+            vals: DOMRefCell::new(vec![]),
+        }, global, TestBindingIterableBinding::Wrap)
+    }
+
+    pub fn Constructor(global: GlobalRef) -> Fallible<Root<TestBindingIterable>> {
+        Ok(TestBindingIterable::new(global))
+    }
+}
+
+impl TestBindingIterableMethods for TestBindingIterable {
+    fn Add(&self, v: DOMString) { self.vals.borrow_mut().push(v); }
+    fn Length(&self) -> u32 { self.vals.borrow().len() as u32 }
+    fn GetItem(&self, n: u32) -> DOMString { self.vals.borrow().get(n as usize).unwrap().clone() }
+    fn IndexedGetter(&self, n: u32, found: &mut bool) -> DOMString {
+        let s = self.GetItem(n);
+        *found = true;
+        s
+    }
+}
diff --git a/components/script/dom/webidls/TestBindingIterable.webidl b/components/script/dom/webidls/TestBindingIterable.webidl
new file mode 100644
index 00000000000..c9e61074eed
--- /dev/null
+++ b/components/script/dom/webidls/TestBindingIterable.webidl
@@ -0,0 +1,14 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+// This interface is entirely internal to Servo, and should not be accessible to
+// web pages.
+
+[Pref="dom.testbinding.enabled", Exposed=(Window,Worker), Constructor]
+interface TestBindingIterable {
+  void add(DOMString arg);
+  readonly attribute unsigned long length;
+  getter DOMString getItem(unsigned long index);
+  iterable<DOMString>;
+};