From 782cfd427a650887a43c78c90cef74869277e7e2 Mon Sep 17 00:00:00 2001
From: Craig Bassett <craig.bassett@gmail.com>
Date: Fri, 25 Jun 2021 10:59:43 -0500
Subject: [PATCH] feat: allow custom gcode to run with thermal presets

Signed-off-by: Craig Bassett <craig.bassett@gmail.com>
---
 .../PresetDialog.vue}                         | 11 ++++++++++
 .../PresetSettings.vue}                       |  9 ++++----
 .../widgets/thermals/TemperatureCard.vue      | 22 ++++++++++++-------
 src/locales/en.yaml                           |  1 +
 src/store/config/types.ts                     |  1 +
 src/views/Settings.vue                        |  6 ++---
 6 files changed, 34 insertions(+), 16 deletions(-)
 rename src/components/settings/{TempPresetDialog.vue => presets/PresetDialog.vue} (92%)
 rename src/components/settings/{TemperaturePresetSettings.vue => presets/PresetSettings.vue} (95%)

diff --git a/src/components/settings/TempPresetDialog.vue b/src/components/settings/presets/PresetDialog.vue
similarity index 92%
rename from src/components/settings/TempPresetDialog.vue
rename to src/components/settings/presets/PresetDialog.vue
index 1a41a47dbf..6ba61c6a03 100644
--- a/src/components/settings/TempPresetDialog.vue
+++ b/src/components/settings/presets/PresetDialog.vue
@@ -82,6 +82,17 @@
           <v-divider :key="i + 'fand'"></v-divider>
         </template>
 
+        <app-setting :title="$t('app.setting.label.thermal_preset_gcode')">
+          <v-textarea
+            v-model="preset.gcode"
+            rows="2"
+            hide-details="auto"
+            class="mb-2"
+            outlined
+          >
+          </v-textarea>
+        </app-setting>
+
         <v-card-actions>
           <v-spacer></v-spacer>
           <app-btn color="warning" text @click="$emit('input', false)" type="button">{{ $t('app.general.btn.cancel') }}</app-btn>
diff --git a/src/components/settings/TemperaturePresetSettings.vue b/src/components/settings/presets/PresetSettings.vue
similarity index 95%
rename from src/components/settings/TemperaturePresetSettings.vue
rename to src/components/settings/presets/PresetSettings.vue
index e01e208a00..7ab3aff48b 100644
--- a/src/components/settings/TemperaturePresetSettings.vue
+++ b/src/components/settings/presets/PresetSettings.vue
@@ -49,12 +49,12 @@
         <v-divider :key="preset.id" v-if="i < presets.length - 1 && presets.length > 0"></v-divider>
       </template>
 
-      <temperature-preset-dialog
+      <preset-dialog
         v-if="dialogState.preset"
         v-model="dialogState.active"
         :preset="dialogState.preset"
         @save="handleSavePreset"
-      ></temperature-preset-dialog>
+      ></preset-dialog>
 
     </v-card>
   </div>
@@ -62,16 +62,15 @@
 
 <script lang="ts">
 import { Component, Mixins } from 'vue-property-decorator'
+import PresetDialog from './PresetDialog.vue'
 import { TemperaturePreset } from '@/store/config/types'
 import { Fan, Heater } from '@/store/printer/types'
 import StateMixin from '@/mixins/state'
 import { VForm } from '@/types/vuetify'
 
-import TemperaturePresetDialog from './TempPresetDialog.vue'
-
 @Component({
   components: {
-    TemperaturePresetDialog
+    PresetDialog
   }
 })
 export default class TemperaturePresetSettings extends Mixins(StateMixin) {
diff --git a/src/components/widgets/thermals/TemperatureCard.vue b/src/components/widgets/thermals/TemperatureCard.vue
index 7cb8fd79a9..c0baf03ca6 100644
--- a/src/components/widgets/thermals/TemperatureCard.vue
+++ b/src/components/widgets/thermals/TemperatureCard.vue
@@ -118,16 +118,22 @@ export default class TemperatureCard extends Mixins(StateMixin) {
   }
 
   handleApplyPreset (preset: TemperaturePreset) {
-    if (preset && preset.values) {
-      for (const key in preset.values) {
-        const item = preset.values[key]
-        if (item.type === 'heater' && item.active && item.value > -1) {
-          this.sendGcode(`SET_HEATER_TEMPERATURE HEATER=${key} TARGET=${item.value}`)
-        }
-        if (item.type === 'fan' && item.active && item.value > -1) {
-          this.sendGcode(`SET_TEMPERATURE_FAN_TARGET TEMPERATURE_FAN=${key} TARGET=${item.value}`)
+    if (preset) {
+      if (preset.values) {
+        for (const key in preset.values) {
+          const item = preset.values[key]
+          if (item.type === 'heater' && item.active && item.value > -1) {
+            this.sendGcode(`SET_HEATER_TEMPERATURE HEATER=${key} TARGET=${item.value}`)
+          }
+          if (item.type === 'fan' && item.active && item.value > -1) {
+            this.sendGcode(`SET_TEMPERATURE_FAN_TARGET TEMPERATURE_FAN=${key} TARGET=${item.value}`)
+          }
         }
       }
+
+      if (preset.gcode) {
+        this.sendGcode(preset.gcode)
+      }
     }
   }
 
diff --git a/src/locales/en.yaml b/src/locales/en.yaml
index 4631fc8003..98aa95389b 100644
--- a/src/locales/en.yaml
+++ b/src/locales/en.yaml
@@ -371,6 +371,7 @@ app:
       retraction_icon_size: Retraction Icon Size
       show_animations: Show animations
       theme_preset: Community preset
+      thermal_preset_gcode: GCode
       thermal_preset_name: Preset Name
       time_estimates: Time estimates
       toggle: Toggle
diff --git a/src/store/config/types.ts b/src/store/config/types.ts
index 517094b950..551e9bc891 100644
--- a/src/store/config/types.ts
+++ b/src/store/config/types.ts
@@ -112,6 +112,7 @@ export interface TemperaturePreset {
   id: number;
   name: string;
   values: TemperaturePresetValues;
+  gcode?: string;
 }
 
 export interface TemperaturePresetValues {
diff --git a/src/views/Settings.vue b/src/views/Settings.vue
index 0e28f11065..32b86ec388 100644
--- a/src/views/Settings.vue
+++ b/src/views/Settings.vue
@@ -9,7 +9,7 @@
         <macro-categories></macro-categories>
         <cameras></cameras>
         <toolhead-settings></toolhead-settings>
-        <temperature-preset-settings></temperature-preset-settings>
+        <preset-settings></preset-settings>
         <gcode-preview-settings></gcode-preview-settings>
         <version-settings v-if="supportsVersions"></version-settings>
       </div>
@@ -23,7 +23,7 @@ import StateMixin from '@/mixins/state'
 
 import MacroCategories from '@/components/settings/macros/MacroCategories.vue'
 import GeneralSettings from '@/components/settings/GeneralSettings.vue'
-import TemperaturePresetSettings from '@/components/settings/TemperaturePresetSettings.vue'
+import PresetSettings from '@/components/settings/presets/PresetSettings.vue'
 import Cameras from '@/components/settings/cameras/Cameras.vue'
 import ToolheadSettings from '@/components/settings/ToolheadSettings.vue'
 import ThemeSettings from '@/components/settings/ThemeSettings.vue'
@@ -35,7 +35,7 @@ import AuthSettings from '@/components/settings/auth/AuthSettings.vue'
   components: {
     MacroCategories,
     GeneralSettings,
-    TemperaturePresetSettings,
+    PresetSettings,
     Cameras,
     ToolheadSettings,
     ThemeSettings,