From bff8947e43dbdf3b004b597bf30cde24616807f3 Mon Sep 17 00:00:00 2001
From: Corey Farwell <coreyf@rwell.org>
Date: Sat, 9 Jan 2016 13:10:26 -0500
Subject: Implement HTMLFormElement::Elements

Fixes #8566
---
 components/script/dom/htmlformelement.rs | 60 ++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

(limited to 'components/script/dom/htmlformelement.rs')

diff --git a/components/script/dom/htmlformelement.rs b/components/script/dom/htmlformelement.rs
index 512a0d6a56c..072231d4054 100644
--- a/components/script/dom/htmlformelement.rs
+++ b/components/script/dom/htmlformelement.rs
@@ -20,10 +20,14 @@ use dom::element::Element;
 use dom::event::{EventBubbles, EventCancelable};
 use dom::eventtarget::EventTarget;
 use dom::htmlbuttonelement::HTMLButtonElement;
+use dom::htmlcollection::CollectionFilter;
 use dom::htmldatalistelement::HTMLDataListElement;
 use dom::htmlelement::HTMLElement;
+use dom::htmlfieldsetelement::HTMLFieldSetElement;
+use dom::htmlformcontrolscollection::HTMLFormControlsCollection;
 use dom::htmlinputelement::HTMLInputElement;
 use dom::htmlobjectelement::HTMLObjectElement;
+use dom::htmloutputelement::HTMLOutputElement;
 use dom::htmlselectelement::HTMLSelectElement;
 use dom::htmltextareaelement::HTMLTextAreaElement;
 use dom::node::{Node, document_from_node, window_from_node};
@@ -135,6 +139,62 @@ impl HTMLFormElementMethods for HTMLFormElement {
     fn Reset(&self) {
         self.reset(ResetFrom::FromFormResetMethod);
     }
+
+    // https://html.spec.whatwg.org/multipage/#dom-form-elements
+    fn Elements(&self) -> Root<HTMLFormControlsCollection> {
+        #[derive(JSTraceable, HeapSizeOf)]
+        struct ElementsFilter {
+            form: Root<HTMLFormElement>
+        }
+        impl CollectionFilter for ElementsFilter {
+            fn filter<'a>(&self, elem: &'a Element, _root: &'a Node) -> bool {
+                let form_owner = match elem.upcast::<Node>().type_id() {
+                    NodeTypeId::Element(ElementTypeId::HTMLElement(t)) => {
+                        match t {
+                            HTMLElementTypeId::HTMLButtonElement => {
+                                elem.downcast::<HTMLButtonElement>().unwrap().form_owner()
+                            }
+                            HTMLElementTypeId::HTMLFieldSetElement => {
+                                elem.downcast::<HTMLFieldSetElement>().unwrap().form_owner()
+                            }
+                            HTMLElementTypeId::HTMLInputElement => {
+                                let input_elem = elem.downcast::<HTMLInputElement>().unwrap();
+                                if input_elem.type_() == atom!("image") {
+                                    return false;
+                                }
+                                input_elem.form_owner()
+                            }
+                            HTMLElementTypeId::HTMLObjectElement => {
+                                elem.downcast::<HTMLObjectElement>().unwrap().form_owner()
+                            }
+                            HTMLElementTypeId::HTMLOutputElement => {
+                                elem.downcast::<HTMLOutputElement>().unwrap().form_owner()
+                            }
+                            HTMLElementTypeId::HTMLSelectElement => {
+                                elem.downcast::<HTMLSelectElement>().unwrap().form_owner()
+                            }
+                            HTMLElementTypeId::HTMLTextAreaElement => {
+                                elem.downcast::<HTMLTextAreaElement>().unwrap().form_owner()
+                            }
+                            _ => {
+                                debug_assert!(!elem.downcast::<HTMLElement>().unwrap().is_listed_element());
+                                return false;
+                            }
+                        }
+                    }
+                    _ => return false,
+                };
+
+                match form_owner {
+                    Some(form_owner) => form_owner == self.form,
+                    None => false,
+                }
+            }
+        }
+        let filter = box ElementsFilter { form: Root::from_ref(self) };
+        let window = window_from_node(self);
+        HTMLFormControlsCollection::new(window.r(), self.upcast(), filter)
+    }
 }
 
 #[derive(Copy, Clone, HeapSizeOf, PartialEq)]
-- 
cgit v1.2.3