Allow overriding charge limit and charge full

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

Author: JohnAZoidberg

framework_lib/src/chromium_ec/mod.rs

@@ -279,6 +279,26 @@ impl CrosEc {
         Ok((status.microphone == 1, status.camera == 1))
     }
 
+    /// Let charge fully once, if currently a charge limit is set
+    pub fn charge_limit_override(&self) -> EcResult<()> {
+        let params = &[ChargeLimitControlModes::Override as u8, 0, 0];
+        let data = self.send_command(EcCommands::ChargeLimitControl as u16, 0, params)?;
+
+        util::assert_win_len(data.len(), 0);
+
+        Ok(())
+    }
+
+    /// Disable all charge limits
+    pub fn charge_limit_disable(&self) -> EcResult<()> {
+        let params = &[ChargeLimitControlModes::Disable as u8, 0, 0];
+        let data = self.send_command(EcCommands::ChargeLimitControl as u16, 0, params)?;
+
+        util::assert_win_len(data.len(), 0);
+
+        Ok(())
+    }
+
     pub fn set_charge_limit(&self, min: u8, max: u8) -> EcResult<()> {
         // Sending bytes manually because the Set command, as opposed to the Get command,
         // does not return any data
@@ -310,8 +330,8 @@ impl CrosEc {
     pub fn set_fp_led_level(&self, level: FpLedBrightnessLevel) -> EcResult<()> {
         // Sending bytes manually because the Set command, as opposed to the Get command,
         // does not return any data
-        let limits = &[level as u8, 0x00];
-        let data = self.send_command(EcCommands::FpLedLevelControl as u16, 0, limits)?;
+        let params = &[level as u8, 0x00];
+        let data = self.send_command(EcCommands::FpLedLevelControl as u16, 0, params)?;
 
         util::assert_win_len(data.len(), 0);
 

framework_lib/src/commandline/clap_std.rs

@@ -124,7 +124,15 @@ struct ClapCli {
     #[arg(long)]
     input_deck_mode: Option<InputDeckModeArg>,
 
-    /// Get or set max charge limit
+    /// Temporarily remove the charge limit and charge full
+    #[arg(long)]
+    charge_full: bool,
+
+    /// Remove min/max charge limit (Overwritten by BIOS on reboot)
+    #[arg(long)]
+    charge_limit_disable: bool,
+
+    /// Get or set max charge limit (Overwritten by BIOS on reboot)
     #[arg(long)]
     charge_limit: Option<Option<u8>>,
 
@@ -250,6 +258,8 @@ pub fn parse(args: &[String]) -> Cli {
         intrusion: args.intrusion,
         inputmodules: args.inputmodules,
         input_deck_mode: args.input_deck_mode,
+        charge_full: args.charge_full,
+        charge_limit_disable: args.charge_limit_disable,
         charge_limit: args.charge_limit,
         fp_brightness: args.fp_brightness,
         kblight: args.kblight,

framework_lib/src/commandline/mod.rs

@@ -147,6 +147,8 @@ pub struct Cli {
     pub intrusion: bool,
     pub inputmodules: bool,
     pub input_deck_mode: Option<InputDeckModeArg>,
+    pub charge_full: bool,
+    pub charge_limit_disable: bool,
     pub charge_limit: Option<Option<u8>>,
     pub fp_brightness: Option<Option<FpBrightnessArg>>,
     pub kblight: Option<Option<u8>>,
@@ -718,6 +720,10 @@ pub fn run_with_args(args: &Cli, _allupdate: bool) -> i32 {
     } else if let Some(mode) = &args.input_deck_mode {
         println!("Set mode to: {:?}", mode);
         ec.set_input_deck_mode((*mode).into()).unwrap();
+    } else if args.charge_full {
+        print_err(ec.charge_limit_override());
+    } else if args.charge_limit_disable {
+        print_err(ec.charge_limit_disable());
     } else if let Some(maybe_limit) = args.charge_limit {
         print_err(handle_charge_limit(&ec, maybe_limit));
     } else if let Some(maybe_brightness) = &args.fp_brightness {

framework_lib/src/commandline/uefi.rs

@@ -81,6 +81,8 @@ pub fn parse(args: &[String]) -> Cli {
         intrusion: false,
         inputmodules: false,
         input_deck_mode: None,
+        charge_full: false,
+        charge_limit_disable: false,
         charge_limit: None,
         fp_brightness: None,
         kblight: None,
@@ -172,6 +174,12 @@ pub fn parse(args: &[String]) -> Cli {
                 None
             };
             found_an_option = true;
+        } else if arg == "--charge-full" {
+            cli.charge_full = true;
+            found_an_option = true;
+        } else if arg == "--charge-limit-override" {
+            cli.charge_limit_disable = true;
+            found_an_option = true;
         } else if arg == "--charge-limit" {
             cli.charge_limit = if args.len() > i + 1 {
                 if let Ok(percent) = args[i + 1].parse::<u8>() {