Add --rgbkbd commandline

Works like this:
```
> cargo build && sudo ./target/debug/framework_tool --rgbkbd 0 16711680 65280 255
```

Signed-off-by: Daniel Schaefer <dhs@frame.work>

Author: JohnAZoidberg

framework_lib/src/commandline/clap_std.rs

@@ -3,6 +3,7 @@
 //! as well as on the UEFI shell tool.
 use clap::Parser;
 
+use crate::chromium_ec::commands::EC_RGBKBD_MAX_KEY_COUNT;
 use crate::chromium_ec::CrosEcDriverType;
 use crate::commandline::{
     Cli, ConsoleArg, FpBrightnessArg, HardwareDeviceType, InputDeckModeArg, RebootEcArg,
@@ -145,6 +146,13 @@ struct ClapCli {
     #[arg(long)]
     kblight: Option<Option<u8>>,
 
+    /// Set the color of <key> to <RGB>. Multiple colors for adjacent keys can be set at once.
+    /// <key> <RGB> [<RGB> ...]
+    /// Example: 0 0xFF000 0x00FF00 0x0000FF
+    #[clap(num_args = 2..EC_RGBKBD_MAX_KEY_COUNT)]
+    #[arg(long)]
+    rgbkbd: Vec<u64>,
+
     /// Set tablet mode override
     #[clap(value_enum)]
     #[arg(long)]
@@ -269,6 +277,7 @@ pub fn parse(args: &[String]) -> Cli {
         get_gpio: args.get_gpio,
         fp_brightness: args.fp_brightness,
         kblight: args.kblight,
+        rgbkbd: args.rgbkbd,
         tablet_mode: args.tablet_mode,
         console: args.console,
         reboot_ec: args.reboot_ec,

framework_lib/src/commandline/mod.rs

@@ -35,6 +35,7 @@ use crate::chromium_ec;
 use crate::chromium_ec::commands::DeckStateMode;
 use crate::chromium_ec::commands::FpLedBrightnessLevel;
 use crate::chromium_ec::commands::RebootEcCmd;
+use crate::chromium_ec::commands::RgbS;
 use crate::chromium_ec::commands::TabletModeOverride;
 use crate::chromium_ec::EcResponseStatus;
 use crate::chromium_ec::{print_err, EcFlashType};
@@ -161,6 +162,7 @@ pub struct Cli {
     pub get_gpio: Option<String>,
     pub fp_brightness: Option<Option<FpBrightnessArg>>,
     pub kblight: Option<Option<u8>>,
+    pub rgbkbd: Vec<u64>,
     pub tablet_mode: Option<TabletModeArg>,
     pub console: Option<ConsoleArg>,
     pub reboot_ec: Option<RebootEcArg>,
@@ -746,6 +748,21 @@ pub fn run_with_args(args: &Cli, _allupdate: bool) -> i32 {
     } else if let Some(Some(kblight)) = args.kblight {
         assert!(kblight <= 100);
         ec.set_keyboard_backlight(kblight);
+    } else if !args.rgbkbd.is_empty() {
+        if args.rgbkbd.len() < 2 {
+            println!(
+                "Must provide at least 2 arguments. Provided only: {}",
+                args.rgbkbd.len()
+            );
+        } else {
+            let start_key = args.rgbkbd[0] as u8;
+            let colors = args.rgbkbd[1..].iter().map(|color| RgbS {
+                r: ((color & 0x00FF0000) >> 16) as u8,
+                g: ((color & 0x0000FF00) >> 8) as u8,
+                b: (color & 0x000000FF) as u8,
+            });
+            ec.rgbkbd_set_color(start_key, colors.collect()).unwrap();
+        }
     } else if let Some(None) = args.kblight {
         print!("Keyboard backlight: ");
         if let Some(percentage) = print_err(ec.get_keyboard_backlight()) {

framework_lib/src/commandline/uefi.rs

@@ -86,6 +86,7 @@ pub fn parse(args: &[String]) -> Cli {
         get_gpio: None,
         fp_brightness: None,
         kblight: None,
+        rgbkbd: Vec<64>,
         tablet_mode: None,
         console: None,
         reboot_ec: None,
@@ -215,6 +216,18 @@ pub fn parse(args: &[String]) -> Cli {
                 Some(None)
             };
             found_an_option = true;
+        } else if arg == "--rgbkbd" {
+            cli.rgbkbd = if args.len() > i + 2 {
+                let colors = Vec::<u64>::new();
+                for color_i in i + 1..args.len() {
+                    // TODO: Fail parsing instead of unwrap()
+                    colors.push(args[color_i].parse::<u64>().unwrap());
+                }
+                colors
+            } else {
+                println!("--rgbkbd requires at least arguments, the start key and an RGB value");
+                Vec::new()
+            }
         } else if arg == "--tablet-mode" {
             cli.tablet_mode = if args.len() > i + 1 {
                 let tablet_mode_arg = &args[i + 1];