From 42eb47891b121120d11e701c68ea53b8dba75419 Mon Sep 17 00:00:00 2001
From: Pedro Lamas <pedrolamas@gmail.com>
Date: Wed, 8 Jan 2025 15:22:25 +0000
Subject: [PATCH] feat: control printer power via Moonraker

Signed-off-by: Pedro Lamas <pedrolamas@gmail.com>
---
 src/components/common/KlippyStatusCard.vue  | 15 ++++-
 src/components/common/SystemControl.vue     | 25 ++++++++
 src/components/layout/AppBar.vue            |  8 ++-
 src/components/settings/GeneralSettings.vue | 64 +++++++++++++++++++--
 src/locales/en.yaml                         |  7 +++
 src/mixins/state.ts                         | 13 +++++
 src/store/config/state.ts                   |  1 +
 src/store/config/types.ts                   |  1 +
 8 files changed, 127 insertions(+), 7 deletions(-)

diff --git a/src/components/common/KlippyStatusCard.vue b/src/components/common/KlippyStatusCard.vue
index 5857790223..ab5c207299 100644
--- a/src/components/common/KlippyStatusCard.vue
+++ b/src/components/common/KlippyStatusCard.vue
@@ -18,7 +18,20 @@
         >
           <v-row>
             <v-col
-              v-if="klippyStateMessage !== 'Printer is ready'"
+              v-if="printerPoweredOff"
+              cols="12"
+            >
+              <v-alert
+                text
+                dense
+                type="error"
+                class="ma-0"
+              >
+                <span v-html="$t('app.general.error.printer_powered_off')" />
+              </v-alert>
+            </v-col>
+            <v-col
+              v-else-if="klippyStateMessage !== 'Printer is ready'"
               cols="12"
             >
               <v-alert
diff --git a/src/components/common/SystemControl.vue b/src/components/common/SystemControl.vue
index 3360b63bc8..7ada63d811 100644
--- a/src/components/common/SystemControl.vue
+++ b/src/components/common/SystemControl.vue
@@ -34,6 +34,24 @@
       </v-tooltip>
     </template>
 
+    <template v-else-if="printerPoweredOff">
+      <v-tooltip bottom>
+        <template #activator="{ on, attrs }">
+          <app-btn
+            v-bind="attrs"
+            block
+            color="primary"
+            class="mb-2"
+            v-on="on"
+            @click="printerPowerOn"
+          >
+            {{ $t('app.general.btn.power_on_printer') }}
+          </app-btn>
+        </template>
+        <span>{{ $t('app.general.tooltip.power_on_printer') }}</span>
+      </v-tooltip>
+    </template>
+
     <template v-else>
       <v-tooltip bottom>
         <template #activator="{ on, attrs }">
@@ -87,6 +105,7 @@ import { Component, Mixins } from 'vue-property-decorator'
 import FilesMixin from '@/mixins/files'
 import StateMixin from '@/mixins/state'
 import ServicesMixin from '@/mixins/services'
