1 files changed, 83 insertions, 0 deletions
diff --git a/components/script/dom/xrinputsource.rs b/components/script/dom/xrinputsource.rs
new file mode 100644
index 00000000000..2b623864d56
--- /dev/null
+++ b/components/script/dom/xrinputsource.rs
@@ -0,0 +1,83 @@
+/* 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 https://mozilla.org/MPL/2.0/. */
+
+use crate::dom::bindings::cell::DomRefCell;
+use crate::dom::bindings::codegen::Bindings::XRInputSourceBinding;
+use crate::dom::bindings::codegen::Bindings::XRInputSourceBinding::{
+    XRHandedness, XRInputSourceMethods,
+};
+use crate::dom::bindings::reflector::{reflect_dom_object, DomObject, Reflector};
+use crate::dom::bindings::root::{Dom, DomRoot, MutNullableDom};
+use crate::dom::globalscope::GlobalScope;
+use crate::dom::xrsession::XRSession;
+use crate::dom::xrspace::XRSpace;
+use dom_struct::dom_struct;
+use webvr_traits::{WebVRGamepadData, WebVRGamepadHand, WebVRGamepadState, WebVRPose};
+
+#[dom_struct]
+pub struct XRInputSource {
+    reflector: Reflector,
+    session: Dom<XRSession>,
+    #[ignore_malloc_size_of = "Defined in rust-webvr"]
+    data: WebVRGamepadData,
+    #[ignore_malloc_size_of = "Defined in rust-webvr"]
+    state: DomRefCell<WebVRGamepadState>,
+    target_ray_space: MutNullableDom<XRSpace>,
+}
+
+impl XRInputSource {
+    pub fn new_inherited(
+        session: &XRSession,
+        data: WebVRGamepadData,
+        state: WebVRGamepadState,
+    ) -> XRInputSource {
+        XRInputSource {
+            reflector: Reflector::new(),
+            session: Dom::from_ref(session),
+            data,
+            state: DomRefCell::new(state),
+            target_ray_space: Default::default(),
+        }
+    }
+
+    pub fn new(
+        global: &GlobalScope,
+        session: &XRSession,
+        data: WebVRGamepadData,
+        state: WebVRGamepadState,
+    ) -> DomRoot<XRInputSource> {
+        reflect_dom_object(
+            Box::new(XRInputSource::new_inherited(session, data, state)),
+            global,
+            XRInputSourceBinding::Wrap,
+        )
+    }
+
+    pub fn update_state(&self, state: WebVRGamepadState) {
+        *self.state.borrow_mut() = state;
+    }
+
+    pub fn pose(&self) -> WebVRPose {
+        self.state.borrow().pose
+    }
+}
+
+impl XRInputSourceMethods for XRInputSource {
+    /// https://immersive-web.github.io/webxr/#dom-xrinputsource-handedness
+    fn Handedness(&self) -> XRHandedness {
+        match self.data.hand {
+            WebVRGamepadHand::Unknown => XRHandedness::None,
+            WebVRGamepadHand::Left => XRHandedness::Left,
+            WebVRGamepadHand::Right => XRHandedness::Right,
+        }
+    }
+
+    /// https://immersive-web.github.io/webxr/#dom-xrinputsource-targetrayspace
+    fn TargetRaySpace(&self) -> DomRoot<XRSpace> {
+        self.target_ray_space.or_init(|| {
+            let global = self.global();
+            XRSpace::new_inputspace(&global, &self.session, &self)
+        })
+    }
+}