From e5219fb8be9ac55587d6f260aae4f40ed089bf91 Mon Sep 17 00:00:00 2001
From: Jason2866 <24528715+Jason2866@users.noreply.github.com>
Date: Sat, 12 Nov 2022 17:12:15 +0100
Subject: [PATCH] ues zopfli to gz firmware

---
 pio-tools/gzip-firmware.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/pio-tools/gzip-firmware.py b/pio-tools/gzip-firmware.py
index 080448a30..26a00abe8 100644
--- a/pio-tools/gzip-firmware.py
+++ b/pio-tools/gzip-firmware.py
@@ -1,11 +1,17 @@
 Import("env")
 import os
 import shutil
-import gzip
 import pathlib
-
 import tasmotapiolib
 
+# Upgrade pip
+env.Execute("$PYTHONEXE -m pip install --upgrade pip")
+# Install zopfli gz compressor from the PyPi registry
+env.Execute("$PYTHONEXE -m pip install zopfli")
+
+# Import zoepfli compress
+from zopfli.gzip import compress
+import gzip
 
 def map_gzip(source, target, env):
     # create string with location and file names based on variant
@@ -45,8 +51,9 @@ if env["PIOPLATFORM"] != "espressif32":
 
         # write gzip firmware file
         with open(bin_file, "rb") as fp:
-            with gzip.open(gzip_file, "wb", compresslevel=9) as f:
-                shutil.copyfileobj(fp, f)
+            with open(gzip_file, "wb") as f:
+                zopfli_gz = compress(fp.read())
+                f.write(zopfli_gz)
 
         ORG_FIRMWARE_SIZE = bin_file.stat().st_size
         GZ_FIRMWARE_SIZE = gzip_file.stat().st_size