From b0e7277826f8227bf2c2ffc6587f5b05869a3a32 Mon Sep 17 00:00:00 2001
From: neroist <pmpiga1@gmail.com>
Date: Wed, 13 Mar 2024 07:16:50 -0400
Subject: [PATCH] add `--all` flag and add new alias for brightness cmd

---
 src/commands/brightness.nim |  6 +++++-
 src/commands/color.nim      |  8 ++++++--
 src/commands/color_temp.nim |  6 +++++-
 src/commands/picker.nim     |  6 +++++-
 src/commands/rgb.nim        |  6 +++++-
 src/commands/state.nim      |  6 +++++-
 src/commands/turn.nim       | 10 ++++++---
 src/common/vars.nim         |  3 +++
 src/nova.nim                | 41 ++++++++++++++++++++++++++-----------
 9 files changed, 70 insertions(+), 22 deletions(-)

diff --git a/src/commands/brightness.nim b/src/commands/brightness.nim
index 7102355..daffcd6 100644
--- a/src/commands/brightness.nim
+++ b/src/commands/brightness.nim
@@ -7,10 +7,14 @@ import puppy
 
 import ../common
 
-proc brightness*(device: int = 0; brightness = -1; output: bool = on): int =
+proc brightness*(device = 0, brightness: int = -1; output = on, all: bool = false): int =
   ## Set device brightness
 
   if not isSetup(output) or not checkDevices(device, output = output): return
+  
+  if all:
+    for i in 0..<numDevices-1:
+      discard brightness(i, brightness, output)
 
   let 
     apiKey = readFile(keyFile)
diff --git a/src/commands/color.nim b/src/commands/color.nim
index 605d6ad..e252b87 100644
--- a/src/commands/color.nim
+++ b/src/commands/color.nim
@@ -10,7 +10,7 @@ import puppy
 
 import ../common
 
-proc color*(device: int = 0; color: string = ""; output: bool = on): string =
+proc color*(device: int = 0; color: string = ""; output = on, all: bool = false): string =
   ## Set device color with an HTML/hex color code.
   ## NOTE: when called with no parameters, the device's current color will be #000000 if:
   ## 
@@ -19,7 +19,11 @@ proc color*(device: int = 0; color: string = ""; output: bool = on): string =
   ## 3. A scene is playing on the device.
 
   if not isSetup(output) or not checkDevices(device, output = output): return
-
+  
+  if all:
+    for i in 0..<numDevices-1:
+      discard color(i, color, output)
+  
   let 
     apiKey = readFile(keyFile)
     devices = parseJson readFile(devicesFile)
diff --git a/src/commands/color_temp.nim b/src/commands/color_temp.nim
index 89e90d9..d25dea9 100644
--- a/src/commands/color_temp.nim
+++ b/src/commands/color_temp.nim
@@ -29,10 +29,14 @@ func kelvinToRgb*(temp: int): tuple[r, g, b: range[0..255]] =
   else:
     result.b = int (138.5177312231 * ln(temp - 10) - 305.0447927307).clamp(0.0, 255.0)
 
-proc colorTemp*(device: int = 0; output: bool = on; temperature: int = -1): int =
+proc colorTemp*(device = 0, temperature: int = -1; output = on, all: bool = false): int =
   ## Set device color temperature in kelvin
 
   if not isSetup(output) or not checkDevices(device, output = output): return
+  
+  if all:
+    for i in 0..<numDevices-1:
+      discard colorTemp(i, temperature, output)
 
   let 
     apiKey = readFile(keyFile) 
diff --git a/src/commands/picker.nim b/src/commands/picker.nim
index 2312853..9d658d4 100644
--- a/src/commands/picker.nim
+++ b/src/commands/picker.nim
@@ -7,9 +7,13 @@ import tinydialogs
 import ../common
 import ./color
 
-proc picker*(device: int = 0; setProperty: bool = true; output: bool = on) = 
+proc picker*(device: int = 0; setProperty = true, output: bool = on, all: bool = false) = 
   ## Pick a color through a GUI (your OS's default color picker dialog)
 
