framework_lib: Add support for RgbKbdSetColor HC

Signed-off-by: Daniel Schaefer <[email protected]>

Author: JohnAZoidberg

framework_lib/src/chromium_ec/command.rs

@@ -46,6 +46,8 @@ pub enum EcCommands {
     RebootEc = 0x00D2,
     /// Get information about PD controller power
     UsbPdPowerInfo = 0x0103,
+    RgbKbdSetColor = 0x013A,
+    RgbKbd = 0x013B,
 
     // Framework specific commands
     /// Configure the behavior of the flash notify

framework_lib/src/chromium_ec/commands.rs

@@ -518,6 +518,37 @@ impl EcRequest<EcResponseUsbPdPowerInfo> for EcRequestUsbPdPowerInfo {
     }
 }
 
+// TODO: Actually 128, but if we go above ~80 EC returns REQUEST_TRUNCATED
+// At least when I use the portio driver
+pub const EC_RGBKBD_MAX_KEY_COUNT: usize = 64;
+
+#[repr(C, packed)]
+#[derive(Default, Clone, Copy, Debug)]
+pub struct RgbS {
+    pub r: u8,
+    pub g: u8,
+    pub b: u8,
+}
+
+#[repr(C, packed)]
+pub struct EcRequestRgbKbdSetColor {
+    /// Specifies the starting key ID whose color is being changed
+    pub start_key: u8,
+    /// Specifies # of elements in color
+    pub length: u8,
+    /// RGB color data array of length up to MAX_KEY_COUNT
+    pub color: [RgbS; EC_RGBKBD_MAX_KEY_COUNT],
+}
+
+#[repr(C, packed)]
+pub struct EcResponseRgbKbdSetColor {}
+
+impl EcRequest<EcResponseRgbKbdSetColor> for EcRequestRgbKbdSetColor {
+    fn command_id() -> EcCommands {
+        EcCommands::RgbKbdSetColor
+    }
+}
+
 // --- Framework Specific commands ---
 
 #[repr(C, packed)]

framework_lib/src/chromium_ec/mod.rs

@@ -921,6 +921,21 @@ impl CrosEc {
         let res = request.send_command(self)?;
         Ok(res.val == 1)
     }
+
+    pub fn rgbkbd_set_color(&self, start_key: u8, colors: Vec<RgbS>) -> EcResult<()> {
+        let mut request = EcRequestRgbKbdSetColor {
+            start_key,
+            length: colors.len() as u8,
+            color: [(); EC_RGBKBD_MAX_KEY_COUNT].map(|()| Default::default()),
+        };
+
+        for (i, color) in colors.iter().enumerate() {
+            request.color[i] = *color;
+        }
+
+        let _res = request.send_command(self)?;
+        Ok(())
+    }
 }
 
 #[cfg_attr(not(feature = "uefi"), derive(clap::ValueEnum))]