ESP32 刷 WLED：把一条桌面灯分段、存预设，再接入 HA

# SPDX-License-Identifier: MIT
"""Read-only checks. Never opens a serial port, flashes or sends network commands."""
import argparse
import hashlib
import json
import pathlib
import struct
import subprocess
import sys

BASE = pathlib.Path(__file__).resolve().parent

def check_payloads():
    plan = json.loads((BASE / "settings-plan.json").read_text())
    assert plan["wledType"] == 22 and plan["length"] == 30
    assert plan["gpio"] == 12 and plan["maxCurrentMA"] == 500
    for name in ("warm-state.json", "cool-state.json"):
        state = json.loads((BASE / name).read_text())
        assert state["bri"] == 32 and state["on"] is True
        assert [(s["id"], s["start"], s["stop"]) for s in state["seg"]] == [(0, 0, 15), (1, 15, 30)]
        assert all(s["fx"] == 0 and len(s["col"][0]) == 3 for s in state["seg"])
    print("PASS local JSON setup plan and two segment payloads. Not a WLED parser/runtime check.")

def check_assets(folder):
    lock = json.loads((BASE / "assets.lock.json").read_text())
    assert [p["offset"] for p in lock["parts"]] == [0, 0x8000, 0x10000]
    assert lock["parts"][0]["bytes"] <= 0x8000
    assert 0x8000 + lock["parts"][1]["bytes"] <= 0x9000
    for part in lock["parts"]:
        path = folder / part["name"]
        content = path.read_bytes()
        assert len(content) == part["bytes"], path
        assert hashlib.sha256(content).hexdigest() == part["sha256"], path
        assert part["offset"] + len(content) <= lock["flashBytes"]
        print("PASS hash/size", part["name"], len(content))
        if part["name"] != "partitions_s3_8m.bin":
            subprocess.run([sys.executable, "-m", "esptool", "image-info", str(path)], check=True)
    raw = (folder / "partitions_s3_8m.bin").read_bytes()
    entries = []
    found_md5 = False
    for i in range(0, len(raw), 32):
        row = raw[i:i + 32]
        magic = struct.unpack_from("<H", row)[0]
        if magic == 0xFFFF:
            break
        if magic == 0xEBEB:
            assert hashlib.md5(raw[:i]).digest() == row[16:32], "partition MD5"
            found_md5 = True
            break
        assert magic == 0x50AA, "partition magic"
        _, kind, subtype, offset, size, label, flags = struct.unpack("<HBBII16sI", row)
        entries.append((label.rstrip(b"\0").decode(), kind, subtype, offset, size, flags))
    assert found_md5, "partition MD5 entry missing"
    ordered = sorted(entries, key=lambda e: e[3])
    assert all(a[3] + a[4] <= b[3] for a, b in zip(ordered, ordered[1:])), "overlapping partitions"
    app = next(e for e in entries if e[0] == "app0")
    assert app[3] == 0x10000 and app[4] == 0x200000
    assert lock["parts"][2]["bytes"] < app[4], "application exceeds OTA slot"
    assert max(e[3] + e[4] for e in entries) == 0x800000
    print("PASS partition MD5, 2MiB app0 slot, non-overlap with boot/partition files, 8MiB end.")
    print("No firmware compilation, USB/flash, PSRAM, radio, strip or HA runtime was tested.")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--assets-dir", type=pathlib.Path, help="Existing downloaded official files; no download occurs")
    args = parser.parse_args()
    check_payloads()
    if args.assets_dir:
        check_assets(args.assets_dir)
    else:
        print("Asset checks skipped: supply --assets-dir after downloading the three pinned files.")

{
  "on": true,
  "bri": 32,
  "transition": 7,
  "mainseg": 0,
  "seg": [
    {
      "id": 0,
      "start": 0,
      "stop": 15,
      "grp": 1,
      "spc": 0,
      "of": 0,
      "on": true,
      "bri": 255,
      "col": [
        [
          255,
          128,
          16
        ],
        [
          0,
          0,
          0
        ],
        [
          0,
          0,
          0
        ]
      ],
      "fx": 0,
      "sx": 128,
      "ix": 128,
      "pal": 0,
      "sel": true
    },
    {
      "id": 1,
      "start": 15,
      "stop": 30,
      "grp": 1,
      "spc": 0,
      "of": 0,
      "on": true,
      "bri": 255,
      "col": [
        [
          96,
          32,
          0
        ],
        [
          0,
          0,
          0
        ],
        [
          0,
          0,
          0
        ]
      ],
      "fx": 0,
      "sx": 128,
      "ix": 128,
      "pal": 0,
      "sel": false
    }
  ]
}

curl --fail --max-time 5 -H 'Content-Type: application/json' --data-binary @warm-state.json http://192.0.2.1/json/state
curl --fail --max-time 5 -H 'Content-Type: application/json' --data '{"psave":1,"n":"DeskWarm","ib":true,"sb":true}' http://192.0.2.1/json/state

# Replace this entity with the real WLED Preset select entity after integration.
# Merge packages into an existing HA configuration; do not duplicate top-level keys.
script:
  h08_desk_warm:
    alias: H08 DeskWarm preset
    sequence:
      - action: select.select_option
        target:
          entity_id: select.replace_with_real_wled_preset
        data:
          option: DeskWarm
  h08_desk_cool:
    alias: H08 DeskCool preset
    sequence:
      - action: select.select_option
        target:
          entity_id: select.replace_with_real_wled_preset
        data:
          option: DeskCool