+  if all:
+    for i in 0..<numDevices:
+      picker(i, setProperty, output)
+
   let 
     pickedColor = colorChooser(
       "Pick a color", 
diff --git a/src/commands/rgb.nim b/src/commands/rgb.nim
index 900c7a5..55a60e5 100644
--- a/src/commands/rgb.nim
+++ b/src/commands/rgb.nim
@@ -8,7 +8,7 @@ import puppy
 
 import ../common
 
-proc rgb*(rgb: seq[int] = @[-1, -1, -1]; device: int = 0; output: bool = on): tuple[r, g, b: int] =
+proc rgb*(rgb: seq[int] = @[-1, -1, -1]; device: int = 0; output = on, all: bool = false): tuple[r, g, b: int] =
   ## Same as command `color` but uses rgb instead of HTML codes, although it doesn't support random colors.
   ## 
   ## NOTE: when called with no parameters, the device's current color will be rgb(0, 0, 0) if:
@@ -18,6 +18,10 @@ proc rgb*(rgb: seq[int] = @[-1, -1, -1]; device: int = 0; output: bool = on): tu
 
   if not isSetup(output) or not checkDevices(device, output=output): return
 
+  if all:
+    for i in 0..<numDevices:
+      discard rgb(rgb, i, output)
+
   var rgb = rgb
 
   if len(rgb) != 3:
diff --git a/src/commands/state.nim b/src/commands/state.nim
index f0ce11e..fc95f8b 100644
--- a/src/commands/state.nim
+++ b/src/commands/state.nim
@@ -7,11 +7,15 @@ import std/json
 import ./color_temp
 import ../common
 
-proc state*(device: int = 0) =
+proc state*(device: int = 0; all: bool = false) =
   ## Output state of device
 
   if not isSetup(true) or not checkDevices(device, output=true): return
 
+  if all:
+    for i in 0..<numDevices-1:
+      state(i)
+
   var
     colorJson = %* {"r": 0, "g": 0, "b": 0}
     colorTem = 0 
diff --git a/src/commands/turn.nim b/src/commands/turn.nim
index ee2d70c..babfdb6 100644
--- a/src/commands/turn.nim
+++ b/src/commands/turn.nim
@@ -11,14 +11,18 @@ func toggle*(str: string): string =
   elif str == "off": "on"
   else: str
 
-proc turn*(device: int = 0; state: string = ""; toggle: bool = false, output: bool = on): string =
+proc turn*(device: int = 0; state: string = ""; toggle = false, output = on, all: bool = false): string =
   ## Turn device on or off
 
   if not isSetup(output) or (not checkDevices(device, output = output)): return
 
-  let apiKey = readFile(keyFile)
+  if all:
+    for i in 0..<numDevices-1:
+      discard turn(i, state, toggle, output)
+
+  let 
+    apiKey = readFile(keyFile)
 
-  let
     resp = parseJson readFile(devicesFile)
     (deviceAddr, model) = getDeviceInfo(resp, device)
 
diff --git a/src/common/vars.nim b/src/common/vars.nim
index 35f1968..c282f95 100644
--- a/src/common/vars.nim
+++ b/src/common/vars.nim
@@ -9,6 +9,9 @@ const
   OutputHelp* = "Whether or not the command will produce output."
     ## "Output" option help text.
   
+  AllHelp* = "If enabled, the command will run on all devices. Ignores `device` option."
+    ## "All" option help text.
+  
   NotSetupErrorMsg* = "Nova is not setup properly. Use the command `nova setup` to setup Nova."
     ## Error message when Nova is not setup properly.
   
diff --git a/src/nova.nim b/src/nova.nim
index 0f1972f..11102d1 100644
--- a/src/nova.nim
+++ b/src/nova.nim
@@ -46,7 +46,8 @@ dispatchMulti(
       "toggle": "Whether or not to toggle the power state of the device (if its on turn it off and " &
                 "vice-versa). This flag takes precedence over the `state` option.",
       "device": $DeviceHelp,
-      "output": $OutputHelp
+      "output": $OutputHelp,
+      "all": $AllHelp
     },
     noAutoEcho = true
   ],
@@ -56,10 +57,17 @@ dispatchMulti(
       "brightness": "The brightness you want to set on the device. Supports values 1-100 only. " &
                     "If left blank, the command will print the current brightness of the device.",
       "device": $DeviceHelp,
-      "output": $OutputHelp
+      "output": $OutputHelp,
+      "all": $AllHelp
     },
     noAutoEcho = true
   ],
+  [
+    brightness.brightness,
+    cmdName = "b",
+    doc = "Alias for 'brighness' command",
+    noAutoEcho = true
+  ],
   [
     color.color,
     help = {
@@ -68,7 +76,8 @@ dispatchMulti(
         "If left blank, will return the current color of the device. " &
         "If `color` is \"rand\" or \"random\" a random color will be displayed on the device",
       "device": $DeviceHelp,
-      "output": $OutputHelp
+      "output": $OutputHelp,
+      "all": $AllHelp
     },
     noAutoEcho = true
   ],
@@ -79,19 +88,26 @@ dispatchMulti(
       "temperature": "The color temperature you want to set on the device. " &
                       "Has to be in the valid range your Govee device supports.",
       "device": $DeviceHelp,
-      "output": $OutputHelp
+      "output": $OutputHelp,
+      "all": $AllHelp
     },
     noAutoEcho = true
   ],
   [
     state.state,
-    help = {"device": $DeviceHelp}
+    help = {
+      "device": $DeviceHelp,
+      "all": $AllHelp
+    }
   ],
   [
     state.state,
     cmdName = "device",
-    doc = "Alias for state",
-    help = {"device": $DeviceHelp}
+    doc = "Alias for 'state' command",
+    help = {
+      "device": $DeviceHelp,
+      "all": $AllHelp
+    }
   ],
   [
     rgb.rgb,
@@ -100,17 +116,18 @@ dispatchMulti(
       "rgb": "The color you want to set on the device in an RGB format. " &
               "Has to be 3 numbers seperated by a space. " &
               "If left blank, the command will print the current color in an RGB function.",
-      "output": $OutputHelp
+      "output": $OutputHelp,
+      "all": $AllHelp
     },
-    noAutoEcho = true,
-    cmdName = "rgb"
+    noAutoEcho = true
   ],
   [
     picker.picker,
     help = {
       "device": $DeviceHelp,
       "output": $OutputHelp,
-      "set_property": "Whether or not to set `device`'s color to the color chosen."
+      "set_property": "Whether or not to set `device`'s color to the color chosen.",
+      "all": $AllHelp
     }
   ],
   [
@@ -129,7 +146,7 @@ dispatchMulti(
     license,
     help = {
       "browser": "Whether or not to open the license in the default " &
-                  "browser, or to just print the license text to the terminal"
+                 "browser, or to just print the license text to the terminal"
     }
   ],
   [docs]