From 3c13fb1b8d19b509adf08b438ed5ce0ec1c01721 Mon Sep 17 00:00:00 2001
From: Jason2866 <24528715+Jason2866@users.noreply.github.com>
Date: Wed, 4 May 2022 19:44:41 +0200
Subject: [PATCH] fix solo1 safemode

---
 pio-tools/post_esp32.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/pio-tools/post_esp32.py b/pio-tools/post_esp32.py
index d149d9405..782eed772 100644
--- a/pio-tools/post_esp32.py
+++ b/pio-tools/post_esp32.py
@@ -36,8 +36,13 @@ FRAMEWORK_DIR = platform.get_package_dir("framework-arduinoespressif32")
 variants_dir = join(FRAMEWORK_DIR, "variants", "tasmota")
 
 def esp32_fetch_safemode_bin(chip):
-    safemode_fw_url = "https://github.com/arendst/Tasmota-firmware/raw/main/firmware/tasmota32/tasmota" + chip[3:] + "-safemode.bin"
-    safemode_fw_name = join(variants_dir,"tasmota" + chip[3:] + "-safemode.bin")
+    if "solo1" in env.subst("$BUILD_DIR"):
+        safemode_fw_url = "https://github.com/arendst/Tasmota-firmware/raw/main/firmware/tasmota32/tasmota32solo1-safemode.bin"
+        safemode_fw_name = join(variants_dir,"tasmota32solo1-safemode.bin")
+    else:
+        safemode_fw_url = "https://github.com/arendst/Tasmota-firmware/raw/main/firmware/tasmota32/tasmota" + chip[3:] + "-safemode.bin"
+        safemode_fw_name = join(variants_dir,"tasmota" + chip[3:] + "-safemode.bin")
+
     if(exists(safemode_fw_name)):
         print("Safemode binary already in place.")
         return