From 03e0c8082124b075c3ea4ffe3253858bb48848b7 Mon Sep 17 00:00:00 2001
From: ReFil <31960031+ReFil@users.noreply.github.com>
Date: Tue, 24 Sep 2024 13:07:03 +0100
Subject: [PATCH] feat(ext power): Support multiple ext_power GPIOS

---
 app/src/ext_power_generic.c | 37 +++++++++++++++++++++++++------------
 1 file changed, 25 insertions(+), 12 deletions(-)

diff --git a/app/src/ext_power_generic.c b/app/src/ext_power_generic.c
index 17b3ba64026..3975e038a0d 100644
--- a/app/src/ext_power_generic.c
+++ b/app/src/ext_power_generic.c
@@ -22,7 +22,8 @@
 LOG_MODULE_DECLARE(zmk, CONFIG_ZMK_LOG_LEVEL);
 
 struct ext_power_generic_config {
-    const struct gpio_dt_spec control;
+    const struct gpio_dt_spec *control;
+    const size_t control_gpios_count;
     const uint16_t init_delay_ms;
 };
 
@@ -59,9 +60,12 @@ static int ext_power_generic_enable(const struct device *dev) {
     struct ext_power_generic_data *data = dev->data;
     const struct ext_power_generic_config *config = dev->config;
 
-    if (gpio_pin_set_dt(&config->control, 1)) {
-        LOG_WRN("Failed to set ext-power control pin");
-        return -EIO;
+    for (int i = 0; i < config->control_gpios_count; i++) {
+        const struct gpio_dt_spec *gpio = &config->control[i];
+        if (gpio_pin_set_dt(gpio, 1)) {
+            LOG_WRN("Failed to set ext-power control pin %d", i);
+            return -EIO;
+        }
     }
     data->status = true;
     return ext_power_save_state();
@@ -71,10 +75,12 @@ static int ext_power_generic_disable(const struct device *dev) {
     struct ext_power_generic_data *data = dev->data;
     const struct ext_power_generic_config *config = dev->config;
 
-    if (gpio_pin_set_dt(&config->control, 0)) {
-        LOG_WRN("Failed to set ext-power control pin");
-        LOG_WRN("Failed to clear ext-power control pin");
-        return -EIO;
+    for (int i = 0; i < config->control_gpios_count; i++) {
+        const struct gpio_dt_spec *gpio = &config->control[i];
+        if (gpio_pin_set_dt(gpio, 0)) {
+            LOG_WRN("Failed to clear ext-power control pin %d", i);
+            return -EIO;
+        }
     }
     data->status = false;
     return ext_power_save_state();
@@ -144,9 +150,12 @@ SETTINGS_STATIC_HANDLER_DEFINE(ext_power, "ext_power/state", NULL, ext_power_set
 static int ext_power_generic_init(const struct device *dev) {
     const struct ext_power_generic_config *config = dev->config;
 
-    if (gpio_pin_configure_dt(&config->control, GPIO_OUTPUT_INACTIVE)) {
-        LOG_ERR("Failed to configure ext-power control pin");
-        return -EIO;
+    for (int i = 0; i < config->control_gpios_count; i++) {
+        const struct gpio_dt_spec *gpio = &config->control[i];
+        if (gpio_pin_configure_dt(gpio, GPIO_OUTPUT_INACTIVE)) {
+            LOG_ERR("Failed to configure ext-power control pin %d", i);
+            return -EIO;
+        }
     }
 
 #if IS_ENABLED(CONFIG_SETTINGS)
@@ -178,8 +187,12 @@ static int ext_power_generic_pm_action(const struct device *dev, enum pm_device_
 }
 #endif /* CONFIG_PM_DEVICE */
 
+static const struct gpio_dt_spec ext_power_control_gpios[DT_INST_PROP_LEN(0, control_gpios)] = {
+    DT_INST_FOREACH_PROP_ELEM_SEP(0, control_gpios, GPIO_DT_SPEC_GET_BY_IDX, (, ))};
+
 static const struct ext_power_generic_config config = {
-    .control = GPIO_DT_SPEC_INST_GET(0, control_gpios),
+    .control = ext_power_control_gpios,
+    .control_gpios_count = DT_INST_PROP_LEN(0, control_gpios),
     .init_delay_ms = DT_INST_PROP_OR(0, init_delay_ms, 0)};
 
 static struct ext_power_generic_data data = {