1 files changed, 132 insertions, 0 deletions
diff --git a/components/script/dom/xrinputsource.rs b/components/script/dom/xrinputsource.rs
new file mode 100644
index 00000000000..248fb6d737a
--- /dev/null
+++ b/components/script/dom/xrinputsource.rs
@@ -0,0 +1,132 @@
+/* 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::codegen::Bindings::XRInputSourceBinding::{
+    XRHandedness, XRInputSourceMethods, XRTargetRayMode,
+};
+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::xrhand::XRHand;
+use crate::dom::xrsession::XRSession;
+use crate::dom::xrspace::XRSpace;
+use crate::realms::enter_realm;
+use crate::script_runtime::JSContext;
+use dom_struct::dom_struct;
+use js::conversions::ToJSValConvertible;
+use js::jsapi::Heap;
+use js::jsval::{JSVal, UndefinedValue};
+use webxr_api::{Handedness, InputId, InputSource, TargetRayMode};
+
+#[dom_struct]
+pub struct XRInputSource {
+    reflector: Reflector,
+    session: Dom<XRSession>,
+    #[ignore_malloc_size_of = "Defined in rust-webxr"]
+    info: InputSource,
+    target_ray_space: MutNullableDom<XRSpace>,
+    grip_space: MutNullableDom<XRSpace>,
+    hand: MutNullableDom<XRHand>,
+    #[ignore_malloc_size_of = "mozjs"]
+    profiles: Heap<JSVal>,
+}
+
+impl XRInputSource {
+    pub fn new_inherited(session: &XRSession, info: InputSource) -> XRInputSource {
+        XRInputSource {
+            reflector: Reflector::new(),
+            session: Dom::from_ref(session),
+            info,
+            target_ray_space: Default::default(),
+            grip_space: Default::default(),
+            hand: Default::default(),
+            profiles: Heap::default(),
+        }
+    }
+
+    #[allow(unsafe_code)]
+    pub fn new(
+        global: &GlobalScope,
+        session: &XRSession,
+        info: InputSource,
+    ) -> DomRoot<XRInputSource> {
+        let source = reflect_dom_object(
+            Box::new(XRInputSource::new_inherited(session, info)),
+            global,
+        );
+
+        let _ac = enter_realm(&*global);
+        let cx = global.get_cx();
+        unsafe {
+            rooted!(in(*cx) let mut profiles = UndefinedValue());
+            source.info.profiles.to_jsval(*cx, profiles.handle_mut());
+            source.profiles.set(profiles.get());
+        }
+        source
+    }
+
+    pub fn id(&self) -> InputId {
+        self.info.id
+    }
+
+    pub fn session(&self) -> &XRSession {
+        &self.session
+    }
+}
+
+impl XRInputSourceMethods for XRInputSource {
+    /// https://immersive-web.github.io/webxr/#dom-xrinputsource-handedness
+    fn Handedness(&self) -> XRHandedness {
+        match self.info.handedness {
+            Handedness::None => XRHandedness::None,
+            Handedness::Left => XRHandedness::Left,
+            Handedness::Right => XRHandedness::Right,
+        }
+    }
+
+    /// https://immersive-web.github.io/webxr/#dom-xrinputsource-targetraymode
+    fn TargetRayMode(&self) -> XRTargetRayMode {
+        match self.info.target_ray_mode {
+            TargetRayMode::Gaze => XRTargetRayMode::Gaze,
+            TargetRayMode::TrackedPointer => XRTargetRayMode::Tracked_pointer,
+            TargetRayMode::Screen => XRTargetRayMode::Screen,
+        }
+    }
+
+    /// 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, false)
+        })
+    }
+
+    /// https://immersive-web.github.io/webxr/#dom-xrinputsource-gripspace
+    fn GetGripSpace(&self) -> Option<DomRoot<XRSpace>> {
+        if self.info.supports_grip {
+            Some(self.grip_space.or_init(|| {
+                let global = self.global();
+                XRSpace::new_inputspace(&global, &self.session, &self, true)
+            }))
+        } else {
+            None
+        }
+    }
+    // https://immersive-web.github.io/webxr/#dom-xrinputsource-profiles
+    fn Profiles(&self, _cx: JSContext) -> JSVal {
+        self.profiles.get()
+    }
+
+    // https://github.com/immersive-web/webxr-hands-input/blob/master/explainer.md
+    fn GetHand(&self) -> Option<DomRoot<XRHand>> {
+        if let Some(ref hand) = self.info.hand_support {
+            Some(
+                self.hand
+                    .or_init(|| XRHand::new(&self.global(), &self, hand.clone())),
+            )
+        } else {
+            None
+        }
+    }
+}