head-management: add vrr-state-v1 extension

src/ifs/head_management/head_management_macros.rs

@@ -399,4 +399,5 @@ declare_extensions! {
     mode_setter_v1: ModeSetterV1,
     physical_display_info_v1: PhysicalDisplayInfoV1,
     non_desktop_info_v1: NonDesktopInfoV1,
+    vrr_state_v1: VrrStateV1,
 }

src/ifs/head_management/jay_head_ext.rs

@@ -9,3 +9,4 @@ pub(super) mod jay_head_ext_mode_info_v1;
 pub(super) mod jay_head_ext_mode_setter_v1;
 pub(super) mod jay_head_ext_non_desktop_info_v1;
 pub(super) mod jay_head_ext_physical_display_info_v1;
+pub(super) mod jay_head_ext_vrr_state_v1;

src/ifs/head_management/jay_head_ext/jay_head_ext_vrr_state_v1.rs

@@ -0,0 +1,58 @@
+use {
+    crate::{
+        ifs::head_management::HeadState,
+        wire::{
+            jay_head_ext_vrr_state_v1::{
+                Capable, Enabled, JayHeadExtVrrStateV1RequestHandler, Reset,
+            },
+            jay_head_manager_ext_vrr_state_v1::JayHeadManagerExtVrrStateV1RequestHandler,
+        },
+    },
+    std::rc::Rc,
+};
+
+impl_vrr_state_v1! {
+    version = 1,
+    after_announce = after_announce,
+    after_transaction = after_transaction,
+}
+
+impl HeadName {
+    fn after_announce(&self, shared: &HeadState) {
+        self.send_state(shared);
+    }
+
+    fn after_transaction(&self, shared: &HeadState, tran: &HeadState) {
+        let shared_capable = shared.monitor_info.as_ref().map(|m| m.vrr_capable);
+        let tran_capable = tran.monitor_info.as_ref().map(|m| m.vrr_capable);
+        if (shared.vrr, shared_capable) != (tran.vrr, tran_capable) {
+            self.send_state(shared);
+        }
+    }
+
+    pub(in super::super) fn send_state(&self, state: &HeadState) {
+        self.client.event(Reset { self_id: self.id });
+        if let Some(mi) = &state.monitor_info
+            && mi.vrr_capable
+        {
+            self.client.event(Capable { self_id: self.id });
+        }
+        if state.vrr {
+            self.client.event(Enabled { self_id: self.id });
+        }
+    }
+}
+
+impl JayHeadManagerExtVrrStateV1RequestHandler for MgrName {
+    type Error = ErrorName;
+
+    mgr_common_req!();
+}
+
+impl JayHeadExtVrrStateV1RequestHandler for HeadName {
+    type Error = ErrorName;
+
+    head_common_req!();
+}
+
+error!();