From 811d140124c8421ae0f437d65e43b1da88d6fc97 Mon Sep 17 00:00:00 2001
From: Jason2866 <24528715+Jason2866@users.noreply.github.com>
Date: Sun, 2 Jan 2022 15:14:54 +0100
Subject: [PATCH] prep one file firmware esp32

For easy initial flashing. Flashing at address 0x0 with esptool.py
Generated in folder `.pio..../firmware-factory.bin` (Not used at the moment!)

## Checklist:
  - [x] The pull request is done against the latest development branch
  - [x] Only relevant files were touched
  - [x] Only one feature/fix was added per PR and the code change compiles without warnings
  - [x] The code change is tested and works with Tasmota core ESP8266 V.2.7.4.9
  - [x] The code change is tested and works with Tasmota core ESP32 V.2.0.2
  - [x] I accept the [CLA](https://github.com/arendst/Tasmota/blob/development/CONTRIBUTING.md#contributor-license-agreement-cla).

_NOTE: The code change must pass CI tests. **Your PR cannot be merged unless tests pass**_
---
 pio-tools/post_esp32.py  | 30 ++++++++++++++++++++++++++++++
 platformio_tasmota32.ini |  1 +
 2 files changed, 31 insertions(+)
 create mode 100644 pio-tools/post_esp32.py

diff --git a/pio-tools/post_esp32.py b/pio-tools/post_esp32.py
new file mode 100644
index 000000000..904d24394
--- /dev/null
+++ b/pio-tools/post_esp32.py
@@ -0,0 +1,30 @@
+# From: https://github.com/letscontrolit/ESPEasy/blob/mega/tools/pio/post_esp32.py
+# Thanks TD-er :)
+
+Import("env")
+
+def esp32_create_factory_bin(source, target, env):
+    #print("Generating one file initial flash firmware for esp32 units")
+    offset = 0x0
+    new_file_name = env.subst("$BUILD_DIR/${PROGNAME}-factory.bin")
+    sections = env.subst(env.get('FLASH_EXTRA_IMAGES'))
+    new_file = open(new_file_name,"wb")
+    print("    {} | {}".format("Offset", "File"))
+    for section in sections:
+      sect_adr,sect_file = section.split(" ",1)
+      print(" -  {} | {}".format(sect_adr,sect_file))
+      source = open(sect_file,"rb")
+      new_file.seek(int(sect_adr,0)-offset)
+      new_file.write(source.read());
+      source.close()
+
+    firmware_name = env.subst("$BUILD_DIR/${PROGNAME}.bin")
+    firmware_start = 0x10000-offset
+    firmware = open(firmware_name,"rb")
+    print(" - {} | {}".format(hex(firmware_start),firmware_name))
+    new_file.seek(firmware_start)
+    new_file.write(firmware.read())
+    new_file.close()
+    firmware.close()
+
+env.AddPostAction("$BUILD_DIR/${PROGNAME}.bin", esp32_create_factory_bin)
diff --git a/platformio_tasmota32.ini b/platformio_tasmota32.ini
index f70c70d3b..1d59d8f18 100644
--- a/platformio_tasmota32.ini
+++ b/platformio_tasmota32.ini
@@ -31,6 +31,7 @@ build_flags                 = ${esp_defaults.build_flags}
                               ; wrappers for the crash-recorder
                               -Wl,--wrap=panicHandler -Wl,--wrap=xt_unhandled_exception
 extra_scripts               = pre:pio-tools/add_c_flags.py
+                              post:pio-tools/post_esp32.py
                               ${esp_defaults.extra_scripts}
 
 [core32]