mirror of
https://github.com/wled/WLED.git
synced 2025-05-10 14:38:41 +00:00
0b8721c25e
Monkey-patch PlatformIO to intercept the build process after library dependencies are loaded, but before the build is fully analyzed. This lets us enforce libArchive=False for usermods without making that setting global across all libraries. The rest of the fixup code is integrated at the same call site for simplicity.
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
Import('env')
|
|
from pathlib import Path # For OS-agnostic path manipulation
|
|
|
|
usermod_dir = Path(env["PROJECT_DIR"]) / "usermods"
|
|
|
|
def find_usermod(mod: str):
|
|
"""Locate this library in the usermods folder.
|
|
We do this to avoid needing to rename a bunch of folders;
|
|
this could be removed later
|
|
"""
|
|
# Check name match
|
|
mp = usermod_dir / mod
|
|
if mp.exists():
|
|
return mp
|
|
mp = usermod_dir / f"usermod_v2_{mod}"
|
|
if mp.exists():
|
|
return mp
|
|
raise RuntimeError(f"Couldn't locate module {mod} in usermods directory!")
|
|
|
|
usermods = env.GetProjectOption("custom_usermods","")
|
|
if usermods:
|
|
proj = env.GetProjectConfig()
|
|
deps = env.GetProjectOption('lib_deps')
|
|
src_dir = proj.get("platformio", "src_dir")
|
|
src_dir = src_dir.replace('\\','/')
|
|
mod_paths = {mod: find_usermod(mod) for mod in usermods.split(" ")}
|
|
usermods = [f"{mod} = symlink://{path}" for mod, path in mod_paths.items()]
|
|
proj.set("env:" + env['PIOENV'], 'lib_deps', deps + usermods)
|
|
|
|
|
|
# Monkey-patch ConfigureProjectLibBuilder to mark up the dependencies
|
|
# Save the old value
|
|
cpl = env.ConfigureProjectLibBuilder
|
|
# Our new wrapper
|
|
def cpl_wrapper(env):
|
|
result = cpl.clone(env)()
|
|
# Update usermod properties
|
|
lib_builders = env.GetLibBuilders()
|
|
um_deps = [dep for dep in lib_builders if usermod_dir in Path(dep.src_dir).parents]
|
|
other_deps = [dep for dep in lib_builders if usermod_dir not in Path(dep.src_dir).parents]
|
|
for um in um_deps:
|
|
# Add include paths for all non-usermod dependencies
|
|
for dep in other_deps:
|
|
for dir in dep.get_include_dirs():
|
|
um.env.PrependUnique(CPPPATH=dir)
|
|
# Add the wled folder to the include path
|
|
um.env.PrependUnique(CPPPATH=env["PROJECT_SRC_DIR"])
|
|
# Make sure we link directly, not through an archive
|
|
# Archives drop the .dtor table section we need
|
|
build = um._manifest.get("build", {})
|
|
build["libArchive"] = False
|
|
um._manifest["build"] = build
|
|
return result
|
|
|
|
# Replace the old one with ours
|
|
env.AddMethod(cpl_wrapper, "ConfigureProjectLibBuilder")
|