+import { SocketActions } from '@/api/socketActions'
 
 @Component({})
 export default class SystemControl extends Mixins(StateMixin, FilesMixin, ServicesMixin) {
@@ -97,5 +116,11 @@ export default class SystemControl extends Mixins(StateMixin, FilesMixin, Servic
   getMoonrakerLog () {
     this.downloadFile('moonraker.log', '')
   }
+
+  printerPowerOn () {
+    const printerPowerDevice: string = this.$store.state.config.uiSettings.general.printerPowerDevice ?? 'printer'
+
+    SocketActions.machineDevicePowerToggle(printerPowerDevice, 'on')
+  }
 }
 </script>
diff --git a/src/components/layout/AppBar.vue b/src/components/layout/AppBar.vue
index 5f7254e926..60f08123a5 100644
--- a/src/components/layout/AppBar.vue
+++ b/src/components/layout/AppBar.vue
@@ -265,6 +265,8 @@ export default class AppBar extends Mixins(StateMixin, ServicesMixin, FilesMixin
       case 'klipper': {
         const device = this.$store.getters['printer/getPinByName'](name) as OutputPin | undefined
 
+        if (!device) return null
+
         return {
           type,
           name: device?.prettyName ?? name,
@@ -273,11 +275,13 @@ export default class AppBar extends Mixins(StateMixin, ServicesMixin, FilesMixin
       }
 
       default: {
-        const device = this.$store.getters['power/getDeviceByName'](topNavPowerToggle) as Device
+        const device = this.$store.getters['power/getDeviceByName'](topNavPowerToggle) as Device | undefined
+
+        if (!device) return null
 
         return {
           type: 'moonraker' as const,
-          name: topNavPowerToggle,
+          name: this.$filters.prettyCase(topNavPowerToggle),
           device
         }
       }
diff --git a/src/components/settings/GeneralSettings.vue b/src/components/settings/GeneralSettings.vue
index a264215d84..5f4bfb4fe3 100644
--- a/src/components/settings/GeneralSettings.vue
+++ b/src/components/settings/GeneralSettings.vue
@@ -99,6 +99,19 @@
 
       <v-divider />
 
+      <app-setting :title="$t('app.setting.label.printer_power_device')">
+        <v-select
+          v-model="printerPowerDevice"
+          filled
+          dense
+          single-line
+          hide-details="auto"
+          :items="printerPowerDevicesList"
+        />
+      </app-setting>
+
+      <v-divider />
+
       <app-setting :title="$t('app.setting.label.power_toggle_in_top_nav')">
         <v-select
           v-model="topNavPowerToggle"
@@ -106,7 +119,7 @@
           dense
           single-line
           hide-details="auto"
-          :items="[{ text: $tc('app.setting.label.none'), value: null }, ...powerDevicesList]"
+          :items="topNavPowerToggleDevicesList"
         />
       </app-setting>
 
@@ -373,6 +386,39 @@ export default class GeneralSettings extends Mixins(StateMixin) {
     })
   }
 
+  get printerPowerDevice (): string | null {
+    return this.$store.state.config.uiSettings.general.printerPowerDevice
+  }
+
+  set printerPowerDevice (value: string | null) {
+    this.$store.dispatch('config/saveByPath', {
+      path: 'uiSettings.general.printerPowerDevice',
+      value,
+      server: true
+    })
+  }
+
+  get printerPowerDevicesList () {
+    const devices = this.$store.getters['power/getDevices'] as Device[]
+
+    const deviceEntries = devices.map(device => ({
+      text: `${this.$filters.prettyCase(device.device)} (${device.type})`,
+      value: device.device
+    }))
+
+    const autoDeviceName = devices.some(device => device.device.toLowerCase() === 'printer')
+      ? 'Printer'
+      : this.$tc('app.setting.label.none')
+
+    return [
+      {
+        text: `${this.$tc('app.setting.label.auto')} (${autoDeviceName})`,
+        value: null
+      },
+      ...deviceEntries
+    ]
+  }
+
   get topNavPowerToggle (): string | null {
     return this.$store.state.config.uiSettings.general.topNavPowerToggle
   }
@@ -385,12 +431,15 @@ export default class GeneralSettings extends Mixins(StateMixin) {
     })
   }
 
-  get powerDevicesList () {
+  get topNavPowerToggleDevicesList () {
     const devices = this.$store.getters['power/getDevices'] as Device[]
     const deviceEntries = devices.length
       ? [
           { header: 'Moonraker' },
-          ...devices.map(device => ({ text: device.device, value: device.device }))
+          ...devices.map(device => ({
+            text: `${this.$filters.prettyCase(device.device)} (${device.type})`,
+            value: device.device
+          }))
         ]
       : []
 
@@ -398,11 +447,18 @@ export default class GeneralSettings extends Mixins(StateMixin) {
     const pinEntries = pins.length
       ? [
           { header: 'Klipper' },
-          ...pins.map(outputPin => ({ text: outputPin.prettyName, value: `${outputPin.name}:klipper` }))
+          ...pins.map(outputPin => ({
+            text: outputPin.prettyName,
+            value: `${outputPin.name}:klipper`
+          }))
         ]
       : []
 
     return [
+      {
+        text: this.$tc('app.setting.label.none'),
+        value: null
+      },
       ...deviceEntries,
       ...pinEntries
     ]
diff --git a/src/locales/en.yaml b/src/locales/en.yaml
index 59d62c85a8..4fb678284a 100644
--- a/src/locales/en.yaml
+++ b/src/locales/en.yaml
@@ -183,6 +183,7 @@ app:
       more_information: More information
       multiply: Multiply
       pause: Pause
+      power_on_printer: Power On Printer
       preheat: Preheat
       presets: Presets
       preview_gcode: Preview Gcode
@@ -238,6 +239,9 @@ app:
       failed_components: >-
         Moonraker has failed plugins, please check your logs, update your
         configuration and restart moonraker.
+      printer_powered_off: >-
+        The printer is currently powered off.<br/><br/>
+        To power on the printer, please click the button on the left.
     label:
       accel_to_decel: Accel to Decel
       acceleration: Acceleration
@@ -480,6 +484,7 @@ app:
       estop: Emergency Stop
       managed_by_moonraker: Managed by your moonraker configuration
       notifications: Notifications
+      power_on_printer: Turns on the power to the printer
       reload_klipper: Reloads klipper configuration.
       reload_restart_klipper: Reloads klipper configuration and restarts MCU's.
       restart_klipper: Restarts the klipper system service.
@@ -557,6 +562,7 @@ app:
       auto_follow_on_file_load: Automatically follow progress on file load
       auto_load_on_print_start: Automatically load file on print start
       auto_load_mobile_on_print_start: Automatically load file on mobile devices
+      auto: Auto
       axes: Axes
       camera_flip_x: Flip horizontally
       camera_flip_y: Flip vertically
@@ -627,6 +633,7 @@ app:
       print_in_progress_layout: Print in Progress layout
       print_progress_calculation: Print Progress calculation
       printer_name: Printer Name
+      printer_power_device: Printer power device
       reset: Reset settings
       retraction_icon_size: Retraction Icon Size
       right_y: Right Y-Axis
diff --git a/src/mixins/state.ts b/src/mixins/state.ts
index 94de31d489..79d7a9f453 100644
--- a/src/mixins/state.ts
+++ b/src/mixins/state.ts
@@ -2,6 +2,7 @@ import Vue from 'vue'
 import { SocketActions } from '@/api/socketActions'
 import { Component } from 'vue-property-decorator'
 import type { Macro } from '@/store/macros/types'
+import type { Device } from '@/store/power/types'
 
 @Component
 export default class StateMixin extends Vue {
@@ -74,6 +75,18 @@ export default class StateMixin extends Vue {
     return this.printerState.toLowerCase() === 'printing'
   }
 
+  get printerPoweredOff (): boolean {
+    if (this.klippyConnected) {
+      return false
+    }
+
+    const printerPowerDevice: string = this.$store.state.config.uiSettings.general.printerPowerDevice ?? 'printer'
+
+    const device = this.$store.getters['power/getDeviceByName'](printerPowerDevice) as Device | undefined
+
+    return device?.status === 'off'
+  }
+
   /**
    * Indicates if we have a valid wait(s).
    * Supports a single string or a list of.
diff --git a/src/store/config/state.ts b/src/store/config/state.ts
index 81673d2d15..05014a854e 100644
--- a/src/store/config/state.ts
+++ b/src/store/config/state.ts
@@ -58,6 +58,7 @@ export const defaultState = (): ConfigState => {
         showUploadAndPrint: true,
         flipConsoleLayout: false,
         cameraFullscreenAction: 'embed',
+        printerPowerDevice: null,
         topNavPowerToggle: null,
         showManualProbeDialogAutomatically: true,
         showBedScrewsAdjustDialogAutomatically: true,
diff --git a/src/store/config/types.ts b/src/store/config/types.ts
index 6c470f7880..4687929d47 100644
--- a/src/store/config/types.ts
+++ b/src/store/config/types.ts
@@ -94,6 +94,7 @@ export interface GeneralConfig {
   showUploadAndPrint: boolean;
   flipConsoleLayout: boolean;
   cameraFullscreenAction: CameraFullscreenAction;
+  printerPowerDevice: null | string;
   topNavPowerToggle: null | string;
   showManualProbeDialogAutomatically: boolean;
   showBedScrewsAdjustDialogAutomatically: boolean;