mirror of
https://github.com/vicliu624/trail-mate.git
synced 2026-07-01 17:41:40 +00:00
liu weikai
302 lines
10 KiB
Python
302 lines
10 KiB
Python
Import("env")
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import json
|
|
import re
|
|
|
|
print("[pio] pre: configuring build environment")
|
|
|
|
project_dir = env.get("PROJECT_DIR")
|
|
pio_platform = (env.get("PIOPLATFORM") or "").lower()
|
|
is_esp32_env = "espressif32" in pio_platform
|
|
is_nrf52_env = "nordicnrf52" in pio_platform
|
|
is_gat562_env = env.get("PIOENV") == "gat562_mesh_evb_pro"
|
|
|
|
|
|
def read_text_best_effort(path):
|
|
encodings = ("utf-16", "utf-8-sig", "utf-8")
|
|
last_error = None
|
|
for encoding in encodings:
|
|
try:
|
|
with open(path, "r", encoding=encoding) as fp:
|
|
return fp.read()
|
|
except UnicodeError as exc:
|
|
last_error = exc
|
|
if last_error:
|
|
raise last_error
|
|
with open(path, "r", encoding="utf-8") as fp:
|
|
return fp.read()
|
|
|
|
|
|
def extract_version_from_changelog():
|
|
changelog_path = os.path.join(project_dir, "CHANGELOG.md")
|
|
if not os.path.exists(changelog_path):
|
|
return None
|
|
|
|
text = read_text_best_effort(changelog_path)
|
|
match = re.search(r"^## \[([0-9][^\]]*)\]", text, re.MULTILINE)
|
|
return match.group(1).strip() if match else None
|
|
|
|
|
|
def extract_version_from_ci_tag():
|
|
github_ref = os.environ.get("GITHUB_REF", "")
|
|
github_ref_name = os.environ.get("GITHUB_REF_NAME", "")
|
|
|
|
tag_name = ""
|
|
if github_ref.startswith("refs/tags/"):
|
|
tag_name = github_ref.rsplit("/", 1)[-1]
|
|
elif github_ref_name:
|
|
tag_name = github_ref_name
|
|
|
|
if not tag_name:
|
|
return None
|
|
return tag_name[1:] if tag_name.startswith("v") else tag_name
|
|
|
|
|
|
def resolve_project_version():
|
|
version = extract_version_from_ci_tag()
|
|
if version:
|
|
return version
|
|
|
|
version = extract_version_from_changelog()
|
|
if version:
|
|
return version
|
|
|
|
return "unknown"
|
|
|
|
|
|
def update_library_build_metadata(library_json_path, desired_updates, description):
|
|
if not os.path.exists(library_json_path):
|
|
print(f"[pio] pre: {description} metadata not found yet: {library_json_path}")
|
|
return
|
|
|
|
with open(library_json_path, "r", encoding="utf-8") as fp:
|
|
library_json = json.load(fp)
|
|
|
|
build_section = dict(library_json.get("build", {}))
|
|
changed = False
|
|
|
|
for key, value in desired_updates.items():
|
|
if build_section.get(key) != value:
|
|
build_section[key] = value
|
|
changed = True
|
|
|
|
if not changed:
|
|
print(f"[pio] pre: {description} metadata already trimmed")
|
|
return
|
|
|
|
library_json["build"] = build_section
|
|
|
|
with open(library_json_path, "w", encoding="utf-8", newline="\n") as fp:
|
|
json.dump(library_json, fp, indent=4)
|
|
fp.write("\n")
|
|
|
|
print(f"[pio] pre: Trimmed {description} build metadata")
|
|
|
|
|
|
def configure_radiolib_for_gat562():
|
|
if not is_gat562_env:
|
|
return
|
|
|
|
project_dir = env.get("PROJECT_DIR")
|
|
radiolib_dir = os.path.join(project_dir, ".pio", "libdeps", env.get("PIOENV"), "RadioLib")
|
|
library_json_path = os.path.join(radiolib_dir, "library.json")
|
|
desired_src_filter = [
|
|
"-<*>",
|
|
"+<Hal.cpp>",
|
|
"+<Module.cpp>",
|
|
"+<hal/Arduino/ArduinoHal.cpp>",
|
|
"+<protocols/PhysicalLayer/*.cpp>",
|
|
"+<protocols/Print/*.cpp>",
|
|
"+<utils/*.cpp>",
|
|
"+<modules/SX126x/SX1262.cpp>",
|
|
"+<modules/SX126x/SX126x.cpp>",
|
|
"+<modules/SX126x/SX126x_LR_FHSS.cpp>",
|
|
"+<modules/SX126x/SX126x_commands.cpp>",
|
|
"+<modules/SX126x/SX126x_config.cpp>",
|
|
]
|
|
update_library_build_metadata(
|
|
library_json_path,
|
|
{"srcFilter": desired_src_filter},
|
|
"RadioLib for gat562_mesh_evb_pro",
|
|
)
|
|
|
|
|
|
def configure_crypto_for_gat562():
|
|
if not is_gat562_env:
|
|
return
|
|
|
|
crypto_dir = os.path.join(project_dir, ".pio", "libdeps", env.get("PIOENV"), "Crypto")
|
|
library_json_path = os.path.join(crypto_dir, "library.json")
|
|
desired_src_filter = [
|
|
"-<*>",
|
|
"+<AES128.cpp>",
|
|
"+<AES256.cpp>",
|
|
"+<AESCommon.cpp>",
|
|
"+<BigNumberUtil.cpp>",
|
|
"+<BlockCipher.cpp>",
|
|
"+<CTR.cpp>",
|
|
"+<ChaCha.cpp>",
|
|
"+<Cipher.cpp>",
|
|
"+<Crypto.cpp>",
|
|
"+<Curve25519.cpp>",
|
|
"+<Hash.cpp>",
|
|
"+<RNG.cpp>",
|
|
"+<SHA256.cpp>",
|
|
]
|
|
update_library_build_metadata(
|
|
library_json_path,
|
|
{"srcFilter": desired_src_filter},
|
|
"Crypto for gat562_mesh_evb_pro",
|
|
)
|
|
|
|
|
|
def configure_nrf52_framework_libraries():
|
|
if not is_gat562_env or not is_nrf52_env:
|
|
return
|
|
|
|
platform = env.PioPlatform()
|
|
framework_dir = platform.get_package_dir("framework-arduinoadafruitnrf52")
|
|
if not framework_dir:
|
|
return
|
|
|
|
tinyusb_cppdefines = [
|
|
("CFG_TUD_CDC", 1),
|
|
("CFG_TUD_MSC", 0),
|
|
("CFG_TUD_HID", 0),
|
|
("CFG_TUD_MIDI", 0),
|
|
("CFG_TUD_VENDOR", 0),
|
|
("CFG_TUD_VIDEO", 0),
|
|
("CFG_TUD_VIDEO_STREAMING", 0),
|
|
]
|
|
env.AppendUnique(CPPDEFINES=tinyusb_cppdefines)
|
|
print("[pio] pre: Restricted TinyUSB device classes for gat562_mesh_evb_pro")
|
|
|
|
bluefruit_json_path = os.path.join(framework_dir, "libraries", "Bluefruit52Lib", "library.json")
|
|
bluefruit_src_filter = [
|
|
"-<*>",
|
|
"+<BLEAdvertising.cpp>",
|
|
"+<BLECentral.cpp>",
|
|
"+<BLECharacteristic.cpp>",
|
|
"+<BLEClientCharacteristic.cpp>",
|
|
"+<BLEClientService.cpp>",
|
|
"+<BLEConnection.cpp>",
|
|
"+<BLEDiscovery.cpp>",
|
|
"+<BLEGatt.cpp>",
|
|
"+<BLEPeriph.cpp>",
|
|
"+<BLEScanner.cpp>",
|
|
"+<BLESecurity.cpp>",
|
|
"+<BLEService.cpp>",
|
|
"+<BLEUuid.cpp>",
|
|
"+<bluefruit.cpp>",
|
|
"+<utility/*.cpp>",
|
|
]
|
|
update_library_build_metadata(
|
|
bluefruit_json_path,
|
|
{"srcFilter": bluefruit_src_filter},
|
|
"Bluefruit52Lib for gat562_mesh_evb_pro",
|
|
)
|
|
|
|
tinyusb_json_path = os.path.join(framework_dir, "libraries", "Adafruit_TinyUSB_Arduino", "library.json")
|
|
tinyusb_src_filter = [
|
|
"-<*>",
|
|
"+<tusb.c>",
|
|
"+<arduino/Adafruit_TinyUSB_API.cpp>",
|
|
"+<arduino/Adafruit_USBD_CDC.cpp>",
|
|
"+<arduino/Adafruit_USBD_Device.cpp>",
|
|
"+<arduino/Adafruit_USBD_Interface.cpp>",
|
|
"+<arduino/ports/nrf/Adafruit_TinyUSB_nrf.cpp>",
|
|
"+<class/cdc/cdc_device.c>",
|
|
"+<common/tusb_fifo.c>",
|
|
"+<device/usbd.c>",
|
|
"+<device/usbd_control.c>",
|
|
"+<portable/nordic/nrf5x/dcd_nrf5x.c>",
|
|
]
|
|
update_library_build_metadata(
|
|
tinyusb_json_path,
|
|
{"srcFilter": tinyusb_src_filter, "libArchive": False},
|
|
"Adafruit_TinyUSB_Arduino for gat562_mesh_evb_pro",
|
|
)
|
|
|
|
|
|
def inject_project_version_define():
|
|
version = resolve_project_version()
|
|
escaped_version = version.replace("\\", "\\\\").replace('"', '\\"')
|
|
env.AppendUnique(CPPDEFINES=[("TRAIL_MATE_FIRMWARE_VERSION", '\\"' + escaped_version + '\\"')])
|
|
print(f"[pio] pre: Injected firmware version: {version}")
|
|
|
|
|
|
configure_radiolib_for_gat562()
|
|
configure_crypto_for_gat562()
|
|
configure_nrf52_framework_libraries()
|
|
inject_project_version_define()
|
|
|
|
# Only ESP Arduino builds need the shared LVGL config under platform/esp.
|
|
if is_esp32_env:
|
|
ui_dir = os.path.join(project_dir, "platform", "esp", "arduino_common", "include")
|
|
env.Append(CPPFLAGS=["-I" + ui_dir])
|
|
env.Append(CCFLAGS=["-I" + ui_dir])
|
|
env.Append(CPPDEFINES=["LV_CONF_INCLUDE_SIMPLE"])
|
|
print(f"[pio] pre: Added LVGL config include path: {ui_dir}")
|
|
|
|
# Nordic builds should use their own shared runtime headers instead of falling
|
|
# back to ESP include roots via unrelated global injection.
|
|
if is_nrf52_env:
|
|
nrf_ui_dir = os.path.join(project_dir, "platform", "nrf52", "arduino_common", "include")
|
|
env.Append(CPPFLAGS=["-I" + nrf_ui_dir])
|
|
env.Append(CCFLAGS=["-I" + nrf_ui_dir])
|
|
print(f"[pio] pre: Added nRF52 runtime include path: {nrf_ui_dir}")
|
|
|
|
# On nordicnrf52/Windows builds, `${platformio.packages_dir}` inside `build_flags`
|
|
# can be rewritten into an invalid builder-local include path. Inject the
|
|
# nRF52 framework paths here using the resolved absolute package location so
|
|
# framework libraries like `Wire` can include `<Adafruit_TinyUSB.h>`.
|
|
if is_nrf52_env:
|
|
platform = env.PioPlatform()
|
|
framework_dir = platform.get_package_dir("framework-arduinoadafruitnrf52")
|
|
if framework_dir:
|
|
framework_include_candidates = [
|
|
os.path.join(framework_dir, "libraries", "Adafruit_TinyUSB_Arduino", "src"),
|
|
os.path.join(framework_dir, "libraries", "Adafruit_TinyUSB_Arduino", "src", "arduino"),
|
|
os.path.join(framework_dir, "libraries", "SPI"),
|
|
os.path.join(framework_dir, "libraries", "Wire"),
|
|
os.path.join(framework_dir, "libraries", "Bluefruit52Lib", "src"),
|
|
os.path.join(framework_dir, "libraries", "Bluefruit52Lib", "src", "services"),
|
|
os.path.join(framework_dir, "libraries", "Adafruit_nRFCrypto", "src"),
|
|
os.path.join(framework_dir, "libraries", "Adafruit_LittleFS", "src"),
|
|
os.path.join(framework_dir, "libraries", "InternalFileSytem", "src"),
|
|
]
|
|
existing_cpppath = set(env.get("CPPPATH", []))
|
|
include_paths = [path for path in framework_include_candidates if os.path.isdir(path)]
|
|
missing_paths = [path for path in include_paths if path not in existing_cpppath]
|
|
if missing_paths:
|
|
env.Append(CPPPATH=missing_paths)
|
|
print(f"[pio] pre: Added nRF52 framework include paths: {', '.join(missing_paths)}")
|
|
|
|
# Generate protobuf files if .proto files exist
|
|
proto_dir = os.path.join(project_dir, "lib", "meshtastic_protobufs")
|
|
generate_script = os.path.join(project_dir, "scripts", "generate_protobuf.py")
|
|
|
|
if os.path.exists(generate_script) and os.path.exists(proto_dir):
|
|
print("[pio] pre: Generating protobuf files...")
|
|
try:
|
|
result = subprocess.run([sys.executable, generate_script],
|
|
cwd=project_dir, capture_output=True, text=True)
|
|
if result.returncode == 0:
|
|
print("[pio] pre: Protobuf generation successful")
|
|
else:
|
|
print(f"[pio] pre: Protobuf generation warning: {result.stderr}")
|
|
except Exception as e:
|
|
print(f"[pio] pre: Protobuf generation error: {e}")
|
|
|
|
# Add generated protobuf include path
|
|
generated_dir = os.path.join(project_dir, "src", "chat", "infra", "meshtastic", "generated")
|
|
if os.path.exists(generated_dir):
|
|
env.Append(CPPFLAGS=["-I" + generated_dir])
|
|
env.Append(CCFLAGS=["-I" + generated_dir])
|
|
print(f"[pio] pre: Added protobuf generated include path: {generated_dir}")
|
|
|
|
# Example hook: add a build flag to verify hook execution
|
|
env.Append(CPPDEFINES=["PIO_PRE_HOOK=1"])
|