From 143e070cdc1c93e2f39984b203e3d957f24e1a80 Mon Sep 17 00:00:00 2001
From: Francois Steyn <fsteyn@tfyre.co.za>
Date: Wed, 11 Dec 2024 18:10:40 +0100
Subject: [PATCH] feat: add x1e support

---
 .../com/tfyre/bambu/printer/BambuConst.java   | 21 ++++++++++++-------
 .../view/dashboard/DashboardPrinter.java      |  4 ++--
 2 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/bambu/src/main/java/com/tfyre/bambu/printer/BambuConst.java b/bambu/src/main/java/com/tfyre/bambu/printer/BambuConst.java
index ac41b27..a485720 100644
--- a/bambu/src/main/java/com/tfyre/bambu/printer/BambuConst.java
+++ b/bambu/src/main/java/com/tfyre/bambu/printer/BambuConst.java
@@ -279,25 +279,32 @@ public String getHtmlColor() {
     }
 
     public enum PrinterModel {
-        UNKNOWN("unknown"),
-        A1("a1"),
-        A1MINI("a1mini"),
-        P1P("p1p"),
-        P1S("p1s"),
-        X1C("x1c");
+        UNKNOWN("unknown", false),
+        A1("a1", false),
+        A1MINI("a1mini", false),
+        P1P("p1p", false),
+        P1S("p1s", false),
+        X1C("x1c", true),
+        X1E("x1e", true);
 
         private static final Map<String, PrinterModel> MAP = EnumSet.allOf(PrinterModel.class).stream().collect(Collectors.toMap(PrinterModel::getModel, Function.identity()));
 
         private final String model;
+        private final boolean temperature;
 
-        private PrinterModel(final String model) {
+        private PrinterModel(final String model, final boolean temperature) {
             this.model = model;
+            this.temperature = temperature;
         }
 
         public String getModel() {
             return model;
         }
 
+        public boolean isTemperature() {
+            return temperature;
+        }
+
         public static Optional<PrinterModel> fromModel(final String model) {
             return Optional.ofNullable(MAP.get(model));
         }
diff --git a/bambu/src/main/java/com/tfyre/bambu/view/dashboard/DashboardPrinter.java b/bambu/src/main/java/com/tfyre/bambu/view/dashboard/DashboardPrinter.java
index ac2f26b..224debc 100644
--- a/bambu/src/main/java/com/tfyre/bambu/view/dashboard/DashboardPrinter.java
+++ b/bambu/src/main/java/com/tfyre/bambu/view/dashboard/DashboardPrinter.java
@@ -609,7 +609,7 @@ private FlexLayout buildStatus() {
                 wrapTemperature(getBadge("Bed", bedImage, bed, bedTarget), () -> (int) temperatureBed, BambuConst.TEMPERATURE_MAX_BED, BambuConst::gcodeTargetTemperatureBed)
         );
 
-        if (printer.getModel() == BambuConst.PrinterModel.X1C) {
+        if (printer.getModel().isTemperature()) {
             result.add(getBadge("Frame", frameImage, frame));
         }
 
@@ -632,7 +632,7 @@ private Div buildAmsHeader(final AmsHeader header) {
                 new Span(header.id()),
                 newSpan("filler")
         );
-        if (printer.getModel() == BambuConst.PrinterModel.X1C) {
+        if (printer.getModel().isTemperature()) {
             result.add(header.temperature());
         }
         result.add(header.humidity());