ccgx: Add enable_ports function

To disable or enable all ports of a PD controller

framwork_tool --pd-disable 0
framwork_tool --pd-enable 1

framwork_tool --pd-disable 0
framwork_tool --pd-enable 1

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

Author: JohnAZoidberg

framework_lib/src/ccgx/device.rs

@@ -366,4 +366,10 @@ impl PdController {
         )?;
         Ok(())
     }
+
+    pub fn enable_ports(&self, enable: bool) -> EcResult<()> {
+        let mask = if enable { 0b11 } else { 0b00 };
+        self.ccgx_write(ControlRegisters::PdPortsEnable, &[mask])?;
+        Ok(())
+    }
 }

framework_lib/src/commandline/clap_std.rs

@@ -87,6 +87,14 @@ struct ClapCli {
     #[arg(long)]
     pd_reset: Option<u8>,
 
+    /// Disable all ports on a specific PD controller (for debugging only)
+    #[arg(long)]
+    pd_disable: Option<u8>,
+
+    /// Enable all ports on a specific PD controller (for debugging only)
+    #[arg(long)]
+    pd_enable: Option<u8>,
+
     /// Show details about connected DP or HDMI Expansion Cards
     #[arg(long)]
     dp_hdmi_info: bool,
@@ -381,6 +389,8 @@ pub fn parse(args: &[String]) -> Cli {
         pdports: args.pdports,
         pd_info: args.pd_info,
         pd_reset: args.pd_reset,
+        pd_disable: args.pd_disable,
+        pd_enable: args.pd_enable,
         dp_hdmi_info: args.dp_hdmi_info,
         dp_hdmi_update: args
             .dp_hdmi_update

framework_lib/src/commandline/mod.rs

@@ -157,6 +157,8 @@ pub struct Cli {
     pub privacy: bool,
     pub pd_info: bool,
     pub pd_reset: Option<u8>,
+    pub pd_disable: Option<u8>,
+    pub pd_enable: Option<u8>,
     pub dp_hdmi_info: bool,
     pub dp_hdmi_update: Option<String>,
     pub audio_card_info: bool,
@@ -1011,6 +1013,28 @@ pub fn run_with_args(args: &Cli, _allupdate: bool) -> i32 {
                 Ok(())
             }
         });
+    } else if let Some(pd) = args.pd_disable {
+        println!("Disabling PD {}...", pd);
+        print_err(match pd {
+            0 => PdController::new(PdPort::Left01, ec.clone()).enable_ports(false),
+            1 => PdController::new(PdPort::Right23, ec.clone()).enable_ports(false),
+            2 => PdController::new(PdPort::Back, ec.clone()).enable_ports(false),
+            _ => {
+                error!("PD {} does not exist", pd);
+                Ok(())
+            }
+        });
+    } else if let Some(pd) = args.pd_enable {
+        println!("Enabling PD {}...", pd);
+        print_err(match pd {
+            0 => PdController::new(PdPort::Left01, ec.clone()).enable_ports(true),
+            1 => PdController::new(PdPort::Right23, ec.clone()).enable_ports(true),
+            2 => PdController::new(PdPort::Back, ec.clone()).enable_ports(true),
+            _ => {
+                error!("PD {} does not exist", pd);
+                Ok(())
+            }
+        });
     } else if args.dp_hdmi_info {
         #[cfg(feature = "hidapi")]
         print_dp_hdmi_details(true);

framework_lib/src/commandline/uefi.rs

@@ -71,6 +71,8 @@ pub fn parse(args: &[String]) -> Cli {
         pdports: false,
         pd_info: false,
         pd_reset: None,
+        pd_disable: None,
+        pd_enable: None,
         dp_hdmi_info: false,
         dp_hdmi_update: None,
         audio_card_info: false,
@@ -535,6 +537,38 @@ pub fn parse(args: &[String]) -> Cli {
                 None
             };
             found_an_option = true;
+        } else if arg == "--pd-disable" {
+            cli.pd_reset = if args.len() > i + 1 {
+                if let Ok(pd) = args[i + 1].parse::<u8>() {
+                    Some(pd)
+                } else {
+                    println!(
+                        "Invalid value for --pd-disable: '{}'. Must be 0 or 1.",
+                        args[i + 1],
+                    );
+                    None
+                }
+            } else {
+                println!("--pd-disable requires specifying the PD controller");
+                None
+            };
+            found_an_option = true;
+        } else if arg == "--pd-enable" {
+            cli.pd_enable = if args.len() > i + 1 {
+                if let Ok(pd) = args[i + 1].parse::<u8>() {
+                    Some(pd)
+                } else {
+                    println!(
+                        "Invalid value for --pd-enable: '{}'. Must be 0 or 1.",
+                        args[i + 1],
+                    );
+                    None
+                }
+            } else {
+                println!("--pd-enable requires specifying the PD controller");
+                None
+            };
+            found_an_option = true;
         } else if arg == "--privacy" {
             cli.privacy = true;
             found_an_option = true;