Address review: cover all source suffixes, pass PIOPLATFORM through, link library -Wl flags

esphome/arduino8266/component.py

@@ -50,9 +50,10 @@ class ArduinoLibrary:
     # Extra compile flags private to this library's own sources
     flags: list[str] = field(default_factory=list)
     # Link inputs the library contributes (-L dirs / -l libs, e.g. from
-    # precompiled vendor blobs)
+    # precompiled vendor blobs) and -Wl, options for the firmware link
     link_dirs: list[Path] = field(default_factory=list)
     link_libs: list[str] = field(default_factory=list)
+    link_flags: list[str] = field(default_factory=list)
 
 
 def _library_info(name: str, read_path: Path, data: dict) -> ArduinoLibrary:
@@ -84,6 +85,8 @@ def _library_info(name: str, read_path: Path, data: dict) -> ArduinoLibrary:
             lib.link_dirs.append((read_path / tok[2:]).resolve())
         elif tok.startswith("-l"):
             lib.link_libs.append(tok[2:])
+        elif tok.startswith("-Wl,"):
+            lib.link_flags.append(tok)
         else:
             lib.flags.append(tok)
 
@@ -148,7 +151,7 @@ def resolve_libraries(framework_path: Path) -> list[ArduinoLibrary]:
             bundled.append(_bundled_library(framework_path, name))
 
     def _emit(component: ConvertedLibrary) -> None:
-        apply_extra_script(component, "esp8266")
+        apply_extra_script(component, "esp8266", pio_platform=ESP8266_PLATFORM)
         converted.append(
             _library_info(
                 component.get_require_name(), component.source_dir, component.data

esphome/build_gen/arduino8266.py

@@ -42,7 +42,22 @@ from esphome.core import CORE, EsphomeError
 from esphome.framework_helpers import get_project_cxx_compile_flags
 from esphome.helpers import mkdir_p, write_file_if_changed
 
-_RULE_FOR_SUFFIX = {".c": "cc", ".cpp": "cxx", ".S": "asm"}
+# Compile rule per source suffix; keys must cover SRC_FILE_EXTENSIONS so any
+# source a library manifest selects has a rule (pinned by a drift test).
+_RULE_FOR_SUFFIX = {
+    ".c": "cc",
+    ".cpp": "cxx",
+    ".cc": "cxx",
+    ".cxx": "cxx",
+    ".c++": "cxx",
+    ".S": "asm",
+    ".spp": "asm",
+    ".SPP": "asm",
+    ".sx": "asm",
+    ".s": "asm",
+    ".asm": "asm",
+    ".ASM": "asm",
+}
 
 # Always excluded from the core build: ESPHome uses its own native OTA
 # backend, so the Arduino Updater (and its 228-byte global) never links.
@@ -437,6 +452,7 @@ def write_project(paths: dict[str, Path]) -> bool:
     if esp8266_data.get(KEY_SCANF_FLOAT):
         link_flags += ["-u", "_scanf_float"]
     link_flags += project_link_flags
+    link_flags += [flag for lib in libraries for flag in lib.link_flags]
     flash_ld = f"testing_{flash_ld_name}" if CORE.testing_mode else flash_ld_name
     link_flags += ["-T", flash_ld]
 

esphome/espidf/extra_script.py

@@ -42,7 +42,9 @@ _LOGGER = logging.getLogger(__name__)
 
 
 def apply_extra_script(
-    component: ConvertedLibrary, idf_target: str | Callable[[], str]
+    component: ConvertedLibrary,
+    idf_target: str | Callable[[], str],
+    pio_platform: str = "espressif32",
 ) -> None:
     """Run a library's PIO ``extraScript`` and fold its captured env vars into
     ``component.data["build"]["flags"]`` so the backend's -L/-l/-D extraction
@@ -50,6 +52,7 @@ def apply_extra_script(
 
     ``idf_target`` may be a callable so a backend whose target lookup needs
     build state (the esp32 variant) resolves it only when a script will run.
+    ``pio_platform`` is exposed to the script as PlatformIO's ``PIOPLATFORM``.
     """
     extra_script = component.data.get("build", {}).get("extraScript")
     if not extra_script:
@@ -64,7 +67,10 @@ def apply_extra_script(
     if callable(idf_target):
         idf_target = idf_target()
     result = run_extra_script(
-        script_path, library_dir=source_path, idf_target=idf_target
+        script_path,
+        library_dir=source_path,
+        idf_target=idf_target,
+        pio_platform=pio_platform,
     )
     extra_flags = captured_as_build_flags(result, library_dir=source_path)
     if not extra_flags:
@@ -101,10 +107,10 @@ class _FakeSConsEnv:
     ``AttributeError`` and abort the script.
     """
 
-    def __init__(self, *, board_mcu: str, pio_env: str) -> None:
+    def __init__(self, *, board_mcu: str, pio_env: str, pio_platform: str) -> None:
         self._vars: dict[str, str] = {
             "BOARD_MCU": board_mcu,
-            "PIOPLATFORM": "espressif32",
+            "PIOPLATFORM": pio_platform,
             "PIOENV": pio_env,
         }
         self.result = ExtraScriptResult()
@@ -132,7 +138,11 @@ class _FakeSConsEnv:
 
 
 def run_extra_script(
-    script_path: Path, *, library_dir: Path, idf_target: str
+    script_path: Path,
+    *,
+    library_dir: Path,
+    idf_target: str,
+    pio_platform: str = "espressif32",
 ) -> ExtraScriptResult:
     """Execute ``script_path`` with a fake SCons env and return captured vars.
 
@@ -147,7 +157,11 @@ def run_extra_script(
     an empty result — extra-scripts are best-effort, and an unsupported
     script shouldn't block the build.
     """
-    env = _FakeSConsEnv(board_mcu=idf_target, pio_env=f"esphome_{idf_target}")
+    env = _FakeSConsEnv(
+        board_mcu=idf_target,
+        pio_env=f"esphome_{idf_target}",
+        pio_platform=pio_platform,
+    )
     code = compile(script_path.read_text(encoding="utf-8"), str(script_path), "exec")
     old_cwd = Path.cwd()
     try: