60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
Import("env")
|
|
import os
|
|
import gzip
|
|
from shutil import copyfileobj, rmtree
|
|
from pathlib import Path
|
|
import subprocess
|
|
|
|
|
|
|
|
|
|
revision = (
|
|
subprocess.check_output(["git", "rev-parse", "HEAD"])
|
|
.strip()
|
|
.decode("utf-8")
|
|
)
|
|
|
|
def gzip_file(input_file, output_file):
|
|
with open(input_file, 'rb') as f_in:
|
|
with gzip.open(output_file, 'wb') as f_out:
|
|
copyfileobj(f_in, f_out)
|
|
|
|
def process_directory(input_dir, output_dir):
|
|
if os.path.exists(output_dir):
|
|
rmtree(output_dir)
|
|
for root, dirs, files in os.walk(input_dir):
|
|
relative_path = os.path.relpath(root, input_dir)
|
|
output_root = os.path.join(output_dir, relative_path)
|
|
|
|
Path(output_root).mkdir(parents=True, exist_ok=True)
|
|
|
|
for file in files:
|
|
# if file.endswith(('.html', '.css', '.js')):
|
|
input_file_path = os.path.join(root, file)
|
|
output_file_path = os.path.join(output_root, file + '.gz')
|
|
gzip_file(input_file_path, output_file_path)
|
|
print(f'Compressed: {input_file_path} -> {output_file_path}')
|
|
file_path = os.path.join(output_dir, "fs_hash.txt")
|
|
with open(file_path, "w") as file:
|
|
file.write(revision)
|
|
|
|
|
|
# Build web interface before building FS
|
|
def before_buildfs(source, target, env):
|
|
env.Execute("cd data && yarn && yarn postinstall && yarn build")
|
|
input_directory = 'data/dist'
|
|
output_directory = 'data/build_gz'
|
|
process_directory(input_directory, output_directory)
|
|
|
|
flash_size = env.BoardConfig().get("upload.flash_size", "4MB")
|
|
fs_image_name = f"littlefs_{flash_size}"
|
|
env.Replace(ESP32_FS_IMAGE_NAME=fs_image_name)
|
|
env.Replace(ESP8266_FS_IMAGE_NAME=fs_image_name)
|
|
|
|
os.environ["PUBLIC_BASE_URL"] = ""
|
|
fs_name = env.get("ESP32_FS_IMAGE_NAME", "littlefs.bin")
|
|
# Or alternatively:
|
|
# fs_name = env.get("FSTOOLNAME", "littlefs.bin")
|
|
|
|
# Use the variable in the pre-action
|
|
env.AddPreAction(f"$BUILD_DIR/{fs_name}.bin", before_buildfs)
|