mirror of
				https://github.com/esphome/esphome.git
				synced 2025-10-31 23:21:54 +00:00 
			
		
		
		
	
							
								
								
									
										2
									
								
								.github/workflows/ci-docker.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/workflows/ci-docker.yml
									
									
									
									
										vendored
									
									
								
							| @@ -49,7 +49,7 @@ jobs: | |||||||
|         with: |         with: | ||||||
|           python-version: "3.10" |           python-version: "3.10" | ||||||
|       - name: Set up Docker Buildx |       - name: Set up Docker Buildx | ||||||
|         uses: docker/setup-buildx-action@v3.10.0 |         uses: docker/setup-buildx-action@v3.11.1 | ||||||
|  |  | ||||||
|       - name: Set TAG |       - name: Set TAG | ||||||
|         run: | |         run: | | ||||||
|   | |||||||
							
								
								
									
										39
									
								
								.github/workflows/ci.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										39
									
								
								.github/workflows/ci.yml
									
									
									
									
										vendored
									
									
								
							| @@ -214,17 +214,51 @@ jobs: | |||||||
|         if: matrix.os == 'windows-latest' |         if: matrix.os == 'windows-latest' | ||||||
|         run: | |         run: | | ||||||
|           ./venv/Scripts/activate |           ./venv/Scripts/activate | ||||||
|           pytest -vv --cov-report=xml --tb=native -n auto tests |           pytest -vv --cov-report=xml --tb=native -n auto tests --ignore=tests/integration/ | ||||||
|       - name: Run pytest |       - name: Run pytest | ||||||
|         if: matrix.os == 'ubuntu-latest' || matrix.os == 'macOS-latest' |         if: matrix.os == 'ubuntu-latest' || matrix.os == 'macOS-latest' | ||||||
|         run: | |         run: | | ||||||
|           . venv/bin/activate |           . venv/bin/activate | ||||||
|           pytest -vv --cov-report=xml --tb=native -n auto tests |           pytest -vv --cov-report=xml --tb=native -n auto tests --ignore=tests/integration/ | ||||||
|       - name: Upload coverage to Codecov |       - name: Upload coverage to Codecov | ||||||
|         uses: codecov/codecov-action@v5.4.3 |         uses: codecov/codecov-action@v5.4.3 | ||||||
|         with: |         with: | ||||||
|           token: ${{ secrets.CODECOV_TOKEN }} |           token: ${{ secrets.CODECOV_TOKEN }} | ||||||
|  |  | ||||||
|  |   integration-tests: | ||||||
|  |     name: Run integration tests | ||||||
|  |     runs-on: ubuntu-latest | ||||||
|  |     needs: | ||||||
|  |       - common | ||||||
|  |     steps: | ||||||
|  |       - name: Check out code from GitHub | ||||||
|  |         uses: actions/checkout@v4.2.2 | ||||||
|  |       - name: Set up Python 3.13 | ||||||
|  |         id: python | ||||||
|  |         uses: actions/setup-python@v5.6.0 | ||||||
|  |         with: | ||||||
|  |           python-version: "3.13" | ||||||
|  |       - name: Restore Python virtual environment | ||||||
|  |         id: cache-venv | ||||||
|  |         uses: actions/cache@v4.2.3 | ||||||
|  |         with: | ||||||
|  |           path: venv | ||||||
|  |           key: ${{ runner.os }}-${{ steps.python.outputs.python-version }}-venv-${{ needs.common.outputs.cache-key }} | ||||||
|  |       - name: Create Python virtual environment | ||||||
|  |         if: steps.cache-venv.outputs.cache-hit != 'true' | ||||||
|  |         run: | | ||||||
|  |           python -m venv venv | ||||||
|  |           . venv/bin/activate | ||||||
|  |           python --version | ||||||
|  |           pip install -r requirements.txt -r requirements_test.txt | ||||||
|  |           pip install -e . | ||||||
|  |       - name: Register matcher | ||||||
|  |         run: echo "::add-matcher::.github/workflows/matchers/pytest.json" | ||||||
|  |       - name: Run integration tests | ||||||
|  |         run: | | ||||||
|  |           . venv/bin/activate | ||||||
|  |           pytest -vv --no-cov --tb=native -n auto tests/integration/ | ||||||
|  |  | ||||||
|   clang-format: |   clang-format: | ||||||
|     name: Check clang-format |     name: Check clang-format | ||||||
|     runs-on: ubuntu-24.04 |     runs-on: ubuntu-24.04 | ||||||
| @@ -494,6 +528,7 @@ jobs: | |||||||
|       - flake8 |       - flake8 | ||||||
|       - pylint |       - pylint | ||||||
|       - pytest |       - pytest | ||||||
|  |       - integration-tests | ||||||
|       - pyupgrade |       - pyupgrade | ||||||
|       - clang-tidy |       - clang-tidy | ||||||
|       - list-components |       - list-components | ||||||
|   | |||||||
							
								
								
									
										23
									
								
								.github/workflows/lock.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										23
									
								
								.github/workflows/lock.yml
									
									
									
									
										vendored
									
									
								
							| @@ -1,28 +1,11 @@ | |||||||
| --- | --- | ||||||
| name: Lock | name: Lock closed issues and PRs | ||||||
|  |  | ||||||
| on: | on: | ||||||
|   schedule: |   schedule: | ||||||
|     - cron: "30 0 * * *" |     - cron: "30 0 * * *"  # Run daily at 00:30 UTC | ||||||
|   workflow_dispatch: |   workflow_dispatch: | ||||||
|  |  | ||||||
| permissions: |  | ||||||
|   issues: write |  | ||||||
|   pull-requests: write |  | ||||||
|  |  | ||||||
| concurrency: |  | ||||||
|   group: lock |  | ||||||
|  |  | ||||||
| jobs: | jobs: | ||||||
|   lock: |   lock: | ||||||
|     runs-on: ubuntu-latest |     uses: esphome/workflows/.github/workflows/lock.yml@main | ||||||
|     steps: |  | ||||||
|       - uses: dessant/lock-threads@v5.0.1 |  | ||||||
|         with: |  | ||||||
|           pr-inactive-days: "1" |  | ||||||
|           pr-lock-reason: "" |  | ||||||
|           exclude-any-pr-labels: keep-open |  | ||||||
|  |  | ||||||
|           issue-inactive-days: "7" |  | ||||||
|           issue-lock-reason: "" |  | ||||||
|           exclude-any-issue-labels: keep-open |  | ||||||
|   | |||||||
							
								
								
									
										4
									
								
								.github/workflows/release.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										4
									
								
								.github/workflows/release.yml
									
									
									
									
										vendored
									
									
								
							| @@ -99,7 +99,7 @@ jobs: | |||||||
|           python-version: "3.10" |           python-version: "3.10" | ||||||
|  |  | ||||||
|       - name: Set up Docker Buildx |       - name: Set up Docker Buildx | ||||||
|         uses: docker/setup-buildx-action@v3.10.0 |         uses: docker/setup-buildx-action@v3.11.1 | ||||||
|  |  | ||||||
|       - name: Log in to docker hub |       - name: Log in to docker hub | ||||||
|         uses: docker/login-action@v3.4.0 |         uses: docker/login-action@v3.4.0 | ||||||
| @@ -178,7 +178,7 @@ jobs: | |||||||
|           merge-multiple: true |           merge-multiple: true | ||||||
|  |  | ||||||
|       - name: Set up Docker Buildx |       - name: Set up Docker Buildx | ||||||
|         uses: docker/setup-buildx-action@v3.10.0 |         uses: docker/setup-buildx-action@v3.11.1 | ||||||
|  |  | ||||||
|       - name: Log in to docker hub |       - name: Log in to docker hub | ||||||
|         if: matrix.registry == 'dockerhub' |         if: matrix.registry == 'dockerhub' | ||||||
|   | |||||||
| @@ -1,10 +1,18 @@ | |||||||
| --- | --- | ||||||
| # See https://pre-commit.com for more information | # See https://pre-commit.com for more information | ||||||
| # See https://pre-commit.com/hooks.html for more hooks | # See https://pre-commit.com/hooks.html for more hooks | ||||||
|  |  | ||||||
|  | ci: | ||||||
|  |   autoupdate_commit_msg: 'pre-commit: autoupdate' | ||||||
|  |   autoupdate_schedule: weekly | ||||||
|  |   autofix_prs: false | ||||||
|  |   # Skip hooks that have issues in pre-commit CI environment | ||||||
|  |   skip: [pylint, yamllint] | ||||||
|  |  | ||||||
| repos: | repos: | ||||||
|   - repo: https://github.com/astral-sh/ruff-pre-commit |   - repo: https://github.com/astral-sh/ruff-pre-commit | ||||||
|     # Ruff version. |     # Ruff version. | ||||||
|     rev: v0.11.10 |     rev: v0.12.2 | ||||||
|     hooks: |     hooks: | ||||||
|       # Run the linter. |       # Run the linter. | ||||||
|       - id: ruff |       - id: ruff | ||||||
| @@ -12,7 +20,7 @@ repos: | |||||||
|       # Run the formatter. |       # Run the formatter. | ||||||
|       - id: ruff-format |       - id: ruff-format | ||||||
|   - repo: https://github.com/PyCQA/flake8 |   - repo: https://github.com/PyCQA/flake8 | ||||||
|     rev: 7.2.0 |     rev: 7.3.0 | ||||||
|     hooks: |     hooks: | ||||||
|       - id: flake8 |       - id: flake8 | ||||||
|         additional_dependencies: |         additional_dependencies: | ||||||
|   | |||||||
							
								
								
									
										12
									
								
								CODEOWNERS
									
									
									
									
									
								
							
							
						
						
									
										12
									
								
								CODEOWNERS
									
									
									
									
									
								
							| @@ -87,6 +87,7 @@ esphome/components/bp1658cj/* @Cossid | |||||||
| esphome/components/bp5758d/* @Cossid | esphome/components/bp5758d/* @Cossid | ||||||
| esphome/components/button/* @esphome/core | esphome/components/button/* @esphome/core | ||||||
| esphome/components/bytebuffer/* @clydebarrow | esphome/components/bytebuffer/* @clydebarrow | ||||||
|  | esphome/components/camera/* @DT-art1 @bdraco | ||||||
| esphome/components/canbus/* @danielschramm @mvturnho | esphome/components/canbus/* @danielschramm @mvturnho | ||||||
| esphome/components/cap1188/* @mreditor97 | esphome/components/cap1188/* @mreditor97 | ||||||
| esphome/components/captive_portal/* @OttoWinter | esphome/components/captive_portal/* @OttoWinter | ||||||
| @@ -124,6 +125,7 @@ esphome/components/dht/* @OttoWinter | |||||||
| esphome/components/display_menu_base/* @numo68 | esphome/components/display_menu_base/* @numo68 | ||||||
| esphome/components/dps310/* @kbx81 | esphome/components/dps310/* @kbx81 | ||||||
| esphome/components/ds1307/* @badbadc0ffee | esphome/components/ds1307/* @badbadc0ffee | ||||||
|  | esphome/components/ds2484/* @mrk-its | ||||||
| esphome/components/dsmr/* @glmnet @zuidwijk | esphome/components/dsmr/* @glmnet @zuidwijk | ||||||
| esphome/components/duty_time/* @dudanov | esphome/components/duty_time/* @dudanov | ||||||
| esphome/components/ee895/* @Stock-M | esphome/components/ee895/* @Stock-M | ||||||
| @@ -146,6 +148,7 @@ esphome/components/esp32_ble_client/* @jesserockz | |||||||
| esphome/components/esp32_ble_server/* @Rapsssito @clydebarrow @jesserockz | esphome/components/esp32_ble_server/* @Rapsssito @clydebarrow @jesserockz | ||||||
| esphome/components/esp32_camera_web_server/* @ayufan | esphome/components/esp32_camera_web_server/* @ayufan | ||||||
| esphome/components/esp32_can/* @Sympatron | esphome/components/esp32_can/* @Sympatron | ||||||
|  | esphome/components/esp32_hosted/* @swoboda1337 | ||||||
| esphome/components/esp32_improv/* @jesserockz | esphome/components/esp32_improv/* @jesserockz | ||||||
| esphome/components/esp32_rmt/* @jesserockz | esphome/components/esp32_rmt/* @jesserockz | ||||||
| esphome/components/esp32_rmt_led_strip/* @jesserockz | esphome/components/esp32_rmt_led_strip/* @jesserockz | ||||||
| @@ -167,6 +170,7 @@ esphome/components/ft5x06/* @clydebarrow | |||||||
| esphome/components/ft63x6/* @gpambrozio | esphome/components/ft63x6/* @gpambrozio | ||||||
| esphome/components/gcja5/* @gcormier | esphome/components/gcja5/* @gcormier | ||||||
| esphome/components/gdk101/* @Szewcson | esphome/components/gdk101/* @Szewcson | ||||||
|  | esphome/components/gl_r01_i2c/* @pkejval | ||||||
| esphome/components/globals/* @esphome/core | esphome/components/globals/* @esphome/core | ||||||
| esphome/components/gp2y1010au0f/* @zry98 | esphome/components/gp2y1010au0f/* @zry98 | ||||||
| esphome/components/gp8403/* @jesserockz | esphome/components/gp8403/* @jesserockz | ||||||
| @@ -247,9 +251,11 @@ esphome/components/libretiny_pwm/* @kuba2k2 | |||||||
| esphome/components/light/* @esphome/core | esphome/components/light/* @esphome/core | ||||||
| esphome/components/lightwaverf/* @max246 | esphome/components/lightwaverf/* @max246 | ||||||
| esphome/components/lilygo_t5_47/touchscreen/* @jesserockz | esphome/components/lilygo_t5_47/touchscreen/* @jesserockz | ||||||
|  | esphome/components/ln882x/* @lamauny | ||||||
| esphome/components/lock/* @esphome/core | esphome/components/lock/* @esphome/core | ||||||
| esphome/components/logger/* @esphome/core | esphome/components/logger/* @esphome/core | ||||||
| esphome/components/logger/select/* @clydebarrow | esphome/components/logger/select/* @clydebarrow | ||||||
|  | esphome/components/lps22/* @nagisa | ||||||
| esphome/components/ltr390/* @latonita @sjtrny | esphome/components/ltr390/* @latonita @sjtrny | ||||||
| esphome/components/ltr501/* @latonita | esphome/components/ltr501/* @latonita | ||||||
| esphome/components/ltr_als_ps/* @latonita | esphome/components/ltr_als_ps/* @latonita | ||||||
| @@ -323,6 +329,7 @@ esphome/components/one_wire/* @ssieb | |||||||
| esphome/components/online_image/* @clydebarrow @guillempages | esphome/components/online_image/* @clydebarrow @guillempages | ||||||
| esphome/components/opentherm/* @olegtarasov | esphome/components/opentherm/* @olegtarasov | ||||||
| esphome/components/openthread/* @mrene | esphome/components/openthread/* @mrene | ||||||
|  | esphome/components/opt3001/* @ccutrer | ||||||
| esphome/components/ota/* @esphome/core | esphome/components/ota/* @esphome/core | ||||||
| esphome/components/output/* @esphome/core | esphome/components/output/* @esphome/core | ||||||
| esphome/components/packet_transport/* @clydebarrow | esphome/components/packet_transport/* @clydebarrow | ||||||
| @@ -330,6 +337,7 @@ esphome/components/pca6416a/* @Mat931 | |||||||
| esphome/components/pca9554/* @clydebarrow @hwstar | esphome/components/pca9554/* @clydebarrow @hwstar | ||||||
| esphome/components/pcf85063/* @brogon | esphome/components/pcf85063/* @brogon | ||||||
| esphome/components/pcf8563/* @KoenBreeman | esphome/components/pcf8563/* @KoenBreeman | ||||||
|  | esphome/components/pi4ioe5v6408/* @jesserockz | ||||||
| esphome/components/pid/* @OttoWinter | esphome/components/pid/* @OttoWinter | ||||||
| esphome/components/pipsolar/* @andreashergert1984 | esphome/components/pipsolar/* @andreashergert1984 | ||||||
| esphome/components/pm1006/* @habbie | esphome/components/pm1006/* @habbie | ||||||
| @@ -436,6 +444,8 @@ esphome/components/sun/* @OttoWinter | |||||||
| esphome/components/sun_gtil2/* @Mat931 | esphome/components/sun_gtil2/* @Mat931 | ||||||
| esphome/components/switch/* @esphome/core | esphome/components/switch/* @esphome/core | ||||||
| esphome/components/switch/binary_sensor/* @ssieb | esphome/components/switch/binary_sensor/* @ssieb | ||||||
|  | esphome/components/sx126x/* @swoboda1337 | ||||||
|  | esphome/components/sx127x/* @swoboda1337 | ||||||
| esphome/components/syslog/* @clydebarrow | esphome/components/syslog/* @clydebarrow | ||||||
| esphome/components/t6615/* @tylermenezes | esphome/components/t6615/* @tylermenezes | ||||||
| esphome/components/tc74/* @sethgirvan | esphome/components/tc74/* @sethgirvan | ||||||
| @@ -494,6 +504,7 @@ esphome/components/voice_assistant/* @jesserockz @kahrendt | |||||||
| esphome/components/wake_on_lan/* @clydebarrow @willwill2will54 | esphome/components/wake_on_lan/* @clydebarrow @willwill2will54 | ||||||
| esphome/components/watchdog/* @oarcher | esphome/components/watchdog/* @oarcher | ||||||
| esphome/components/waveshare_epaper/* @clydebarrow | esphome/components/waveshare_epaper/* @clydebarrow | ||||||
|  | esphome/components/web_server/ota/* @esphome/core | ||||||
| esphome/components/web_server_base/* @OttoWinter | esphome/components/web_server_base/* @OttoWinter | ||||||
| esphome/components/web_server_idf/* @dentra | esphome/components/web_server_idf/* @dentra | ||||||
| esphome/components/weikai/* @DrCoolZic | esphome/components/weikai/* @DrCoolZic | ||||||
| @@ -520,6 +531,7 @@ esphome/components/xiaomi_lywsd03mmc/* @ahpohl | |||||||
| esphome/components/xiaomi_mhoc303/* @drug123 | esphome/components/xiaomi_mhoc303/* @drug123 | ||||||
| esphome/components/xiaomi_mhoc401/* @vevsvevs | esphome/components/xiaomi_mhoc401/* @vevsvevs | ||||||
| esphome/components/xiaomi_rtcgq02lm/* @jesserockz | esphome/components/xiaomi_rtcgq02lm/* @jesserockz | ||||||
|  | esphome/components/xiaomi_xmwsdj04mmc/* @medusalix | ||||||
| esphome/components/xl9535/* @mreditor97 | esphome/components/xl9535/* @mreditor97 | ||||||
| esphome/components/xpt2046/touchscreen/* @nielsnl68 @numo68 | esphome/components/xpt2046/touchscreen/* @nielsnl68 @numo68 | ||||||
| esphome/components/xxtea/* @clydebarrow | esphome/components/xxtea/* @clydebarrow | ||||||
|   | |||||||
							
								
								
									
										2
									
								
								Doxyfile
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								Doxyfile
									
									
									
									
									
								
							| @@ -48,7 +48,7 @@ PROJECT_NAME           = ESPHome | |||||||
| # could be handy for archiving the generated documentation or if some version | # could be handy for archiving the generated documentation or if some version | ||||||
| # control system is used. | # control system is used. | ||||||
|  |  | ||||||
| PROJECT_NUMBER         = 2025.6.3 | PROJECT_NUMBER         = 2025.7.0 | ||||||
|  |  | ||||||
| # Using the PROJECT_BRIEF tag one can provide an optional one line description | # Using the PROJECT_BRIEF tag one can provide an optional one line description | ||||||
| # for a project that appears at the top of each page and should give viewer a | # for a project that appears at the top of each page and should give viewer a | ||||||
|   | |||||||
| @@ -34,11 +34,9 @@ from esphome.const import ( | |||||||
|     CONF_PORT, |     CONF_PORT, | ||||||
|     CONF_SUBSTITUTIONS, |     CONF_SUBSTITUTIONS, | ||||||
|     CONF_TOPIC, |     CONF_TOPIC, | ||||||
|     PLATFORM_BK72XX, |  | ||||||
|     PLATFORM_ESP32, |     PLATFORM_ESP32, | ||||||
|     PLATFORM_ESP8266, |     PLATFORM_ESP8266, | ||||||
|     PLATFORM_RP2040, |     PLATFORM_RP2040, | ||||||
|     PLATFORM_RTL87XX, |  | ||||||
|     SECRETS_FILES, |     SECRETS_FILES, | ||||||
| ) | ) | ||||||
| from esphome.core import CORE, EsphomeError, coroutine | from esphome.core import CORE, EsphomeError, coroutine | ||||||
| @@ -354,7 +352,7 @@ def upload_program(config, args, host): | |||||||
|         if CORE.target_platform in (PLATFORM_RP2040): |         if CORE.target_platform in (PLATFORM_RP2040): | ||||||
|             return upload_using_platformio(config, args.device) |             return upload_using_platformio(config, args.device) | ||||||
|  |  | ||||||
|         if CORE.target_platform in (PLATFORM_BK72XX, PLATFORM_RTL87XX): |         if CORE.is_libretiny: | ||||||
|             return upload_using_platformio(config, host) |             return upload_using_platformio(config, host) | ||||||
|  |  | ||||||
|         return 1  # Unknown target platform |         return 1  # Unknown target platform | ||||||
|   | |||||||
| @@ -22,6 +22,7 @@ from esphome.cpp_generator import (  # noqa: F401 | |||||||
|     TemplateArguments, |     TemplateArguments, | ||||||
|     add, |     add, | ||||||
|     add_build_flag, |     add_build_flag, | ||||||
|  |     add_build_unflag, | ||||||
|     add_define, |     add_define, | ||||||
|     add_global, |     add_global, | ||||||
|     add_library, |     add_library, | ||||||
| @@ -34,6 +35,7 @@ from esphome.cpp_generator import (  # noqa: F401 | |||||||
|     process_lambda, |     process_lambda, | ||||||
|     progmem_array, |     progmem_array, | ||||||
|     safe_exp, |     safe_exp, | ||||||
|  |     set_cpp_standard, | ||||||
|     statement, |     statement, | ||||||
|     static_const_array, |     static_const_array, | ||||||
|     templatable, |     templatable, | ||||||
|   | |||||||
| @@ -4,6 +4,7 @@ | |||||||
| #include "esphome/core/helpers.h" | #include "esphome/core/helpers.h" | ||||||
| #include "esphome/core/log.h" | #include "esphome/core/log.h" | ||||||
| #include <cmath> | #include <cmath> | ||||||
|  | #include <numbers> | ||||||
|  |  | ||||||
| #ifdef USE_ESP8266 | #ifdef USE_ESP8266 | ||||||
| #include <core_esp8266_waveform.h> | #include <core_esp8266_waveform.h> | ||||||
| @@ -193,18 +194,17 @@ void AcDimmer::setup() { | |||||||
|   setTimer1Callback(&timer_interrupt); |   setTimer1Callback(&timer_interrupt); | ||||||
| #endif | #endif | ||||||
| #ifdef USE_ESP32 | #ifdef USE_ESP32 | ||||||
|   // 80 Divider -> 1 count=1µs |   // timer frequency of 1mhz | ||||||
|   dimmer_timer = timerBegin(0, 80, true); |   dimmer_timer = timerBegin(1000000); | ||||||
|   timerAttachInterrupt(dimmer_timer, &AcDimmerDataStore::s_timer_intr, true); |   timerAttachInterrupt(dimmer_timer, &AcDimmerDataStore::s_timer_intr); | ||||||
|   // For ESP32, we can't use dynamic interval calculation because the timerX functions |   // For ESP32, we can't use dynamic interval calculation because the timerX functions | ||||||
|   // are not callable from ISR (placed in flash storage). |   // are not callable from ISR (placed in flash storage). | ||||||
|   // Here we just use an interrupt firing every 50 µs. |   // Here we just use an interrupt firing every 50 µs. | ||||||
|   timerAlarmWrite(dimmer_timer, 50, true); |   timerAlarm(dimmer_timer, 50, true, 0); | ||||||
|   timerAlarmEnable(dimmer_timer); |  | ||||||
| #endif | #endif | ||||||
| } | } | ||||||
| void AcDimmer::write_state(float state) { | void AcDimmer::write_state(float state) { | ||||||
|   state = std::acos(1 - (2 * state)) / 3.14159;  // RMS power compensation |   state = std::acos(1 - (2 * state)) / std::numbers::pi;  // RMS power compensation | ||||||
|   auto new_value = static_cast<uint16_t>(roundf(state * 65535)); |   auto new_value = static_cast<uint16_t>(roundf(state * 65535)); | ||||||
|   if (new_value != 0 && this->store_.value == 0) |   if (new_value != 0 && this->store_.value == 0) | ||||||
|     this->store_.init_cycle = this->init_with_half_cycle_; |     this->store_.init_cycle = this->init_with_half_cycle_; | ||||||
|   | |||||||
| @@ -10,8 +10,15 @@ from esphome.components.esp32.const import ( | |||||||
|     VARIANT_ESP32S2, |     VARIANT_ESP32S2, | ||||||
|     VARIANT_ESP32S3, |     VARIANT_ESP32S3, | ||||||
| ) | ) | ||||||
|  | from esphome.config_helpers import filter_source_files_from_platform | ||||||
| import esphome.config_validation as cv | import esphome.config_validation as cv | ||||||
| from esphome.const import CONF_ANALOG, CONF_INPUT, CONF_NUMBER, PLATFORM_ESP8266 | from esphome.const import ( | ||||||
|  |     CONF_ANALOG, | ||||||
|  |     CONF_INPUT, | ||||||
|  |     CONF_NUMBER, | ||||||
|  |     PLATFORM_ESP8266, | ||||||
|  |     PlatformFramework, | ||||||
|  | ) | ||||||
| from esphome.core import CORE | from esphome.core import CORE | ||||||
|  |  | ||||||
| CODEOWNERS = ["@esphome/core"] | CODEOWNERS = ["@esphome/core"] | ||||||
| @@ -229,3 +236,20 @@ def validate_adc_pin(value): | |||||||
|         )(value) |         )(value) | ||||||
|  |  | ||||||
|     raise NotImplementedError |     raise NotImplementedError | ||||||
|  |  | ||||||
|  |  | ||||||
|  | FILTER_SOURCE_FILES = filter_source_files_from_platform( | ||||||
|  |     { | ||||||
|  |         "adc_sensor_esp32.cpp": { | ||||||
|  |             PlatformFramework.ESP32_ARDUINO, | ||||||
|  |             PlatformFramework.ESP32_IDF, | ||||||
|  |         }, | ||||||
|  |         "adc_sensor_esp8266.cpp": {PlatformFramework.ESP8266_ARDUINO}, | ||||||
|  |         "adc_sensor_rp2040.cpp": {PlatformFramework.RP2040_ARDUINO}, | ||||||
|  |         "adc_sensor_libretiny.cpp": { | ||||||
|  |             PlatformFramework.BK72XX_ARDUINO, | ||||||
|  |             PlatformFramework.RTL87XX_ARDUINO, | ||||||
|  |             PlatformFramework.LN882X_ARDUINO, | ||||||
|  |         }, | ||||||
|  |     } | ||||||
|  | ) | ||||||
|   | |||||||
| @@ -15,8 +15,7 @@ namespace adc { | |||||||
|  |  | ||||||
| #ifdef USE_ESP32 | #ifdef USE_ESP32 | ||||||
| // clang-format off | // clang-format off | ||||||
| #if (ESP_IDF_VERSION_MAJOR == 4 && ESP_IDF_VERSION >= ESP_IDF_VERSION_VAL(4, 4, 7)) || \ | #if (ESP_IDF_VERSION_MAJOR == 5 && \ | ||||||
|     (ESP_IDF_VERSION_MAJOR == 5 && \ |  | ||||||
|      ((ESP_IDF_VERSION_MINOR == 0 && ESP_IDF_VERSION_PATCH >= 5) || \ |      ((ESP_IDF_VERSION_MINOR == 0 && ESP_IDF_VERSION_PATCH >= 5) || \ | ||||||
|       (ESP_IDF_VERSION_MINOR == 1 && ESP_IDF_VERSION_PATCH >= 3) || \ |       (ESP_IDF_VERSION_MINOR == 1 && ESP_IDF_VERSION_PATCH >= 3) || \ | ||||||
|       (ESP_IDF_VERSION_MINOR >= 2)) \ |       (ESP_IDF_VERSION_MINOR >= 2)) \ | ||||||
| @@ -28,19 +27,24 @@ static const adc_atten_t ADC_ATTEN_DB_12_COMPAT = ADC_ATTEN_DB_11; | |||||||
| #endif | #endif | ||||||
| #endif  // USE_ESP32 | #endif  // USE_ESP32 | ||||||
|  |  | ||||||
| enum class SamplingMode : uint8_t { AVG = 0, MIN = 1, MAX = 2 }; | enum class SamplingMode : uint8_t { | ||||||
|  |   AVG = 0, | ||||||
|  |   MIN = 1, | ||||||
|  |   MAX = 2, | ||||||
|  | }; | ||||||
|  |  | ||||||
| const LogString *sampling_mode_to_str(SamplingMode mode); | const LogString *sampling_mode_to_str(SamplingMode mode); | ||||||
|  |  | ||||||
| class Aggregator { | class Aggregator { | ||||||
|  public: |  public: | ||||||
|  |   Aggregator(SamplingMode mode); | ||||||
|   void add_sample(uint32_t value); |   void add_sample(uint32_t value); | ||||||
|   uint32_t aggregate(); |   uint32_t aggregate(); | ||||||
|   Aggregator(SamplingMode mode); |  | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   SamplingMode mode_{SamplingMode::AVG}; |  | ||||||
|   uint32_t aggr_{0}; |   uint32_t aggr_{0}; | ||||||
|   uint32_t samples_{0}; |   uint32_t samples_{0}; | ||||||
|  |   SamplingMode mode_{SamplingMode::AVG}; | ||||||
| }; | }; | ||||||
|  |  | ||||||
| class ADCSensor : public sensor::Sensor, public PollingComponent, public voltage_sampler::VoltageSampler { | class ADCSensor : public sensor::Sensor, public PollingComponent, public voltage_sampler::VoltageSampler { | ||||||
| @@ -81,9 +85,9 @@ class ADCSensor : public sensor::Sensor, public PollingComponent, public voltage | |||||||
| #endif  // USE_RP2040 | #endif  // USE_RP2040 | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   InternalGPIOPin *pin_; |  | ||||||
|   bool output_raw_{false}; |  | ||||||
|   uint8_t sample_count_{1}; |   uint8_t sample_count_{1}; | ||||||
|  |   bool output_raw_{false}; | ||||||
|  |   InternalGPIOPin *pin_; | ||||||
|   SamplingMode sampling_mode_{SamplingMode::AVG}; |   SamplingMode sampling_mode_{SamplingMode::AVG}; | ||||||
|  |  | ||||||
| #ifdef USE_RP2040 | #ifdef USE_RP2040 | ||||||
| @@ -95,11 +99,7 @@ class ADCSensor : public sensor::Sensor, public PollingComponent, public voltage | |||||||
|   adc1_channel_t channel1_{ADC1_CHANNEL_MAX}; |   adc1_channel_t channel1_{ADC1_CHANNEL_MAX}; | ||||||
|   adc2_channel_t channel2_{ADC2_CHANNEL_MAX}; |   adc2_channel_t channel2_{ADC2_CHANNEL_MAX}; | ||||||
|   bool autorange_{false}; |   bool autorange_{false}; | ||||||
| #if ESP_IDF_VERSION_MAJOR >= 5 |  | ||||||
|   esp_adc_cal_characteristics_t cal_characteristics_[SOC_ADC_ATTEN_NUM] = {}; |   esp_adc_cal_characteristics_t cal_characteristics_[SOC_ADC_ATTEN_NUM] = {}; | ||||||
| #else |  | ||||||
|   esp_adc_cal_characteristics_t cal_characteristics_[ADC_ATTEN_MAX] = {}; |  | ||||||
| #endif  // ESP_IDF_VERSION_MAJOR |  | ||||||
| #endif  // USE_ESP32 | #endif  // USE_ESP32 | ||||||
| }; | }; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -61,7 +61,7 @@ uint32_t Aggregator::aggregate() { | |||||||
|  |  | ||||||
| void ADCSensor::update() { | void ADCSensor::update() { | ||||||
|   float value_v = this->sample(); |   float value_v = this->sample(); | ||||||
|   ESP_LOGV(TAG, "'%s': Got voltage=%.4fV", this->get_name().c_str(), value_v); |   ESP_LOGV(TAG, "'%s': Voltage=%.4fV", this->get_name().c_str(), value_v); | ||||||
|   this->publish_state(value_v); |   this->publish_state(value_v); | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -55,32 +55,40 @@ void ADCSensor::setup() { | |||||||
| } | } | ||||||
|  |  | ||||||
| void ADCSensor::dump_config() { | void ADCSensor::dump_config() { | ||||||
|  |   static const char *const ATTEN_AUTO_STR = "auto"; | ||||||
|  |   static const char *const ATTEN_0DB_STR = "0 db"; | ||||||
|  |   static const char *const ATTEN_2_5DB_STR = "2.5 db"; | ||||||
|  |   static const char *const ATTEN_6DB_STR = "6 db"; | ||||||
|  |   static const char *const ATTEN_12DB_STR = "12 db"; | ||||||
|  |   const char *atten_str = ATTEN_AUTO_STR; | ||||||
|  |  | ||||||
|   LOG_SENSOR("", "ADC Sensor", this); |   LOG_SENSOR("", "ADC Sensor", this); | ||||||
|   LOG_PIN("  Pin: ", this->pin_); |   LOG_PIN("  Pin: ", this->pin_); | ||||||
|   if (this->autorange_) { |  | ||||||
|     ESP_LOGCONFIG(TAG, "  Attenuation: auto"); |   if (!this->autorange_) { | ||||||
|   } else { |  | ||||||
|     switch (this->attenuation_) { |     switch (this->attenuation_) { | ||||||
|       case ADC_ATTEN_DB_0: |       case ADC_ATTEN_DB_0: | ||||||
|         ESP_LOGCONFIG(TAG, "  Attenuation: 0db"); |         atten_str = ATTEN_0DB_STR; | ||||||
|         break; |         break; | ||||||
|       case ADC_ATTEN_DB_2_5: |       case ADC_ATTEN_DB_2_5: | ||||||
|         ESP_LOGCONFIG(TAG, "  Attenuation: 2.5db"); |         atten_str = ATTEN_2_5DB_STR; | ||||||
|         break; |         break; | ||||||
|       case ADC_ATTEN_DB_6: |       case ADC_ATTEN_DB_6: | ||||||
|         ESP_LOGCONFIG(TAG, "  Attenuation: 6db"); |         atten_str = ATTEN_6DB_STR; | ||||||
|         break; |         break; | ||||||
|       case ADC_ATTEN_DB_12_COMPAT: |       case ADC_ATTEN_DB_12_COMPAT: | ||||||
|         ESP_LOGCONFIG(TAG, "  Attenuation: 12db"); |         atten_str = ATTEN_12DB_STR; | ||||||
|         break; |         break; | ||||||
|       default:  // This is to satisfy the unused ADC_ATTEN_MAX |       default:  // This is to satisfy the unused ADC_ATTEN_MAX | ||||||
|         break; |         break; | ||||||
|     } |     } | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   ESP_LOGCONFIG(TAG, |   ESP_LOGCONFIG(TAG, | ||||||
|  |                 "  Attenuation: %s\n" | ||||||
|                 "  Samples: %i\n" |                 "  Samples: %i\n" | ||||||
|                 "  Sampling mode: %s", |                 "  Sampling mode: %s", | ||||||
|                 this->sample_count_, LOG_STR_ARG(sampling_mode_to_str(this->sampling_mode_))); |                 atten_str, this->sample_count_, LOG_STR_ARG(sampling_mode_to_str(this->sampling_mode_))); | ||||||
|   LOG_UPDATE_INTERVAL(this); |   LOG_UPDATE_INTERVAL(this); | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -85,8 +85,6 @@ class ADE7880 : public i2c::I2CDevice, public PollingComponent { | |||||||
|  |  | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   ADE7880Store store_{}; |   ADE7880Store store_{}; | ||||||
|   InternalGPIOPin *irq0_pin_{nullptr}; |   InternalGPIOPin *irq0_pin_{nullptr}; | ||||||
|   | |||||||
| @@ -49,7 +49,6 @@ class ADS1115Component : public Component, public i2c::I2CDevice { | |||||||
|   void setup() override; |   void setup() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   /// HARDWARE_LATE setup priority |   /// HARDWARE_LATE setup priority | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   void set_continuous_mode(bool continuous_mode) { continuous_mode_ = continuous_mode; } |   void set_continuous_mode(bool continuous_mode) { continuous_mode_ = continuous_mode; } | ||||||
|  |  | ||||||
|   /// Helper method to request a measurement from a sensor. |   /// Helper method to request a measurement from a sensor. | ||||||
|   | |||||||
| @@ -34,7 +34,6 @@ class ADS1118 : public Component, | |||||||
|   ADS1118() = default; |   ADS1118() = default; | ||||||
|   void setup() override; |   void setup() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   /// Helper method to request a measurement from a sensor. |   /// Helper method to request a measurement from a sensor. | ||||||
|   float request_measurement(ADS1118Multiplexer multiplexer, ADS1118Gain gain, bool temperature_mode); |   float request_measurement(ADS1118Multiplexer multiplexer, ADS1118Gain gain, bool temperature_mode); | ||||||
|  |  | ||||||
|   | |||||||
| @@ -31,8 +31,6 @@ class AGS10Component : public PollingComponent, public i2c::I2CDevice { | |||||||
|  |  | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   /** |   /** | ||||||
|    * Modifies target address of AGS10. |    * Modifies target address of AGS10. | ||||||
|    * |    * | ||||||
|   | |||||||
| @@ -66,7 +66,6 @@ class AIC3204 : public audio_dac::AudioDac, public Component, public i2c::I2CDev | |||||||
|  public: |  public: | ||||||
|   void setup() override; |   void setup() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   bool set_mute_off() override; |   bool set_mute_off() override; | ||||||
|   bool set_mute_on() override; |   bool set_mute_on() override; | ||||||
|   | |||||||
| @@ -14,8 +14,8 @@ from esphome.const import ( | |||||||
|     CONF_WEB_SERVER, |     CONF_WEB_SERVER, | ||||||
| ) | ) | ||||||
| from esphome.core import CORE, coroutine_with_priority | from esphome.core import CORE, coroutine_with_priority | ||||||
|  | from esphome.core.entity_helpers import entity_duplicate_validator, setup_entity | ||||||
| from esphome.cpp_generator import MockObjClass | from esphome.cpp_generator import MockObjClass | ||||||
| from esphome.cpp_helpers import setup_entity |  | ||||||
|  |  | ||||||
| CODEOWNERS = ["@grahambrown11", "@hwstar"] | CODEOWNERS = ["@grahambrown11", "@hwstar"] | ||||||
| IS_PLATFORM_COMPONENT = True | IS_PLATFORM_COMPONENT = True | ||||||
| @@ -149,6 +149,9 @@ _ALARM_CONTROL_PANEL_SCHEMA = ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | _ALARM_CONTROL_PANEL_SCHEMA.add_extra(entity_duplicate_validator("alarm_control_panel")) | ||||||
|  |  | ||||||
|  |  | ||||||
| def alarm_control_panel_schema( | def alarm_control_panel_schema( | ||||||
|     class_: MockObjClass, |     class_: MockObjClass, | ||||||
|     *, |     *, | ||||||
| @@ -190,7 +193,7 @@ ALARM_CONTROL_PANEL_CONDITION_SCHEMA = maybe_simple_id( | |||||||
|  |  | ||||||
|  |  | ||||||
| async def setup_alarm_control_panel_core_(var, config): | async def setup_alarm_control_panel_core_(var, config): | ||||||
|     await setup_entity(var, config) |     await setup_entity(var, config, "alarm_control_panel") | ||||||
|     for conf in config.get(CONF_ON_STATE, []): |     for conf in config.get(CONF_ON_STATE, []): | ||||||
|         trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var) |         trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var) | ||||||
|         await automation.build_automation(trigger, [], conf) |         await automation.build_automation(trigger, [], conf) | ||||||
|   | |||||||
| @@ -41,7 +41,6 @@ class Alpha3 : public esphome::ble_client::BLEClientNode, public PollingComponen | |||||||
|   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, |   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, | ||||||
|                            esp_ble_gattc_cb_param_t *param) override; |                            esp_ble_gattc_cb_param_t *param) override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   void set_flow_sensor(sensor::Sensor *sensor) { this->flow_sensor_ = sensor; } |   void set_flow_sensor(sensor::Sensor *sensor) { this->flow_sensor_ = sensor; } | ||||||
|   void set_head_sensor(sensor::Sensor *sensor) { this->head_sensor_ = sensor; } |   void set_head_sensor(sensor::Sensor *sensor) { this->head_sensor_ = sensor; } | ||||||
|   void set_power_sensor(sensor::Sensor *sensor) { this->power_sensor_ = sensor; } |   void set_power_sensor(sensor::Sensor *sensor) { this->power_sensor_ = sensor; } | ||||||
|   | |||||||
| @@ -22,7 +22,6 @@ class Am43Component : public cover::Cover, public esphome::ble_client::BLEClient | |||||||
|   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, |   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, | ||||||
|                            esp_ble_gattc_cb_param_t *param) override; |                            esp_ble_gattc_cb_param_t *param) override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   cover::CoverTraits get_traits() override; |   cover::CoverTraits get_traits() override; | ||||||
|   void set_pin(uint16_t pin) { this->pin_ = pin; } |   void set_pin(uint16_t pin) { this->pin_ = pin; } | ||||||
|   void set_invert_position(bool invert_position) { this->invert_position_ = invert_position; } |   void set_invert_position(bool invert_position) { this->invert_position_ = invert_position; } | ||||||
|   | |||||||
| @@ -22,7 +22,6 @@ class Am43 : public esphome::ble_client::BLEClientNode, public PollingComponent | |||||||
|   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, |   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, | ||||||
|                            esp_ble_gattc_cb_param_t *param) override; |                            esp_ble_gattc_cb_param_t *param) override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   void set_battery(sensor::Sensor *battery) { battery_ = battery; } |   void set_battery(sensor::Sensor *battery) { battery_ = battery; } | ||||||
|   void set_illuminance(sensor::Sensor *illuminance) { illuminance_ = illuminance; } |   void set_illuminance(sensor::Sensor *illuminance) { illuminance_ = illuminance; } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -12,8 +12,6 @@ class AnalogThresholdBinarySensor : public Component, public binary_sensor::Bina | |||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   void setup() override; |   void setup() override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   void set_sensor(sensor::Sensor *analog_sensor); |   void set_sensor(sensor::Sensor *analog_sensor); | ||||||
|   template<typename T> void set_upper_threshold(T upper_threshold) { this->upper_threshold_ = upper_threshold; } |   template<typename T> void set_upper_threshold(T upper_threshold) { this->upper_threshold_ = upper_threshold; } | ||||||
|   template<typename T> void set_lower_threshold(T lower_threshold) { this->lower_threshold_ = lower_threshold; } |   template<typename T> void set_lower_threshold(T lower_threshold) { this->lower_threshold_ = lower_threshold; } | ||||||
|   | |||||||
| @@ -17,7 +17,11 @@ void Anova::setup() { | |||||||
|   this->current_request_ = 0; |   this->current_request_ = 0; | ||||||
| } | } | ||||||
|  |  | ||||||
| void Anova::loop() {} | void Anova::loop() { | ||||||
|  |   // Parent BLEClientNode has a loop() method, but this component uses | ||||||
|  |   // polling via update() and BLE callbacks so loop isn't needed | ||||||
|  |   this->disable_loop(); | ||||||
|  | } | ||||||
|  |  | ||||||
| void Anova::control(const ClimateCall &call) { | void Anova::control(const ClimateCall &call) { | ||||||
|   if (call.get_mode().has_value()) { |   if (call.get_mode().has_value()) { | ||||||
|   | |||||||
| @@ -26,7 +26,6 @@ class Anova : public climate::Climate, public esphome::ble_client::BLEClientNode | |||||||
|   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, |   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, | ||||||
|                            esp_ble_gattc_cb_param_t *param) override; |                            esp_ble_gattc_cb_param_t *param) override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   climate::ClimateTraits traits() override { |   climate::ClimateTraits traits() override { | ||||||
|     auto traits = climate::ClimateTraits(); |     auto traits = climate::ClimateTraits(); | ||||||
|     traits.set_supports_current_temperature(true); |     traits.set_supports_current_temperature(true); | ||||||
|   | |||||||
| @@ -23,7 +23,7 @@ void APDS9960::setup() { | |||||||
|     return; |     return; | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   if (id != 0xAB && id != 0x9C && id != 0xA8) {  // APDS9960 all should have one of these IDs |   if (id != 0xAB && id != 0x9C && id != 0xA8 && id != 0x9E) {  // APDS9960 all should have one of these IDs | ||||||
|     this->error_code_ = WRONG_ID; |     this->error_code_ = WRONG_ID; | ||||||
|     this->mark_failed(); |     this->mark_failed(); | ||||||
|     return; |     return; | ||||||
|   | |||||||
| @@ -3,6 +3,7 @@ import base64 | |||||||
| from esphome import automation | from esphome import automation | ||||||
| from esphome.automation import Condition | from esphome.automation import Condition | ||||||
| import esphome.codegen as cg | import esphome.codegen as cg | ||||||
|  | from esphome.config_helpers import get_logger_level | ||||||
| import esphome.config_validation as cv | import esphome.config_validation as cv | ||||||
| from esphome.const import ( | from esphome.const import ( | ||||||
|     CONF_ACTION, |     CONF_ACTION, | ||||||
| @@ -23,8 +24,9 @@ from esphome.const import ( | |||||||
|     CONF_TRIGGER_ID, |     CONF_TRIGGER_ID, | ||||||
|     CONF_VARIABLES, |     CONF_VARIABLES, | ||||||
| ) | ) | ||||||
| from esphome.core import coroutine_with_priority | from esphome.core import CORE, coroutine_with_priority | ||||||
|  |  | ||||||
|  | DOMAIN = "api" | ||||||
| DEPENDENCIES = ["network"] | DEPENDENCIES = ["network"] | ||||||
| AUTO_LOAD = ["socket"] | AUTO_LOAD = ["socket"] | ||||||
| CODEOWNERS = ["@OttoWinter"] | CODEOWNERS = ["@OttoWinter"] | ||||||
| @@ -50,6 +52,7 @@ SERVICE_ARG_NATIVE_TYPES = { | |||||||
| } | } | ||||||
| CONF_ENCRYPTION = "encryption" | CONF_ENCRYPTION = "encryption" | ||||||
| CONF_BATCH_DELAY = "batch_delay" | CONF_BATCH_DELAY = "batch_delay" | ||||||
|  | CONF_CUSTOM_SERVICES = "custom_services" | ||||||
|  |  | ||||||
|  |  | ||||||
| def validate_encryption_key(value): | def validate_encryption_key(value): | ||||||
| @@ -110,9 +113,11 @@ CONFIG_SCHEMA = cv.All( | |||||||
|             ): ACTIONS_SCHEMA, |             ): ACTIONS_SCHEMA, | ||||||
|             cv.Exclusive(CONF_ACTIONS, group_of_exclusion=CONF_ACTIONS): ACTIONS_SCHEMA, |             cv.Exclusive(CONF_ACTIONS, group_of_exclusion=CONF_ACTIONS): ACTIONS_SCHEMA, | ||||||
|             cv.Optional(CONF_ENCRYPTION): _encryption_schema, |             cv.Optional(CONF_ENCRYPTION): _encryption_schema, | ||||||
|             cv.Optional( |             cv.Optional(CONF_BATCH_DELAY, default="100ms"): cv.All( | ||||||
|                 CONF_BATCH_DELAY, default="100ms" |                 cv.positive_time_period_milliseconds, | ||||||
|             ): cv.positive_time_period_milliseconds, |                 cv.Range(max=cv.TimePeriod(milliseconds=65535)), | ||||||
|  |             ), | ||||||
|  |             cv.Optional(CONF_CUSTOM_SERVICES, default=False): cv.boolean, | ||||||
|             cv.Optional(CONF_ON_CLIENT_CONNECTED): automation.validate_automation( |             cv.Optional(CONF_ON_CLIENT_CONNECTED): automation.validate_automation( | ||||||
|                 single=True |                 single=True | ||||||
|             ), |             ), | ||||||
| @@ -131,11 +136,18 @@ async def to_code(config): | |||||||
|     await cg.register_component(var, config) |     await cg.register_component(var, config) | ||||||
|  |  | ||||||
|     cg.add(var.set_port(config[CONF_PORT])) |     cg.add(var.set_port(config[CONF_PORT])) | ||||||
|  |     if config[CONF_PASSWORD]: | ||||||
|  |         cg.add_define("USE_API_PASSWORD") | ||||||
|         cg.add(var.set_password(config[CONF_PASSWORD])) |         cg.add(var.set_password(config[CONF_PASSWORD])) | ||||||
|     cg.add(var.set_reboot_timeout(config[CONF_REBOOT_TIMEOUT])) |     cg.add(var.set_reboot_timeout(config[CONF_REBOOT_TIMEOUT])) | ||||||
|     cg.add(var.set_batch_delay(config[CONF_BATCH_DELAY])) |     cg.add(var.set_batch_delay(config[CONF_BATCH_DELAY])) | ||||||
|  |  | ||||||
|     for conf in config.get(CONF_ACTIONS, []): |     # Set USE_API_SERVICES if any services are enabled | ||||||
|  |     if config.get(CONF_ACTIONS) or config[CONF_CUSTOM_SERVICES]: | ||||||
|  |         cg.add_define("USE_API_SERVICES") | ||||||
|  |  | ||||||
|  |     if actions := config.get(CONF_ACTIONS, []): | ||||||
|  |         for conf in actions: | ||||||
|             template_args = [] |             template_args = [] | ||||||
|             func_args = [] |             func_args = [] | ||||||
|             service_arg_names = [] |             service_arg_names = [] | ||||||
| @@ -152,6 +164,7 @@ async def to_code(config): | |||||||
|             await automation.build_automation(trigger, func_args, conf) |             await automation.build_automation(trigger, func_args, conf) | ||||||
|  |  | ||||||
|     if CONF_ON_CLIENT_CONNECTED in config: |     if CONF_ON_CLIENT_CONNECTED in config: | ||||||
|  |         cg.add_define("USE_API_CLIENT_CONNECTED_TRIGGER") | ||||||
|         await automation.build_automation( |         await automation.build_automation( | ||||||
|             var.get_client_connected_trigger(), |             var.get_client_connected_trigger(), | ||||||
|             [(cg.std_string, "client_info"), (cg.std_string, "client_address")], |             [(cg.std_string, "client_info"), (cg.std_string, "client_address")], | ||||||
| @@ -159,6 +172,7 @@ async def to_code(config): | |||||||
|         ) |         ) | ||||||
|  |  | ||||||
|     if CONF_ON_CLIENT_DISCONNECTED in config: |     if CONF_ON_CLIENT_DISCONNECTED in config: | ||||||
|  |         cg.add_define("USE_API_CLIENT_DISCONNECTED_TRIGGER") | ||||||
|         await automation.build_automation( |         await automation.build_automation( | ||||||
|             var.get_client_disconnected_trigger(), |             var.get_client_disconnected_trigger(), | ||||||
|             [(cg.std_string, "client_info"), (cg.std_string, "client_address")], |             [(cg.std_string, "client_info"), (cg.std_string, "client_address")], | ||||||
| @@ -177,7 +191,7 @@ async def to_code(config): | |||||||
|             # and plaintext disabled. Only a factory reset can remove it. |             # and plaintext disabled. Only a factory reset can remove it. | ||||||
|             cg.add_define("USE_API_PLAINTEXT") |             cg.add_define("USE_API_PLAINTEXT") | ||||||
|         cg.add_define("USE_API_NOISE") |         cg.add_define("USE_API_NOISE") | ||||||
|         cg.add_library("esphome/noise-c", "0.1.6") |         cg.add_library("esphome/noise-c", "0.1.10") | ||||||
|     else: |     else: | ||||||
|         cg.add_define("USE_API_PLAINTEXT") |         cg.add_define("USE_API_PLAINTEXT") | ||||||
|  |  | ||||||
| @@ -306,3 +320,25 @@ async def homeassistant_tag_scanned_to_code(config, action_id, template_arg, arg | |||||||
| @automation.register_condition("api.connected", APIConnectedCondition, {}) | @automation.register_condition("api.connected", APIConnectedCondition, {}) | ||||||
| async def api_connected_to_code(config, condition_id, template_arg, args): | async def api_connected_to_code(config, condition_id, template_arg, args): | ||||||
|     return cg.new_Pvariable(condition_id, template_arg) |     return cg.new_Pvariable(condition_id, template_arg) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | def FILTER_SOURCE_FILES() -> list[str]: | ||||||
|  |     """Filter out api_pb2_dump.cpp when proto message dumping is not enabled | ||||||
|  |     and user_services.cpp when no services are defined.""" | ||||||
|  |     files_to_filter = [] | ||||||
|  |  | ||||||
|  |     # api_pb2_dump.cpp is only needed when HAS_PROTO_MESSAGE_DUMP is defined | ||||||
|  |     # This is a particularly large file that still needs to be opened and read | ||||||
|  |     # all the way to the end even when ifdef'd out | ||||||
|  |     # | ||||||
|  |     # HAS_PROTO_MESSAGE_DUMP is defined when ESPHOME_LOG_HAS_VERY_VERBOSE is set, | ||||||
|  |     # which happens when the logger level is VERY_VERBOSE | ||||||
|  |     if get_logger_level() != "VERY_VERBOSE": | ||||||
|  |         files_to_filter.append("api_pb2_dump.cpp") | ||||||
|  |  | ||||||
|  |     # user_services.cpp is only needed when services are defined | ||||||
|  |     config = CORE.config.get(DOMAIN, {}) | ||||||
|  |     if config and not config.get(CONF_ACTIONS) and not config[CONF_CUSTOM_SERVICES]: | ||||||
|  |         files_to_filter.append("user_services.cpp") | ||||||
|  |  | ||||||
|  |     return files_to_filter | ||||||
|   | |||||||
| @@ -188,6 +188,17 @@ message DeviceInfoRequest { | |||||||
|   // Empty |   // Empty | ||||||
| } | } | ||||||
|  |  | ||||||
|  | message AreaInfo { | ||||||
|  |   uint32 area_id = 1; | ||||||
|  |   string name = 2; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | message DeviceInfo { | ||||||
|  |   uint32 device_id = 1; | ||||||
|  |   string name = 2; | ||||||
|  |   uint32 area_id = 3; | ||||||
|  | } | ||||||
|  |  | ||||||
| message DeviceInfoResponse { | message DeviceInfoResponse { | ||||||
|   option (id) = 10; |   option (id) = 10; | ||||||
|   option (source) = SOURCE_SERVER; |   option (source) = SOURCE_SERVER; | ||||||
| @@ -236,6 +247,12 @@ message DeviceInfoResponse { | |||||||
|  |  | ||||||
|   // Supports receiving and saving api encryption key |   // Supports receiving and saving api encryption key | ||||||
|   bool api_encryption_supported = 19; |   bool api_encryption_supported = 19; | ||||||
|  |  | ||||||
|  |   repeated DeviceInfo devices = 20; | ||||||
|  |   repeated AreaInfo areas = 21; | ||||||
|  |  | ||||||
|  |   // Top-level area info to phase out suggested_area | ||||||
|  |   AreaInfo area = 22; | ||||||
| } | } | ||||||
|  |  | ||||||
| message ListEntitiesRequest { | message ListEntitiesRequest { | ||||||
| @@ -280,6 +297,7 @@ message ListEntitiesBinarySensorResponse { | |||||||
|   bool disabled_by_default = 7; |   bool disabled_by_default = 7; | ||||||
|   string icon = 8; |   string icon = 8; | ||||||
|   EntityCategory entity_category = 9; |   EntityCategory entity_category = 9; | ||||||
|  |   uint32 device_id = 10; | ||||||
| } | } | ||||||
| message BinarySensorStateResponse { | message BinarySensorStateResponse { | ||||||
|   option (id) = 21; |   option (id) = 21; | ||||||
| @@ -293,6 +311,7 @@ message BinarySensorStateResponse { | |||||||
|   // If the binary sensor does not have a valid state yet. |   // If the binary sensor does not have a valid state yet. | ||||||
|   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller |   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller | ||||||
|   bool missing_state = 3; |   bool missing_state = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== COVER ==================== | // ==================== COVER ==================== | ||||||
| @@ -315,6 +334,7 @@ message ListEntitiesCoverResponse { | |||||||
|   string icon = 10; |   string icon = 10; | ||||||
|   EntityCategory entity_category = 11; |   EntityCategory entity_category = 11; | ||||||
|   bool supports_stop = 12; |   bool supports_stop = 12; | ||||||
|  |   uint32 device_id = 13; | ||||||
| } | } | ||||||
|  |  | ||||||
| enum LegacyCoverState { | enum LegacyCoverState { | ||||||
| @@ -341,6 +361,7 @@ message CoverStateResponse { | |||||||
|   float position = 3; |   float position = 3; | ||||||
|   float tilt = 4; |   float tilt = 4; | ||||||
|   CoverOperation current_operation = 5; |   CoverOperation current_operation = 5; | ||||||
|  |   uint32 device_id = 6; | ||||||
| } | } | ||||||
|  |  | ||||||
| enum LegacyCoverCommand { | enum LegacyCoverCommand { | ||||||
| @@ -353,6 +374,7 @@ message CoverCommandRequest { | |||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_COVER"; |   option (ifdef) = "USE_COVER"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|  |  | ||||||
| @@ -366,6 +388,7 @@ message CoverCommandRequest { | |||||||
|   bool has_tilt = 6; |   bool has_tilt = 6; | ||||||
|   float tilt = 7; |   float tilt = 7; | ||||||
|   bool stop = 8; |   bool stop = 8; | ||||||
|  |   uint32 device_id = 9; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== FAN ==================== | // ==================== FAN ==================== | ||||||
| @@ -388,6 +411,7 @@ message ListEntitiesFanResponse { | |||||||
|   string icon = 10; |   string icon = 10; | ||||||
|   EntityCategory entity_category = 11; |   EntityCategory entity_category = 11; | ||||||
|   repeated string supported_preset_modes = 12; |   repeated string supported_preset_modes = 12; | ||||||
|  |   uint32 device_id = 13; | ||||||
| } | } | ||||||
| enum FanSpeed { | enum FanSpeed { | ||||||
|   FAN_SPEED_LOW = 0; |   FAN_SPEED_LOW = 0; | ||||||
| @@ -412,12 +436,14 @@ message FanStateResponse { | |||||||
|   FanDirection direction = 5; |   FanDirection direction = 5; | ||||||
|   int32 speed_level = 6; |   int32 speed_level = 6; | ||||||
|   string preset_mode = 7; |   string preset_mode = 7; | ||||||
|  |   uint32 device_id = 8; | ||||||
| } | } | ||||||
| message FanCommandRequest { | message FanCommandRequest { | ||||||
|   option (id) = 31; |   option (id) = 31; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_FAN"; |   option (ifdef) = "USE_FAN"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bool has_state = 2; |   bool has_state = 2; | ||||||
| @@ -432,6 +458,7 @@ message FanCommandRequest { | |||||||
|   int32 speed_level = 11; |   int32 speed_level = 11; | ||||||
|   bool has_preset_mode = 12; |   bool has_preset_mode = 12; | ||||||
|   string preset_mode = 13; |   string preset_mode = 13; | ||||||
|  |   uint32 device_id = 14; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== LIGHT ==================== | // ==================== LIGHT ==================== | ||||||
| @@ -471,6 +498,7 @@ message ListEntitiesLightResponse { | |||||||
|   bool disabled_by_default = 13; |   bool disabled_by_default = 13; | ||||||
|   string icon = 14; |   string icon = 14; | ||||||
|   EntityCategory entity_category = 15; |   EntityCategory entity_category = 15; | ||||||
|  |   uint32 device_id = 16; | ||||||
| } | } | ||||||
| message LightStateResponse { | message LightStateResponse { | ||||||
|   option (id) = 24; |   option (id) = 24; | ||||||
| @@ -492,12 +520,14 @@ message LightStateResponse { | |||||||
|   float cold_white = 12; |   float cold_white = 12; | ||||||
|   float warm_white = 13; |   float warm_white = 13; | ||||||
|   string effect = 9; |   string effect = 9; | ||||||
|  |   uint32 device_id = 14; | ||||||
| } | } | ||||||
| message LightCommandRequest { | message LightCommandRequest { | ||||||
|   option (id) = 32; |   option (id) = 32; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_LIGHT"; |   option (ifdef) = "USE_LIGHT"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bool has_state = 2; |   bool has_state = 2; | ||||||
| @@ -526,6 +556,7 @@ message LightCommandRequest { | |||||||
|   uint32 flash_length = 17; |   uint32 flash_length = 17; | ||||||
|   bool has_effect = 18; |   bool has_effect = 18; | ||||||
|   string effect = 19; |   string effect = 19; | ||||||
|  |   uint32 device_id = 28; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== SENSOR ==================== | // ==================== SENSOR ==================== | ||||||
| @@ -563,6 +594,7 @@ message ListEntitiesSensorResponse { | |||||||
|   SensorLastResetType legacy_last_reset_type = 11; |   SensorLastResetType legacy_last_reset_type = 11; | ||||||
|   bool disabled_by_default = 12; |   bool disabled_by_default = 12; | ||||||
|   EntityCategory entity_category = 13; |   EntityCategory entity_category = 13; | ||||||
|  |   uint32 device_id = 14; | ||||||
| } | } | ||||||
| message SensorStateResponse { | message SensorStateResponse { | ||||||
|   option (id) = 25; |   option (id) = 25; | ||||||
| @@ -576,6 +608,7 @@ message SensorStateResponse { | |||||||
|   // If the sensor does not have a valid state yet. |   // If the sensor does not have a valid state yet. | ||||||
|   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller |   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller | ||||||
|   bool missing_state = 3; |   bool missing_state = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== SWITCH ==================== | // ==================== SWITCH ==================== | ||||||
| @@ -595,6 +628,7 @@ message ListEntitiesSwitchResponse { | |||||||
|   bool disabled_by_default = 7; |   bool disabled_by_default = 7; | ||||||
|   EntityCategory entity_category = 8; |   EntityCategory entity_category = 8; | ||||||
|   string device_class = 9; |   string device_class = 9; | ||||||
|  |   uint32 device_id = 10; | ||||||
| } | } | ||||||
| message SwitchStateResponse { | message SwitchStateResponse { | ||||||
|   option (id) = 26; |   option (id) = 26; | ||||||
| @@ -605,15 +639,18 @@ message SwitchStateResponse { | |||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bool state = 2; |   bool state = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
| message SwitchCommandRequest { | message SwitchCommandRequest { | ||||||
|   option (id) = 33; |   option (id) = 33; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_SWITCH"; |   option (ifdef) = "USE_SWITCH"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bool state = 2; |   bool state = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== TEXT SENSOR ==================== | // ==================== TEXT SENSOR ==================== | ||||||
| @@ -632,6 +669,7 @@ message ListEntitiesTextSensorResponse { | |||||||
|   bool disabled_by_default = 6; |   bool disabled_by_default = 6; | ||||||
|   EntityCategory entity_category = 7; |   EntityCategory entity_category = 7; | ||||||
|   string device_class = 8; |   string device_class = 8; | ||||||
|  |   uint32 device_id = 9; | ||||||
| } | } | ||||||
| message TextSensorStateResponse { | message TextSensorStateResponse { | ||||||
|   option (id) = 27; |   option (id) = 27; | ||||||
| @@ -645,6 +683,7 @@ message TextSensorStateResponse { | |||||||
|   // If the text sensor does not have a valid state yet. |   // If the text sensor does not have a valid state yet. | ||||||
|   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller |   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller | ||||||
|   bool missing_state = 3; |   bool missing_state = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== SUBSCRIBE LOGS ==================== | // ==================== SUBSCRIBE LOGS ==================== | ||||||
| @@ -768,18 +807,21 @@ enum ServiceArgType { | |||||||
|   SERVICE_ARG_TYPE_STRING_ARRAY = 7; |   SERVICE_ARG_TYPE_STRING_ARRAY = 7; | ||||||
| } | } | ||||||
| message ListEntitiesServicesArgument { | message ListEntitiesServicesArgument { | ||||||
|  |   option (ifdef) = "USE_API_SERVICES"; | ||||||
|   string name = 1; |   string name = 1; | ||||||
|   ServiceArgType type = 2; |   ServiceArgType type = 2; | ||||||
| } | } | ||||||
| message ListEntitiesServicesResponse { | message ListEntitiesServicesResponse { | ||||||
|   option (id) = 41; |   option (id) = 41; | ||||||
|   option (source) = SOURCE_SERVER; |   option (source) = SOURCE_SERVER; | ||||||
|  |   option (ifdef) = "USE_API_SERVICES"; | ||||||
|  |  | ||||||
|   string name = 1; |   string name = 1; | ||||||
|   fixed32 key = 2; |   fixed32 key = 2; | ||||||
|   repeated ListEntitiesServicesArgument args = 3; |   repeated ListEntitiesServicesArgument args = 3; | ||||||
| } | } | ||||||
| message ExecuteServiceArgument { | message ExecuteServiceArgument { | ||||||
|  |   option (ifdef) = "USE_API_SERVICES"; | ||||||
|   bool bool_ = 1; |   bool bool_ = 1; | ||||||
|   int32 legacy_int = 2; |   int32 legacy_int = 2; | ||||||
|   float float_ = 3; |   float float_ = 3; | ||||||
| @@ -795,6 +837,7 @@ message ExecuteServiceRequest { | |||||||
|   option (id) = 42; |   option (id) = 42; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (ifdef) = "USE_API_SERVICES"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   repeated ExecuteServiceArgument args = 2; |   repeated ExecuteServiceArgument args = 2; | ||||||
| @@ -805,7 +848,7 @@ message ListEntitiesCameraResponse { | |||||||
|   option (id) = 43; |   option (id) = 43; | ||||||
|   option (base_class) = "InfoResponseProtoMessage"; |   option (base_class) = "InfoResponseProtoMessage"; | ||||||
|   option (source) = SOURCE_SERVER; |   option (source) = SOURCE_SERVER; | ||||||
|   option (ifdef) = "USE_ESP32_CAMERA"; |   option (ifdef) = "USE_CAMERA"; | ||||||
|  |  | ||||||
|   string object_id = 1; |   string object_id = 1; | ||||||
|   fixed32 key = 2; |   fixed32 key = 2; | ||||||
| @@ -814,21 +857,24 @@ message ListEntitiesCameraResponse { | |||||||
|   bool disabled_by_default = 5; |   bool disabled_by_default = 5; | ||||||
|   string icon = 6; |   string icon = 6; | ||||||
|   EntityCategory entity_category = 7; |   EntityCategory entity_category = 7; | ||||||
|  |   uint32 device_id = 8; | ||||||
| } | } | ||||||
|  |  | ||||||
| message CameraImageResponse { | message CameraImageResponse { | ||||||
|   option (id) = 44; |   option (id) = 44; | ||||||
|  |   option (base_class) = "StateResponseProtoMessage"; | ||||||
|   option (source) = SOURCE_SERVER; |   option (source) = SOURCE_SERVER; | ||||||
|   option (ifdef) = "USE_ESP32_CAMERA"; |   option (ifdef) = "USE_CAMERA"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bytes data = 2; |   bytes data = 2; | ||||||
|   bool done = 3; |   bool done = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
| message CameraImageRequest { | message CameraImageRequest { | ||||||
|   option (id) = 45; |   option (id) = 45; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_ESP32_CAMERA"; |   option (ifdef) = "USE_CAMERA"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |  | ||||||
|   bool single = 1; |   bool single = 1; | ||||||
| @@ -916,6 +962,7 @@ message ListEntitiesClimateResponse { | |||||||
|   bool supports_target_humidity = 23; |   bool supports_target_humidity = 23; | ||||||
|   float visual_min_humidity = 24; |   float visual_min_humidity = 24; | ||||||
|   float visual_max_humidity = 25; |   float visual_max_humidity = 25; | ||||||
|  |   uint32 device_id = 26; | ||||||
| } | } | ||||||
| message ClimateStateResponse { | message ClimateStateResponse { | ||||||
|   option (id) = 47; |   option (id) = 47; | ||||||
| @@ -940,12 +987,14 @@ message ClimateStateResponse { | |||||||
|   string custom_preset = 13; |   string custom_preset = 13; | ||||||
|   float current_humidity = 14; |   float current_humidity = 14; | ||||||
|   float target_humidity = 15; |   float target_humidity = 15; | ||||||
|  |   uint32 device_id = 16; | ||||||
| } | } | ||||||
| message ClimateCommandRequest { | message ClimateCommandRequest { | ||||||
|   option (id) = 48; |   option (id) = 48; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_CLIMATE"; |   option (ifdef) = "USE_CLIMATE"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bool has_mode = 2; |   bool has_mode = 2; | ||||||
| @@ -971,6 +1020,7 @@ message ClimateCommandRequest { | |||||||
|   string custom_preset = 21; |   string custom_preset = 21; | ||||||
|   bool has_target_humidity = 22; |   bool has_target_humidity = 22; | ||||||
|   float target_humidity = 23; |   float target_humidity = 23; | ||||||
|  |   uint32 device_id = 24; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== NUMBER ==================== | // ==================== NUMBER ==================== | ||||||
| @@ -999,6 +1049,7 @@ message ListEntitiesNumberResponse { | |||||||
|   string unit_of_measurement = 11; |   string unit_of_measurement = 11; | ||||||
|   NumberMode mode = 12; |   NumberMode mode = 12; | ||||||
|   string device_class = 13; |   string device_class = 13; | ||||||
|  |   uint32 device_id = 14; | ||||||
| } | } | ||||||
| message NumberStateResponse { | message NumberStateResponse { | ||||||
|   option (id) = 50; |   option (id) = 50; | ||||||
| @@ -1012,15 +1063,18 @@ message NumberStateResponse { | |||||||
|   // If the number does not have a valid state yet. |   // If the number does not have a valid state yet. | ||||||
|   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller |   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller | ||||||
|   bool missing_state = 3; |   bool missing_state = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
| message NumberCommandRequest { | message NumberCommandRequest { | ||||||
|   option (id) = 51; |   option (id) = 51; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_NUMBER"; |   option (ifdef) = "USE_NUMBER"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   float state = 2; |   float state = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== SELECT ==================== | // ==================== SELECT ==================== | ||||||
| @@ -1039,6 +1093,7 @@ message ListEntitiesSelectResponse { | |||||||
|   repeated string options = 6; |   repeated string options = 6; | ||||||
|   bool disabled_by_default = 7; |   bool disabled_by_default = 7; | ||||||
|   EntityCategory entity_category = 8; |   EntityCategory entity_category = 8; | ||||||
|  |   uint32 device_id = 9; | ||||||
| } | } | ||||||
| message SelectStateResponse { | message SelectStateResponse { | ||||||
|   option (id) = 53; |   option (id) = 53; | ||||||
| @@ -1052,15 +1107,18 @@ message SelectStateResponse { | |||||||
|   // If the select does not have a valid state yet. |   // If the select does not have a valid state yet. | ||||||
|   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller |   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller | ||||||
|   bool missing_state = 3; |   bool missing_state = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
| message SelectCommandRequest { | message SelectCommandRequest { | ||||||
|   option (id) = 54; |   option (id) = 54; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_SELECT"; |   option (ifdef) = "USE_SELECT"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   string state = 2; |   string state = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== SIREN ==================== | // ==================== SIREN ==================== | ||||||
| @@ -1081,6 +1139,7 @@ message ListEntitiesSirenResponse { | |||||||
|   bool supports_duration = 8; |   bool supports_duration = 8; | ||||||
|   bool supports_volume = 9; |   bool supports_volume = 9; | ||||||
|   EntityCategory entity_category = 10; |   EntityCategory entity_category = 10; | ||||||
|  |   uint32 device_id = 11; | ||||||
| } | } | ||||||
| message SirenStateResponse { | message SirenStateResponse { | ||||||
|   option (id) = 56; |   option (id) = 56; | ||||||
| @@ -1091,12 +1150,14 @@ message SirenStateResponse { | |||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bool state = 2; |   bool state = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
| message SirenCommandRequest { | message SirenCommandRequest { | ||||||
|   option (id) = 57; |   option (id) = 57; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_SIREN"; |   option (ifdef) = "USE_SIREN"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bool has_state = 2; |   bool has_state = 2; | ||||||
| @@ -1107,6 +1168,7 @@ message SirenCommandRequest { | |||||||
|   uint32 duration = 7; |   uint32 duration = 7; | ||||||
|   bool has_volume = 8; |   bool has_volume = 8; | ||||||
|   float volume = 9; |   float volume = 9; | ||||||
|  |   uint32 device_id = 10; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== LOCK ==================== | // ==================== LOCK ==================== | ||||||
| @@ -1144,6 +1206,7 @@ message ListEntitiesLockResponse { | |||||||
|  |  | ||||||
|   // Not yet implemented: |   // Not yet implemented: | ||||||
|   string code_format = 11; |   string code_format = 11; | ||||||
|  |   uint32 device_id = 12; | ||||||
| } | } | ||||||
| message LockStateResponse { | message LockStateResponse { | ||||||
|   option (id) = 59; |   option (id) = 59; | ||||||
| @@ -1153,18 +1216,21 @@ message LockStateResponse { | |||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   LockState state = 2; |   LockState state = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
| message LockCommandRequest { | message LockCommandRequest { | ||||||
|   option (id) = 60; |   option (id) = 60; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_LOCK"; |   option (ifdef) = "USE_LOCK"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   LockCommand command = 2; |   LockCommand command = 2; | ||||||
|  |  | ||||||
|   // Not yet implemented: |   // Not yet implemented: | ||||||
|   bool has_code = 3; |   bool has_code = 3; | ||||||
|   string code = 4; |   string code = 4; | ||||||
|  |   uint32 device_id = 5; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== BUTTON ==================== | // ==================== BUTTON ==================== | ||||||
| @@ -1183,14 +1249,17 @@ message ListEntitiesButtonResponse { | |||||||
|   bool disabled_by_default = 6; |   bool disabled_by_default = 6; | ||||||
|   EntityCategory entity_category = 7; |   EntityCategory entity_category = 7; | ||||||
|   string device_class = 8; |   string device_class = 8; | ||||||
|  |   uint32 device_id = 9; | ||||||
| } | } | ||||||
| message ButtonCommandRequest { | message ButtonCommandRequest { | ||||||
|   option (id) = 62; |   option (id) = 62; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_BUTTON"; |   option (ifdef) = "USE_BUTTON"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|  |   uint32 device_id = 2; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== MEDIA PLAYER ==================== | // ==================== MEDIA PLAYER ==================== | ||||||
| @@ -1238,6 +1307,8 @@ message ListEntitiesMediaPlayerResponse { | |||||||
|   bool supports_pause = 8; |   bool supports_pause = 8; | ||||||
|  |  | ||||||
|   repeated MediaPlayerSupportedFormat supported_formats = 9; |   repeated MediaPlayerSupportedFormat supported_formats = 9; | ||||||
|  |  | ||||||
|  |   uint32 device_id = 10; | ||||||
| } | } | ||||||
| message MediaPlayerStateResponse { | message MediaPlayerStateResponse { | ||||||
|   option (id) = 64; |   option (id) = 64; | ||||||
| @@ -1249,12 +1320,14 @@ message MediaPlayerStateResponse { | |||||||
|   MediaPlayerState state = 2; |   MediaPlayerState state = 2; | ||||||
|   float volume = 3; |   float volume = 3; | ||||||
|   bool muted = 4; |   bool muted = 4; | ||||||
|  |   uint32 device_id = 5; | ||||||
| } | } | ||||||
| message MediaPlayerCommandRequest { | message MediaPlayerCommandRequest { | ||||||
|   option (id) = 65; |   option (id) = 65; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_MEDIA_PLAYER"; |   option (ifdef) = "USE_MEDIA_PLAYER"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|  |  | ||||||
| @@ -1269,6 +1342,7 @@ message MediaPlayerCommandRequest { | |||||||
|  |  | ||||||
|   bool has_announcement = 8; |   bool has_announcement = 8; | ||||||
|   bool announcement = 9; |   bool announcement = 9; | ||||||
|  |   uint32 device_id = 10; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== BLUETOOTH ==================== | // ==================== BLUETOOTH ==================== | ||||||
| @@ -1778,6 +1852,7 @@ message ListEntitiesAlarmControlPanelResponse { | |||||||
|   uint32 supported_features = 8; |   uint32 supported_features = 8; | ||||||
|   bool requires_code = 9; |   bool requires_code = 9; | ||||||
|   bool requires_code_to_arm = 10; |   bool requires_code_to_arm = 10; | ||||||
|  |   uint32 device_id = 11; | ||||||
| } | } | ||||||
|  |  | ||||||
| message AlarmControlPanelStateResponse { | message AlarmControlPanelStateResponse { | ||||||
| @@ -1788,6 +1863,7 @@ message AlarmControlPanelStateResponse { | |||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   AlarmControlPanelState state = 2; |   AlarmControlPanelState state = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
|  |  | ||||||
| message AlarmControlPanelCommandRequest { | message AlarmControlPanelCommandRequest { | ||||||
| @@ -1795,9 +1871,11 @@ message AlarmControlPanelCommandRequest { | |||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_ALARM_CONTROL_PANEL"; |   option (ifdef) = "USE_ALARM_CONTROL_PANEL"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   AlarmControlPanelStateCommand command = 2; |   AlarmControlPanelStateCommand command = 2; | ||||||
|   string code = 3; |   string code = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ===================== TEXT ===================== | // ===================== TEXT ===================== | ||||||
| @@ -1823,6 +1901,7 @@ message ListEntitiesTextResponse { | |||||||
|   uint32 max_length = 9; |   uint32 max_length = 9; | ||||||
|   string pattern = 10; |   string pattern = 10; | ||||||
|   TextMode mode = 11; |   TextMode mode = 11; | ||||||
|  |   uint32 device_id = 12; | ||||||
| } | } | ||||||
| message TextStateResponse { | message TextStateResponse { | ||||||
|   option (id) = 98; |   option (id) = 98; | ||||||
| @@ -1836,15 +1915,18 @@ message TextStateResponse { | |||||||
|   // If the Text does not have a valid state yet. |   // If the Text does not have a valid state yet. | ||||||
|   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller |   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller | ||||||
|   bool missing_state = 3; |   bool missing_state = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
| message TextCommandRequest { | message TextCommandRequest { | ||||||
|   option (id) = 99; |   option (id) = 99; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_TEXT"; |   option (ifdef) = "USE_TEXT"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   string state = 2; |   string state = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
|  |  | ||||||
|  |  | ||||||
| @@ -1863,6 +1945,7 @@ message ListEntitiesDateResponse { | |||||||
|   string icon = 5; |   string icon = 5; | ||||||
|   bool disabled_by_default = 6; |   bool disabled_by_default = 6; | ||||||
|   EntityCategory entity_category = 7; |   EntityCategory entity_category = 7; | ||||||
|  |   uint32 device_id = 8; | ||||||
| } | } | ||||||
| message DateStateResponse { | message DateStateResponse { | ||||||
|   option (id) = 101; |   option (id) = 101; | ||||||
| @@ -1878,17 +1961,20 @@ message DateStateResponse { | |||||||
|   uint32 year = 3; |   uint32 year = 3; | ||||||
|   uint32 month = 4; |   uint32 month = 4; | ||||||
|   uint32 day = 5; |   uint32 day = 5; | ||||||
|  |   uint32 device_id = 6; | ||||||
| } | } | ||||||
| message DateCommandRequest { | message DateCommandRequest { | ||||||
|   option (id) = 102; |   option (id) = 102; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_DATETIME_DATE"; |   option (ifdef) = "USE_DATETIME_DATE"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   uint32 year = 2; |   uint32 year = 2; | ||||||
|   uint32 month = 3; |   uint32 month = 3; | ||||||
|   uint32 day = 4; |   uint32 day = 4; | ||||||
|  |   uint32 device_id = 5; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== DATETIME TIME ==================== | // ==================== DATETIME TIME ==================== | ||||||
| @@ -1906,6 +1992,7 @@ message ListEntitiesTimeResponse { | |||||||
|   string icon = 5; |   string icon = 5; | ||||||
|   bool disabled_by_default = 6; |   bool disabled_by_default = 6; | ||||||
|   EntityCategory entity_category = 7; |   EntityCategory entity_category = 7; | ||||||
|  |   uint32 device_id = 8; | ||||||
| } | } | ||||||
| message TimeStateResponse { | message TimeStateResponse { | ||||||
|   option (id) = 104; |   option (id) = 104; | ||||||
| @@ -1921,17 +2008,20 @@ message TimeStateResponse { | |||||||
|   uint32 hour = 3; |   uint32 hour = 3; | ||||||
|   uint32 minute = 4; |   uint32 minute = 4; | ||||||
|   uint32 second = 5; |   uint32 second = 5; | ||||||
|  |   uint32 device_id = 6; | ||||||
| } | } | ||||||
| message TimeCommandRequest { | message TimeCommandRequest { | ||||||
|   option (id) = 105; |   option (id) = 105; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_DATETIME_TIME"; |   option (ifdef) = "USE_DATETIME_TIME"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   uint32 hour = 2; |   uint32 hour = 2; | ||||||
|   uint32 minute = 3; |   uint32 minute = 3; | ||||||
|   uint32 second = 4; |   uint32 second = 4; | ||||||
|  |   uint32 device_id = 5; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== EVENT ==================== | // ==================== EVENT ==================== | ||||||
| @@ -1952,6 +2042,7 @@ message ListEntitiesEventResponse { | |||||||
|   string device_class = 8; |   string device_class = 8; | ||||||
|  |  | ||||||
|   repeated string event_types = 9; |   repeated string event_types = 9; | ||||||
|  |   uint32 device_id = 10; | ||||||
| } | } | ||||||
| message EventResponse { | message EventResponse { | ||||||
|   option (id) = 108; |   option (id) = 108; | ||||||
| @@ -1961,6 +2052,7 @@ message EventResponse { | |||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   string event_type = 2; |   string event_type = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== VALVE ==================== | // ==================== VALVE ==================== | ||||||
| @@ -1983,6 +2075,7 @@ message ListEntitiesValveResponse { | |||||||
|   bool assumed_state = 9; |   bool assumed_state = 9; | ||||||
|   bool supports_position = 10; |   bool supports_position = 10; | ||||||
|   bool supports_stop = 11; |   bool supports_stop = 11; | ||||||
|  |   uint32 device_id = 12; | ||||||
| } | } | ||||||
|  |  | ||||||
| enum ValveOperation { | enum ValveOperation { | ||||||
| @@ -2000,6 +2093,7 @@ message ValveStateResponse { | |||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   float position = 2; |   float position = 2; | ||||||
|   ValveOperation current_operation = 3; |   ValveOperation current_operation = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
|  |  | ||||||
| message ValveCommandRequest { | message ValveCommandRequest { | ||||||
| @@ -2007,11 +2101,13 @@ message ValveCommandRequest { | |||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_VALVE"; |   option (ifdef) = "USE_VALVE"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   bool has_position = 2; |   bool has_position = 2; | ||||||
|   float position = 3; |   float position = 3; | ||||||
|   bool stop = 4; |   bool stop = 4; | ||||||
|  |   uint32 device_id = 5; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== DATETIME DATETIME ==================== | // ==================== DATETIME DATETIME ==================== | ||||||
| @@ -2029,6 +2125,7 @@ message ListEntitiesDateTimeResponse { | |||||||
|   string icon = 5; |   string icon = 5; | ||||||
|   bool disabled_by_default = 6; |   bool disabled_by_default = 6; | ||||||
|   EntityCategory entity_category = 7; |   EntityCategory entity_category = 7; | ||||||
|  |   uint32 device_id = 8; | ||||||
| } | } | ||||||
| message DateTimeStateResponse { | message DateTimeStateResponse { | ||||||
|   option (id) = 113; |   option (id) = 113; | ||||||
| @@ -2042,15 +2139,18 @@ message DateTimeStateResponse { | |||||||
|   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller |   // Equivalent to `!obj->has_state()` - inverse logic to make state packets smaller | ||||||
|   bool missing_state = 2; |   bool missing_state = 2; | ||||||
|   fixed32 epoch_seconds = 3; |   fixed32 epoch_seconds = 3; | ||||||
|  |   uint32 device_id = 4; | ||||||
| } | } | ||||||
| message DateTimeCommandRequest { | message DateTimeCommandRequest { | ||||||
|   option (id) = 114; |   option (id) = 114; | ||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_DATETIME_DATETIME"; |   option (ifdef) = "USE_DATETIME_DATETIME"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   fixed32 epoch_seconds = 2; |   fixed32 epoch_seconds = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
|  |  | ||||||
| // ==================== UPDATE ==================== | // ==================== UPDATE ==================== | ||||||
| @@ -2069,6 +2169,7 @@ message ListEntitiesUpdateResponse { | |||||||
|   bool disabled_by_default = 6; |   bool disabled_by_default = 6; | ||||||
|   EntityCategory entity_category = 7; |   EntityCategory entity_category = 7; | ||||||
|   string device_class = 8; |   string device_class = 8; | ||||||
|  |   uint32 device_id = 9; | ||||||
| } | } | ||||||
| message UpdateStateResponse { | message UpdateStateResponse { | ||||||
|   option (id) = 117; |   option (id) = 117; | ||||||
| @@ -2087,6 +2188,7 @@ message UpdateStateResponse { | |||||||
|   string title = 8; |   string title = 8; | ||||||
|   string release_summary = 9; |   string release_summary = 9; | ||||||
|   string release_url = 10; |   string release_url = 10; | ||||||
|  |   uint32 device_id = 11; | ||||||
| } | } | ||||||
| enum UpdateCommand { | enum UpdateCommand { | ||||||
|   UPDATE_COMMAND_NONE = 0; |   UPDATE_COMMAND_NONE = 0; | ||||||
| @@ -2098,7 +2200,9 @@ message UpdateCommandRequest { | |||||||
|   option (source) = SOURCE_CLIENT; |   option (source) = SOURCE_CLIENT; | ||||||
|   option (ifdef) = "USE_UPDATE"; |   option (ifdef) = "USE_UPDATE"; | ||||||
|   option (no_delay) = true; |   option (no_delay) = true; | ||||||
|  |   option (base_class) = "CommandProtoMessage"; | ||||||
|  |  | ||||||
|   fixed32 key = 1; |   fixed32 key = 1; | ||||||
|   UpdateCommand command = 2; |   UpdateCommand command = 2; | ||||||
|  |   uint32 device_id = 3; | ||||||
| } | } | ||||||
|   | |||||||
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							| @@ -18,10 +18,13 @@ namespace api { | |||||||
|  |  | ||||||
| // Keepalive timeout in milliseconds | // Keepalive timeout in milliseconds | ||||||
| static constexpr uint32_t KEEPALIVE_TIMEOUT_MS = 60000; | static constexpr uint32_t KEEPALIVE_TIMEOUT_MS = 60000; | ||||||
|  | // Maximum number of entities to process in a single batch during initial state/info sending | ||||||
|  | static constexpr size_t MAX_INITIAL_PER_BATCH = 20; | ||||||
|  |  | ||||||
| class APIConnection : public APIServerConnection { | class APIConnection : public APIServerConnection { | ||||||
|  public: |  public: | ||||||
|   friend class APIServer; |   friend class APIServer; | ||||||
|  |   friend class ListEntitiesIterator; | ||||||
|   APIConnection(std::unique_ptr<socket::Socket> socket, APIServer *parent); |   APIConnection(std::unique_ptr<socket::Socket> socket, APIServer *parent); | ||||||
|   virtual ~APIConnection(); |   virtual ~APIConnection(); | ||||||
|  |  | ||||||
| @@ -30,102 +33,83 @@ class APIConnection : public APIServerConnection { | |||||||
|  |  | ||||||
|   bool send_list_info_done() { |   bool send_list_info_done() { | ||||||
|     return this->schedule_message_(nullptr, &APIConnection::try_send_list_info_done, |     return this->schedule_message_(nullptr, &APIConnection::try_send_list_info_done, | ||||||
|                                    ListEntitiesDoneResponse::MESSAGE_TYPE); |                                    ListEntitiesDoneResponse::MESSAGE_TYPE, ListEntitiesDoneResponse::ESTIMATED_SIZE); | ||||||
|   } |   } | ||||||
| #ifdef USE_BINARY_SENSOR | #ifdef USE_BINARY_SENSOR | ||||||
|   bool send_binary_sensor_state(binary_sensor::BinarySensor *binary_sensor); |   bool send_binary_sensor_state(binary_sensor::BinarySensor *binary_sensor); | ||||||
|   void send_binary_sensor_info(binary_sensor::BinarySensor *binary_sensor); |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_COVER | #ifdef USE_COVER | ||||||
|   bool send_cover_state(cover::Cover *cover); |   bool send_cover_state(cover::Cover *cover); | ||||||
|   void send_cover_info(cover::Cover *cover); |  | ||||||
|   void cover_command(const CoverCommandRequest &msg) override; |   void cover_command(const CoverCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_FAN | #ifdef USE_FAN | ||||||
|   bool send_fan_state(fan::Fan *fan); |   bool send_fan_state(fan::Fan *fan); | ||||||
|   void send_fan_info(fan::Fan *fan); |  | ||||||
|   void fan_command(const FanCommandRequest &msg) override; |   void fan_command(const FanCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LIGHT | #ifdef USE_LIGHT | ||||||
|   bool send_light_state(light::LightState *light); |   bool send_light_state(light::LightState *light); | ||||||
|   void send_light_info(light::LightState *light); |  | ||||||
|   void light_command(const LightCommandRequest &msg) override; |   void light_command(const LightCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SENSOR | #ifdef USE_SENSOR | ||||||
|   bool send_sensor_state(sensor::Sensor *sensor); |   bool send_sensor_state(sensor::Sensor *sensor); | ||||||
|   void send_sensor_info(sensor::Sensor *sensor); |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SWITCH | #ifdef USE_SWITCH | ||||||
|   bool send_switch_state(switch_::Switch *a_switch); |   bool send_switch_state(switch_::Switch *a_switch); | ||||||
|   void send_switch_info(switch_::Switch *a_switch); |  | ||||||
|   void switch_command(const SwitchCommandRequest &msg) override; |   void switch_command(const SwitchCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT_SENSOR | #ifdef USE_TEXT_SENSOR | ||||||
|   bool send_text_sensor_state(text_sensor::TextSensor *text_sensor); |   bool send_text_sensor_state(text_sensor::TextSensor *text_sensor); | ||||||
|   void send_text_sensor_info(text_sensor::TextSensor *text_sensor); |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_ESP32_CAMERA | #ifdef USE_CAMERA | ||||||
|   void set_camera_state(std::shared_ptr<esp32_camera::CameraImage> image); |   void set_camera_state(std::shared_ptr<camera::CameraImage> image); | ||||||
|   void send_camera_info(esp32_camera::ESP32Camera *camera); |  | ||||||
|   void camera_image(const CameraImageRequest &msg) override; |   void camera_image(const CameraImageRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_CLIMATE | #ifdef USE_CLIMATE | ||||||
|   bool send_climate_state(climate::Climate *climate); |   bool send_climate_state(climate::Climate *climate); | ||||||
|   void send_climate_info(climate::Climate *climate); |  | ||||||
|   void climate_command(const ClimateCommandRequest &msg) override; |   void climate_command(const ClimateCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_NUMBER | #ifdef USE_NUMBER | ||||||
|   bool send_number_state(number::Number *number); |   bool send_number_state(number::Number *number); | ||||||
|   void send_number_info(number::Number *number); |  | ||||||
|   void number_command(const NumberCommandRequest &msg) override; |   void number_command(const NumberCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_DATE | #ifdef USE_DATETIME_DATE | ||||||
|   bool send_date_state(datetime::DateEntity *date); |   bool send_date_state(datetime::DateEntity *date); | ||||||
|   void send_date_info(datetime::DateEntity *date); |  | ||||||
|   void date_command(const DateCommandRequest &msg) override; |   void date_command(const DateCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_TIME | #ifdef USE_DATETIME_TIME | ||||||
|   bool send_time_state(datetime::TimeEntity *time); |   bool send_time_state(datetime::TimeEntity *time); | ||||||
|   void send_time_info(datetime::TimeEntity *time); |  | ||||||
|   void time_command(const TimeCommandRequest &msg) override; |   void time_command(const TimeCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_DATETIME | #ifdef USE_DATETIME_DATETIME | ||||||
|   bool send_datetime_state(datetime::DateTimeEntity *datetime); |   bool send_datetime_state(datetime::DateTimeEntity *datetime); | ||||||
|   void send_datetime_info(datetime::DateTimeEntity *datetime); |  | ||||||
|   void datetime_command(const DateTimeCommandRequest &msg) override; |   void datetime_command(const DateTimeCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT | #ifdef USE_TEXT | ||||||
|   bool send_text_state(text::Text *text); |   bool send_text_state(text::Text *text); | ||||||
|   void send_text_info(text::Text *text); |  | ||||||
|   void text_command(const TextCommandRequest &msg) override; |   void text_command(const TextCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SELECT | #ifdef USE_SELECT | ||||||
|   bool send_select_state(select::Select *select); |   bool send_select_state(select::Select *select); | ||||||
|   void send_select_info(select::Select *select); |  | ||||||
|   void select_command(const SelectCommandRequest &msg) override; |   void select_command(const SelectCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_BUTTON | #ifdef USE_BUTTON | ||||||
|   void send_button_info(button::Button *button); |  | ||||||
|   void button_command(const ButtonCommandRequest &msg) override; |   void button_command(const ButtonCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LOCK | #ifdef USE_LOCK | ||||||
|   bool send_lock_state(lock::Lock *a_lock); |   bool send_lock_state(lock::Lock *a_lock); | ||||||
|   void send_lock_info(lock::Lock *a_lock); |  | ||||||
|   void lock_command(const LockCommandRequest &msg) override; |   void lock_command(const LockCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_VALVE | #ifdef USE_VALVE | ||||||
|   bool send_valve_state(valve::Valve *valve); |   bool send_valve_state(valve::Valve *valve); | ||||||
|   void send_valve_info(valve::Valve *valve); |  | ||||||
|   void valve_command(const ValveCommandRequest &msg) override; |   void valve_command(const ValveCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_MEDIA_PLAYER | #ifdef USE_MEDIA_PLAYER | ||||||
|   bool send_media_player_state(media_player::MediaPlayer *media_player); |   bool send_media_player_state(media_player::MediaPlayer *media_player); | ||||||
|   void send_media_player_info(media_player::MediaPlayer *media_player); |  | ||||||
|   void media_player_command(const MediaPlayerCommandRequest &msg) override; |   void media_player_command(const MediaPlayerCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
|   bool try_send_log_message(int level, const char *tag, const char *line); |   bool try_send_log_message(int level, const char *tag, const char *line, size_t message_len); | ||||||
|   void send_homeassistant_service_call(const HomeassistantServiceResponse &call) { |   void send_homeassistant_service_call(const HomeassistantServiceResponse &call) { | ||||||
|     if (!this->service_call_subscription_) |     if (!this->flags_.service_call_subscription) | ||||||
|       return; |       return; | ||||||
|     this->send_message(call); |     this->send_message(call); | ||||||
|   } |   } | ||||||
| @@ -167,26 +151,22 @@ class APIConnection : public APIServerConnection { | |||||||
|  |  | ||||||
| #ifdef USE_ALARM_CONTROL_PANEL | #ifdef USE_ALARM_CONTROL_PANEL | ||||||
|   bool send_alarm_control_panel_state(alarm_control_panel::AlarmControlPanel *a_alarm_control_panel); |   bool send_alarm_control_panel_state(alarm_control_panel::AlarmControlPanel *a_alarm_control_panel); | ||||||
|   void send_alarm_control_panel_info(alarm_control_panel::AlarmControlPanel *a_alarm_control_panel); |  | ||||||
|   void alarm_control_panel_command(const AlarmControlPanelCommandRequest &msg) override; |   void alarm_control_panel_command(const AlarmControlPanelCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_EVENT | #ifdef USE_EVENT | ||||||
|   void send_event(event::Event *event, const std::string &event_type); |   void send_event(event::Event *event, const std::string &event_type); | ||||||
|   void send_event_info(event::Event *event); |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_UPDATE | #ifdef USE_UPDATE | ||||||
|   bool send_update_state(update::UpdateEntity *update); |   bool send_update_state(update::UpdateEntity *update); | ||||||
|   void send_update_info(update::UpdateEntity *update); |  | ||||||
|   void update_command(const UpdateCommandRequest &msg) override; |   void update_command(const UpdateCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|   void on_disconnect_response(const DisconnectResponse &value) override; |   void on_disconnect_response(const DisconnectResponse &value) override; | ||||||
|   void on_ping_response(const PingResponse &value) override { |   void on_ping_response(const PingResponse &value) override { | ||||||
|     // we initiated ping |     // we initiated ping | ||||||
|     this->ping_retries_ = 0; |     this->flags_.sent_ping = false; | ||||||
|     this->sent_ping_ = false; |  | ||||||
|   } |   } | ||||||
|   void on_home_assistant_state_response(const HomeAssistantStateResponse &msg) override; |   void on_home_assistant_state_response(const HomeAssistantStateResponse &msg) override; | ||||||
| #ifdef USE_HOMEASSISTANT_TIME | #ifdef USE_HOMEASSISTANT_TIME | ||||||
| @@ -199,30 +179,35 @@ class APIConnection : public APIServerConnection { | |||||||
|   DeviceInfoResponse device_info(const DeviceInfoRequest &msg) override; |   DeviceInfoResponse device_info(const DeviceInfoRequest &msg) override; | ||||||
|   void list_entities(const ListEntitiesRequest &msg) override { this->list_entities_iterator_.begin(); } |   void list_entities(const ListEntitiesRequest &msg) override { this->list_entities_iterator_.begin(); } | ||||||
|   void subscribe_states(const SubscribeStatesRequest &msg) override { |   void subscribe_states(const SubscribeStatesRequest &msg) override { | ||||||
|     this->state_subscription_ = true; |     this->flags_.state_subscription = true; | ||||||
|     this->initial_state_iterator_.begin(); |     this->initial_state_iterator_.begin(); | ||||||
|   } |   } | ||||||
|   void subscribe_logs(const SubscribeLogsRequest &msg) override { |   void subscribe_logs(const SubscribeLogsRequest &msg) override { | ||||||
|     this->log_subscription_ = msg.level; |     this->flags_.log_subscription = msg.level; | ||||||
|     if (msg.dump_config) |     if (msg.dump_config) | ||||||
|       App.schedule_dump_config(); |       App.schedule_dump_config(); | ||||||
|   } |   } | ||||||
|   void subscribe_homeassistant_services(const SubscribeHomeassistantServicesRequest &msg) override { |   void subscribe_homeassistant_services(const SubscribeHomeassistantServicesRequest &msg) override { | ||||||
|     this->service_call_subscription_ = true; |     this->flags_.service_call_subscription = true; | ||||||
|   } |   } | ||||||
|   void subscribe_home_assistant_states(const SubscribeHomeAssistantStatesRequest &msg) override; |   void subscribe_home_assistant_states(const SubscribeHomeAssistantStatesRequest &msg) override; | ||||||
|   GetTimeResponse get_time(const GetTimeRequest &msg) override { |   GetTimeResponse get_time(const GetTimeRequest &msg) override { | ||||||
|     // TODO |     // TODO | ||||||
|     return {}; |     return {}; | ||||||
|   } |   } | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   void execute_service(const ExecuteServiceRequest &msg) override; |   void execute_service(const ExecuteServiceRequest &msg) override; | ||||||
|  | #endif | ||||||
| #ifdef USE_API_NOISE | #ifdef USE_API_NOISE | ||||||
|   NoiseEncryptionSetKeyResponse noise_encryption_set_key(const NoiseEncryptionSetKeyRequest &msg) override; |   NoiseEncryptionSetKeyResponse noise_encryption_set_key(const NoiseEncryptionSetKeyRequest &msg) override; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|   bool is_authenticated() override { return this->connection_state_ == ConnectionState::AUTHENTICATED; } |   bool is_authenticated() override { | ||||||
|  |     return static_cast<ConnectionState>(this->flags_.connection_state) == ConnectionState::AUTHENTICATED; | ||||||
|  |   } | ||||||
|   bool is_connection_setup() override { |   bool is_connection_setup() override { | ||||||
|     return this->connection_state_ == ConnectionState ::CONNECTED || this->is_authenticated(); |     return static_cast<ConnectionState>(this->flags_.connection_state) == ConnectionState::CONNECTED || | ||||||
|  |            this->is_authenticated(); | ||||||
|   } |   } | ||||||
|   void on_fatal_error() override; |   void on_fatal_error() override; | ||||||
|   void on_unauthenticated_access() override; |   void on_unauthenticated_access() override; | ||||||
| @@ -273,9 +258,15 @@ class APIConnection : public APIServerConnection { | |||||||
|   } |   } | ||||||
|  |  | ||||||
|   bool try_to_clear_buffer(bool log_out_of_space); |   bool try_to_clear_buffer(bool log_out_of_space); | ||||||
|   bool send_buffer(ProtoWriteBuffer buffer, uint16_t message_type) override; |   bool send_buffer(ProtoWriteBuffer buffer, uint8_t message_type) override; | ||||||
|  |  | ||||||
|   std::string get_client_combined_info() const { return this->client_combined_info_; } |   std::string get_client_combined_info() const { | ||||||
|  |     if (this->client_info_ == this->client_peername_) { | ||||||
|  |       // Before Hello message, both are the same (just IP:port) | ||||||
|  |       return this->client_info_; | ||||||
|  |     } | ||||||
|  |     return this->client_info_ + " (" + this->client_peername_ + ")"; | ||||||
|  |   } | ||||||
|  |  | ||||||
|   // Buffer allocator methods for batch processing |   // Buffer allocator methods for batch processing | ||||||
|   ProtoWriteBuffer allocate_single_message_buffer(uint16_t size); |   ProtoWriteBuffer allocate_single_message_buffer(uint16_t size); | ||||||
| @@ -295,17 +286,42 @@ class APIConnection : public APIServerConnection { | |||||||
|     response.icon = entity->get_icon(); |     response.icon = entity->get_icon(); | ||||||
|     response.disabled_by_default = entity->is_disabled_by_default(); |     response.disabled_by_default = entity->is_disabled_by_default(); | ||||||
|     response.entity_category = static_cast<enums::EntityCategory>(entity->get_entity_category()); |     response.entity_category = static_cast<enums::EntityCategory>(entity->get_entity_category()); | ||||||
|  | #ifdef USE_DEVICES | ||||||
|  |     response.device_id = entity->get_device_id(); | ||||||
|  | #endif | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // Helper function to fill common entity state fields |   // Helper function to fill common entity state fields | ||||||
|   static void fill_entity_state_base(esphome::EntityBase *entity, StateResponseProtoMessage &response) { |   static void fill_entity_state_base(esphome::EntityBase *entity, StateResponseProtoMessage &response) { | ||||||
|     response.key = entity->get_object_id_hash(); |     response.key = entity->get_object_id_hash(); | ||||||
|  | #ifdef USE_DEVICES | ||||||
|  |     response.device_id = entity->get_device_id(); | ||||||
|  | #endif | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // Non-template helper to encode any ProtoMessage |   // Non-template helper to encode any ProtoMessage | ||||||
|   static uint16_t encode_message_to_buffer(ProtoMessage &msg, uint16_t message_type, APIConnection *conn, |   static uint16_t encode_message_to_buffer(ProtoMessage &msg, uint8_t message_type, APIConnection *conn, | ||||||
|                                            uint32_t remaining_size, bool is_single); |                                            uint32_t remaining_size, bool is_single); | ||||||
|  |  | ||||||
|  | #ifdef USE_VOICE_ASSISTANT | ||||||
|  |   // Helper to check voice assistant validity and connection ownership | ||||||
|  |   inline bool check_voice_assistant_api_connection_() const; | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |   // Helper method to process multiple entities from an iterator in a batch | ||||||
|  |   template<typename Iterator> void process_iterator_batch_(Iterator &iterator) { | ||||||
|  |     size_t initial_size = this->deferred_batch_.size(); | ||||||
|  |     while (!iterator.completed() && (this->deferred_batch_.size() - initial_size) < MAX_INITIAL_PER_BATCH) { | ||||||
|  |       iterator.advance(); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // If the batch is full, process it immediately | ||||||
|  |     // Note: iterator.advance() already calls schedule_batch_() via schedule_message_() | ||||||
|  |     if (this->deferred_batch_.size() >= MAX_INITIAL_PER_BATCH) { | ||||||
|  |       this->process_batch_(); | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  |  | ||||||
| #ifdef USE_BINARY_SENSOR | #ifdef USE_BINARY_SENSOR | ||||||
|   static uint16_t try_send_binary_sensor_state(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, |   static uint16_t try_send_binary_sensor_state(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, | ||||||
|                                                bool is_single); |                                                bool is_single); | ||||||
| @@ -416,7 +432,7 @@ class APIConnection : public APIServerConnection { | |||||||
|   static uint16_t try_send_update_info(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, |   static uint16_t try_send_update_info(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, | ||||||
|                                        bool is_single); |                                        bool is_single); | ||||||
| #endif | #endif | ||||||
| #ifdef USE_ESP32_CAMERA | #ifdef USE_CAMERA | ||||||
|   static uint16_t try_send_camera_info(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, |   static uint16_t try_send_camera_info(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, | ||||||
|                                        bool is_single); |                                        bool is_single); | ||||||
| #endif | #endif | ||||||
| @@ -429,127 +445,82 @@ class APIConnection : public APIServerConnection { | |||||||
|   static uint16_t try_send_disconnect_request(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, |   static uint16_t try_send_disconnect_request(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, | ||||||
|                                               bool is_single); |                                               bool is_single); | ||||||
|  |  | ||||||
|   // Helper function to get estimated message size for buffer pre-allocation |   // Batch message method for ping requests | ||||||
|   static uint16_t get_estimated_message_size(uint16_t message_type); |   static uint16_t try_send_ping_request(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, | ||||||
|  |                                         bool is_single); | ||||||
|  |  | ||||||
|   enum class ConnectionState { |   // === Optimal member ordering for 32-bit systems === | ||||||
|     WAITING_FOR_HELLO, |  | ||||||
|     CONNECTED, |  | ||||||
|     AUTHENTICATED, |  | ||||||
|   } connection_state_{ConnectionState::WAITING_FOR_HELLO}; |  | ||||||
|  |  | ||||||
|   bool remove_{false}; |  | ||||||
|  |  | ||||||
|  |   // Group 1: Pointers (4 bytes each on 32-bit) | ||||||
|   std::unique_ptr<APIFrameHelper> helper_; |   std::unique_ptr<APIFrameHelper> helper_; | ||||||
|  |  | ||||||
|   std::string client_info_; |  | ||||||
|   std::string client_peername_; |  | ||||||
|   std::string client_combined_info_; |  | ||||||
|   uint32_t client_api_version_major_{0}; |  | ||||||
|   uint32_t client_api_version_minor_{0}; |  | ||||||
| #ifdef USE_ESP32_CAMERA |  | ||||||
|   esp32_camera::CameraImageReader image_reader_; |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
|   bool state_subscription_{false}; |  | ||||||
|   int log_subscription_{ESPHOME_LOG_LEVEL_NONE}; |  | ||||||
|   uint32_t last_traffic_; |  | ||||||
|   uint32_t next_ping_retry_{0}; |  | ||||||
|   uint8_t ping_retries_{0}; |  | ||||||
|   bool sent_ping_{false}; |  | ||||||
|   bool service_call_subscription_{false}; |  | ||||||
|   bool next_close_ = false; |  | ||||||
|   APIServer *parent_; |   APIServer *parent_; | ||||||
|  |  | ||||||
|  |   // Group 2: Larger objects (must be 4-byte aligned) | ||||||
|  |   // These contain vectors/pointers internally, so putting them early ensures good alignment | ||||||
|   InitialStateIterator initial_state_iterator_; |   InitialStateIterator initial_state_iterator_; | ||||||
|   ListEntitiesIterator list_entities_iterator_; |   ListEntitiesIterator list_entities_iterator_; | ||||||
|  | #ifdef USE_CAMERA | ||||||
|  |   std::unique_ptr<camera::CameraImageReader> image_reader_; | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |   // Group 3: Strings (12 bytes each on 32-bit, 4-byte aligned) | ||||||
|  |   std::string client_info_; | ||||||
|  |   std::string client_peername_; | ||||||
|  |  | ||||||
|  |   // Group 4: 4-byte types | ||||||
|  |   uint32_t last_traffic_; | ||||||
|   int state_subs_at_ = -1; |   int state_subs_at_ = -1; | ||||||
|  |  | ||||||
|   // Function pointer type for message encoding |   // Function pointer type for message encoding | ||||||
|   using MessageCreatorPtr = uint16_t (*)(EntityBase *, APIConnection *, uint32_t remaining_size, bool is_single); |   using MessageCreatorPtr = uint16_t (*)(EntityBase *, APIConnection *, uint32_t remaining_size, bool is_single); | ||||||
|  |  | ||||||
|   // Optimized MessageCreator class using union dispatch |  | ||||||
|   class MessageCreator { |   class MessageCreator { | ||||||
|    public: |    public: | ||||||
|     // Constructor for function pointer (message_type = 0) |     // Constructor for function pointer | ||||||
|     MessageCreator(MessageCreatorPtr ptr) : message_type_(0) { data_.ptr = ptr; } |     MessageCreator(MessageCreatorPtr ptr) { data_.function_ptr = ptr; } | ||||||
|  |  | ||||||
|     // Constructor for string state capture |     // Constructor for string state capture | ||||||
|     MessageCreator(const std::string &value, uint16_t msg_type) : message_type_(msg_type) { |     explicit MessageCreator(const std::string &str_value) { data_.string_ptr = new std::string(str_value); } | ||||||
|       data_.string_ptr = new std::string(value); |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Destructor |     // No destructor - cleanup must be called explicitly with message_type | ||||||
|     ~MessageCreator() { |  | ||||||
|       // Clean up string data for string-based message types |  | ||||||
|       if (uses_string_data_()) { |  | ||||||
|         delete data_.string_ptr; |  | ||||||
|       } |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Copy constructor |     // Delete copy operations - MessageCreator should only be moved | ||||||
|     MessageCreator(const MessageCreator &other) : message_type_(other.message_type_) { |     MessageCreator(const MessageCreator &other) = delete; | ||||||
|       if (message_type_ == 0) { |     MessageCreator &operator=(const MessageCreator &other) = delete; | ||||||
|         data_.ptr = other.data_.ptr; |  | ||||||
|       } else if (uses_string_data_()) { |  | ||||||
|         data_.string_ptr = new std::string(*other.data_.string_ptr); |  | ||||||
|       } else { |  | ||||||
|         data_ = other.data_;  // For POD types |  | ||||||
|       } |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Move constructor |     // Move constructor | ||||||
|     MessageCreator(MessageCreator &&other) noexcept : data_(other.data_), message_type_(other.message_type_) { |     MessageCreator(MessageCreator &&other) noexcept : data_(other.data_) { other.data_.function_ptr = nullptr; } | ||||||
|       other.message_type_ = 0;  // Reset other to function pointer type |  | ||||||
|       other.data_.ptr = nullptr; |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Assignment operators (needed for batch deduplication) |  | ||||||
|     MessageCreator &operator=(const MessageCreator &other) { |  | ||||||
|       if (this != &other) { |  | ||||||
|         // Clean up current string data if needed |  | ||||||
|         if (uses_string_data_()) { |  | ||||||
|           delete data_.string_ptr; |  | ||||||
|         } |  | ||||||
|         // Copy new data |  | ||||||
|         message_type_ = other.message_type_; |  | ||||||
|         if (other.message_type_ == 0) { |  | ||||||
|           data_.ptr = other.data_.ptr; |  | ||||||
|         } else if (other.uses_string_data_()) { |  | ||||||
|           data_.string_ptr = new std::string(*other.data_.string_ptr); |  | ||||||
|         } else { |  | ||||||
|           data_ = other.data_; |  | ||||||
|         } |  | ||||||
|       } |  | ||||||
|       return *this; |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|  |     // Move assignment | ||||||
|     MessageCreator &operator=(MessageCreator &&other) noexcept { |     MessageCreator &operator=(MessageCreator &&other) noexcept { | ||||||
|       if (this != &other) { |       if (this != &other) { | ||||||
|         // Clean up current string data if needed |         // IMPORTANT: Caller must ensure cleanup() was called if this contains a string! | ||||||
|         if (uses_string_data_()) { |         // In our usage, this happens in add_item() deduplication and vector::erase() | ||||||
|           delete data_.string_ptr; |  | ||||||
|         } |  | ||||||
|         // Move data |  | ||||||
|         message_type_ = other.message_type_; |  | ||||||
|         data_ = other.data_; |         data_ = other.data_; | ||||||
|         // Reset other to safe state |         other.data_.function_ptr = nullptr; | ||||||
|         other.message_type_ = 0; |  | ||||||
|         other.data_.ptr = nullptr; |  | ||||||
|       } |       } | ||||||
|       return *this; |       return *this; | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     // Call operator |     // Call operator - uses message_type to determine union type | ||||||
|     uint16_t operator()(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, bool is_single) const; |     uint16_t operator()(EntityBase *entity, APIConnection *conn, uint32_t remaining_size, bool is_single, | ||||||
|  |                         uint8_t message_type) const; | ||||||
|  |  | ||||||
|  |     // Manual cleanup method - must be called before destruction for string types | ||||||
|  |     void cleanup(uint8_t message_type) { | ||||||
|  | #ifdef USE_EVENT | ||||||
|  |       if (message_type == EventResponse::MESSAGE_TYPE && data_.string_ptr != nullptr) { | ||||||
|  |         delete data_.string_ptr; | ||||||
|  |         data_.string_ptr = nullptr; | ||||||
|  |       } | ||||||
|  | #endif | ||||||
|  |     } | ||||||
|  |  | ||||||
|    private: |    private: | ||||||
|     // Helper to check if this message type uses heap-allocated strings |     union Data { | ||||||
|     bool uses_string_data_() const { return message_type_ == EventResponse::MESSAGE_TYPE; } |       MessageCreatorPtr function_ptr; | ||||||
|     union CreatorData { |       std::string *string_ptr; | ||||||
|       MessageCreatorPtr ptr;    // 8 bytes |     } data_;  // 4 bytes on 32-bit, 8 bytes on 64-bit - same as before | ||||||
|       std::string *string_ptr;  // 8 bytes |  | ||||||
|     } data_;                    // 8 bytes |  | ||||||
|     uint16_t message_type_;     // 2 bytes (0 = function ptr, >0 = state capture) |  | ||||||
|   }; |   }; | ||||||
|  |  | ||||||
|   // Generic batching mechanism for both state updates and entity info |   // Generic batching mechanism for both state updates and entity info | ||||||
| @@ -557,33 +528,96 @@ class APIConnection : public APIServerConnection { | |||||||
|     struct BatchItem { |     struct BatchItem { | ||||||
|       EntityBase *entity;      // Entity pointer |       EntityBase *entity;      // Entity pointer | ||||||
|       MessageCreator creator;  // Function that creates the message when needed |       MessageCreator creator;  // Function that creates the message when needed | ||||||
|       uint16_t message_type;   // Message type for overhead calculation |       uint8_t message_type;    // Message type for overhead calculation (max 255) | ||||||
|  |       uint8_t estimated_size;  // Estimated message size (max 255 bytes) | ||||||
|  |  | ||||||
|       // Constructor for creating BatchItem |       // Constructor for creating BatchItem | ||||||
|       BatchItem(EntityBase *entity, MessageCreator creator, uint16_t message_type) |       BatchItem(EntityBase *entity, MessageCreator creator, uint8_t message_type, uint8_t estimated_size) | ||||||
|           : entity(entity), creator(std::move(creator)), message_type(message_type) {} |           : entity(entity), creator(std::move(creator)), message_type(message_type), estimated_size(estimated_size) {} | ||||||
|     }; |     }; | ||||||
|  |  | ||||||
|     std::vector<BatchItem> items; |     std::vector<BatchItem> items; | ||||||
|     uint32_t batch_start_time{0}; |     uint32_t batch_start_time{0}; | ||||||
|     bool batch_scheduled{false}; |  | ||||||
|  |  | ||||||
|  |    private: | ||||||
|  |     // Helper to cleanup items from the beginning | ||||||
|  |     void cleanup_items_(size_t count) { | ||||||
|  |       for (size_t i = 0; i < count; i++) { | ||||||
|  |         items[i].creator.cleanup(items[i].message_type); | ||||||
|  |       } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |    public: | ||||||
|     DeferredBatch() { |     DeferredBatch() { | ||||||
|       // Pre-allocate capacity for typical batch sizes to avoid reallocation |       // Pre-allocate capacity for typical batch sizes to avoid reallocation | ||||||
|       items.reserve(8); |       items.reserve(8); | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     ~DeferredBatch() { | ||||||
|  |       // Ensure cleanup of any remaining items | ||||||
|  |       clear(); | ||||||
|  |     } | ||||||
|  |  | ||||||
|     // Add item to the batch |     // Add item to the batch | ||||||
|     void add_item(EntityBase *entity, MessageCreator creator, uint16_t message_type); |     void add_item(EntityBase *entity, MessageCreator creator, uint8_t message_type, uint8_t estimated_size); | ||||||
|  |     // Add item to the front of the batch (for high priority messages like ping) | ||||||
|  |     void add_item_front(EntityBase *entity, MessageCreator creator, uint8_t message_type, uint8_t estimated_size); | ||||||
|  |  | ||||||
|  |     // Clear all items with proper cleanup | ||||||
|     void clear() { |     void clear() { | ||||||
|  |       cleanup_items_(items.size()); | ||||||
|       items.clear(); |       items.clear(); | ||||||
|       batch_scheduled = false; |  | ||||||
|       batch_start_time = 0; |       batch_start_time = 0; | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     // Remove processed items from the front with proper cleanup | ||||||
|  |     void remove_front(size_t count) { | ||||||
|  |       cleanup_items_(count); | ||||||
|  |       items.erase(items.begin(), items.begin() + count); | ||||||
|  |     } | ||||||
|  |  | ||||||
|     bool empty() const { return items.empty(); } |     bool empty() const { return items.empty(); } | ||||||
|  |     size_t size() const { return items.size(); } | ||||||
|  |     const BatchItem &operator[](size_t index) const { return items[index]; } | ||||||
|   }; |   }; | ||||||
|  |  | ||||||
|  |   // DeferredBatch here (16 bytes, 4-byte aligned) | ||||||
|   DeferredBatch deferred_batch_; |   DeferredBatch deferred_batch_; | ||||||
|  |  | ||||||
|  |   // ConnectionState enum for type safety | ||||||
|  |   enum class ConnectionState : uint8_t { | ||||||
|  |     WAITING_FOR_HELLO = 0, | ||||||
|  |     CONNECTED = 1, | ||||||
|  |     AUTHENTICATED = 2, | ||||||
|  |   }; | ||||||
|  |  | ||||||
|  |   // Group 5: Pack all small members together to minimize padding | ||||||
|  |   // This group starts at a 4-byte boundary after DeferredBatch | ||||||
|  |   struct APIFlags { | ||||||
|  |     // Connection state only needs 2 bits (3 states) | ||||||
|  |     uint8_t connection_state : 2; | ||||||
|  |     // Log subscription needs 3 bits (log levels 0-7) | ||||||
|  |     uint8_t log_subscription : 3; | ||||||
|  |     // Boolean flags (1 bit each) | ||||||
|  |     uint8_t remove : 1; | ||||||
|  |     uint8_t state_subscription : 1; | ||||||
|  |     uint8_t sent_ping : 1; | ||||||
|  |  | ||||||
|  |     uint8_t service_call_subscription : 1; | ||||||
|  |     uint8_t next_close : 1; | ||||||
|  |     uint8_t batch_scheduled : 1; | ||||||
|  |     uint8_t batch_first_message : 1;          // For batch buffer allocation | ||||||
|  |     uint8_t should_try_send_immediately : 1;  // True after initial states are sent | ||||||
|  | #ifdef HAS_PROTO_MESSAGE_DUMP | ||||||
|  |     uint8_t log_only_mode : 1; | ||||||
|  | #endif | ||||||
|  |   } flags_{};  // 2 bytes total | ||||||
|  |  | ||||||
|  |   // 2-byte types immediately after flags_ (no padding between them) | ||||||
|  |   uint16_t client_api_version_major_{0}; | ||||||
|  |   uint16_t client_api_version_minor_{0}; | ||||||
|  |   // Total: 2 (flags) + 2 + 2 = 6 bytes, then 2 bytes padding to next 4-byte boundary | ||||||
|  |  | ||||||
|   uint32_t get_batch_delay_ms_() const; |   uint32_t get_batch_delay_ms_() const; | ||||||
|   // Message will use 8 more bytes than the minimum size, and typical |   // Message will use 8 more bytes than the minimum size, and typical | ||||||
|   // MTU is 1500. Sometimes users will see as low as 1460 MTU. |   // MTU is 1500. Sometimes users will see as low as 1460 MTU. | ||||||
| @@ -596,23 +630,72 @@ class APIConnection : public APIServerConnection { | |||||||
|   // to send in one go. This is the maximum size of a single packet |   // to send in one go. This is the maximum size of a single packet | ||||||
|   // that can be sent over the network. |   // that can be sent over the network. | ||||||
|   // This is to avoid fragmentation of the packet. |   // This is to avoid fragmentation of the packet. | ||||||
|   static constexpr size_t MAX_PACKET_SIZE = 1390;  // MTU |   static constexpr size_t MAX_BATCH_PACKET_SIZE = 1390;  // MTU | ||||||
|  |  | ||||||
|   bool schedule_batch_(); |   bool schedule_batch_(); | ||||||
|   void process_batch_(); |   void process_batch_(); | ||||||
|  |   void clear_batch_() { | ||||||
|  |     this->deferred_batch_.clear(); | ||||||
|  |     this->flags_.batch_scheduled = false; | ||||||
|  |   } | ||||||
|  |  | ||||||
|   // State for batch buffer allocation | #ifdef HAS_PROTO_MESSAGE_DUMP | ||||||
|   bool batch_first_message_{false}; |   // Helper to log a proto message from a MessageCreator object | ||||||
|  |   void log_proto_message_(EntityBase *entity, const MessageCreator &creator, uint8_t message_type) { | ||||||
|  |     this->flags_.log_only_mode = true; | ||||||
|  |     creator(entity, this, MAX_BATCH_PACKET_SIZE, true, message_type); | ||||||
|  |     this->flags_.log_only_mode = false; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   void log_batch_item_(const DeferredBatch::BatchItem &item) { | ||||||
|  |     // Use the helper to log the message | ||||||
|  |     this->log_proto_message_(item.entity, item.creator, item.message_type); | ||||||
|  |   } | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |   // Helper method to send a message either immediately or via batching | ||||||
|  |   bool send_message_smart_(EntityBase *entity, MessageCreatorPtr creator, uint8_t message_type, | ||||||
|  |                            uint8_t estimated_size) { | ||||||
|  |     // Try to send immediately if: | ||||||
|  |     // 1. We should try to send immediately (should_try_send_immediately = true) | ||||||
|  |     // 2. Batch delay is 0 (user has opted in to immediate sending) | ||||||
|  |     // 3. Buffer has space available | ||||||
|  |     if (this->flags_.should_try_send_immediately && this->get_batch_delay_ms_() == 0 && | ||||||
|  |         this->helper_->can_write_without_blocking()) { | ||||||
|  |       // Now actually encode and send | ||||||
|  |       if (creator(entity, this, MAX_BATCH_PACKET_SIZE, true) && | ||||||
|  |           this->send_buffer(ProtoWriteBuffer{&this->parent_->get_shared_buffer_ref()}, message_type)) { | ||||||
|  | #ifdef HAS_PROTO_MESSAGE_DUMP | ||||||
|  |         // Log the message in verbose mode | ||||||
|  |         this->log_proto_message_(entity, MessageCreator(creator), message_type); | ||||||
|  | #endif | ||||||
|  |         return true; | ||||||
|  |       } | ||||||
|  |  | ||||||
|  |       // If immediate send failed, fall through to batching | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Fall back to scheduled batching | ||||||
|  |     return this->schedule_message_(entity, creator, message_type, estimated_size); | ||||||
|  |   } | ||||||
|  |  | ||||||
|   // Helper function to schedule a deferred message with known message type |   // Helper function to schedule a deferred message with known message type | ||||||
|   bool schedule_message_(EntityBase *entity, MessageCreator creator, uint16_t message_type) { |   bool schedule_message_(EntityBase *entity, MessageCreator creator, uint8_t message_type, uint8_t estimated_size) { | ||||||
|     this->deferred_batch_.add_item(entity, std::move(creator), message_type); |     this->deferred_batch_.add_item(entity, std::move(creator), message_type, estimated_size); | ||||||
|     return this->schedule_batch_(); |     return this->schedule_batch_(); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // Overload for function pointers (for info messages and current state reads) |   // Overload for function pointers (for info messages and current state reads) | ||||||
|   bool schedule_message_(EntityBase *entity, MessageCreatorPtr function_ptr, uint16_t message_type) { |   bool schedule_message_(EntityBase *entity, MessageCreatorPtr function_ptr, uint8_t message_type, | ||||||
|     return schedule_message_(entity, MessageCreator(function_ptr), message_type); |                          uint8_t estimated_size) { | ||||||
|  |     return schedule_message_(entity, MessageCreator(function_ptr), message_type, estimated_size); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   // Helper function to schedule a high priority message at the front of the batch | ||||||
|  |   bool schedule_message_front_(EntityBase *entity, MessageCreatorPtr function_ptr, uint8_t message_type, | ||||||
|  |                                uint8_t estimated_size) { | ||||||
|  |     this->deferred_batch_.add_item_front(entity, MessageCreator(function_ptr), message_type, estimated_size); | ||||||
|  |     return this->schedule_batch_(); | ||||||
|   } |   } | ||||||
| }; | }; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -5,7 +5,6 @@ | |||||||
| #include "esphome/core/helpers.h" | #include "esphome/core/helpers.h" | ||||||
| #include "esphome/core/log.h" | #include "esphome/core/log.h" | ||||||
| #include "proto.h" | #include "proto.h" | ||||||
| #include "api_pb2_size.h" |  | ||||||
| #include <cstring> | #include <cstring> | ||||||
| #include <cinttypes> | #include <cinttypes> | ||||||
|  |  | ||||||
| @@ -66,6 +65,17 @@ const char *api_error_to_str(APIError err) { | |||||||
|   return "UNKNOWN"; |   return "UNKNOWN"; | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Default implementation for loop - handles sending buffered data | ||||||
|  | APIError APIFrameHelper::loop() { | ||||||
|  |   if (!this->tx_buf_.empty()) { | ||||||
|  |     APIError err = try_send_tx_buf_(); | ||||||
|  |     if (err != APIError::OK && err != APIError::WOULD_BLOCK) { | ||||||
|  |       return err; | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  |   return APIError::OK;  // Convert WOULD_BLOCK to OK to avoid connection termination | ||||||
|  | } | ||||||
|  |  | ||||||
| // Helper method to buffer data from IOVs | // Helper method to buffer data from IOVs | ||||||
| void APIFrameHelper::buffer_data_from_iov_(const struct iovec *iov, int iovcnt, uint16_t total_write_len) { | void APIFrameHelper::buffer_data_from_iov_(const struct iovec *iov, int iovcnt, uint16_t total_write_len) { | ||||||
|   SendBuffer buffer; |   SendBuffer buffer; | ||||||
| @@ -214,6 +224,22 @@ APIError APIFrameHelper::init_common_() { | |||||||
| } | } | ||||||
|  |  | ||||||
| #define HELPER_LOG(msg, ...) ESP_LOGVV(TAG, "%s: " msg, this->info_.c_str(), ##__VA_ARGS__) | #define HELPER_LOG(msg, ...) ESP_LOGVV(TAG, "%s: " msg, this->info_.c_str(), ##__VA_ARGS__) | ||||||
|  |  | ||||||
|  | APIError APIFrameHelper::handle_socket_read_result_(ssize_t received) { | ||||||
|  |   if (received == -1) { | ||||||
|  |     if (errno == EWOULDBLOCK || errno == EAGAIN) { | ||||||
|  |       return APIError::WOULD_BLOCK; | ||||||
|  |     } | ||||||
|  |     state_ = State::FAILED; | ||||||
|  |     HELPER_LOG("Socket read failed with errno %d", errno); | ||||||
|  |     return APIError::SOCKET_READ_FAILED; | ||||||
|  |   } else if (received == 0) { | ||||||
|  |     state_ = State::FAILED; | ||||||
|  |     HELPER_LOG("Connection closed"); | ||||||
|  |     return APIError::CONNECTION_CLOSED; | ||||||
|  |   } | ||||||
|  |   return APIError::OK; | ||||||
|  | } | ||||||
| // uncomment to log raw packets | // uncomment to log raw packets | ||||||
| //#define HELPER_LOG_PACKETS | //#define HELPER_LOG_PACKETS | ||||||
|  |  | ||||||
| @@ -274,17 +300,21 @@ APIError APINoiseFrameHelper::init() { | |||||||
| } | } | ||||||
| /// Run through handshake messages (if in that phase) | /// Run through handshake messages (if in that phase) | ||||||
| APIError APINoiseFrameHelper::loop() { | APIError APINoiseFrameHelper::loop() { | ||||||
|  |   // During handshake phase, process as many actions as possible until we can't progress | ||||||
|  |   // socket_->ready() stays true until next main loop, but state_action() will return | ||||||
|  |   // WOULD_BLOCK when no more data is available to read | ||||||
|  |   while (state_ != State::DATA && this->socket_->ready()) { | ||||||
|     APIError err = state_action_(); |     APIError err = state_action_(); | ||||||
|     if (err != APIError::OK && err != APIError::WOULD_BLOCK) { |     if (err != APIError::OK && err != APIError::WOULD_BLOCK) { | ||||||
|       return err; |       return err; | ||||||
|     } |     } | ||||||
|   if (!this->tx_buf_.empty()) { |     if (err == APIError::WOULD_BLOCK) { | ||||||
|     err = try_send_tx_buf_(); |       break; | ||||||
|     if (err != APIError::OK && err != APIError::WOULD_BLOCK) { |  | ||||||
|       return err; |  | ||||||
|     } |     } | ||||||
|   } |   } | ||||||
|   return APIError::OK;  // Convert WOULD_BLOCK to OK to avoid connection termination |  | ||||||
|  |   // Use base class implementation for buffer sending | ||||||
|  |   return APIFrameHelper::loop(); | ||||||
| } | } | ||||||
|  |  | ||||||
| /** Read a packet into the rx_buf_. If successful, stores frame data in the frame parameter | /** Read a packet into the rx_buf_. If successful, stores frame data in the frame parameter | ||||||
| @@ -312,17 +342,9 @@ APIError APINoiseFrameHelper::try_read_frame_(ParsedFrame *frame) { | |||||||
|     // no header information yet |     // no header information yet | ||||||
|     uint8_t to_read = 3 - rx_header_buf_len_; |     uint8_t to_read = 3 - rx_header_buf_len_; | ||||||
|     ssize_t received = this->socket_->read(&rx_header_buf_[rx_header_buf_len_], to_read); |     ssize_t received = this->socket_->read(&rx_header_buf_[rx_header_buf_len_], to_read); | ||||||
|     if (received == -1) { |     APIError err = handle_socket_read_result_(received); | ||||||
|       if (errno == EWOULDBLOCK || errno == EAGAIN) { |     if (err != APIError::OK) { | ||||||
|         return APIError::WOULD_BLOCK; |       return err; | ||||||
|       } |  | ||||||
|       state_ = State::FAILED; |  | ||||||
|       HELPER_LOG("Socket read failed with errno %d", errno); |  | ||||||
|       return APIError::SOCKET_READ_FAILED; |  | ||||||
|     } else if (received == 0) { |  | ||||||
|       state_ = State::FAILED; |  | ||||||
|       HELPER_LOG("Connection closed"); |  | ||||||
|       return APIError::CONNECTION_CLOSED; |  | ||||||
|     } |     } | ||||||
|     rx_header_buf_len_ += static_cast<uint8_t>(received); |     rx_header_buf_len_ += static_cast<uint8_t>(received); | ||||||
|     if (static_cast<uint8_t>(received) != to_read) { |     if (static_cast<uint8_t>(received) != to_read) { | ||||||
| @@ -330,17 +352,15 @@ APIError APINoiseFrameHelper::try_read_frame_(ParsedFrame *frame) { | |||||||
|       return APIError::WOULD_BLOCK; |       return APIError::WOULD_BLOCK; | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     if (rx_header_buf_[0] != 0x01) { | ||||||
|  |       state_ = State::FAILED; | ||||||
|  |       HELPER_LOG("Bad indicator byte %u", rx_header_buf_[0]); | ||||||
|  |       return APIError::BAD_INDICATOR; | ||||||
|  |     } | ||||||
|     // header reading done |     // header reading done | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // read body |   // read body | ||||||
|   uint8_t indicator = rx_header_buf_[0]; |  | ||||||
|   if (indicator != 0x01) { |  | ||||||
|     state_ = State::FAILED; |  | ||||||
|     HELPER_LOG("Bad indicator byte %u", indicator); |  | ||||||
|     return APIError::BAD_INDICATOR; |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   uint16_t msg_size = (((uint16_t) rx_header_buf_[1]) << 8) | rx_header_buf_[2]; |   uint16_t msg_size = (((uint16_t) rx_header_buf_[1]) << 8) | rx_header_buf_[2]; | ||||||
|  |  | ||||||
|   if (state_ != State::DATA && msg_size > 128) { |   if (state_ != State::DATA && msg_size > 128) { | ||||||
| @@ -359,17 +379,9 @@ APIError APINoiseFrameHelper::try_read_frame_(ParsedFrame *frame) { | |||||||
|     // more data to read |     // more data to read | ||||||
|     uint16_t to_read = msg_size - rx_buf_len_; |     uint16_t to_read = msg_size - rx_buf_len_; | ||||||
|     ssize_t received = this->socket_->read(&rx_buf_[rx_buf_len_], to_read); |     ssize_t received = this->socket_->read(&rx_buf_[rx_buf_len_], to_read); | ||||||
|     if (received == -1) { |     APIError err = handle_socket_read_result_(received); | ||||||
|       if (errno == EWOULDBLOCK || errno == EAGAIN) { |     if (err != APIError::OK) { | ||||||
|         return APIError::WOULD_BLOCK; |       return err; | ||||||
|       } |  | ||||||
|       state_ = State::FAILED; |  | ||||||
|       HELPER_LOG("Socket read failed with errno %d", errno); |  | ||||||
|       return APIError::SOCKET_READ_FAILED; |  | ||||||
|     } else if (received == 0) { |  | ||||||
|       state_ = State::FAILED; |  | ||||||
|       HELPER_LOG("Connection closed"); |  | ||||||
|       return APIError::CONNECTION_CLOSED; |  | ||||||
|     } |     } | ||||||
|     rx_buf_len_ += static_cast<uint16_t>(received); |     rx_buf_len_ += static_cast<uint16_t>(received); | ||||||
|     if (static_cast<uint16_t>(received) != to_read) { |     if (static_cast<uint16_t>(received) != to_read) { | ||||||
| @@ -586,10 +598,6 @@ APIError APINoiseFrameHelper::read_packet(ReadPacketBuffer *buffer) { | |||||||
|     return APIError::BAD_DATA_PACKET; |     return APIError::BAD_DATA_PACKET; | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // uint16_t type; |  | ||||||
|   // uint16_t data_len; |  | ||||||
|   // uint8_t *data; |  | ||||||
|   // uint8_t *padding;  zero or more bytes to fill up the rest of the packet |  | ||||||
|   uint16_t type = (((uint16_t) msg_data[0]) << 8) | msg_data[1]; |   uint16_t type = (((uint16_t) msg_data[0]) << 8) | msg_data[1]; | ||||||
|   uint16_t data_len = (((uint16_t) msg_data[2]) << 8) | msg_data[3]; |   uint16_t data_len = (((uint16_t) msg_data[2]) << 8) | msg_data[3]; | ||||||
|   if (data_len > msg_size - 4) { |   if (data_len > msg_size - 4) { | ||||||
| @@ -604,21 +612,15 @@ APIError APINoiseFrameHelper::read_packet(ReadPacketBuffer *buffer) { | |||||||
|   buffer->type = type; |   buffer->type = type; | ||||||
|   return APIError::OK; |   return APIError::OK; | ||||||
| } | } | ||||||
| APIError APINoiseFrameHelper::write_protobuf_packet(uint16_t type, ProtoWriteBuffer buffer) { | APIError APINoiseFrameHelper::write_protobuf_packet(uint8_t type, ProtoWriteBuffer buffer) { | ||||||
|   std::vector<uint8_t> *raw_buffer = buffer.get_buffer(); |  | ||||||
|   uint16_t payload_len = static_cast<uint16_t>(raw_buffer->size() - frame_header_padding_); |  | ||||||
|  |  | ||||||
|   // Resize to include MAC space (required for Noise encryption) |   // Resize to include MAC space (required for Noise encryption) | ||||||
|   raw_buffer->resize(raw_buffer->size() + frame_footer_size_); |   buffer.get_buffer()->resize(buffer.get_buffer()->size() + frame_footer_size_); | ||||||
|  |   PacketInfo packet{type, 0, | ||||||
|   // Use write_protobuf_packets with a single packet |                     static_cast<uint16_t>(buffer.get_buffer()->size() - frame_header_padding_ - frame_footer_size_)}; | ||||||
|   std::vector<PacketInfo> packets; |   return write_protobuf_packets(buffer, std::span<const PacketInfo>(&packet, 1)); | ||||||
|   packets.emplace_back(type, 0, payload_len); |  | ||||||
|  |  | ||||||
|   return write_protobuf_packets(buffer, packets); |  | ||||||
| } | } | ||||||
|  |  | ||||||
| APIError APINoiseFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer, const std::vector<PacketInfo> &packets) { | APIError APINoiseFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer, std::span<const PacketInfo> packets) { | ||||||
|   APIError aerr = state_action_(); |   APIError aerr = state_action_(); | ||||||
|   if (aerr != APIError::OK) { |   if (aerr != APIError::OK) { | ||||||
|     return aerr; |     return aerr; | ||||||
| @@ -633,18 +635,15 @@ APIError APINoiseFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer, co | |||||||
|   } |   } | ||||||
|  |  | ||||||
|   std::vector<uint8_t> *raw_buffer = buffer.get_buffer(); |   std::vector<uint8_t> *raw_buffer = buffer.get_buffer(); | ||||||
|  |   uint8_t *buffer_data = raw_buffer->data();  // Cache buffer pointer | ||||||
|  |  | ||||||
|   this->reusable_iovs_.clear(); |   this->reusable_iovs_.clear(); | ||||||
|   this->reusable_iovs_.reserve(packets.size()); |   this->reusable_iovs_.reserve(packets.size()); | ||||||
|  |  | ||||||
|   // We need to encrypt each packet in place |   // We need to encrypt each packet in place | ||||||
|   for (const auto &packet : packets) { |   for (const auto &packet : packets) { | ||||||
|     uint16_t type = packet.message_type; |  | ||||||
|     uint16_t offset = packet.offset; |  | ||||||
|     uint16_t payload_len = packet.payload_size; |  | ||||||
|     uint16_t msg_len = 4 + payload_len;  // type(2) + data_len(2) + payload |  | ||||||
|  |  | ||||||
|     // The buffer already has padding at offset |     // The buffer already has padding at offset | ||||||
|     uint8_t *buf_start = raw_buffer->data() + offset; |     uint8_t *buf_start = buffer_data + packet.offset; | ||||||
|  |  | ||||||
|     // Write noise header |     // Write noise header | ||||||
|     buf_start[0] = 0x01;  // indicator |     buf_start[0] = 0x01;  // indicator | ||||||
| @@ -652,10 +651,10 @@ APIError APINoiseFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer, co | |||||||
|  |  | ||||||
|     // Write message header (to be encrypted) |     // Write message header (to be encrypted) | ||||||
|     const uint8_t msg_offset = 3; |     const uint8_t msg_offset = 3; | ||||||
|     buf_start[msg_offset + 0] = (uint8_t) (type >> 8);         // type high byte |     buf_start[msg_offset] = static_cast<uint8_t>(packet.message_type >> 8);      // type high byte | ||||||
|     buf_start[msg_offset + 1] = (uint8_t) type;                // type low byte |     buf_start[msg_offset + 1] = static_cast<uint8_t>(packet.message_type);       // type low byte | ||||||
|     buf_start[msg_offset + 2] = (uint8_t) (payload_len >> 8);  // data_len high byte |     buf_start[msg_offset + 2] = static_cast<uint8_t>(packet.payload_size >> 8);  // data_len high byte | ||||||
|     buf_start[msg_offset + 3] = (uint8_t) payload_len;         // data_len low byte |     buf_start[msg_offset + 3] = static_cast<uint8_t>(packet.payload_size);       // data_len low byte | ||||||
|     // payload data is already in the buffer starting at offset + 7 |     // payload data is already in the buffer starting at offset + 7 | ||||||
|  |  | ||||||
|     // Make sure we have space for MAC |     // Make sure we have space for MAC | ||||||
| @@ -664,7 +663,8 @@ APIError APINoiseFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer, co | |||||||
|     // Encrypt the message in place |     // Encrypt the message in place | ||||||
|     NoiseBuffer mbuf; |     NoiseBuffer mbuf; | ||||||
|     noise_buffer_init(mbuf); |     noise_buffer_init(mbuf); | ||||||
|     noise_buffer_set_inout(mbuf, buf_start + msg_offset, msg_len, msg_len + frame_footer_size_); |     noise_buffer_set_inout(mbuf, buf_start + msg_offset, 4 + packet.payload_size, | ||||||
|  |                            4 + packet.payload_size + frame_footer_size_); | ||||||
|  |  | ||||||
|     int err = noise_cipherstate_encrypt(send_cipher_, &mbuf); |     int err = noise_cipherstate_encrypt(send_cipher_, &mbuf); | ||||||
|     if (err != 0) { |     if (err != 0) { | ||||||
| @@ -674,14 +674,12 @@ APIError APINoiseFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer, co | |||||||
|     } |     } | ||||||
|  |  | ||||||
|     // Fill in the encrypted size |     // Fill in the encrypted size | ||||||
|     buf_start[1] = (uint8_t) (mbuf.size >> 8); |     buf_start[1] = static_cast<uint8_t>(mbuf.size >> 8); | ||||||
|     buf_start[2] = (uint8_t) mbuf.size; |     buf_start[2] = static_cast<uint8_t>(mbuf.size); | ||||||
|  |  | ||||||
|     // Add iovec for this encrypted packet |     // Add iovec for this encrypted packet | ||||||
|     struct iovec iov; |     this->reusable_iovs_.push_back( | ||||||
|     iov.iov_base = buf_start; |         {buf_start, static_cast<size_t>(3 + mbuf.size)});  // indicator + size + encrypted data | ||||||
|     iov.iov_len = 3 + mbuf.size;  // indicator + size + encrypted data |  | ||||||
|     this->reusable_iovs_.push_back(iov); |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // Send all encrypted packets in one writev call |   // Send all encrypted packets in one writev call | ||||||
| @@ -822,18 +820,12 @@ APIError APIPlaintextFrameHelper::init() { | |||||||
|   state_ = State::DATA; |   state_ = State::DATA; | ||||||
|   return APIError::OK; |   return APIError::OK; | ||||||
| } | } | ||||||
| /// Not used for plaintext |  | ||||||
| APIError APIPlaintextFrameHelper::loop() { | APIError APIPlaintextFrameHelper::loop() { | ||||||
|   if (state_ != State::DATA) { |   if (state_ != State::DATA) { | ||||||
|     return APIError::BAD_STATE; |     return APIError::BAD_STATE; | ||||||
|   } |   } | ||||||
|   if (!this->tx_buf_.empty()) { |   // Use base class implementation for buffer sending | ||||||
|     APIError err = try_send_tx_buf_(); |   return APIFrameHelper::loop(); | ||||||
|     if (err != APIError::OK && err != APIError::WOULD_BLOCK) { |  | ||||||
|       return err; |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|   return APIError::OK;  // Convert WOULD_BLOCK to OK to avoid connection termination |  | ||||||
| } | } | ||||||
|  |  | ||||||
| /** Read a packet into the rx_buf_. If successful, stores frame data in the frame parameter | /** Read a packet into the rx_buf_. If successful, stores frame data in the frame parameter | ||||||
| @@ -862,17 +854,9 @@ APIError APIPlaintextFrameHelper::try_read_frame_(ParsedFrame *frame) { | |||||||
|     // Try to get to at least 3 bytes total (indicator + 2 varint bytes), then read one byte at a time |     // Try to get to at least 3 bytes total (indicator + 2 varint bytes), then read one byte at a time | ||||||
|     ssize_t received = |     ssize_t received = | ||||||
|         this->socket_->read(&rx_header_buf_[rx_header_buf_pos_], rx_header_buf_pos_ < 3 ? 3 - rx_header_buf_pos_ : 1); |         this->socket_->read(&rx_header_buf_[rx_header_buf_pos_], rx_header_buf_pos_ < 3 ? 3 - rx_header_buf_pos_ : 1); | ||||||
|     if (received == -1) { |     APIError err = handle_socket_read_result_(received); | ||||||
|       if (errno == EWOULDBLOCK || errno == EAGAIN) { |     if (err != APIError::OK) { | ||||||
|         return APIError::WOULD_BLOCK; |       return err; | ||||||
|       } |  | ||||||
|       state_ = State::FAILED; |  | ||||||
|       HELPER_LOG("Socket read failed with errno %d", errno); |  | ||||||
|       return APIError::SOCKET_READ_FAILED; |  | ||||||
|     } else if (received == 0) { |  | ||||||
|       state_ = State::FAILED; |  | ||||||
|       HELPER_LOG("Connection closed"); |  | ||||||
|       return APIError::CONNECTION_CLOSED; |  | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     // If this was the first read, validate the indicator byte |     // If this was the first read, validate the indicator byte | ||||||
| @@ -956,17 +940,9 @@ APIError APIPlaintextFrameHelper::try_read_frame_(ParsedFrame *frame) { | |||||||
|     // more data to read |     // more data to read | ||||||
|     uint16_t to_read = rx_header_parsed_len_ - rx_buf_len_; |     uint16_t to_read = rx_header_parsed_len_ - rx_buf_len_; | ||||||
|     ssize_t received = this->socket_->read(&rx_buf_[rx_buf_len_], to_read); |     ssize_t received = this->socket_->read(&rx_buf_[rx_buf_len_], to_read); | ||||||
|     if (received == -1) { |     APIError err = handle_socket_read_result_(received); | ||||||
|       if (errno == EWOULDBLOCK || errno == EAGAIN) { |     if (err != APIError::OK) { | ||||||
|         return APIError::WOULD_BLOCK; |       return err; | ||||||
|       } |  | ||||||
|       state_ = State::FAILED; |  | ||||||
|       HELPER_LOG("Socket read failed with errno %d", errno); |  | ||||||
|       return APIError::SOCKET_READ_FAILED; |  | ||||||
|     } else if (received == 0) { |  | ||||||
|       state_ = State::FAILED; |  | ||||||
|       HELPER_LOG("Connection closed"); |  | ||||||
|       return APIError::CONNECTION_CLOSED; |  | ||||||
|     } |     } | ||||||
|     rx_buf_len_ += static_cast<uint16_t>(received); |     rx_buf_len_ += static_cast<uint16_t>(received); | ||||||
|     if (static_cast<uint16_t>(received) != to_read) { |     if (static_cast<uint16_t>(received) != to_read) { | ||||||
| @@ -1025,19 +1001,12 @@ APIError APIPlaintextFrameHelper::read_packet(ReadPacketBuffer *buffer) { | |||||||
|   buffer->type = rx_header_parsed_type_; |   buffer->type = rx_header_parsed_type_; | ||||||
|   return APIError::OK; |   return APIError::OK; | ||||||
| } | } | ||||||
| APIError APIPlaintextFrameHelper::write_protobuf_packet(uint16_t type, ProtoWriteBuffer buffer) { | APIError APIPlaintextFrameHelper::write_protobuf_packet(uint8_t type, ProtoWriteBuffer buffer) { | ||||||
|   std::vector<uint8_t> *raw_buffer = buffer.get_buffer(); |   PacketInfo packet{type, 0, static_cast<uint16_t>(buffer.get_buffer()->size() - frame_header_padding_)}; | ||||||
|   uint16_t payload_len = static_cast<uint16_t>(raw_buffer->size() - frame_header_padding_); |   return write_protobuf_packets(buffer, std::span<const PacketInfo>(&packet, 1)); | ||||||
|  |  | ||||||
|   // Use write_protobuf_packets with a single packet |  | ||||||
|   std::vector<PacketInfo> packets; |  | ||||||
|   packets.emplace_back(type, 0, payload_len); |  | ||||||
|  |  | ||||||
|   return write_protobuf_packets(buffer, packets); |  | ||||||
| } | } | ||||||
|  |  | ||||||
| APIError APIPlaintextFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer, | APIError APIPlaintextFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer, std::span<const PacketInfo> packets) { | ||||||
|                                                          const std::vector<PacketInfo> &packets) { |  | ||||||
|   if (state_ != State::DATA) { |   if (state_ != State::DATA) { | ||||||
|     return APIError::BAD_STATE; |     return APIError::BAD_STATE; | ||||||
|   } |   } | ||||||
| @@ -1047,17 +1016,15 @@ APIError APIPlaintextFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer | |||||||
|   } |   } | ||||||
|  |  | ||||||
|   std::vector<uint8_t> *raw_buffer = buffer.get_buffer(); |   std::vector<uint8_t> *raw_buffer = buffer.get_buffer(); | ||||||
|  |   uint8_t *buffer_data = raw_buffer->data();  // Cache buffer pointer | ||||||
|  |  | ||||||
|   this->reusable_iovs_.clear(); |   this->reusable_iovs_.clear(); | ||||||
|   this->reusable_iovs_.reserve(packets.size()); |   this->reusable_iovs_.reserve(packets.size()); | ||||||
|  |  | ||||||
|   for (const auto &packet : packets) { |   for (const auto &packet : packets) { | ||||||
|     uint16_t type = packet.message_type; |  | ||||||
|     uint16_t offset = packet.offset; |  | ||||||
|     uint16_t payload_len = packet.payload_size; |  | ||||||
|  |  | ||||||
|     // Calculate varint sizes for header layout |     // Calculate varint sizes for header layout | ||||||
|     uint8_t size_varint_len = api::ProtoSize::varint(static_cast<uint32_t>(payload_len)); |     uint8_t size_varint_len = api::ProtoSize::varint(static_cast<uint32_t>(packet.payload_size)); | ||||||
|     uint8_t type_varint_len = api::ProtoSize::varint(static_cast<uint32_t>(type)); |     uint8_t type_varint_len = api::ProtoSize::varint(static_cast<uint32_t>(packet.message_type)); | ||||||
|     uint8_t total_header_len = 1 + size_varint_len + type_varint_len; |     uint8_t total_header_len = 1 + size_varint_len + type_varint_len; | ||||||
|  |  | ||||||
|     // Calculate where to start writing the header |     // Calculate where to start writing the header | ||||||
| @@ -1085,23 +1052,20 @@ APIError APIPlaintextFrameHelper::write_protobuf_packets(ProtoWriteBuffer buffer | |||||||
|     // |     // | ||||||
|     // The message starts at offset + frame_header_padding_ |     // The message starts at offset + frame_header_padding_ | ||||||
|     // So we write the header starting at offset + frame_header_padding_ - total_header_len |     // So we write the header starting at offset + frame_header_padding_ - total_header_len | ||||||
|     uint8_t *buf_start = raw_buffer->data() + offset; |     uint8_t *buf_start = buffer_data + packet.offset; | ||||||
|     uint32_t header_offset = frame_header_padding_ - total_header_len; |     uint32_t header_offset = frame_header_padding_ - total_header_len; | ||||||
|  |  | ||||||
|     // Write the plaintext header |     // Write the plaintext header | ||||||
|     buf_start[header_offset] = 0x00;  // indicator |     buf_start[header_offset] = 0x00;  // indicator | ||||||
|  |  | ||||||
|     // Encode size varint directly into buffer |     // Encode varints directly into buffer | ||||||
|     ProtoVarInt(payload_len).encode_to_buffer_unchecked(buf_start + header_offset + 1, size_varint_len); |     ProtoVarInt(packet.payload_size).encode_to_buffer_unchecked(buf_start + header_offset + 1, size_varint_len); | ||||||
|  |     ProtoVarInt(packet.message_type) | ||||||
|     // Encode type varint directly into buffer |         .encode_to_buffer_unchecked(buf_start + header_offset + 1 + size_varint_len, type_varint_len); | ||||||
|     ProtoVarInt(type).encode_to_buffer_unchecked(buf_start + header_offset + 1 + size_varint_len, type_varint_len); |  | ||||||
|  |  | ||||||
|     // Add iovec for this packet (header + payload) |     // Add iovec for this packet (header + payload) | ||||||
|     struct iovec iov; |     this->reusable_iovs_.push_back( | ||||||
|     iov.iov_base = buf_start + header_offset; |         {buf_start + header_offset, static_cast<size_t>(total_header_len + packet.payload_size)}); | ||||||
|     iov.iov_len = total_header_len + payload_len; |  | ||||||
|     this->reusable_iovs_.push_back(iov); |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // Send all packets in one writev call |   // Send all packets in one writev call | ||||||
|   | |||||||
| @@ -2,6 +2,7 @@ | |||||||
| #include <cstdint> | #include <cstdint> | ||||||
| #include <deque> | #include <deque> | ||||||
| #include <limits> | #include <limits> | ||||||
|  | #include <span> | ||||||
| #include <utility> | #include <utility> | ||||||
| #include <vector> | #include <vector> | ||||||
|  |  | ||||||
| @@ -29,16 +30,14 @@ struct ReadPacketBuffer { | |||||||
|  |  | ||||||
| // Packed packet info structure to minimize memory usage | // Packed packet info structure to minimize memory usage | ||||||
| struct PacketInfo { | struct PacketInfo { | ||||||
|   uint16_t message_type;  // 2 bytes |   uint16_t offset;        // Offset in buffer where message starts | ||||||
|   uint16_t offset;        // 2 bytes (sufficient for packet size ~1460 bytes) |   uint16_t payload_size;  // Size of the message payload | ||||||
|   uint16_t payload_size;  // 2 bytes (up to 65535 bytes) |   uint8_t message_type;   // Message type (0-255) | ||||||
|   uint16_t padding;       // 2 byte (for alignment) |  | ||||||
|  |  | ||||||
|   PacketInfo(uint16_t type, uint16_t off, uint16_t size) |   PacketInfo(uint8_t type, uint16_t off, uint16_t size) : offset(off), payload_size(size), message_type(type) {} | ||||||
|       : message_type(type), offset(off), payload_size(size), padding(0) {} |  | ||||||
| }; | }; | ||||||
|  |  | ||||||
| enum class APIError : int { | enum class APIError : uint16_t { | ||||||
|   OK = 0, |   OK = 0, | ||||||
|   WOULD_BLOCK = 1001, |   WOULD_BLOCK = 1001, | ||||||
|   BAD_HANDSHAKE_PACKET_LEN = 1002, |   BAD_HANDSHAKE_PACKET_LEN = 1002, | ||||||
| @@ -74,7 +73,7 @@ class APIFrameHelper { | |||||||
|   } |   } | ||||||
|   virtual ~APIFrameHelper() = default; |   virtual ~APIFrameHelper() = default; | ||||||
|   virtual APIError init() = 0; |   virtual APIError init() = 0; | ||||||
|   virtual APIError loop() = 0; |   virtual APIError loop(); | ||||||
|   virtual APIError read_packet(ReadPacketBuffer *buffer) = 0; |   virtual APIError read_packet(ReadPacketBuffer *buffer) = 0; | ||||||
|   bool can_write_without_blocking() { return state_ == State::DATA && tx_buf_.empty(); } |   bool can_write_without_blocking() { return state_ == State::DATA && tx_buf_.empty(); } | ||||||
|   std::string getpeername() { return socket_->getpeername(); } |   std::string getpeername() { return socket_->getpeername(); } | ||||||
| @@ -97,11 +96,11 @@ class APIFrameHelper { | |||||||
|   } |   } | ||||||
|   // Give this helper a name for logging |   // Give this helper a name for logging | ||||||
|   void set_log_info(std::string info) { info_ = std::move(info); } |   void set_log_info(std::string info) { info_ = std::move(info); } | ||||||
|   virtual APIError write_protobuf_packet(uint16_t type, ProtoWriteBuffer buffer) = 0; |   virtual APIError write_protobuf_packet(uint8_t type, ProtoWriteBuffer buffer) = 0; | ||||||
|   // Write multiple protobuf packets in a single operation |   // Write multiple protobuf packets in a single operation | ||||||
|   // packets contains (message_type, offset, length) for each message in the buffer |   // packets contains (message_type, offset, length) for each message in the buffer | ||||||
|   // The buffer contains all messages with appropriate padding before each |   // The buffer contains all messages with appropriate padding before each | ||||||
|   virtual APIError write_protobuf_packets(ProtoWriteBuffer buffer, const std::vector<PacketInfo> &packets) = 0; |   virtual APIError write_protobuf_packets(ProtoWriteBuffer buffer, std::span<const PacketInfo> packets) = 0; | ||||||
|   // Get the frame header padding required by this protocol |   // Get the frame header padding required by this protocol | ||||||
|   virtual uint8_t frame_header_padding() = 0; |   virtual uint8_t frame_header_padding() = 0; | ||||||
|   // Get the frame footer size required by this protocol |   // Get the frame footer size required by this protocol | ||||||
| @@ -125,38 +124,6 @@ class APIFrameHelper { | |||||||
|     const uint8_t *current_data() const { return data.data() + offset; } |     const uint8_t *current_data() const { return data.data() + offset; } | ||||||
|   }; |   }; | ||||||
|  |  | ||||||
|   // Queue of data buffers to be sent |  | ||||||
|   std::deque<SendBuffer> tx_buf_; |  | ||||||
|  |  | ||||||
|   // Common state enum for all frame helpers |  | ||||||
|   // Note: Not all states are used by all implementations |  | ||||||
|   // - INITIALIZE: Used by both Noise and Plaintext |  | ||||||
|   // - CLIENT_HELLO, SERVER_HELLO, HANDSHAKE: Only used by Noise protocol |  | ||||||
|   // - DATA: Used by both Noise and Plaintext |  | ||||||
|   // - CLOSED: Used by both Noise and Plaintext |  | ||||||
|   // - FAILED: Used by both Noise and Plaintext |  | ||||||
|   // - EXPLICIT_REJECT: Only used by Noise protocol |  | ||||||
|   enum class State { |  | ||||||
|     INITIALIZE = 1, |  | ||||||
|     CLIENT_HELLO = 2,  // Noise only |  | ||||||
|     SERVER_HELLO = 3,  // Noise only |  | ||||||
|     HANDSHAKE = 4,     // Noise only |  | ||||||
|     DATA = 5, |  | ||||||
|     CLOSED = 6, |  | ||||||
|     FAILED = 7, |  | ||||||
|     EXPLICIT_REJECT = 8,  // Noise only |  | ||||||
|   }; |  | ||||||
|  |  | ||||||
|   // Current state of the frame helper |  | ||||||
|   State state_{State::INITIALIZE}; |  | ||||||
|  |  | ||||||
|   // Helper name for logging |  | ||||||
|   std::string info_; |  | ||||||
|  |  | ||||||
|   // Socket for communication |  | ||||||
|   socket::Socket *socket_{nullptr}; |  | ||||||
|   std::unique_ptr<socket::Socket> socket_owned_; |  | ||||||
|  |  | ||||||
|   // Common implementation for writing raw data to socket |   // Common implementation for writing raw data to socket | ||||||
|   APIError write_raw_(const struct iovec *iov, int iovcnt); |   APIError write_raw_(const struct iovec *iov, int iovcnt); | ||||||
|  |  | ||||||
| @@ -169,18 +136,47 @@ class APIFrameHelper { | |||||||
|   APIError write_raw_(const struct iovec *iov, int iovcnt, socket::Socket *socket, std::vector<uint8_t> &tx_buf, |   APIError write_raw_(const struct iovec *iov, int iovcnt, socket::Socket *socket, std::vector<uint8_t> &tx_buf, | ||||||
|                       const std::string &info, StateEnum &state, StateEnum failed_state); |                       const std::string &info, StateEnum &state, StateEnum failed_state); | ||||||
|  |  | ||||||
|  |   // Pointers first (4 bytes each) | ||||||
|  |   socket::Socket *socket_{nullptr}; | ||||||
|  |   std::unique_ptr<socket::Socket> socket_owned_; | ||||||
|  |  | ||||||
|  |   // Common state enum for all frame helpers | ||||||
|  |   // Note: Not all states are used by all implementations | ||||||
|  |   // - INITIALIZE: Used by both Noise and Plaintext | ||||||
|  |   // - CLIENT_HELLO, SERVER_HELLO, HANDSHAKE: Only used by Noise protocol | ||||||
|  |   // - DATA: Used by both Noise and Plaintext | ||||||
|  |   // - CLOSED: Used by both Noise and Plaintext | ||||||
|  |   // - FAILED: Used by both Noise and Plaintext | ||||||
|  |   // - EXPLICIT_REJECT: Only used by Noise protocol | ||||||
|  |   enum class State : uint8_t { | ||||||
|  |     INITIALIZE = 1, | ||||||
|  |     CLIENT_HELLO = 2,  // Noise only | ||||||
|  |     SERVER_HELLO = 3,  // Noise only | ||||||
|  |     HANDSHAKE = 4,     // Noise only | ||||||
|  |     DATA = 5, | ||||||
|  |     CLOSED = 6, | ||||||
|  |     FAILED = 7, | ||||||
|  |     EXPLICIT_REJECT = 8,  // Noise only | ||||||
|  |   }; | ||||||
|  |  | ||||||
|  |   // Containers (size varies, but typically 12+ bytes on 32-bit) | ||||||
|  |   std::deque<SendBuffer> tx_buf_; | ||||||
|  |   std::string info_; | ||||||
|  |   std::vector<struct iovec> reusable_iovs_; | ||||||
|  |   std::vector<uint8_t> rx_buf_; | ||||||
|  |  | ||||||
|  |   // Group smaller types together | ||||||
|  |   uint16_t rx_buf_len_ = 0; | ||||||
|  |   State state_{State::INITIALIZE}; | ||||||
|   uint8_t frame_header_padding_{0}; |   uint8_t frame_header_padding_{0}; | ||||||
|   uint8_t frame_footer_size_{0}; |   uint8_t frame_footer_size_{0}; | ||||||
|  |   // 5 bytes total, 3 bytes padding | ||||||
|   // Reusable IOV array for write_protobuf_packets to avoid repeated allocations |  | ||||||
|   std::vector<struct iovec> reusable_iovs_; |  | ||||||
|  |  | ||||||
|   // Receive buffer for reading frame data |  | ||||||
|   std::vector<uint8_t> rx_buf_; |  | ||||||
|   uint16_t rx_buf_len_ = 0; |  | ||||||
|  |  | ||||||
|   // Common initialization for both plaintext and noise protocols |   // Common initialization for both plaintext and noise protocols | ||||||
|   APIError init_common_(); |   APIError init_common_(); | ||||||
|  |  | ||||||
|  |   // Helper method to handle socket read results | ||||||
|  |   APIError handle_socket_read_result_(ssize_t received); | ||||||
| }; | }; | ||||||
|  |  | ||||||
| #ifdef USE_API_NOISE | #ifdef USE_API_NOISE | ||||||
| @@ -199,8 +195,8 @@ class APINoiseFrameHelper : public APIFrameHelper { | |||||||
|   APIError init() override; |   APIError init() override; | ||||||
|   APIError loop() override; |   APIError loop() override; | ||||||
|   APIError read_packet(ReadPacketBuffer *buffer) override; |   APIError read_packet(ReadPacketBuffer *buffer) override; | ||||||
|   APIError write_protobuf_packet(uint16_t type, ProtoWriteBuffer buffer) override; |   APIError write_protobuf_packet(uint8_t type, ProtoWriteBuffer buffer) override; | ||||||
|   APIError write_protobuf_packets(ProtoWriteBuffer buffer, const std::vector<PacketInfo> &packets) override; |   APIError write_protobuf_packets(ProtoWriteBuffer buffer, std::span<const PacketInfo> packets) override; | ||||||
|   // Get the frame header padding required by this protocol |   // Get the frame header padding required by this protocol | ||||||
|   uint8_t frame_header_padding() override { return frame_header_padding_; } |   uint8_t frame_header_padding() override { return frame_header_padding_; } | ||||||
|   // Get the frame footer size required by this protocol |   // Get the frame footer size required by this protocol | ||||||
| @@ -213,19 +209,28 @@ class APINoiseFrameHelper : public APIFrameHelper { | |||||||
|   APIError init_handshake_(); |   APIError init_handshake_(); | ||||||
|   APIError check_handshake_finished_(); |   APIError check_handshake_finished_(); | ||||||
|   void send_explicit_handshake_reject_(const std::string &reason); |   void send_explicit_handshake_reject_(const std::string &reason); | ||||||
|  |  | ||||||
|  |   // Pointers first (4 bytes each) | ||||||
|  |   NoiseHandshakeState *handshake_{nullptr}; | ||||||
|  |   NoiseCipherState *send_cipher_{nullptr}; | ||||||
|  |   NoiseCipherState *recv_cipher_{nullptr}; | ||||||
|  |  | ||||||
|  |   // Shared pointer (8 bytes on 32-bit = 4 bytes control block pointer + 4 bytes object pointer) | ||||||
|  |   std::shared_ptr<APINoiseContext> ctx_; | ||||||
|  |  | ||||||
|  |   // Vector (12 bytes on 32-bit) | ||||||
|  |   std::vector<uint8_t> prologue_; | ||||||
|  |  | ||||||
|  |   // NoiseProtocolId (size depends on implementation) | ||||||
|  |   NoiseProtocolId nid_; | ||||||
|  |  | ||||||
|  |   // Group small types together | ||||||
|   // Fixed-size header buffer for noise protocol: |   // Fixed-size header buffer for noise protocol: | ||||||
|   // 1 byte for indicator + 2 bytes for message size (16-bit value, not varint) |   // 1 byte for indicator + 2 bytes for message size (16-bit value, not varint) | ||||||
|   // Note: Maximum message size is UINT16_MAX (65535), with a limit of 128 bytes during handshake phase |   // Note: Maximum message size is UINT16_MAX (65535), with a limit of 128 bytes during handshake phase | ||||||
|   uint8_t rx_header_buf_[3]; |   uint8_t rx_header_buf_[3]; | ||||||
|   uint8_t rx_header_buf_len_ = 0; |   uint8_t rx_header_buf_len_ = 0; | ||||||
|  |   // 4 bytes total, no padding | ||||||
|   std::vector<uint8_t> prologue_; |  | ||||||
|  |  | ||||||
|   std::shared_ptr<APINoiseContext> ctx_; |  | ||||||
|   NoiseHandshakeState *handshake_{nullptr}; |  | ||||||
|   NoiseCipherState *send_cipher_{nullptr}; |  | ||||||
|   NoiseCipherState *recv_cipher_{nullptr}; |  | ||||||
|   NoiseProtocolId nid_; |  | ||||||
| }; | }; | ||||||
| #endif  // USE_API_NOISE | #endif  // USE_API_NOISE | ||||||
|  |  | ||||||
| @@ -244,14 +249,20 @@ class APIPlaintextFrameHelper : public APIFrameHelper { | |||||||
|   APIError init() override; |   APIError init() override; | ||||||
|   APIError loop() override; |   APIError loop() override; | ||||||
|   APIError read_packet(ReadPacketBuffer *buffer) override; |   APIError read_packet(ReadPacketBuffer *buffer) override; | ||||||
|   APIError write_protobuf_packet(uint16_t type, ProtoWriteBuffer buffer) override; |   APIError write_protobuf_packet(uint8_t type, ProtoWriteBuffer buffer) override; | ||||||
|   APIError write_protobuf_packets(ProtoWriteBuffer buffer, const std::vector<PacketInfo> &packets) override; |   APIError write_protobuf_packets(ProtoWriteBuffer buffer, std::span<const PacketInfo> packets) override; | ||||||
|   uint8_t frame_header_padding() override { return frame_header_padding_; } |   uint8_t frame_header_padding() override { return frame_header_padding_; } | ||||||
|   // Get the frame footer size required by this protocol |   // Get the frame footer size required by this protocol | ||||||
|   uint8_t frame_footer_size() override { return frame_footer_size_; } |   uint8_t frame_footer_size() override { return frame_footer_size_; } | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   APIError try_read_frame_(ParsedFrame *frame); |   APIError try_read_frame_(ParsedFrame *frame); | ||||||
|  |  | ||||||
|  |   // Group 2-byte aligned types | ||||||
|  |   uint16_t rx_header_parsed_type_ = 0; | ||||||
|  |   uint16_t rx_header_parsed_len_ = 0; | ||||||
|  |  | ||||||
|  |   // Group 1-byte types together | ||||||
|   // Fixed-size header buffer for plaintext protocol: |   // Fixed-size header buffer for plaintext protocol: | ||||||
|   // We now store the indicator byte + the two varints. |   // We now store the indicator byte + the two varints. | ||||||
|   // To match noise protocol's maximum message size (UINT16_MAX = 65535), we need: |   // To match noise protocol's maximum message size (UINT16_MAX = 65535), we need: | ||||||
| @@ -263,8 +274,7 @@ class APIPlaintextFrameHelper : public APIFrameHelper { | |||||||
|   uint8_t rx_header_buf_[6];  // 1 byte indicator + 5 bytes for varints (3 for size + 2 for type) |   uint8_t rx_header_buf_[6];  // 1 byte indicator + 5 bytes for varints (3 for size + 2 for type) | ||||||
|   uint8_t rx_header_buf_pos_ = 0; |   uint8_t rx_header_buf_pos_ = 0; | ||||||
|   bool rx_header_parsed_ = false; |   bool rx_header_parsed_ = false; | ||||||
|   uint16_t rx_header_parsed_type_ = 0; |   // 8 bytes total, no padding needed | ||||||
|   uint16_t rx_header_parsed_len_ = 0; |  | ||||||
| }; | }; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|   | |||||||
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										4432
									
								
								esphome/components/api/api_pb2_dump.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4432
									
								
								esphome/components/api/api_pb2_dump.cpp
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							| @@ -2,9 +2,10 @@ | |||||||
| // See script/api_protobuf/api_protobuf.py | // See script/api_protobuf/api_protobuf.py | ||||||
| #pragma once | #pragma once | ||||||
|  |  | ||||||
| #include "api_pb2.h" |  | ||||||
| #include "esphome/core/defines.h" | #include "esphome/core/defines.h" | ||||||
|  |  | ||||||
|  | #include "api_pb2.h" | ||||||
|  |  | ||||||
| namespace esphome { | namespace esphome { | ||||||
| namespace api { | namespace api { | ||||||
|  |  | ||||||
| @@ -19,7 +20,7 @@ class APIServerConnectionBase : public ProtoService { | |||||||
|  |  | ||||||
|   template<typename T> bool send_message(const T &msg) { |   template<typename T> bool send_message(const T &msg) { | ||||||
| #ifdef HAS_PROTO_MESSAGE_DUMP | #ifdef HAS_PROTO_MESSAGE_DUMP | ||||||
|     this->log_send_message_(T::message_name(), msg.dump()); |     this->log_send_message_(msg.message_name(), msg.dump()); | ||||||
| #endif | #endif | ||||||
|     return this->send_message_(msg, T::MESSAGE_TYPE); |     return this->send_message_(msg, T::MESSAGE_TYPE); | ||||||
|   } |   } | ||||||
| @@ -68,9 +69,11 @@ class APIServerConnectionBase : public ProtoService { | |||||||
|   virtual void on_get_time_request(const GetTimeRequest &value){}; |   virtual void on_get_time_request(const GetTimeRequest &value){}; | ||||||
|   virtual void on_get_time_response(const GetTimeResponse &value){}; |   virtual void on_get_time_response(const GetTimeResponse &value){}; | ||||||
|  |  | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   virtual void on_execute_service_request(const ExecuteServiceRequest &value){}; |   virtual void on_execute_service_request(const ExecuteServiceRequest &value){}; | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #ifdef USE_ESP32_CAMERA | #ifdef USE_CAMERA | ||||||
|   virtual void on_camera_image_request(const CameraImageRequest &value){}; |   virtual void on_camera_image_request(const CameraImageRequest &value){}; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| @@ -199,7 +202,7 @@ class APIServerConnectionBase : public ProtoService { | |||||||
|   virtual void on_update_command_request(const UpdateCommandRequest &value){}; |   virtual void on_update_command_request(const UpdateCommandRequest &value){}; | ||||||
| #endif | #endif | ||||||
|  protected: |  protected: | ||||||
|   bool read_message(uint32_t msg_size, uint32_t msg_type, uint8_t *msg_data) override; |   void read_message(uint32_t msg_size, uint32_t msg_type, uint8_t *msg_data) override; | ||||||
| }; | }; | ||||||
|  |  | ||||||
| class APIServerConnection : public APIServerConnectionBase { | class APIServerConnection : public APIServerConnectionBase { | ||||||
| @@ -215,14 +218,16 @@ class APIServerConnection : public APIServerConnectionBase { | |||||||
|   virtual void subscribe_homeassistant_services(const SubscribeHomeassistantServicesRequest &msg) = 0; |   virtual void subscribe_homeassistant_services(const SubscribeHomeassistantServicesRequest &msg) = 0; | ||||||
|   virtual void subscribe_home_assistant_states(const SubscribeHomeAssistantStatesRequest &msg) = 0; |   virtual void subscribe_home_assistant_states(const SubscribeHomeAssistantStatesRequest &msg) = 0; | ||||||
|   virtual GetTimeResponse get_time(const GetTimeRequest &msg) = 0; |   virtual GetTimeResponse get_time(const GetTimeRequest &msg) = 0; | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   virtual void execute_service(const ExecuteServiceRequest &msg) = 0; |   virtual void execute_service(const ExecuteServiceRequest &msg) = 0; | ||||||
|  | #endif | ||||||
| #ifdef USE_API_NOISE | #ifdef USE_API_NOISE | ||||||
|   virtual NoiseEncryptionSetKeyResponse noise_encryption_set_key(const NoiseEncryptionSetKeyRequest &msg) = 0; |   virtual NoiseEncryptionSetKeyResponse noise_encryption_set_key(const NoiseEncryptionSetKeyRequest &msg) = 0; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_BUTTON | #ifdef USE_BUTTON | ||||||
|   virtual void button_command(const ButtonCommandRequest &msg) = 0; |   virtual void button_command(const ButtonCommandRequest &msg) = 0; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_ESP32_CAMERA | #ifdef USE_CAMERA | ||||||
|   virtual void camera_image(const CameraImageRequest &msg) = 0; |   virtual void camera_image(const CameraImageRequest &msg) = 0; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_CLIMATE | #ifdef USE_CLIMATE | ||||||
| @@ -332,14 +337,16 @@ class APIServerConnection : public APIServerConnectionBase { | |||||||
|   void on_subscribe_homeassistant_services_request(const SubscribeHomeassistantServicesRequest &msg) override; |   void on_subscribe_homeassistant_services_request(const SubscribeHomeassistantServicesRequest &msg) override; | ||||||
|   void on_subscribe_home_assistant_states_request(const SubscribeHomeAssistantStatesRequest &msg) override; |   void on_subscribe_home_assistant_states_request(const SubscribeHomeAssistantStatesRequest &msg) override; | ||||||
|   void on_get_time_request(const GetTimeRequest &msg) override; |   void on_get_time_request(const GetTimeRequest &msg) override; | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   void on_execute_service_request(const ExecuteServiceRequest &msg) override; |   void on_execute_service_request(const ExecuteServiceRequest &msg) override; | ||||||
|  | #endif | ||||||
| #ifdef USE_API_NOISE | #ifdef USE_API_NOISE | ||||||
|   void on_noise_encryption_set_key_request(const NoiseEncryptionSetKeyRequest &msg) override; |   void on_noise_encryption_set_key_request(const NoiseEncryptionSetKeyRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_BUTTON | #ifdef USE_BUTTON | ||||||
|   void on_button_command_request(const ButtonCommandRequest &msg) override; |   void on_button_command_request(const ButtonCommandRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_ESP32_CAMERA | #ifdef USE_CAMERA | ||||||
|   void on_camera_image_request(const CameraImageRequest &msg) override; |   void on_camera_image_request(const CameraImageRequest &msg) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_CLIMATE | #ifdef USE_CLIMATE | ||||||
|   | |||||||
| @@ -1,361 +0,0 @@ | |||||||
| #pragma once |  | ||||||
|  |  | ||||||
| #include "proto.h" |  | ||||||
| #include <cstdint> |  | ||||||
| #include <string> |  | ||||||
|  |  | ||||||
| namespace esphome { |  | ||||||
| namespace api { |  | ||||||
|  |  | ||||||
| class ProtoSize { |  | ||||||
|  public: |  | ||||||
|   /** |  | ||||||
|    * @brief ProtoSize class for Protocol Buffer serialization size calculation |  | ||||||
|    * |  | ||||||
|    * This class provides static methods to calculate the exact byte counts needed |  | ||||||
|    * for encoding various Protocol Buffer field types. All methods are designed to be |  | ||||||
|    * efficient for the common case where many fields have default values. |  | ||||||
|    * |  | ||||||
|    * Implements Protocol Buffer encoding size calculation according to: |  | ||||||
|    * https://protobuf.dev/programming-guides/encoding/ |  | ||||||
|    * |  | ||||||
|    * Key features: |  | ||||||
|    * - Early-return optimization for zero/default values |  | ||||||
|    * - Direct total_size updates to avoid unnecessary additions |  | ||||||
|    * - Specialized handling for different field types according to protobuf spec |  | ||||||
|    * - Templated helpers for repeated fields and messages |  | ||||||
|    */ |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates the size in bytes needed to encode a uint32_t value as a varint |  | ||||||
|    * |  | ||||||
|    * @param value The uint32_t value to calculate size for |  | ||||||
|    * @return The number of bytes needed to encode the value |  | ||||||
|    */ |  | ||||||
|   static inline uint32_t varint(uint32_t value) { |  | ||||||
|     // Optimized varint size calculation using leading zeros |  | ||||||
|     // Each 7 bits requires one byte in the varint encoding |  | ||||||
|     if (value < 128) |  | ||||||
|       return 1;  // 7 bits, common case for small values |  | ||||||
|  |  | ||||||
|     // For larger values, count bytes needed based on the position of the highest bit set |  | ||||||
|     if (value < 16384) { |  | ||||||
|       return 2;  // 14 bits |  | ||||||
|     } else if (value < 2097152) { |  | ||||||
|       return 3;  // 21 bits |  | ||||||
|     } else if (value < 268435456) { |  | ||||||
|       return 4;  // 28 bits |  | ||||||
|     } else { |  | ||||||
|       return 5;  // 32 bits (maximum for uint32_t) |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates the size in bytes needed to encode a uint64_t value as a varint |  | ||||||
|    * |  | ||||||
|    * @param value The uint64_t value to calculate size for |  | ||||||
|    * @return The number of bytes needed to encode the value |  | ||||||
|    */ |  | ||||||
|   static inline uint32_t varint(uint64_t value) { |  | ||||||
|     // Handle common case of values fitting in uint32_t (vast majority of use cases) |  | ||||||
|     if (value <= UINT32_MAX) { |  | ||||||
|       return varint(static_cast<uint32_t>(value)); |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // For larger values, determine size based on highest bit position |  | ||||||
|     if (value < (1ULL << 35)) { |  | ||||||
|       return 5;  // 35 bits |  | ||||||
|     } else if (value < (1ULL << 42)) { |  | ||||||
|       return 6;  // 42 bits |  | ||||||
|     } else if (value < (1ULL << 49)) { |  | ||||||
|       return 7;  // 49 bits |  | ||||||
|     } else if (value < (1ULL << 56)) { |  | ||||||
|       return 8;  // 56 bits |  | ||||||
|     } else if (value < (1ULL << 63)) { |  | ||||||
|       return 9;  // 63 bits |  | ||||||
|     } else { |  | ||||||
|       return 10;  // 64 bits (maximum for uint64_t) |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates the size in bytes needed to encode an int32_t value as a varint |  | ||||||
|    * |  | ||||||
|    * Special handling is needed for negative values, which are sign-extended to 64 bits |  | ||||||
|    * in Protocol Buffers, resulting in a 10-byte varint. |  | ||||||
|    * |  | ||||||
|    * @param value The int32_t value to calculate size for |  | ||||||
|    * @return The number of bytes needed to encode the value |  | ||||||
|    */ |  | ||||||
|   static inline uint32_t varint(int32_t value) { |  | ||||||
|     // Negative values are sign-extended to 64 bits in protocol buffers, |  | ||||||
|     // which always results in a 10-byte varint for negative int32 |  | ||||||
|     if (value < 0) { |  | ||||||
|       return 10;  // Negative int32 is always 10 bytes long |  | ||||||
|     } |  | ||||||
|     // For non-negative values, use the uint32_t implementation |  | ||||||
|     return varint(static_cast<uint32_t>(value)); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates the size in bytes needed to encode an int64_t value as a varint |  | ||||||
|    * |  | ||||||
|    * @param value The int64_t value to calculate size for |  | ||||||
|    * @return The number of bytes needed to encode the value |  | ||||||
|    */ |  | ||||||
|   static inline uint32_t varint(int64_t value) { |  | ||||||
|     // For int64_t, we convert to uint64_t and calculate the size |  | ||||||
|     // This works because the bit pattern determines the encoding size, |  | ||||||
|     // and we've handled negative int32 values as a special case above |  | ||||||
|     return varint(static_cast<uint64_t>(value)); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates the size in bytes needed to encode a field ID and wire type |  | ||||||
|    * |  | ||||||
|    * @param field_id The field identifier |  | ||||||
|    * @param type The wire type value (from the WireType enum in the protobuf spec) |  | ||||||
|    * @return The number of bytes needed to encode the field ID and wire type |  | ||||||
|    */ |  | ||||||
|   static inline uint32_t field(uint32_t field_id, uint32_t type) { |  | ||||||
|     uint32_t tag = (field_id << 3) | (type & 0b111); |  | ||||||
|     return varint(tag); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Common parameters for all add_*_field methods |  | ||||||
|    * |  | ||||||
|    * All add_*_field methods follow these common patterns: |  | ||||||
|    * |  | ||||||
|    * @param total_size Reference to the total message size to update |  | ||||||
|    * @param field_id_size Pre-calculated size of the field ID in bytes |  | ||||||
|    * @param value The value to calculate size for (type varies) |  | ||||||
|    * @param force Whether to calculate size even if the value is default/zero/empty |  | ||||||
|    * |  | ||||||
|    * Each method follows this implementation pattern: |  | ||||||
|    * 1. Skip calculation if value is default (0, false, empty) and not forced |  | ||||||
|    * 2. Calculate the size based on the field's encoding rules |  | ||||||
|    * 3. Add the field_id_size + calculated value size to total_size |  | ||||||
|    */ |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of an int32 field to the total message size |  | ||||||
|    */ |  | ||||||
|   static inline void add_int32_field(uint32_t &total_size, uint32_t field_id_size, int32_t value, bool force = false) { |  | ||||||
|     // Skip calculation if value is zero and not forced |  | ||||||
|     if (value == 0 && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Calculate and directly add to total_size |  | ||||||
|     if (value < 0) { |  | ||||||
|       // Negative values are encoded as 10-byte varints in protobuf |  | ||||||
|       total_size += field_id_size + 10; |  | ||||||
|     } else { |  | ||||||
|       // For non-negative values, use the standard varint size |  | ||||||
|       total_size += field_id_size + varint(static_cast<uint32_t>(value)); |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a uint32 field to the total message size |  | ||||||
|    */ |  | ||||||
|   static inline void add_uint32_field(uint32_t &total_size, uint32_t field_id_size, uint32_t value, |  | ||||||
|                                       bool force = false) { |  | ||||||
|     // Skip calculation if value is zero and not forced |  | ||||||
|     if (value == 0 && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Calculate and directly add to total_size |  | ||||||
|     total_size += field_id_size + varint(value); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a boolean field to the total message size |  | ||||||
|    */ |  | ||||||
|   static inline void add_bool_field(uint32_t &total_size, uint32_t field_id_size, bool value, bool force = false) { |  | ||||||
|     // Skip calculation if value is false and not forced |  | ||||||
|     if (!value && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Boolean fields always use 1 byte when true |  | ||||||
|     total_size += field_id_size + 1; |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a fixed field to the total message size |  | ||||||
|    * |  | ||||||
|    * Fixed fields always take exactly N bytes (4 for fixed32/float, 8 for fixed64/double). |  | ||||||
|    * |  | ||||||
|    * @tparam NumBytes The number of bytes for this fixed field (4 or 8) |  | ||||||
|    * @param is_nonzero Whether the value is non-zero |  | ||||||
|    */ |  | ||||||
|   template<uint32_t NumBytes> |  | ||||||
|   static inline void add_fixed_field(uint32_t &total_size, uint32_t field_id_size, bool is_nonzero, |  | ||||||
|                                      bool force = false) { |  | ||||||
|     // Skip calculation if value is zero and not forced |  | ||||||
|     if (!is_nonzero && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Fixed fields always take exactly NumBytes |  | ||||||
|     total_size += field_id_size + NumBytes; |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of an enum field to the total message size |  | ||||||
|    * |  | ||||||
|    * Enum fields are encoded as uint32 varints. |  | ||||||
|    */ |  | ||||||
|   static inline void add_enum_field(uint32_t &total_size, uint32_t field_id_size, uint32_t value, bool force = false) { |  | ||||||
|     // Skip calculation if value is zero and not forced |  | ||||||
|     if (value == 0 && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Enums are encoded as uint32 |  | ||||||
|     total_size += field_id_size + varint(value); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a sint32 field to the total message size |  | ||||||
|    * |  | ||||||
|    * Sint32 fields use ZigZag encoding, which is more efficient for negative values. |  | ||||||
|    */ |  | ||||||
|   static inline void add_sint32_field(uint32_t &total_size, uint32_t field_id_size, int32_t value, bool force = false) { |  | ||||||
|     // Skip calculation if value is zero and not forced |  | ||||||
|     if (value == 0 && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // ZigZag encoding for sint32: (n << 1) ^ (n >> 31) |  | ||||||
|     uint32_t zigzag = (static_cast<uint32_t>(value) << 1) ^ (static_cast<uint32_t>(value >> 31)); |  | ||||||
|     total_size += field_id_size + varint(zigzag); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of an int64 field to the total message size |  | ||||||
|    */ |  | ||||||
|   static inline void add_int64_field(uint32_t &total_size, uint32_t field_id_size, int64_t value, bool force = false) { |  | ||||||
|     // Skip calculation if value is zero and not forced |  | ||||||
|     if (value == 0 && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Calculate and directly add to total_size |  | ||||||
|     total_size += field_id_size + varint(value); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a uint64 field to the total message size |  | ||||||
|    */ |  | ||||||
|   static inline void add_uint64_field(uint32_t &total_size, uint32_t field_id_size, uint64_t value, |  | ||||||
|                                       bool force = false) { |  | ||||||
|     // Skip calculation if value is zero and not forced |  | ||||||
|     if (value == 0 && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Calculate and directly add to total_size |  | ||||||
|     total_size += field_id_size + varint(value); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a sint64 field to the total message size |  | ||||||
|    * |  | ||||||
|    * Sint64 fields use ZigZag encoding, which is more efficient for negative values. |  | ||||||
|    */ |  | ||||||
|   static inline void add_sint64_field(uint32_t &total_size, uint32_t field_id_size, int64_t value, bool force = false) { |  | ||||||
|     // Skip calculation if value is zero and not forced |  | ||||||
|     if (value == 0 && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // ZigZag encoding for sint64: (n << 1) ^ (n >> 63) |  | ||||||
|     uint64_t zigzag = (static_cast<uint64_t>(value) << 1) ^ (static_cast<uint64_t>(value >> 63)); |  | ||||||
|     total_size += field_id_size + varint(zigzag); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a string/bytes field to the total message size |  | ||||||
|    */ |  | ||||||
|   static inline void add_string_field(uint32_t &total_size, uint32_t field_id_size, const std::string &str, |  | ||||||
|                                       bool force = false) { |  | ||||||
|     // Skip calculation if string is empty and not forced |  | ||||||
|     if (str.empty() && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Calculate and directly add to total_size |  | ||||||
|     const uint32_t str_size = static_cast<uint32_t>(str.size()); |  | ||||||
|     total_size += field_id_size + varint(str_size) + str_size; |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a nested message field to the total message size |  | ||||||
|    * |  | ||||||
|    * This helper function directly updates the total_size reference if the nested size |  | ||||||
|    * is greater than zero or force is true. |  | ||||||
|    * |  | ||||||
|    * @param nested_size The pre-calculated size of the nested message |  | ||||||
|    */ |  | ||||||
|   static inline void add_message_field(uint32_t &total_size, uint32_t field_id_size, uint32_t nested_size, |  | ||||||
|                                        bool force = false) { |  | ||||||
|     // Skip calculation if nested message is empty and not forced |  | ||||||
|     if (nested_size == 0 && !force) { |  | ||||||
|       return;  // No need to update total_size |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // Calculate and directly add to total_size |  | ||||||
|     // Field ID + length varint + nested message content |  | ||||||
|     total_size += field_id_size + varint(nested_size) + nested_size; |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the size of a nested message field to the total message size |  | ||||||
|    * |  | ||||||
|    * This templated version directly takes a message object, calculates its size internally, |  | ||||||
|    * and updates the total_size reference. This eliminates the need for a temporary variable |  | ||||||
|    * at the call site. |  | ||||||
|    * |  | ||||||
|    * @tparam MessageType The type of the nested message (inferred from parameter) |  | ||||||
|    * @param message The nested message object |  | ||||||
|    */ |  | ||||||
|   template<typename MessageType> |  | ||||||
|   static inline void add_message_object(uint32_t &total_size, uint32_t field_id_size, const MessageType &message, |  | ||||||
|                                         bool force = false) { |  | ||||||
|     uint32_t nested_size = 0; |  | ||||||
|     message.calculate_size(nested_size); |  | ||||||
|  |  | ||||||
|     // Use the base implementation with the calculated nested_size |  | ||||||
|     add_message_field(total_size, field_id_size, nested_size, force); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   /** |  | ||||||
|    * @brief Calculates and adds the sizes of all messages in a repeated field to the total message size |  | ||||||
|    * |  | ||||||
|    * This helper processes a vector of message objects, calculating the size for each message |  | ||||||
|    * and adding it to the total size. |  | ||||||
|    * |  | ||||||
|    * @tparam MessageType The type of the nested messages in the vector |  | ||||||
|    * @param messages Vector of message objects |  | ||||||
|    */ |  | ||||||
|   template<typename MessageType> |  | ||||||
|   static inline void add_repeated_message(uint32_t &total_size, uint32_t field_id_size, |  | ||||||
|                                           const std::vector<MessageType> &messages) { |  | ||||||
|     // Skip if the vector is empty |  | ||||||
|     if (messages.empty()) { |  | ||||||
|       return; |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // For repeated fields, always use force=true |  | ||||||
|     for (const auto &message : messages) { |  | ||||||
|       add_message_object(total_size, field_id_size, message, true); |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
| }; |  | ||||||
|  |  | ||||||
| }  // namespace api |  | ||||||
| }  // namespace esphome |  | ||||||
| @@ -47,6 +47,11 @@ void APIServer::setup() { | |||||||
|   } |   } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|  |   // Schedule reboot if no clients connect within timeout | ||||||
|  |   if (this->reboot_timeout_ != 0) { | ||||||
|  |     this->schedule_reboot_timeout_(); | ||||||
|  |   } | ||||||
|  |  | ||||||
|   this->socket_ = socket::socket_ip_loop_monitored(SOCK_STREAM, 0);  // monitored for incoming connections |   this->socket_ = socket::socket_ip_loop_monitored(SOCK_STREAM, 0);  // monitored for incoming connections | ||||||
|   if (this->socket_ == nullptr) { |   if (this->socket_ == nullptr) { | ||||||
|     ESP_LOGW(TAG, "Could not create socket"); |     ESP_LOGW(TAG, "Could not create socket"); | ||||||
| @@ -91,7 +96,8 @@ void APIServer::setup() { | |||||||
|  |  | ||||||
| #ifdef USE_LOGGER | #ifdef USE_LOGGER | ||||||
|   if (logger::global_logger != nullptr) { |   if (logger::global_logger != nullptr) { | ||||||
|     logger::global_logger->add_on_log_callback([this](int level, const char *tag, const char *message) { |     logger::global_logger->add_on_log_callback( | ||||||
|  |         [this](int level, const char *tag, const char *message, size_t message_len) { | ||||||
|           if (this->shutting_down_) { |           if (this->shutting_down_) { | ||||||
|             // Don't try to send logs during shutdown |             // Don't try to send logs during shutdown | ||||||
|             // as it could result in a recursion and |             // as it could result in a recursion and | ||||||
| @@ -99,21 +105,18 @@ void APIServer::setup() { | |||||||
|             return; |             return; | ||||||
|           } |           } | ||||||
|           for (auto &c : this->clients_) { |           for (auto &c : this->clients_) { | ||||||
|         if (!c->remove_) |             if (!c->flags_.remove) | ||||||
|           c->try_send_log_message(level, tag, message); |               c->try_send_log_message(level, tag, message, message_len); | ||||||
|           } |           } | ||||||
|         }); |         }); | ||||||
|   } |   } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|   this->last_connected_ = millis(); | #ifdef USE_CAMERA | ||||||
|  |   if (camera::Camera::instance() != nullptr && !camera::Camera::instance()->is_internal()) { | ||||||
| #ifdef USE_ESP32_CAMERA |     camera::Camera::instance()->add_image_callback([this](const std::shared_ptr<camera::CameraImage> &image) { | ||||||
|   if (esp32_camera::global_esp32_camera != nullptr && !esp32_camera::global_esp32_camera->is_internal()) { |  | ||||||
|     esp32_camera::global_esp32_camera->add_image_callback( |  | ||||||
|         [this](const std::shared_ptr<esp32_camera::CameraImage> &image) { |  | ||||||
|       for (auto &c : this->clients_) { |       for (auto &c : this->clients_) { | ||||||
|             if (!c->remove_) |         if (!c->flags_.remove) | ||||||
|           c->set_camera_state(image); |           c->set_camera_state(image); | ||||||
|       } |       } | ||||||
|     }); |     }); | ||||||
| @@ -121,6 +124,16 @@ void APIServer::setup() { | |||||||
| #endif | #endif | ||||||
| } | } | ||||||
|  |  | ||||||
|  | void APIServer::schedule_reboot_timeout_() { | ||||||
|  |   this->status_set_warning(); | ||||||
|  |   this->set_timeout("api_reboot", this->reboot_timeout_, []() { | ||||||
|  |     if (!global_api_server->is_connected()) { | ||||||
|  |       ESP_LOGE(TAG, "No clients; rebooting"); | ||||||
|  |       App.reboot(); | ||||||
|  |     } | ||||||
|  |   }); | ||||||
|  | } | ||||||
|  |  | ||||||
| void APIServer::loop() { | void APIServer::loop() { | ||||||
|   // Accept new clients only if the socket exists and has incoming connections |   // Accept new clients only if the socket exists and has incoming connections | ||||||
|   if (this->socket_ && this->socket_->ready()) { |   if (this->socket_ && this->socket_->ready()) { | ||||||
| @@ -130,51 +143,63 @@ void APIServer::loop() { | |||||||
|       auto sock = this->socket_->accept_loop_monitored((struct sockaddr *) &source_addr, &addr_len); |       auto sock = this->socket_->accept_loop_monitored((struct sockaddr *) &source_addr, &addr_len); | ||||||
|       if (!sock) |       if (!sock) | ||||||
|         break; |         break; | ||||||
|       ESP_LOGD(TAG, "Accepted %s", sock->getpeername().c_str()); |       ESP_LOGD(TAG, "Accept %s", sock->getpeername().c_str()); | ||||||
|  |  | ||||||
|       auto *conn = new APIConnection(std::move(sock), this); |       auto *conn = new APIConnection(std::move(sock), this); | ||||||
|       this->clients_.emplace_back(conn); |       this->clients_.emplace_back(conn); | ||||||
|       conn->start(); |       conn->start(); | ||||||
|  |  | ||||||
|  |       // Clear warning status and cancel reboot when first client connects | ||||||
|  |       if (this->clients_.size() == 1 && this->reboot_timeout_ != 0) { | ||||||
|  |         this->status_clear_warning(); | ||||||
|  |         this->cancel_timeout("api_reboot"); | ||||||
|  |       } | ||||||
|     } |     } | ||||||
|   } |   } | ||||||
|  |  | ||||||
|  |   if (this->clients_.empty()) { | ||||||
|  |     return; | ||||||
|  |   } | ||||||
|  |  | ||||||
|   // Process clients and remove disconnected ones in a single pass |   // Process clients and remove disconnected ones in a single pass | ||||||
|   if (!this->clients_.empty()) { |   // Check network connectivity once for all clients | ||||||
|  |   if (!network::is_connected()) { | ||||||
|  |     // Network is down - disconnect all clients | ||||||
|  |     for (auto &client : this->clients_) { | ||||||
|  |       client->on_fatal_error(); | ||||||
|  |       ESP_LOGW(TAG, "%s: Network down; disconnect", client->get_client_combined_info().c_str()); | ||||||
|  |     } | ||||||
|  |     // Continue to process and clean up the clients below | ||||||
|  |   } | ||||||
|  |  | ||||||
|   size_t client_index = 0; |   size_t client_index = 0; | ||||||
|   while (client_index < this->clients_.size()) { |   while (client_index < this->clients_.size()) { | ||||||
|     auto &client = this->clients_[client_index]; |     auto &client = this->clients_[client_index]; | ||||||
|  |  | ||||||
|       if (client->remove_) { |     if (!client->flags_.remove) { | ||||||
|         // Handle disconnection |       // Common case: process active client | ||||||
|  |       client->loop(); | ||||||
|  |       client_index++; | ||||||
|  |       continue; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Rare case: handle disconnection | ||||||
|  | #ifdef USE_API_CLIENT_DISCONNECTED_TRIGGER | ||||||
|     this->client_disconnected_trigger_->trigger(client->client_info_, client->client_peername_); |     this->client_disconnected_trigger_->trigger(client->client_info_, client->client_peername_); | ||||||
|         ESP_LOGV(TAG, "Removing connection to %s", client->client_info_.c_str()); | #endif | ||||||
|  |     ESP_LOGV(TAG, "Remove connection %s", client->client_info_.c_str()); | ||||||
|  |  | ||||||
|     // Swap with the last element and pop (avoids expensive vector shifts) |     // Swap with the last element and pop (avoids expensive vector shifts) | ||||||
|     if (client_index < this->clients_.size() - 1) { |     if (client_index < this->clients_.size() - 1) { | ||||||
|       std::swap(this->clients_[client_index], this->clients_.back()); |       std::swap(this->clients_[client_index], this->clients_.back()); | ||||||
|     } |     } | ||||||
|     this->clients_.pop_back(); |     this->clients_.pop_back(); | ||||||
|         // Don't increment client_index since we need to process the swapped element |  | ||||||
|       } else { |  | ||||||
|         // Process active client |  | ||||||
|         client->loop(); |  | ||||||
|         client_index++;  // Move to next client |  | ||||||
|       } |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   if (this->reboot_timeout_ != 0) { |     // Schedule reboot when last client disconnects | ||||||
|     const uint32_t now = millis(); |     if (this->clients_.empty() && this->reboot_timeout_ != 0) { | ||||||
|     if (!this->is_connected()) { |       this->schedule_reboot_timeout_(); | ||||||
|       if (now - this->last_connected_ > this->reboot_timeout_) { |  | ||||||
|         ESP_LOGE(TAG, "No client connected; rebooting"); |  | ||||||
|         App.reboot(); |  | ||||||
|       } |  | ||||||
|       this->status_set_warning(); |  | ||||||
|     } else { |  | ||||||
|       this->last_connected_ = now; |  | ||||||
|       this->status_clear_warning(); |  | ||||||
|     } |     } | ||||||
|  |     // Don't increment client_index since we need to process the swapped element | ||||||
|   } |   } | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -193,6 +218,7 @@ void APIServer::dump_config() { | |||||||
| #endif | #endif | ||||||
| } | } | ||||||
|  |  | ||||||
|  | #ifdef USE_API_PASSWORD | ||||||
| bool APIServer::uses_password() const { return !this->password_.empty(); } | bool APIServer::uses_password() const { return !this->password_.empty(); } | ||||||
|  |  | ||||||
| bool APIServer::check_password(const std::string &password) const { | bool APIServer::check_password(const std::string &password) const { | ||||||
| @@ -223,192 +249,129 @@ bool APIServer::check_password(const std::string &password) const { | |||||||
|  |  | ||||||
|   return result == 0; |   return result == 0; | ||||||
| } | } | ||||||
|  | #endif | ||||||
|  |  | ||||||
| void APIServer::handle_disconnect(APIConnection *conn) {} | void APIServer::handle_disconnect(APIConnection *conn) {} | ||||||
|  |  | ||||||
| #ifdef USE_BINARY_SENSOR | // Macro for entities without extra parameters | ||||||
| void APIServer::on_binary_sensor_update(binary_sensor::BinarySensor *obj, bool state) { | #define API_DISPATCH_UPDATE(entity_type, entity_name) \ | ||||||
|   if (obj->is_internal()) |   void APIServer::on_##entity_name##_update(entity_type *obj) { /* NOLINT(bugprone-macro-parentheses) */ \ | ||||||
|     return; |     if (obj->is_internal()) \ | ||||||
|   for (auto &c : this->clients_) |       return; \ | ||||||
|     c->send_binary_sensor_state(obj); |     for (auto &c : this->clients_) \ | ||||||
|  |       c->send_##entity_name##_state(obj); \ | ||||||
|   } |   } | ||||||
|  |  | ||||||
|  | // Macro for entities with extra parameters (but parameters not used in send) | ||||||
|  | #define API_DISPATCH_UPDATE_IGNORE_PARAMS(entity_type, entity_name, ...) \ | ||||||
|  |   void APIServer::on_##entity_name##_update(entity_type *obj, __VA_ARGS__) { /* NOLINT(bugprone-macro-parentheses) */ \ | ||||||
|  |     if (obj->is_internal()) \ | ||||||
|  |       return; \ | ||||||
|  |     for (auto &c : this->clients_) \ | ||||||
|  |       c->send_##entity_name##_state(obj); \ | ||||||
|  |   } | ||||||
|  |  | ||||||
|  | #ifdef USE_BINARY_SENSOR | ||||||
|  | API_DISPATCH_UPDATE(binary_sensor::BinarySensor, binary_sensor) | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_COVER | #ifdef USE_COVER | ||||||
| void APIServer::on_cover_update(cover::Cover *obj) { | API_DISPATCH_UPDATE(cover::Cover, cover) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_cover_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_FAN | #ifdef USE_FAN | ||||||
| void APIServer::on_fan_update(fan::Fan *obj) { | API_DISPATCH_UPDATE(fan::Fan, fan) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_fan_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_LIGHT | #ifdef USE_LIGHT | ||||||
| void APIServer::on_light_update(light::LightState *obj) { | API_DISPATCH_UPDATE(light::LightState, light) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_light_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_SENSOR | #ifdef USE_SENSOR | ||||||
| void APIServer::on_sensor_update(sensor::Sensor *obj, float state) { | API_DISPATCH_UPDATE_IGNORE_PARAMS(sensor::Sensor, sensor, float state) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_sensor_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_SWITCH | #ifdef USE_SWITCH | ||||||
| void APIServer::on_switch_update(switch_::Switch *obj, bool state) { | API_DISPATCH_UPDATE_IGNORE_PARAMS(switch_::Switch, switch, bool state) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_switch_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_TEXT_SENSOR | #ifdef USE_TEXT_SENSOR | ||||||
| void APIServer::on_text_sensor_update(text_sensor::TextSensor *obj, const std::string &state) { | API_DISPATCH_UPDATE_IGNORE_PARAMS(text_sensor::TextSensor, text_sensor, const std::string &state) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_text_sensor_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_CLIMATE | #ifdef USE_CLIMATE | ||||||
| void APIServer::on_climate_update(climate::Climate *obj) { | API_DISPATCH_UPDATE(climate::Climate, climate) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_climate_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_NUMBER | #ifdef USE_NUMBER | ||||||
| void APIServer::on_number_update(number::Number *obj, float state) { | API_DISPATCH_UPDATE_IGNORE_PARAMS(number::Number, number, float state) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_number_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_DATETIME_DATE | #ifdef USE_DATETIME_DATE | ||||||
| void APIServer::on_date_update(datetime::DateEntity *obj) { | API_DISPATCH_UPDATE(datetime::DateEntity, date) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_date_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_DATETIME_TIME | #ifdef USE_DATETIME_TIME | ||||||
| void APIServer::on_time_update(datetime::TimeEntity *obj) { | API_DISPATCH_UPDATE(datetime::TimeEntity, time) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_time_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_DATETIME_DATETIME | #ifdef USE_DATETIME_DATETIME | ||||||
| void APIServer::on_datetime_update(datetime::DateTimeEntity *obj) { | API_DISPATCH_UPDATE(datetime::DateTimeEntity, datetime) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_datetime_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_TEXT | #ifdef USE_TEXT | ||||||
| void APIServer::on_text_update(text::Text *obj, const std::string &state) { | API_DISPATCH_UPDATE_IGNORE_PARAMS(text::Text, text, const std::string &state) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_text_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_SELECT | #ifdef USE_SELECT | ||||||
| void APIServer::on_select_update(select::Select *obj, const std::string &state, size_t index) { | API_DISPATCH_UPDATE_IGNORE_PARAMS(select::Select, select, const std::string &state, size_t index) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_select_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_LOCK | #ifdef USE_LOCK | ||||||
| void APIServer::on_lock_update(lock::Lock *obj) { | API_DISPATCH_UPDATE(lock::Lock, lock) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_lock_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_VALVE | #ifdef USE_VALVE | ||||||
| void APIServer::on_valve_update(valve::Valve *obj) { | API_DISPATCH_UPDATE(valve::Valve, valve) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_valve_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_MEDIA_PLAYER | #ifdef USE_MEDIA_PLAYER | ||||||
| void APIServer::on_media_player_update(media_player::MediaPlayer *obj) { | API_DISPATCH_UPDATE(media_player::MediaPlayer, media_player) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_media_player_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_EVENT | #ifdef USE_EVENT | ||||||
|  | // Event is a special case - it's the only entity that passes extra parameters to the send method | ||||||
| void APIServer::on_event(event::Event *obj, const std::string &event_type) { | void APIServer::on_event(event::Event *obj, const std::string &event_type) { | ||||||
|  |   if (obj->is_internal()) | ||||||
|  |     return; | ||||||
|   for (auto &c : this->clients_) |   for (auto &c : this->clients_) | ||||||
|     c->send_event(obj, event_type); |     c->send_event(obj, event_type); | ||||||
| } | } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_UPDATE | #ifdef USE_UPDATE | ||||||
|  | // Update is a special case - the method is called on_update, not on_update_update | ||||||
| void APIServer::on_update(update::UpdateEntity *obj) { | void APIServer::on_update(update::UpdateEntity *obj) { | ||||||
|  |   if (obj->is_internal()) | ||||||
|  |     return; | ||||||
|   for (auto &c : this->clients_) |   for (auto &c : this->clients_) | ||||||
|     c->send_update_state(obj); |     c->send_update_state(obj); | ||||||
| } | } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #ifdef USE_ALARM_CONTROL_PANEL | #ifdef USE_ALARM_CONTROL_PANEL | ||||||
| void APIServer::on_alarm_control_panel_update(alarm_control_panel::AlarmControlPanel *obj) { | API_DISPATCH_UPDATE(alarm_control_panel::AlarmControlPanel, alarm_control_panel) | ||||||
|   if (obj->is_internal()) |  | ||||||
|     return; |  | ||||||
|   for (auto &c : this->clients_) |  | ||||||
|     c->send_alarm_control_panel_state(obj); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| float APIServer::get_setup_priority() const { return setup_priority::AFTER_WIFI; } | float APIServer::get_setup_priority() const { return setup_priority::AFTER_WIFI; } | ||||||
|  |  | ||||||
| void APIServer::set_port(uint16_t port) { this->port_ = port; } | void APIServer::set_port(uint16_t port) { this->port_ = port; } | ||||||
|  |  | ||||||
|  | #ifdef USE_API_PASSWORD | ||||||
| void APIServer::set_password(const std::string &password) { this->password_ = password; } | void APIServer::set_password(const std::string &password) { this->password_ = password; } | ||||||
|  | #endif | ||||||
|  |  | ||||||
| void APIServer::set_batch_delay(uint32_t batch_delay) { this->batch_delay_ = batch_delay; } | void APIServer::set_batch_delay(uint16_t batch_delay) { this->batch_delay_ = batch_delay; } | ||||||
|  |  | ||||||
| void APIServer::send_homeassistant_service_call(const HomeassistantServiceResponse &call) { | void APIServer::send_homeassistant_service_call(const HomeassistantServiceResponse &call) { | ||||||
|   for (auto &client : this->clients_) { |   for (auto &client : this->clients_) { | ||||||
| @@ -479,7 +442,7 @@ bool APIServer::save_noise_psk(psk_t psk, bool make_active) { | |||||||
| #ifdef USE_HOMEASSISTANT_TIME | #ifdef USE_HOMEASSISTANT_TIME | ||||||
| void APIServer::request_time() { | void APIServer::request_time() { | ||||||
|   for (auto &client : this->clients_) { |   for (auto &client : this->clients_) { | ||||||
|     if (!client->remove_ && client->is_authenticated()) |     if (!client->flags_.remove && client->is_authenticated()) | ||||||
|       client->send_time_request(); |       client->send_time_request(); | ||||||
|   } |   } | ||||||
| } | } | ||||||
| @@ -503,8 +466,9 @@ void APIServer::on_shutdown() { | |||||||
|   for (auto &c : this->clients_) { |   for (auto &c : this->clients_) { | ||||||
|     if (!c->send_message(DisconnectRequest())) { |     if (!c->send_message(DisconnectRequest())) { | ||||||
|       // If we can't send the disconnect request directly (tx_buffer full), |       // If we can't send the disconnect request directly (tx_buffer full), | ||||||
|       // schedule it in the batch so it will be sent with the 5ms timer |       // schedule it at the front of the batch so it will be sent with priority | ||||||
|       c->schedule_message_(nullptr, &APIConnection::try_send_disconnect_request, DisconnectRequest::MESSAGE_TYPE); |       c->schedule_message_front_(nullptr, &APIConnection::try_send_disconnect_request, DisconnectRequest::MESSAGE_TYPE, | ||||||
|  |                                  DisconnectRequest::ESTIMATED_SIZE); | ||||||
|     } |     } | ||||||
|   } |   } | ||||||
| } | } | ||||||
|   | |||||||
| @@ -12,7 +12,9 @@ | |||||||
| #include "esphome/core/log.h" | #include "esphome/core/log.h" | ||||||
| #include "list_entities.h" | #include "list_entities.h" | ||||||
| #include "subscribe_state.h" | #include "subscribe_state.h" | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
| #include "user_services.h" | #include "user_services.h" | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #include <vector> | #include <vector> | ||||||
|  |  | ||||||
| @@ -35,13 +37,15 @@ class APIServer : public Component, public Controller { | |||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   void on_shutdown() override; |   void on_shutdown() override; | ||||||
|   bool teardown() override; |   bool teardown() override; | ||||||
|  | #ifdef USE_API_PASSWORD | ||||||
|   bool check_password(const std::string &password) const; |   bool check_password(const std::string &password) const; | ||||||
|   bool uses_password() const; |   bool uses_password() const; | ||||||
|   void set_port(uint16_t port); |  | ||||||
|   void set_password(const std::string &password); |   void set_password(const std::string &password); | ||||||
|  | #endif | ||||||
|  |   void set_port(uint16_t port); | ||||||
|   void set_reboot_timeout(uint32_t reboot_timeout); |   void set_reboot_timeout(uint32_t reboot_timeout); | ||||||
|   void set_batch_delay(uint32_t batch_delay); |   void set_batch_delay(uint16_t batch_delay); | ||||||
|   uint32_t get_batch_delay() const { return batch_delay_; } |   uint16_t get_batch_delay() const { return batch_delay_; } | ||||||
|  |  | ||||||
|   // Get reference to shared buffer for API connections |   // Get reference to shared buffer for API connections | ||||||
|   std::vector<uint8_t> &get_shared_buffer_ref() { return shared_write_buffer_; } |   std::vector<uint8_t> &get_shared_buffer_ref() { return shared_write_buffer_; } | ||||||
| @@ -54,7 +58,7 @@ class APIServer : public Component, public Controller { | |||||||
|  |  | ||||||
|   void handle_disconnect(APIConnection *conn); |   void handle_disconnect(APIConnection *conn); | ||||||
| #ifdef USE_BINARY_SENSOR | #ifdef USE_BINARY_SENSOR | ||||||
|   void on_binary_sensor_update(binary_sensor::BinarySensor *obj, bool state) override; |   void on_binary_sensor_update(binary_sensor::BinarySensor *obj) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_COVER | #ifdef USE_COVER | ||||||
|   void on_cover_update(cover::Cover *obj) override; |   void on_cover_update(cover::Cover *obj) override; | ||||||
| @@ -105,7 +109,9 @@ class APIServer : public Component, public Controller { | |||||||
|   void on_media_player_update(media_player::MediaPlayer *obj) override; |   void on_media_player_update(media_player::MediaPlayer *obj) override; | ||||||
| #endif | #endif | ||||||
|   void send_homeassistant_service_call(const HomeassistantServiceResponse &call); |   void send_homeassistant_service_call(const HomeassistantServiceResponse &call); | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   void register_user_service(UserServiceDescriptor *descriptor) { this->user_services_.push_back(descriptor); } |   void register_user_service(UserServiceDescriptor *descriptor) { this->user_services_.push_back(descriptor); } | ||||||
|  | #endif | ||||||
| #ifdef USE_HOMEASSISTANT_TIME | #ifdef USE_HOMEASSISTANT_TIME | ||||||
|   void request_time(); |   void request_time(); | ||||||
| #endif | #endif | ||||||
| @@ -134,27 +140,49 @@ class APIServer : public Component, public Controller { | |||||||
|   void get_home_assistant_state(std::string entity_id, optional<std::string> attribute, |   void get_home_assistant_state(std::string entity_id, optional<std::string> attribute, | ||||||
|                                 std::function<void(std::string)> f); |                                 std::function<void(std::string)> f); | ||||||
|   const std::vector<HomeAssistantStateSubscription> &get_state_subs() const; |   const std::vector<HomeAssistantStateSubscription> &get_state_subs() const; | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   const std::vector<UserServiceDescriptor *> &get_user_services() const { return this->user_services_; } |   const std::vector<UserServiceDescriptor *> &get_user_services() const { return this->user_services_; } | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | #ifdef USE_API_CLIENT_CONNECTED_TRIGGER | ||||||
|   Trigger<std::string, std::string> *get_client_connected_trigger() const { return this->client_connected_trigger_; } |   Trigger<std::string, std::string> *get_client_connected_trigger() const { return this->client_connected_trigger_; } | ||||||
|  | #endif | ||||||
|  | #ifdef USE_API_CLIENT_DISCONNECTED_TRIGGER | ||||||
|   Trigger<std::string, std::string> *get_client_disconnected_trigger() const { |   Trigger<std::string, std::string> *get_client_disconnected_trigger() const { | ||||||
|     return this->client_disconnected_trigger_; |     return this->client_disconnected_trigger_; | ||||||
|   } |   } | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   bool shutting_down_ = false; |   void schedule_reboot_timeout_(); | ||||||
|  |   // Pointers and pointer-like types first (4 bytes each) | ||||||
|   std::unique_ptr<socket::Socket> socket_ = nullptr; |   std::unique_ptr<socket::Socket> socket_ = nullptr; | ||||||
|   uint16_t port_{6053}; | #ifdef USE_API_CLIENT_CONNECTED_TRIGGER | ||||||
|  |   Trigger<std::string, std::string> *client_connected_trigger_ = new Trigger<std::string, std::string>(); | ||||||
|  | #endif | ||||||
|  | #ifdef USE_API_CLIENT_DISCONNECTED_TRIGGER | ||||||
|  |   Trigger<std::string, std::string> *client_disconnected_trigger_ = new Trigger<std::string, std::string>(); | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |   // 4-byte aligned types | ||||||
|   uint32_t reboot_timeout_{300000}; |   uint32_t reboot_timeout_{300000}; | ||||||
|   uint32_t batch_delay_{100}; |  | ||||||
|   uint32_t last_connected_{0}; |   // Vectors and strings (12 bytes each on 32-bit) | ||||||
|   std::vector<std::unique_ptr<APIConnection>> clients_; |   std::vector<std::unique_ptr<APIConnection>> clients_; | ||||||
|  | #ifdef USE_API_PASSWORD | ||||||
|   std::string password_; |   std::string password_; | ||||||
|  | #endif | ||||||
|   std::vector<uint8_t> shared_write_buffer_;  // Shared proto write buffer for all connections |   std::vector<uint8_t> shared_write_buffer_;  // Shared proto write buffer for all connections | ||||||
|   std::vector<HomeAssistantStateSubscription> state_subs_; |   std::vector<HomeAssistantStateSubscription> state_subs_; | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   std::vector<UserServiceDescriptor *> user_services_; |   std::vector<UserServiceDescriptor *> user_services_; | ||||||
|   Trigger<std::string, std::string> *client_connected_trigger_ = new Trigger<std::string, std::string>(); | #endif | ||||||
|   Trigger<std::string, std::string> *client_disconnected_trigger_ = new Trigger<std::string, std::string>(); |  | ||||||
|  |   // Group smaller types together | ||||||
|  |   uint16_t port_{6053}; | ||||||
|  |   uint16_t batch_delay_{100}; | ||||||
|  |   bool shutting_down_ = false; | ||||||
|  |   // 5 bytes used, 3 bytes padding | ||||||
|  |  | ||||||
| #ifdef USE_API_NOISE | #ifdef USE_API_NOISE | ||||||
|   std::shared_ptr<APINoiseContext> noise_ctx_ = std::make_shared<APINoiseContext>(); |   std::shared_ptr<APINoiseContext> noise_ctx_ = std::make_shared<APINoiseContext>(); | ||||||
|   | |||||||
| @@ -4,7 +4,13 @@ import asyncio | |||||||
| from datetime import datetime | from datetime import datetime | ||||||
| import logging | import logging | ||||||
| from typing import TYPE_CHECKING, Any | from typing import TYPE_CHECKING, Any | ||||||
|  | import warnings | ||||||
|  |  | ||||||
|  | # Suppress protobuf version warnings | ||||||
|  | with warnings.catch_warnings(): | ||||||
|  |     warnings.filterwarnings( | ||||||
|  |         "ignore", category=UserWarning, message=".*Protobuf gencode version.*" | ||||||
|  |     ) | ||||||
|     from aioesphomeapi import APIClient, parse_log_message |     from aioesphomeapi import APIClient, parse_log_message | ||||||
|     from aioesphomeapi.log_runner import async_run |     from aioesphomeapi.log_runner import async_run | ||||||
|  |  | ||||||
|   | |||||||
| @@ -3,10 +3,13 @@ | |||||||
| #include <map> | #include <map> | ||||||
| #include "api_server.h" | #include "api_server.h" | ||||||
| #ifdef USE_API | #ifdef USE_API | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
| #include "user_services.h" | #include "user_services.h" | ||||||
|  | #endif | ||||||
| namespace esphome { | namespace esphome { | ||||||
| namespace api { | namespace api { | ||||||
|  |  | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
| template<typename T, typename... Ts> class CustomAPIDeviceService : public UserServiceBase<Ts...> { | template<typename T, typename... Ts> class CustomAPIDeviceService : public UserServiceBase<Ts...> { | ||||||
|  public: |  public: | ||||||
|   CustomAPIDeviceService(const std::string &name, const std::array<std::string, sizeof...(Ts)> &arg_names, T *obj, |   CustomAPIDeviceService(const std::string &name, const std::array<std::string, sizeof...(Ts)> &arg_names, T *obj, | ||||||
| @@ -19,6 +22,7 @@ template<typename T, typename... Ts> class CustomAPIDeviceService : public UserS | |||||||
|   T *obj_; |   T *obj_; | ||||||
|   void (T::*callback_)(Ts...); |   void (T::*callback_)(Ts...); | ||||||
| }; | }; | ||||||
|  | #endif  // USE_API_SERVICES | ||||||
|  |  | ||||||
| class CustomAPIDevice { | class CustomAPIDevice { | ||||||
|  public: |  public: | ||||||
| @@ -46,12 +50,14 @@ class CustomAPIDevice { | |||||||
|    * @param name The name of the service to register. |    * @param name The name of the service to register. | ||||||
|    * @param arg_names The name of the arguments for the service, must match the arguments of the function. |    * @param arg_names The name of the arguments for the service, must match the arguments of the function. | ||||||
|    */ |    */ | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   template<typename T, typename... Ts> |   template<typename T, typename... Ts> | ||||||
|   void register_service(void (T::*callback)(Ts...), const std::string &name, |   void register_service(void (T::*callback)(Ts...), const std::string &name, | ||||||
|                         const std::array<std::string, sizeof...(Ts)> &arg_names) { |                         const std::array<std::string, sizeof...(Ts)> &arg_names) { | ||||||
|     auto *service = new CustomAPIDeviceService<T, Ts...>(name, arg_names, (T *) this, callback);  // NOLINT |     auto *service = new CustomAPIDeviceService<T, Ts...>(name, arg_names, (T *) this, callback);  // NOLINT | ||||||
|     global_api_server->register_user_service(service); |     global_api_server->register_user_service(service); | ||||||
|   } |   } | ||||||
|  | #endif | ||||||
|  |  | ||||||
|   /** Register a custom native API service that will show up in Home Assistant. |   /** Register a custom native API service that will show up in Home Assistant. | ||||||
|    * |    * | ||||||
| @@ -71,10 +77,12 @@ class CustomAPIDevice { | |||||||
|    * @param callback The member function to call when the service is triggered. |    * @param callback The member function to call when the service is triggered. | ||||||
|    * @param name The name of the arguments for the service, must match the arguments of the function. |    * @param name The name of the arguments for the service, must match the arguments of the function. | ||||||
|    */ |    */ | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   template<typename T> void register_service(void (T::*callback)(), const std::string &name) { |   template<typename T> void register_service(void (T::*callback)(), const std::string &name) { | ||||||
|     auto *service = new CustomAPIDeviceService<T>(name, {}, (T *) this, callback);  // NOLINT |     auto *service = new CustomAPIDeviceService<T>(name, {}, (T *) this, callback);  // NOLINT | ||||||
|     global_api_server->register_user_service(service); |     global_api_server->register_user_service(service); | ||||||
|   } |   } | ||||||
|  | #endif | ||||||
|  |  | ||||||
|   /** Subscribe to the state (or attribute state) of an entity from Home Assistant. |   /** Subscribe to the state (or attribute state) of an entity from Home Assistant. | ||||||
|    * |    * | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| #include "list_entities.h" | #include "list_entities.h" | ||||||
| #ifdef USE_API | #ifdef USE_API | ||||||
| #include "api_connection.h" | #include "api_connection.h" | ||||||
|  | #include "api_pb2.h" | ||||||
| #include "esphome/core/application.h" | #include "esphome/core/application.h" | ||||||
| #include "esphome/core/log.h" | #include "esphome/core/log.h" | ||||||
| #include "esphome/core/util.h" | #include "esphome/core/util.h" | ||||||
| @@ -8,153 +9,85 @@ | |||||||
| namespace esphome { | namespace esphome { | ||||||
| namespace api { | namespace api { | ||||||
|  |  | ||||||
|  | // Generate entity handler implementations using macros | ||||||
| #ifdef USE_BINARY_SENSOR | #ifdef USE_BINARY_SENSOR | ||||||
| bool ListEntitiesIterator::on_binary_sensor(binary_sensor::BinarySensor *binary_sensor) { | LIST_ENTITIES_HANDLER(binary_sensor, binary_sensor::BinarySensor, ListEntitiesBinarySensorResponse) | ||||||
|   this->client_->send_binary_sensor_info(binary_sensor); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_COVER | #ifdef USE_COVER | ||||||
| bool ListEntitiesIterator::on_cover(cover::Cover *cover) { | LIST_ENTITIES_HANDLER(cover, cover::Cover, ListEntitiesCoverResponse) | ||||||
|   this->client_->send_cover_info(cover); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_FAN | #ifdef USE_FAN | ||||||
| bool ListEntitiesIterator::on_fan(fan::Fan *fan) { | LIST_ENTITIES_HANDLER(fan, fan::Fan, ListEntitiesFanResponse) | ||||||
|   this->client_->send_fan_info(fan); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LIGHT | #ifdef USE_LIGHT | ||||||
| bool ListEntitiesIterator::on_light(light::LightState *light) { | LIST_ENTITIES_HANDLER(light, light::LightState, ListEntitiesLightResponse) | ||||||
|   this->client_->send_light_info(light); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SENSOR | #ifdef USE_SENSOR | ||||||
| bool ListEntitiesIterator::on_sensor(sensor::Sensor *sensor) { | LIST_ENTITIES_HANDLER(sensor, sensor::Sensor, ListEntitiesSensorResponse) | ||||||
|   this->client_->send_sensor_info(sensor); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SWITCH | #ifdef USE_SWITCH | ||||||
| bool ListEntitiesIterator::on_switch(switch_::Switch *a_switch) { | LIST_ENTITIES_HANDLER(switch, switch_::Switch, ListEntitiesSwitchResponse) | ||||||
|   this->client_->send_switch_info(a_switch); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_BUTTON | #ifdef USE_BUTTON | ||||||
| bool ListEntitiesIterator::on_button(button::Button *button) { | LIST_ENTITIES_HANDLER(button, button::Button, ListEntitiesButtonResponse) | ||||||
|   this->client_->send_button_info(button); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT_SENSOR | #ifdef USE_TEXT_SENSOR | ||||||
| bool ListEntitiesIterator::on_text_sensor(text_sensor::TextSensor *text_sensor) { | LIST_ENTITIES_HANDLER(text_sensor, text_sensor::TextSensor, ListEntitiesTextSensorResponse) | ||||||
|   this->client_->send_text_sensor_info(text_sensor); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LOCK | #ifdef USE_LOCK | ||||||
| bool ListEntitiesIterator::on_lock(lock::Lock *a_lock) { | LIST_ENTITIES_HANDLER(lock, lock::Lock, ListEntitiesLockResponse) | ||||||
|   this->client_->send_lock_info(a_lock); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_VALVE | #ifdef USE_VALVE | ||||||
| bool ListEntitiesIterator::on_valve(valve::Valve *valve) { | LIST_ENTITIES_HANDLER(valve, valve::Valve, ListEntitiesValveResponse) | ||||||
|   this->client_->send_valve_info(valve); | #endif | ||||||
|   return true; | #ifdef USE_CAMERA | ||||||
| } | LIST_ENTITIES_HANDLER(camera, camera::Camera, ListEntitiesCameraResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_CLIMATE | ||||||
|  | LIST_ENTITIES_HANDLER(climate, climate::Climate, ListEntitiesClimateResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_NUMBER | ||||||
|  | LIST_ENTITIES_HANDLER(number, number::Number, ListEntitiesNumberResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_DATETIME_DATE | ||||||
|  | LIST_ENTITIES_HANDLER(date, datetime::DateEntity, ListEntitiesDateResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_DATETIME_TIME | ||||||
|  | LIST_ENTITIES_HANDLER(time, datetime::TimeEntity, ListEntitiesTimeResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_DATETIME_DATETIME | ||||||
|  | LIST_ENTITIES_HANDLER(datetime, datetime::DateTimeEntity, ListEntitiesDateTimeResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_TEXT | ||||||
|  | LIST_ENTITIES_HANDLER(text, text::Text, ListEntitiesTextResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_SELECT | ||||||
|  | LIST_ENTITIES_HANDLER(select, select::Select, ListEntitiesSelectResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_MEDIA_PLAYER | ||||||
|  | LIST_ENTITIES_HANDLER(media_player, media_player::MediaPlayer, ListEntitiesMediaPlayerResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_ALARM_CONTROL_PANEL | ||||||
|  | LIST_ENTITIES_HANDLER(alarm_control_panel, alarm_control_panel::AlarmControlPanel, | ||||||
|  |                       ListEntitiesAlarmControlPanelResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_EVENT | ||||||
|  | LIST_ENTITIES_HANDLER(event, event::Event, ListEntitiesEventResponse) | ||||||
|  | #endif | ||||||
|  | #ifdef USE_UPDATE | ||||||
|  | LIST_ENTITIES_HANDLER(update, update::UpdateEntity, ListEntitiesUpdateResponse) | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|  | // Special cases that don't follow the pattern | ||||||
| bool ListEntitiesIterator::on_end() { return this->client_->send_list_info_done(); } | bool ListEntitiesIterator::on_end() { return this->client_->send_list_info_done(); } | ||||||
|  |  | ||||||
| ListEntitiesIterator::ListEntitiesIterator(APIConnection *client) : client_(client) {} | ListEntitiesIterator::ListEntitiesIterator(APIConnection *client) : client_(client) {} | ||||||
|  |  | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
| bool ListEntitiesIterator::on_service(UserServiceDescriptor *service) { | bool ListEntitiesIterator::on_service(UserServiceDescriptor *service) { | ||||||
|   auto resp = service->encode_list_service_response(); |   auto resp = service->encode_list_service_response(); | ||||||
|   return this->client_->send_message(resp); |   return this->client_->send_message(resp); | ||||||
| } | } | ||||||
|  |  | ||||||
| #ifdef USE_ESP32_CAMERA |  | ||||||
| bool ListEntitiesIterator::on_camera(esp32_camera::ESP32Camera *camera) { |  | ||||||
|   this->client_->send_camera_info(camera); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef USE_CLIMATE |  | ||||||
| bool ListEntitiesIterator::on_climate(climate::Climate *climate) { |  | ||||||
|   this->client_->send_climate_info(climate); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef USE_NUMBER |  | ||||||
| bool ListEntitiesIterator::on_number(number::Number *number) { |  | ||||||
|   this->client_->send_number_info(number); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef USE_DATETIME_DATE |  | ||||||
| bool ListEntitiesIterator::on_date(datetime::DateEntity *date) { |  | ||||||
|   this->client_->send_date_info(date); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef USE_DATETIME_TIME |  | ||||||
| bool ListEntitiesIterator::on_time(datetime::TimeEntity *time) { |  | ||||||
|   this->client_->send_time_info(time); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef USE_DATETIME_DATETIME |  | ||||||
| bool ListEntitiesIterator::on_datetime(datetime::DateTimeEntity *datetime) { |  | ||||||
|   this->client_->send_datetime_info(datetime); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef USE_TEXT |  | ||||||
| bool ListEntitiesIterator::on_text(text::Text *text) { |  | ||||||
|   this->client_->send_text_info(text); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef USE_SELECT |  | ||||||
| bool ListEntitiesIterator::on_select(select::Select *select) { |  | ||||||
|   this->client_->send_select_info(select); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef USE_MEDIA_PLAYER |  | ||||||
| bool ListEntitiesIterator::on_media_player(media_player::MediaPlayer *media_player) { |  | ||||||
|   this->client_->send_media_player_info(media_player); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
| #ifdef USE_ALARM_CONTROL_PANEL |  | ||||||
| bool ListEntitiesIterator::on_alarm_control_panel(alarm_control_panel::AlarmControlPanel *a_alarm_control_panel) { |  | ||||||
|   this->client_->send_alarm_control_panel_info(a_alarm_control_panel); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
| #ifdef USE_EVENT |  | ||||||
| bool ListEntitiesIterator::on_event(event::Event *event) { |  | ||||||
|   this->client_->send_event_info(event); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
| #ifdef USE_UPDATE |  | ||||||
| bool ListEntitiesIterator::on_update(update::UpdateEntity *update) { |  | ||||||
|   this->client_->send_update_info(update); |  | ||||||
|   return true; |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| }  // namespace api | }  // namespace api | ||||||
|   | |||||||
| @@ -9,75 +9,85 @@ namespace api { | |||||||
|  |  | ||||||
| class APIConnection; | class APIConnection; | ||||||
|  |  | ||||||
|  | // Macro for generating ListEntitiesIterator handlers | ||||||
|  | // Calls schedule_message_ with try_send_*_info | ||||||
|  | #define LIST_ENTITIES_HANDLER(entity_type, EntityClass, ResponseType) \ | ||||||
|  |   bool ListEntitiesIterator::on_##entity_type(EntityClass *entity) { /* NOLINT(bugprone-macro-parentheses) */ \ | ||||||
|  |     return this->client_->schedule_message_(entity, &APIConnection::try_send_##entity_type##_info, \ | ||||||
|  |                                             ResponseType::MESSAGE_TYPE, ResponseType::ESTIMATED_SIZE); \ | ||||||
|  |   } | ||||||
|  |  | ||||||
| class ListEntitiesIterator : public ComponentIterator { | class ListEntitiesIterator : public ComponentIterator { | ||||||
|  public: |  public: | ||||||
|   ListEntitiesIterator(APIConnection *client); |   ListEntitiesIterator(APIConnection *client); | ||||||
| #ifdef USE_BINARY_SENSOR | #ifdef USE_BINARY_SENSOR | ||||||
|   bool on_binary_sensor(binary_sensor::BinarySensor *binary_sensor) override; |   bool on_binary_sensor(binary_sensor::BinarySensor *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_COVER | #ifdef USE_COVER | ||||||
|   bool on_cover(cover::Cover *cover) override; |   bool on_cover(cover::Cover *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_FAN | #ifdef USE_FAN | ||||||
|   bool on_fan(fan::Fan *fan) override; |   bool on_fan(fan::Fan *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LIGHT | #ifdef USE_LIGHT | ||||||
|   bool on_light(light::LightState *light) override; |   bool on_light(light::LightState *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SENSOR | #ifdef USE_SENSOR | ||||||
|   bool on_sensor(sensor::Sensor *sensor) override; |   bool on_sensor(sensor::Sensor *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SWITCH | #ifdef USE_SWITCH | ||||||
|   bool on_switch(switch_::Switch *a_switch) override; |   bool on_switch(switch_::Switch *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_BUTTON | #ifdef USE_BUTTON | ||||||
|   bool on_button(button::Button *button) override; |   bool on_button(button::Button *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT_SENSOR | #ifdef USE_TEXT_SENSOR | ||||||
|   bool on_text_sensor(text_sensor::TextSensor *text_sensor) override; |   bool on_text_sensor(text_sensor::TextSensor *entity) override; | ||||||
| #endif | #endif | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
|   bool on_service(UserServiceDescriptor *service) override; |   bool on_service(UserServiceDescriptor *service) override; | ||||||
| #ifdef USE_ESP32_CAMERA | #endif | ||||||
|   bool on_camera(esp32_camera::ESP32Camera *camera) override; | #ifdef USE_CAMERA | ||||||
|  |   bool on_camera(camera::Camera *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_CLIMATE | #ifdef USE_CLIMATE | ||||||
|   bool on_climate(climate::Climate *climate) override; |   bool on_climate(climate::Climate *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_NUMBER | #ifdef USE_NUMBER | ||||||
|   bool on_number(number::Number *number) override; |   bool on_number(number::Number *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_DATE | #ifdef USE_DATETIME_DATE | ||||||
|   bool on_date(datetime::DateEntity *date) override; |   bool on_date(datetime::DateEntity *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_TIME | #ifdef USE_DATETIME_TIME | ||||||
|   bool on_time(datetime::TimeEntity *time) override; |   bool on_time(datetime::TimeEntity *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_DATETIME | #ifdef USE_DATETIME_DATETIME | ||||||
|   bool on_datetime(datetime::DateTimeEntity *datetime) override; |   bool on_datetime(datetime::DateTimeEntity *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT | #ifdef USE_TEXT | ||||||
|   bool on_text(text::Text *text) override; |   bool on_text(text::Text *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SELECT | #ifdef USE_SELECT | ||||||
|   bool on_select(select::Select *select) override; |   bool on_select(select::Select *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LOCK | #ifdef USE_LOCK | ||||||
|   bool on_lock(lock::Lock *a_lock) override; |   bool on_lock(lock::Lock *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_VALVE | #ifdef USE_VALVE | ||||||
|   bool on_valve(valve::Valve *valve) override; |   bool on_valve(valve::Valve *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_MEDIA_PLAYER | #ifdef USE_MEDIA_PLAYER | ||||||
|   bool on_media_player(media_player::MediaPlayer *media_player) override; |   bool on_media_player(media_player::MediaPlayer *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_ALARM_CONTROL_PANEL | #ifdef USE_ALARM_CONTROL_PANEL | ||||||
|   bool on_alarm_control_panel(alarm_control_panel::AlarmControlPanel *a_alarm_control_panel) override; |   bool on_alarm_control_panel(alarm_control_panel::AlarmControlPanel *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_EVENT | #ifdef USE_EVENT | ||||||
|   bool on_event(event::Event *event) override; |   bool on_event(event::Event *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_UPDATE | #ifdef USE_UPDATE | ||||||
|   bool on_update(update::UpdateEntity *update) override; |   bool on_update(update::UpdateEntity *entity) override; | ||||||
| #endif | #endif | ||||||
|   bool on_end() override; |   bool on_end() override; | ||||||
|   bool completed() { return this->state_ == IteratorState::NONE; } |   bool completed() { return this->state_ == IteratorState::NONE; } | ||||||
|   | |||||||
| @@ -4,6 +4,7 @@ | |||||||
| #include "esphome/core/helpers.h" | #include "esphome/core/helpers.h" | ||||||
| #include "esphome/core/log.h" | #include "esphome/core/log.h" | ||||||
|  |  | ||||||
|  | #include <cassert> | ||||||
| #include <vector> | #include <vector> | ||||||
|  |  | ||||||
| #ifdef ESPHOME_LOG_HAS_VERY_VERBOSE | #ifdef ESPHOME_LOG_HAS_VERY_VERBOSE | ||||||
| @@ -59,7 +60,6 @@ class ProtoVarInt { | |||||||
|   uint32_t as_uint32() const { return this->value_; } |   uint32_t as_uint32() const { return this->value_; } | ||||||
|   uint64_t as_uint64() const { return this->value_; } |   uint64_t as_uint64() const { return this->value_; } | ||||||
|   bool as_bool() const { return this->value_; } |   bool as_bool() const { return this->value_; } | ||||||
|   template<typename T> T as_enum() const { return static_cast<T>(this->as_uint32()); } |  | ||||||
|   int32_t as_int32() const { |   int32_t as_int32() const { | ||||||
|     // Not ZigZag encoded |     // Not ZigZag encoded | ||||||
|     return static_cast<int32_t>(this->as_int64()); |     return static_cast<int32_t>(this->as_int64()); | ||||||
| @@ -133,15 +133,24 @@ class ProtoVarInt { | |||||||
|   uint64_t value_; |   uint64_t value_; | ||||||
| }; | }; | ||||||
|  |  | ||||||
|  | // Forward declaration for decode_to_message and encode_to_writer | ||||||
|  | class ProtoMessage; | ||||||
|  |  | ||||||
| class ProtoLengthDelimited { | class ProtoLengthDelimited { | ||||||
|  public: |  public: | ||||||
|   explicit ProtoLengthDelimited(const uint8_t *value, size_t length) : value_(value), length_(length) {} |   explicit ProtoLengthDelimited(const uint8_t *value, size_t length) : value_(value), length_(length) {} | ||||||
|   std::string as_string() const { return std::string(reinterpret_cast<const char *>(this->value_), this->length_); } |   std::string as_string() const { return std::string(reinterpret_cast<const char *>(this->value_), this->length_); } | ||||||
|   template<class C> C as_message() const { |  | ||||||
|     auto msg = C(); |   /** | ||||||
|     msg.decode(this->value_, this->length_); |    * Decode the length-delimited data into an existing ProtoMessage instance. | ||||||
|     return msg; |    * | ||||||
|   } |    * This method allows decoding without templates, enabling use in contexts | ||||||
|  |    * where the message type is not known at compile time. The ProtoMessage's | ||||||
|  |    * decode() method will be called with the raw data and length. | ||||||
|  |    * | ||||||
|  |    * @param msg The ProtoMessage instance to decode into | ||||||
|  |    */ | ||||||
|  |   void decode_to_message(ProtoMessage &msg) const; | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   const uint8_t *const value_; |   const uint8_t *const value_; | ||||||
| @@ -263,9 +272,6 @@ class ProtoWriteBuffer { | |||||||
|     this->write((value >> 48) & 0xFF); |     this->write((value >> 48) & 0xFF); | ||||||
|     this->write((value >> 56) & 0xFF); |     this->write((value >> 56) & 0xFF); | ||||||
|   } |   } | ||||||
|   template<typename T> void encode_enum(uint32_t field_id, T value, bool force = false) { |  | ||||||
|     this->encode_uint32(field_id, static_cast<uint32_t>(value), force); |  | ||||||
|   } |  | ||||||
|   void encode_float(uint32_t field_id, float value, bool force = false) { |   void encode_float(uint32_t field_id, float value, bool force = false) { | ||||||
|     if (value == 0.0f && !force) |     if (value == 0.0f && !force) | ||||||
|       return; |       return; | ||||||
| @@ -306,18 +312,7 @@ class ProtoWriteBuffer { | |||||||
|     } |     } | ||||||
|     this->encode_uint64(field_id, uvalue, force); |     this->encode_uint64(field_id, uvalue, force); | ||||||
|   } |   } | ||||||
|   template<class C> void encode_message(uint32_t field_id, const C &value, bool force = false) { |   void encode_message(uint32_t field_id, const ProtoMessage &value, bool force = false); | ||||||
|     this->encode_field_raw(field_id, 2);  // type 2: Length-delimited message |  | ||||||
|     size_t begin = this->buffer_->size(); |  | ||||||
|  |  | ||||||
|     value.encode(*this); |  | ||||||
|  |  | ||||||
|     const uint32_t nested_length = this->buffer_->size() - begin; |  | ||||||
|     // add size varint |  | ||||||
|     std::vector<uint8_t> var; |  | ||||||
|     ProtoVarInt(nested_length).encode(var); |  | ||||||
|     this->buffer_->insert(this->buffer_->begin() + begin, var.begin(), var.end()); |  | ||||||
|   } |  | ||||||
|   std::vector<uint8_t> *get_buffer() const { return buffer_; } |   std::vector<uint8_t> *get_buffer() const { return buffer_; } | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
| @@ -327,12 +322,15 @@ class ProtoWriteBuffer { | |||||||
| class ProtoMessage { | class ProtoMessage { | ||||||
|  public: |  public: | ||||||
|   virtual ~ProtoMessage() = default; |   virtual ~ProtoMessage() = default; | ||||||
|   virtual void encode(ProtoWriteBuffer buffer) const = 0; |   // Default implementation for messages with no fields | ||||||
|  |   virtual void encode(ProtoWriteBuffer buffer) const {} | ||||||
|   void decode(const uint8_t *buffer, size_t length); |   void decode(const uint8_t *buffer, size_t length); | ||||||
|   virtual void calculate_size(uint32_t &total_size) const = 0; |   // Default implementation for messages with no fields | ||||||
|  |   virtual void calculate_size(uint32_t &total_size) const {} | ||||||
| #ifdef HAS_PROTO_MESSAGE_DUMP | #ifdef HAS_PROTO_MESSAGE_DUMP | ||||||
|   std::string dump() const; |   std::string dump() const; | ||||||
|   virtual void dump_to(std::string &out) const = 0; |   virtual void dump_to(std::string &out) const = 0; | ||||||
|  |   virtual const char *message_name() const { return "unknown"; } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
| @@ -342,6 +340,494 @@ class ProtoMessage { | |||||||
|   virtual bool decode_64bit(uint32_t field_id, Proto64Bit value) { return false; } |   virtual bool decode_64bit(uint32_t field_id, Proto64Bit value) { return false; } | ||||||
| }; | }; | ||||||
|  |  | ||||||
|  | class ProtoSize { | ||||||
|  |  public: | ||||||
|  |   /** | ||||||
|  |    * @brief ProtoSize class for Protocol Buffer serialization size calculation | ||||||
|  |    * | ||||||
|  |    * This class provides static methods to calculate the exact byte counts needed | ||||||
|  |    * for encoding various Protocol Buffer field types. All methods are designed to be | ||||||
|  |    * efficient for the common case where many fields have default values. | ||||||
|  |    * | ||||||
|  |    * Implements Protocol Buffer encoding size calculation according to: | ||||||
|  |    * https://protobuf.dev/programming-guides/encoding/ | ||||||
|  |    * | ||||||
|  |    * Key features: | ||||||
|  |    * - Early-return optimization for zero/default values | ||||||
|  |    * - Direct total_size updates to avoid unnecessary additions | ||||||
|  |    * - Specialized handling for different field types according to protobuf spec | ||||||
|  |    * - Templated helpers for repeated fields and messages | ||||||
|  |    */ | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates the size in bytes needed to encode a uint32_t value as a varint | ||||||
|  |    * | ||||||
|  |    * @param value The uint32_t value to calculate size for | ||||||
|  |    * @return The number of bytes needed to encode the value | ||||||
|  |    */ | ||||||
|  |   static inline uint32_t varint(uint32_t value) { | ||||||
|  |     // Optimized varint size calculation using leading zeros | ||||||
|  |     // Each 7 bits requires one byte in the varint encoding | ||||||
|  |     if (value < 128) | ||||||
|  |       return 1;  // 7 bits, common case for small values | ||||||
|  |  | ||||||
|  |     // For larger values, count bytes needed based on the position of the highest bit set | ||||||
|  |     if (value < 16384) { | ||||||
|  |       return 2;  // 14 bits | ||||||
|  |     } else if (value < 2097152) { | ||||||
|  |       return 3;  // 21 bits | ||||||
|  |     } else if (value < 268435456) { | ||||||
|  |       return 4;  // 28 bits | ||||||
|  |     } else { | ||||||
|  |       return 5;  // 32 bits (maximum for uint32_t) | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates the size in bytes needed to encode a uint64_t value as a varint | ||||||
|  |    * | ||||||
|  |    * @param value The uint64_t value to calculate size for | ||||||
|  |    * @return The number of bytes needed to encode the value | ||||||
|  |    */ | ||||||
|  |   static inline uint32_t varint(uint64_t value) { | ||||||
|  |     // Handle common case of values fitting in uint32_t (vast majority of use cases) | ||||||
|  |     if (value <= UINT32_MAX) { | ||||||
|  |       return varint(static_cast<uint32_t>(value)); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // For larger values, determine size based on highest bit position | ||||||
|  |     if (value < (1ULL << 35)) { | ||||||
|  |       return 5;  // 35 bits | ||||||
|  |     } else if (value < (1ULL << 42)) { | ||||||
|  |       return 6;  // 42 bits | ||||||
|  |     } else if (value < (1ULL << 49)) { | ||||||
|  |       return 7;  // 49 bits | ||||||
|  |     } else if (value < (1ULL << 56)) { | ||||||
|  |       return 8;  // 56 bits | ||||||
|  |     } else if (value < (1ULL << 63)) { | ||||||
|  |       return 9;  // 63 bits | ||||||
|  |     } else { | ||||||
|  |       return 10;  // 64 bits (maximum for uint64_t) | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates the size in bytes needed to encode an int32_t value as a varint | ||||||
|  |    * | ||||||
|  |    * Special handling is needed for negative values, which are sign-extended to 64 bits | ||||||
|  |    * in Protocol Buffers, resulting in a 10-byte varint. | ||||||
|  |    * | ||||||
|  |    * @param value The int32_t value to calculate size for | ||||||
|  |    * @return The number of bytes needed to encode the value | ||||||
|  |    */ | ||||||
|  |   static inline uint32_t varint(int32_t value) { | ||||||
|  |     // Negative values are sign-extended to 64 bits in protocol buffers, | ||||||
|  |     // which always results in a 10-byte varint for negative int32 | ||||||
|  |     if (value < 0) { | ||||||
|  |       return 10;  // Negative int32 is always 10 bytes long | ||||||
|  |     } | ||||||
|  |     // For non-negative values, use the uint32_t implementation | ||||||
|  |     return varint(static_cast<uint32_t>(value)); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates the size in bytes needed to encode an int64_t value as a varint | ||||||
|  |    * | ||||||
|  |    * @param value The int64_t value to calculate size for | ||||||
|  |    * @return The number of bytes needed to encode the value | ||||||
|  |    */ | ||||||
|  |   static inline uint32_t varint(int64_t value) { | ||||||
|  |     // For int64_t, we convert to uint64_t and calculate the size | ||||||
|  |     // This works because the bit pattern determines the encoding size, | ||||||
|  |     // and we've handled negative int32 values as a special case above | ||||||
|  |     return varint(static_cast<uint64_t>(value)); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates the size in bytes needed to encode a field ID and wire type | ||||||
|  |    * | ||||||
|  |    * @param field_id The field identifier | ||||||
|  |    * @param type The wire type value (from the WireType enum in the protobuf spec) | ||||||
|  |    * @return The number of bytes needed to encode the field ID and wire type | ||||||
|  |    */ | ||||||
|  |   static inline uint32_t field(uint32_t field_id, uint32_t type) { | ||||||
|  |     uint32_t tag = (field_id << 3) | (type & 0b111); | ||||||
|  |     return varint(tag); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Common parameters for all add_*_field methods | ||||||
|  |    * | ||||||
|  |    * All add_*_field methods follow these common patterns: | ||||||
|  |    * | ||||||
|  |    * @param total_size Reference to the total message size to update | ||||||
|  |    * @param field_id_size Pre-calculated size of the field ID in bytes | ||||||
|  |    * @param value The value to calculate size for (type varies) | ||||||
|  |    * @param force Whether to calculate size even if the value is default/zero/empty | ||||||
|  |    * | ||||||
|  |    * Each method follows this implementation pattern: | ||||||
|  |    * 1. Skip calculation if value is default (0, false, empty) and not forced | ||||||
|  |    * 2. Calculate the size based on the field's encoding rules | ||||||
|  |    * 3. Add the field_id_size + calculated value size to total_size | ||||||
|  |    */ | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of an int32 field to the total message size | ||||||
|  |    */ | ||||||
|  |   static inline void add_int32_field(uint32_t &total_size, uint32_t field_id_size, int32_t value) { | ||||||
|  |     // Skip calculation if value is zero | ||||||
|  |     if (value == 0) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Calculate and directly add to total_size | ||||||
|  |     if (value < 0) { | ||||||
|  |       // Negative values are encoded as 10-byte varints in protobuf | ||||||
|  |       total_size += field_id_size + 10; | ||||||
|  |     } else { | ||||||
|  |       // For non-negative values, use the standard varint size | ||||||
|  |       total_size += field_id_size + varint(static_cast<uint32_t>(value)); | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of an int32 field to the total message size (repeated field version) | ||||||
|  |    */ | ||||||
|  |   static inline void add_int32_field_repeated(uint32_t &total_size, uint32_t field_id_size, int32_t value) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     if (value < 0) { | ||||||
|  |       // Negative values are encoded as 10-byte varints in protobuf | ||||||
|  |       total_size += field_id_size + 10; | ||||||
|  |     } else { | ||||||
|  |       // For non-negative values, use the standard varint size | ||||||
|  |       total_size += field_id_size + varint(static_cast<uint32_t>(value)); | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a uint32 field to the total message size | ||||||
|  |    */ | ||||||
|  |   static inline void add_uint32_field(uint32_t &total_size, uint32_t field_id_size, uint32_t value) { | ||||||
|  |     // Skip calculation if value is zero | ||||||
|  |     if (value == 0) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Calculate and directly add to total_size | ||||||
|  |     total_size += field_id_size + varint(value); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a uint32 field to the total message size (repeated field version) | ||||||
|  |    */ | ||||||
|  |   static inline void add_uint32_field_repeated(uint32_t &total_size, uint32_t field_id_size, uint32_t value) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     total_size += field_id_size + varint(value); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a boolean field to the total message size | ||||||
|  |    */ | ||||||
|  |   static inline void add_bool_field(uint32_t &total_size, uint32_t field_id_size, bool value) { | ||||||
|  |     // Skip calculation if value is false | ||||||
|  |     if (!value) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Boolean fields always use 1 byte when true | ||||||
|  |     total_size += field_id_size + 1; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a boolean field to the total message size (repeated field version) | ||||||
|  |    */ | ||||||
|  |   static inline void add_bool_field_repeated(uint32_t &total_size, uint32_t field_id_size, bool value) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     // Boolean fields always use 1 byte | ||||||
|  |     total_size += field_id_size + 1; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a fixed field to the total message size | ||||||
|  |    * | ||||||
|  |    * Fixed fields always take exactly N bytes (4 for fixed32/float, 8 for fixed64/double). | ||||||
|  |    * | ||||||
|  |    * @tparam NumBytes The number of bytes for this fixed field (4 or 8) | ||||||
|  |    * @param is_nonzero Whether the value is non-zero | ||||||
|  |    */ | ||||||
|  |   template<uint32_t NumBytes> | ||||||
|  |   static inline void add_fixed_field(uint32_t &total_size, uint32_t field_id_size, bool is_nonzero) { | ||||||
|  |     // Skip calculation if value is zero | ||||||
|  |     if (!is_nonzero) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Fixed fields always take exactly NumBytes | ||||||
|  |     total_size += field_id_size + NumBytes; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of an enum field to the total message size | ||||||
|  |    * | ||||||
|  |    * Enum fields are encoded as uint32 varints. | ||||||
|  |    */ | ||||||
|  |   static inline void add_enum_field(uint32_t &total_size, uint32_t field_id_size, uint32_t value) { | ||||||
|  |     // Skip calculation if value is zero | ||||||
|  |     if (value == 0) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Enums are encoded as uint32 | ||||||
|  |     total_size += field_id_size + varint(value); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of an enum field to the total message size (repeated field version) | ||||||
|  |    * | ||||||
|  |    * Enum fields are encoded as uint32 varints. | ||||||
|  |    */ | ||||||
|  |   static inline void add_enum_field_repeated(uint32_t &total_size, uint32_t field_id_size, uint32_t value) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     // Enums are encoded as uint32 | ||||||
|  |     total_size += field_id_size + varint(value); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a sint32 field to the total message size | ||||||
|  |    * | ||||||
|  |    * Sint32 fields use ZigZag encoding, which is more efficient for negative values. | ||||||
|  |    */ | ||||||
|  |   static inline void add_sint32_field(uint32_t &total_size, uint32_t field_id_size, int32_t value) { | ||||||
|  |     // Skip calculation if value is zero | ||||||
|  |     if (value == 0) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // ZigZag encoding for sint32: (n << 1) ^ (n >> 31) | ||||||
|  |     uint32_t zigzag = (static_cast<uint32_t>(value) << 1) ^ (static_cast<uint32_t>(value >> 31)); | ||||||
|  |     total_size += field_id_size + varint(zigzag); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a sint32 field to the total message size (repeated field version) | ||||||
|  |    * | ||||||
|  |    * Sint32 fields use ZigZag encoding, which is more efficient for negative values. | ||||||
|  |    */ | ||||||
|  |   static inline void add_sint32_field_repeated(uint32_t &total_size, uint32_t field_id_size, int32_t value) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     // ZigZag encoding for sint32: (n << 1) ^ (n >> 31) | ||||||
|  |     uint32_t zigzag = (static_cast<uint32_t>(value) << 1) ^ (static_cast<uint32_t>(value >> 31)); | ||||||
|  |     total_size += field_id_size + varint(zigzag); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of an int64 field to the total message size | ||||||
|  |    */ | ||||||
|  |   static inline void add_int64_field(uint32_t &total_size, uint32_t field_id_size, int64_t value) { | ||||||
|  |     // Skip calculation if value is zero | ||||||
|  |     if (value == 0) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Calculate and directly add to total_size | ||||||
|  |     total_size += field_id_size + varint(value); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of an int64 field to the total message size (repeated field version) | ||||||
|  |    */ | ||||||
|  |   static inline void add_int64_field_repeated(uint32_t &total_size, uint32_t field_id_size, int64_t value) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     total_size += field_id_size + varint(value); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a uint64 field to the total message size | ||||||
|  |    */ | ||||||
|  |   static inline void add_uint64_field(uint32_t &total_size, uint32_t field_id_size, uint64_t value) { | ||||||
|  |     // Skip calculation if value is zero | ||||||
|  |     if (value == 0) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Calculate and directly add to total_size | ||||||
|  |     total_size += field_id_size + varint(value); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a uint64 field to the total message size (repeated field version) | ||||||
|  |    */ | ||||||
|  |   static inline void add_uint64_field_repeated(uint32_t &total_size, uint32_t field_id_size, uint64_t value) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     total_size += field_id_size + varint(value); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a sint64 field to the total message size | ||||||
|  |    * | ||||||
|  |    * Sint64 fields use ZigZag encoding, which is more efficient for negative values. | ||||||
|  |    */ | ||||||
|  |   static inline void add_sint64_field(uint32_t &total_size, uint32_t field_id_size, int64_t value) { | ||||||
|  |     // Skip calculation if value is zero | ||||||
|  |     if (value == 0) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // ZigZag encoding for sint64: (n << 1) ^ (n >> 63) | ||||||
|  |     uint64_t zigzag = (static_cast<uint64_t>(value) << 1) ^ (static_cast<uint64_t>(value >> 63)); | ||||||
|  |     total_size += field_id_size + varint(zigzag); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a sint64 field to the total message size (repeated field version) | ||||||
|  |    * | ||||||
|  |    * Sint64 fields use ZigZag encoding, which is more efficient for negative values. | ||||||
|  |    */ | ||||||
|  |   static inline void add_sint64_field_repeated(uint32_t &total_size, uint32_t field_id_size, int64_t value) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     // ZigZag encoding for sint64: (n << 1) ^ (n >> 63) | ||||||
|  |     uint64_t zigzag = (static_cast<uint64_t>(value) << 1) ^ (static_cast<uint64_t>(value >> 63)); | ||||||
|  |     total_size += field_id_size + varint(zigzag); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a string/bytes field to the total message size | ||||||
|  |    */ | ||||||
|  |   static inline void add_string_field(uint32_t &total_size, uint32_t field_id_size, const std::string &str) { | ||||||
|  |     // Skip calculation if string is empty | ||||||
|  |     if (str.empty()) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Calculate and directly add to total_size | ||||||
|  |     const uint32_t str_size = static_cast<uint32_t>(str.size()); | ||||||
|  |     total_size += field_id_size + varint(str_size) + str_size; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a string/bytes field to the total message size (repeated field version) | ||||||
|  |    */ | ||||||
|  |   static inline void add_string_field_repeated(uint32_t &total_size, uint32_t field_id_size, const std::string &str) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     const uint32_t str_size = static_cast<uint32_t>(str.size()); | ||||||
|  |     total_size += field_id_size + varint(str_size) + str_size; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a nested message field to the total message size | ||||||
|  |    * | ||||||
|  |    * This helper function directly updates the total_size reference if the nested size | ||||||
|  |    * is greater than zero. | ||||||
|  |    * | ||||||
|  |    * @param nested_size The pre-calculated size of the nested message | ||||||
|  |    */ | ||||||
|  |   static inline void add_message_field(uint32_t &total_size, uint32_t field_id_size, uint32_t nested_size) { | ||||||
|  |     // Skip calculation if nested message is empty | ||||||
|  |     if (nested_size == 0) { | ||||||
|  |       return;  // No need to update total_size | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Calculate and directly add to total_size | ||||||
|  |     // Field ID + length varint + nested message content | ||||||
|  |     total_size += field_id_size + varint(nested_size) + nested_size; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a nested message field to the total message size (repeated field version) | ||||||
|  |    * | ||||||
|  |    * @param nested_size The pre-calculated size of the nested message | ||||||
|  |    */ | ||||||
|  |   static inline void add_message_field_repeated(uint32_t &total_size, uint32_t field_id_size, uint32_t nested_size) { | ||||||
|  |     // Always calculate size for repeated fields | ||||||
|  |     // Field ID + length varint + nested message content | ||||||
|  |     total_size += field_id_size + varint(nested_size) + nested_size; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a nested message field to the total message size | ||||||
|  |    * | ||||||
|  |    * This version takes a ProtoMessage object, calculates its size internally, | ||||||
|  |    * and updates the total_size reference. This eliminates the need for a temporary variable | ||||||
|  |    * at the call site. | ||||||
|  |    * | ||||||
|  |    * @param message The nested message object | ||||||
|  |    */ | ||||||
|  |   static inline void add_message_object(uint32_t &total_size, uint32_t field_id_size, const ProtoMessage &message) { | ||||||
|  |     uint32_t nested_size = 0; | ||||||
|  |     message.calculate_size(nested_size); | ||||||
|  |  | ||||||
|  |     // Use the base implementation with the calculated nested_size | ||||||
|  |     add_message_field(total_size, field_id_size, nested_size); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the size of a nested message field to the total message size (repeated field version) | ||||||
|  |    * | ||||||
|  |    * @param message The nested message object | ||||||
|  |    */ | ||||||
|  |   static inline void add_message_object_repeated(uint32_t &total_size, uint32_t field_id_size, | ||||||
|  |                                                  const ProtoMessage &message) { | ||||||
|  |     uint32_t nested_size = 0; | ||||||
|  |     message.calculate_size(nested_size); | ||||||
|  |  | ||||||
|  |     // Use the base implementation with the calculated nested_size | ||||||
|  |     add_message_field_repeated(total_size, field_id_size, nested_size); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /** | ||||||
|  |    * @brief Calculates and adds the sizes of all messages in a repeated field to the total message size | ||||||
|  |    * | ||||||
|  |    * This helper processes a vector of message objects, calculating the size for each message | ||||||
|  |    * and adding it to the total size. | ||||||
|  |    * | ||||||
|  |    * @tparam MessageType The type of the nested messages in the vector | ||||||
|  |    * @param messages Vector of message objects | ||||||
|  |    */ | ||||||
|  |   template<typename MessageType> | ||||||
|  |   static inline void add_repeated_message(uint32_t &total_size, uint32_t field_id_size, | ||||||
|  |                                           const std::vector<MessageType> &messages) { | ||||||
|  |     // Skip if the vector is empty | ||||||
|  |     if (messages.empty()) { | ||||||
|  |       return; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // Use the repeated field version for all messages | ||||||
|  |     for (const auto &message : messages) { | ||||||
|  |       add_message_object_repeated(total_size, field_id_size, message); | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  | }; | ||||||
|  |  | ||||||
|  | // Implementation of encode_message - must be after ProtoMessage is defined | ||||||
|  | inline void ProtoWriteBuffer::encode_message(uint32_t field_id, const ProtoMessage &value, bool force) { | ||||||
|  |   this->encode_field_raw(field_id, 2);  // type 2: Length-delimited message | ||||||
|  |  | ||||||
|  |   // Calculate the message size first | ||||||
|  |   uint32_t msg_length_bytes = 0; | ||||||
|  |   value.calculate_size(msg_length_bytes); | ||||||
|  |  | ||||||
|  |   // Calculate how many bytes the length varint needs | ||||||
|  |   uint32_t varint_length_bytes = ProtoSize::varint(msg_length_bytes); | ||||||
|  |  | ||||||
|  |   // Reserve exact space for the length varint | ||||||
|  |   size_t begin = this->buffer_->size(); | ||||||
|  |   this->buffer_->resize(this->buffer_->size() + varint_length_bytes); | ||||||
|  |  | ||||||
|  |   // Write the length varint directly | ||||||
|  |   ProtoVarInt(msg_length_bytes).encode_to_buffer_unchecked(this->buffer_->data() + begin, varint_length_bytes); | ||||||
|  |  | ||||||
|  |   // Now encode the message content - it will append to the buffer | ||||||
|  |   value.encode(*this); | ||||||
|  |  | ||||||
|  |   // Verify that the encoded size matches what we calculated | ||||||
|  |   assert(this->buffer_->size() == begin + varint_length_bytes + msg_length_bytes); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Implementation of decode_to_message - must be after ProtoMessage is defined | ||||||
|  | inline void ProtoLengthDelimited::decode_to_message(ProtoMessage &msg) const { | ||||||
|  |   msg.decode(this->value_, this->length_); | ||||||
|  | } | ||||||
|  |  | ||||||
| template<typename T> const char *proto_enum_to_string(T value); | template<typename T> const char *proto_enum_to_string(T value); | ||||||
|  |  | ||||||
| class ProtoService { | class ProtoService { | ||||||
| @@ -360,11 +846,11 @@ class ProtoService { | |||||||
|    * @return A ProtoWriteBuffer object with the reserved size. |    * @return A ProtoWriteBuffer object with the reserved size. | ||||||
|    */ |    */ | ||||||
|   virtual ProtoWriteBuffer create_buffer(uint32_t reserve_size) = 0; |   virtual ProtoWriteBuffer create_buffer(uint32_t reserve_size) = 0; | ||||||
|   virtual bool send_buffer(ProtoWriteBuffer buffer, uint16_t message_type) = 0; |   virtual bool send_buffer(ProtoWriteBuffer buffer, uint8_t message_type) = 0; | ||||||
|   virtual bool read_message(uint32_t msg_size, uint32_t msg_type, uint8_t *msg_data) = 0; |   virtual void read_message(uint32_t msg_size, uint32_t msg_type, uint8_t *msg_data) = 0; | ||||||
|  |  | ||||||
|   // Optimized method that pre-allocates buffer based on message size |   // Optimized method that pre-allocates buffer based on message size | ||||||
|   bool send_message_(const ProtoMessage &msg, uint16_t message_type) { |   bool send_message_(const ProtoMessage &msg, uint8_t message_type) { | ||||||
|     uint32_t msg_size = 0; |     uint32_t msg_size = 0; | ||||||
|     msg.calculate_size(msg_size); |     msg.calculate_size(msg_size); | ||||||
|  |  | ||||||
| @@ -377,6 +863,26 @@ class ProtoService { | |||||||
|     // Send the buffer |     // Send the buffer | ||||||
|     return this->send_buffer(buffer, message_type); |     return this->send_buffer(buffer, message_type); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|  |   // Authentication helper methods | ||||||
|  |   bool check_connection_setup_() { | ||||||
|  |     if (!this->is_connection_setup()) { | ||||||
|  |       this->on_no_setup_connection(); | ||||||
|  |       return false; | ||||||
|  |     } | ||||||
|  |     return true; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   bool check_authenticated_() { | ||||||
|  |     if (!this->check_connection_setup_()) { | ||||||
|  |       return false; | ||||||
|  |     } | ||||||
|  |     if (!this->is_authenticated()) { | ||||||
|  |       this->on_unauthenticated_access(); | ||||||
|  |       return false; | ||||||
|  |     } | ||||||
|  |     return true; | ||||||
|  |   } | ||||||
| }; | }; | ||||||
|  |  | ||||||
| }  // namespace api | }  // namespace api | ||||||
|   | |||||||
| @@ -6,73 +6,67 @@ | |||||||
| namespace esphome { | namespace esphome { | ||||||
| namespace api { | namespace api { | ||||||
|  |  | ||||||
|  | // Generate entity handler implementations using macros | ||||||
| #ifdef USE_BINARY_SENSOR | #ifdef USE_BINARY_SENSOR | ||||||
| bool InitialStateIterator::on_binary_sensor(binary_sensor::BinarySensor *binary_sensor) { | INITIAL_STATE_HANDLER(binary_sensor, binary_sensor::BinarySensor) | ||||||
|   return this->client_->send_binary_sensor_state(binary_sensor); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_COVER | #ifdef USE_COVER | ||||||
| bool InitialStateIterator::on_cover(cover::Cover *cover) { return this->client_->send_cover_state(cover); } | INITIAL_STATE_HANDLER(cover, cover::Cover) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_FAN | #ifdef USE_FAN | ||||||
| bool InitialStateIterator::on_fan(fan::Fan *fan) { return this->client_->send_fan_state(fan); } | INITIAL_STATE_HANDLER(fan, fan::Fan) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LIGHT | #ifdef USE_LIGHT | ||||||
| bool InitialStateIterator::on_light(light::LightState *light) { return this->client_->send_light_state(light); } | INITIAL_STATE_HANDLER(light, light::LightState) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SENSOR | #ifdef USE_SENSOR | ||||||
| bool InitialStateIterator::on_sensor(sensor::Sensor *sensor) { return this->client_->send_sensor_state(sensor); } | INITIAL_STATE_HANDLER(sensor, sensor::Sensor) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SWITCH | #ifdef USE_SWITCH | ||||||
| bool InitialStateIterator::on_switch(switch_::Switch *a_switch) { return this->client_->send_switch_state(a_switch); } | INITIAL_STATE_HANDLER(switch, switch_::Switch) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT_SENSOR | #ifdef USE_TEXT_SENSOR | ||||||
| bool InitialStateIterator::on_text_sensor(text_sensor::TextSensor *text_sensor) { | INITIAL_STATE_HANDLER(text_sensor, text_sensor::TextSensor) | ||||||
|   return this->client_->send_text_sensor_state(text_sensor); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_CLIMATE | #ifdef USE_CLIMATE | ||||||
| bool InitialStateIterator::on_climate(climate::Climate *climate) { return this->client_->send_climate_state(climate); } | INITIAL_STATE_HANDLER(climate, climate::Climate) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_NUMBER | #ifdef USE_NUMBER | ||||||
| bool InitialStateIterator::on_number(number::Number *number) { return this->client_->send_number_state(number); } | INITIAL_STATE_HANDLER(number, number::Number) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_DATE | #ifdef USE_DATETIME_DATE | ||||||
| bool InitialStateIterator::on_date(datetime::DateEntity *date) { return this->client_->send_date_state(date); } | INITIAL_STATE_HANDLER(date, datetime::DateEntity) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_TIME | #ifdef USE_DATETIME_TIME | ||||||
| bool InitialStateIterator::on_time(datetime::TimeEntity *time) { return this->client_->send_time_state(time); } | INITIAL_STATE_HANDLER(time, datetime::TimeEntity) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_DATETIME | #ifdef USE_DATETIME_DATETIME | ||||||
| bool InitialStateIterator::on_datetime(datetime::DateTimeEntity *datetime) { | INITIAL_STATE_HANDLER(datetime, datetime::DateTimeEntity) | ||||||
|   return this->client_->send_datetime_state(datetime); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT | #ifdef USE_TEXT | ||||||
| bool InitialStateIterator::on_text(text::Text *text) { return this->client_->send_text_state(text); } | INITIAL_STATE_HANDLER(text, text::Text) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SELECT | #ifdef USE_SELECT | ||||||
| bool InitialStateIterator::on_select(select::Select *select) { return this->client_->send_select_state(select); } | INITIAL_STATE_HANDLER(select, select::Select) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LOCK | #ifdef USE_LOCK | ||||||
| bool InitialStateIterator::on_lock(lock::Lock *a_lock) { return this->client_->send_lock_state(a_lock); } | INITIAL_STATE_HANDLER(lock, lock::Lock) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_VALVE | #ifdef USE_VALVE | ||||||
| bool InitialStateIterator::on_valve(valve::Valve *valve) { return this->client_->send_valve_state(valve); } | INITIAL_STATE_HANDLER(valve, valve::Valve) | ||||||
| #endif | #endif | ||||||
| #ifdef USE_MEDIA_PLAYER | #ifdef USE_MEDIA_PLAYER | ||||||
| bool InitialStateIterator::on_media_player(media_player::MediaPlayer *media_player) { | INITIAL_STATE_HANDLER(media_player, media_player::MediaPlayer) | ||||||
|   return this->client_->send_media_player_state(media_player); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_ALARM_CONTROL_PANEL | #ifdef USE_ALARM_CONTROL_PANEL | ||||||
| bool InitialStateIterator::on_alarm_control_panel(alarm_control_panel::AlarmControlPanel *a_alarm_control_panel) { | INITIAL_STATE_HANDLER(alarm_control_panel, alarm_control_panel::AlarmControlPanel) | ||||||
|   return this->client_->send_alarm_control_panel_state(a_alarm_control_panel); |  | ||||||
| } |  | ||||||
| #endif | #endif | ||||||
| #ifdef USE_UPDATE | #ifdef USE_UPDATE | ||||||
| bool InitialStateIterator::on_update(update::UpdateEntity *update) { return this->client_->send_update_state(update); } | INITIAL_STATE_HANDLER(update, update::UpdateEntity) | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|  | // Special cases (button and event) are already defined inline in subscribe_state.h | ||||||
|  |  | ||||||
| InitialStateIterator::InitialStateIterator(APIConnection *client) : client_(client) {} | InitialStateIterator::InitialStateIterator(APIConnection *client) : client_(client) {} | ||||||
|  |  | ||||||
| }  // namespace api | }  // namespace api | ||||||
|   | |||||||
| @@ -10,71 +10,78 @@ namespace api { | |||||||
|  |  | ||||||
| class APIConnection; | class APIConnection; | ||||||
|  |  | ||||||
|  | // Macro for generating InitialStateIterator handlers | ||||||
|  | // Calls send_*_state | ||||||
|  | #define INITIAL_STATE_HANDLER(entity_type, EntityClass) \ | ||||||
|  |   bool InitialStateIterator::on_##entity_type(EntityClass *entity) { /* NOLINT(bugprone-macro-parentheses) */ \ | ||||||
|  |     return this->client_->send_##entity_type##_state(entity); \ | ||||||
|  |   } | ||||||
|  |  | ||||||
| class InitialStateIterator : public ComponentIterator { | class InitialStateIterator : public ComponentIterator { | ||||||
|  public: |  public: | ||||||
|   InitialStateIterator(APIConnection *client); |   InitialStateIterator(APIConnection *client); | ||||||
| #ifdef USE_BINARY_SENSOR | #ifdef USE_BINARY_SENSOR | ||||||
|   bool on_binary_sensor(binary_sensor::BinarySensor *binary_sensor) override; |   bool on_binary_sensor(binary_sensor::BinarySensor *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_COVER | #ifdef USE_COVER | ||||||
|   bool on_cover(cover::Cover *cover) override; |   bool on_cover(cover::Cover *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_FAN | #ifdef USE_FAN | ||||||
|   bool on_fan(fan::Fan *fan) override; |   bool on_fan(fan::Fan *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LIGHT | #ifdef USE_LIGHT | ||||||
|   bool on_light(light::LightState *light) override; |   bool on_light(light::LightState *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SENSOR | #ifdef USE_SENSOR | ||||||
|   bool on_sensor(sensor::Sensor *sensor) override; |   bool on_sensor(sensor::Sensor *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SWITCH | #ifdef USE_SWITCH | ||||||
|   bool on_switch(switch_::Switch *a_switch) override; |   bool on_switch(switch_::Switch *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_BUTTON | #ifdef USE_BUTTON | ||||||
|   bool on_button(button::Button *button) override { return true; }; |   bool on_button(button::Button *button) override { return true; }; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT_SENSOR | #ifdef USE_TEXT_SENSOR | ||||||
|   bool on_text_sensor(text_sensor::TextSensor *text_sensor) override; |   bool on_text_sensor(text_sensor::TextSensor *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_CLIMATE | #ifdef USE_CLIMATE | ||||||
|   bool on_climate(climate::Climate *climate) override; |   bool on_climate(climate::Climate *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_NUMBER | #ifdef USE_NUMBER | ||||||
|   bool on_number(number::Number *number) override; |   bool on_number(number::Number *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_DATE | #ifdef USE_DATETIME_DATE | ||||||
|   bool on_date(datetime::DateEntity *date) override; |   bool on_date(datetime::DateEntity *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_TIME | #ifdef USE_DATETIME_TIME | ||||||
|   bool on_time(datetime::TimeEntity *time) override; |   bool on_time(datetime::TimeEntity *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_DATETIME_DATETIME | #ifdef USE_DATETIME_DATETIME | ||||||
|   bool on_datetime(datetime::DateTimeEntity *datetime) override; |   bool on_datetime(datetime::DateTimeEntity *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_TEXT | #ifdef USE_TEXT | ||||||
|   bool on_text(text::Text *text) override; |   bool on_text(text::Text *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_SELECT | #ifdef USE_SELECT | ||||||
|   bool on_select(select::Select *select) override; |   bool on_select(select::Select *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_LOCK | #ifdef USE_LOCK | ||||||
|   bool on_lock(lock::Lock *a_lock) override; |   bool on_lock(lock::Lock *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_VALVE | #ifdef USE_VALVE | ||||||
|   bool on_valve(valve::Valve *valve) override; |   bool on_valve(valve::Valve *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_MEDIA_PLAYER | #ifdef USE_MEDIA_PLAYER | ||||||
|   bool on_media_player(media_player::MediaPlayer *media_player) override; |   bool on_media_player(media_player::MediaPlayer *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_ALARM_CONTROL_PANEL | #ifdef USE_ALARM_CONTROL_PANEL | ||||||
|   bool on_alarm_control_panel(alarm_control_panel::AlarmControlPanel *a_alarm_control_panel) override; |   bool on_alarm_control_panel(alarm_control_panel::AlarmControlPanel *entity) override; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_EVENT | #ifdef USE_EVENT | ||||||
|   bool on_event(event::Event *event) override { return true; }; |   bool on_event(event::Event *event) override { return true; }; | ||||||
| #endif | #endif | ||||||
| #ifdef USE_UPDATE | #ifdef USE_UPDATE | ||||||
|   bool on_update(update::UpdateEntity *update) override; |   bool on_update(update::UpdateEntity *entity) override; | ||||||
| #endif | #endif | ||||||
|   bool completed() { return this->state_ == IteratorState::NONE; } |   bool completed() { return this->state_ == IteratorState::NONE; } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -7,6 +7,7 @@ | |||||||
| #include "esphome/core/automation.h" | #include "esphome/core/automation.h" | ||||||
| #include "api_pb2.h" | #include "api_pb2.h" | ||||||
|  |  | ||||||
|  | #ifdef USE_API_SERVICES | ||||||
| namespace esphome { | namespace esphome { | ||||||
| namespace api { | namespace api { | ||||||
|  |  | ||||||
| @@ -73,3 +74,4 @@ template<typename... Ts> class UserServiceTrigger : public UserServiceBase<Ts... | |||||||
|  |  | ||||||
| }  // namespace api | }  // namespace api | ||||||
| }  // namespace esphome | }  // namespace esphome | ||||||
|  | #endif  // USE_API_SERVICES | ||||||
|   | |||||||
| @@ -3,8 +3,6 @@ | |||||||
| #include "esphome/core/component.h" | #include "esphome/core/component.h" | ||||||
| #include "esphome/components/as3935/as3935.h" | #include "esphome/components/as3935/as3935.h" | ||||||
| #include "esphome/components/spi/spi.h" | #include "esphome/components/spi/spi.h" | ||||||
| #include "esphome/components/sensor/sensor.h" |  | ||||||
| #include "esphome/components/binary_sensor/binary_sensor.h" |  | ||||||
|  |  | ||||||
| namespace esphome { | namespace esphome { | ||||||
| namespace as3935_spi { | namespace as3935_spi { | ||||||
|   | |||||||
| @@ -50,7 +50,6 @@ class AS5600Component : public Component, public i2c::I2CDevice { | |||||||
|   void setup() override; |   void setup() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   /// HARDWARE_LATE setup priority |   /// HARDWARE_LATE setup priority | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   // configuration setters |   // configuration setters | ||||||
|   void set_dir_pin(InternalGPIOPin *pin) { this->dir_pin_ = pin; } |   void set_dir_pin(InternalGPIOPin *pin) { this->dir_pin_ = pin; } | ||||||
|   | |||||||
| @@ -5,6 +5,7 @@ from esphome.const import ( | |||||||
|     PLATFORM_BK72XX, |     PLATFORM_BK72XX, | ||||||
|     PLATFORM_ESP32, |     PLATFORM_ESP32, | ||||||
|     PLATFORM_ESP8266, |     PLATFORM_ESP8266, | ||||||
|  |     PLATFORM_LN882X, | ||||||
|     PLATFORM_RTL87XX, |     PLATFORM_RTL87XX, | ||||||
| ) | ) | ||||||
| from esphome.core import CORE, coroutine_with_priority | from esphome.core import CORE, coroutine_with_priority | ||||||
| @@ -14,15 +15,23 @@ CODEOWNERS = ["@OttoWinter"] | |||||||
| CONFIG_SCHEMA = cv.All( | CONFIG_SCHEMA = cv.All( | ||||||
|     cv.Schema({}), |     cv.Schema({}), | ||||||
|     cv.only_with_arduino, |     cv.only_with_arduino, | ||||||
|     cv.only_on([PLATFORM_ESP32, PLATFORM_ESP8266, PLATFORM_BK72XX, PLATFORM_RTL87XX]), |     cv.only_on( | ||||||
|  |         [ | ||||||
|  |             PLATFORM_ESP32, | ||||||
|  |             PLATFORM_ESP8266, | ||||||
|  |             PLATFORM_BK72XX, | ||||||
|  |             PLATFORM_LN882X, | ||||||
|  |             PLATFORM_RTL87XX, | ||||||
|  |         ] | ||||||
|  |     ), | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  |  | ||||||
| @coroutine_with_priority(200.0) | @coroutine_with_priority(200.0) | ||||||
| async def to_code(config): | async def to_code(config): | ||||||
|     if CORE.is_esp32 or CORE.is_libretiny: |     if CORE.is_esp32 or CORE.is_libretiny: | ||||||
|         # https://github.com/esphome/AsyncTCP/blob/master/library.json |         # https://github.com/ESP32Async/AsyncTCP | ||||||
|         cg.add_library("esphome/AsyncTCP-esphome", "2.1.4") |         cg.add_library("ESP32Async/AsyncTCP", "3.4.5") | ||||||
|     elif CORE.is_esp8266: |     elif CORE.is_esp8266: | ||||||
|         # https://github.com/esphome/ESPAsyncTCP |         # https://github.com/ESP32Async/ESPAsyncTCP | ||||||
|         cg.add_library("esphome/ESPAsyncTCP-esphome", "2.0.0") |         cg.add_library("ESP32Async/ESPAsyncTCP", "2.0.0") | ||||||
|   | |||||||
| @@ -25,7 +25,6 @@ class ATCMiThermometer : public Component, public esp32_ble_tracker::ESPBTDevice | |||||||
|  |  | ||||||
|   bool parse_device(const esp32_ble_tracker::ESPBTDevice &device) override; |   bool parse_device(const esp32_ble_tracker::ESPBTDevice &device) override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   void set_temperature(sensor::Sensor *temperature) { temperature_ = temperature; } |   void set_temperature(sensor::Sensor *temperature) { temperature_ = temperature; } | ||||||
|   void set_humidity(sensor::Sensor *humidity) { humidity_ = humidity; } |   void set_humidity(sensor::Sensor *humidity) { humidity_ = humidity; } | ||||||
|   void set_battery_level(sensor::Sensor *battery_level) { battery_level_ = battery_level; } |   void set_battery_level(sensor::Sensor *battery_level) { battery_level_ = battery_level; } | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| #include "atm90e32.h" | #include "atm90e32.h" | ||||||
| #include <cinttypes> | #include <cinttypes> | ||||||
| #include <cmath> | #include <cmath> | ||||||
|  | #include <numbers> | ||||||
| #include "esphome/core/log.h" | #include "esphome/core/log.h" | ||||||
|  |  | ||||||
| namespace esphome { | namespace esphome { | ||||||
| @@ -848,7 +849,7 @@ uint16_t ATM90E32Component::calculate_voltage_threshold(int line_freq, uint16_t | |||||||
|   float nominal_voltage = (line_freq == 60) ? 120.0f : 220.0f; |   float nominal_voltage = (line_freq == 60) ? 120.0f : 220.0f; | ||||||
|   float target_voltage = nominal_voltage * multiplier; |   float target_voltage = nominal_voltage * multiplier; | ||||||
|  |  | ||||||
|   float peak_01v = target_voltage * 100.0f * std::sqrt(2.0f);  // convert RMS → peak, scale to 0.01V |   float peak_01v = target_voltage * 100.0f * std::numbers::sqrt2_v<float>;  // convert RMS → peak, scale to 0.01V | ||||||
|   float divider = (2.0f * ugain) / 32768.0f; |   float divider = (2.0f * ugain) / 32768.0f; | ||||||
|  |  | ||||||
|   float threshold = peak_01v / divider; |   float threshold = peak_01v / divider; | ||||||
|   | |||||||
| @@ -312,7 +312,7 @@ FileDecoderState AudioDecoder::decode_mp3_() { | |||||||
|   if (err) { |   if (err) { | ||||||
|     switch (err) { |     switch (err) { | ||||||
|       case esp_audio_libs::helix_decoder::ERR_MP3_OUT_OF_MEMORY: |       case esp_audio_libs::helix_decoder::ERR_MP3_OUT_OF_MEMORY: | ||||||
|         // Intentional fallthrough |         [[fallthrough]]; | ||||||
|       case esp_audio_libs::helix_decoder::ERR_MP3_NULL_POINTER: |       case esp_audio_libs::helix_decoder::ERR_MP3_NULL_POINTER: | ||||||
|         return FileDecoderState::FAILED; |         return FileDecoderState::FAILED; | ||||||
|         break; |         break; | ||||||
|   | |||||||
| @@ -86,7 +86,7 @@ bool AudioTransferBuffer::reallocate(size_t new_buffer_size) { | |||||||
| bool AudioTransferBuffer::allocate_buffer_(size_t buffer_size) { | bool AudioTransferBuffer::allocate_buffer_(size_t buffer_size) { | ||||||
|   this->buffer_size_ = buffer_size; |   this->buffer_size_ = buffer_size; | ||||||
|  |  | ||||||
|   RAMAllocator<uint8_t> allocator(ExternalRAMAllocator<uint8_t>::ALLOW_FAILURE); |   RAMAllocator<uint8_t> allocator; | ||||||
|  |  | ||||||
|   this->buffer_ = allocator.allocate(this->buffer_size_); |   this->buffer_ = allocator.allocate(this->buffer_size_); | ||||||
|   if (this->buffer_ == nullptr) { |   if (this->buffer_ == nullptr) { | ||||||
| @@ -101,7 +101,7 @@ bool AudioTransferBuffer::allocate_buffer_(size_t buffer_size) { | |||||||
|  |  | ||||||
| void AudioTransferBuffer::deallocate_buffer_() { | void AudioTransferBuffer::deallocate_buffer_() { | ||||||
|   if (this->buffer_ != nullptr) { |   if (this->buffer_ != nullptr) { | ||||||
|     RAMAllocator<uint8_t> allocator(ExternalRAMAllocator<uint8_t>::ALLOW_FAILURE); |     RAMAllocator<uint8_t> allocator; | ||||||
|     allocator.deallocate(this->buffer_, this->buffer_size_); |     allocator.deallocate(this->buffer_, this->buffer_size_); | ||||||
|     this->buffer_ = nullptr; |     this->buffer_ = nullptr; | ||||||
|     this->data_start_ = nullptr; |     this->data_start_ = nullptr; | ||||||
|   | |||||||
| @@ -16,7 +16,6 @@ class BParasite : public Component, public esp32_ble_tracker::ESPBTDeviceListene | |||||||
|  |  | ||||||
|   bool parse_device(const esp32_ble_tracker::ESPBTDevice &device) override; |   bool parse_device(const esp32_ble_tracker::ESPBTDevice &device) override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   void set_battery_voltage(sensor::Sensor *battery_voltage) { battery_voltage_ = battery_voltage; } |   void set_battery_voltage(sensor::Sensor *battery_voltage) { battery_voltage_ = battery_voltage; } | ||||||
|   void set_temperature(sensor::Sensor *temperature) { temperature_ = temperature; } |   void set_temperature(sensor::Sensor *temperature) { temperature_ = temperature; } | ||||||
|   | |||||||
| @@ -480,7 +480,11 @@ void BedJetHub::set_clock(uint8_t hour, uint8_t minute) { | |||||||
|  |  | ||||||
| /* Internal */ | /* Internal */ | ||||||
|  |  | ||||||
| void BedJetHub::loop() {} | void BedJetHub::loop() { | ||||||
|  |   // Parent BLEClientNode has a loop() method, but this component uses | ||||||
|  |   // polling via update() and BLE callbacks so loop isn't needed | ||||||
|  |   this->disable_loop(); | ||||||
|  | } | ||||||
| void BedJetHub::update() { this->dispatch_status_(); } | void BedJetHub::update() { this->dispatch_status_(); } | ||||||
|  |  | ||||||
| void BedJetHub::dump_config() { | void BedJetHub::dump_config() { | ||||||
|   | |||||||
| @@ -83,7 +83,11 @@ void BedJetClimate::reset_state_() { | |||||||
|   this->publish_state(); |   this->publish_state(); | ||||||
| } | } | ||||||
|  |  | ||||||
| void BedJetClimate::loop() {} | void BedJetClimate::loop() { | ||||||
|  |   // This component is controlled via the parent BedJetHub | ||||||
|  |   // Empty loop not needed, disable to save CPU cycles | ||||||
|  |   this->disable_loop(); | ||||||
|  | } | ||||||
|  |  | ||||||
| void BedJetClimate::control(const ClimateCall &call) { | void BedJetClimate::control(const ClimateCall &call) { | ||||||
|   ESP_LOGD(TAG, "Received BedJetClimate::control"); |   ESP_LOGD(TAG, "Received BedJetClimate::control"); | ||||||
|   | |||||||
| @@ -7,11 +7,13 @@ | |||||||
|  |  | ||||||
| extern "C" { | extern "C" { | ||||||
| #include "rtos_pub.h" | #include "rtos_pub.h" | ||||||
| #include "spi.h" | // rtos_pub.h must be included before the rest of the includes | ||||||
|  |  | ||||||
| #include "arm_arch.h" | #include "arm_arch.h" | ||||||
| #include "general_dma_pub.h" | #include "general_dma_pub.h" | ||||||
| #include "gpio_pub.h" | #include "gpio_pub.h" | ||||||
| #include "icu_pub.h" | #include "icu_pub.h" | ||||||
|  | #include "spi.h" | ||||||
| #undef SPI_DAT | #undef SPI_DAT | ||||||
| #undef SPI_BASE | #undef SPI_BASE | ||||||
| }; | }; | ||||||
| @@ -124,7 +126,7 @@ void BekenSPILEDStripLightOutput::setup() { | |||||||
|   size_t buffer_size = this->get_buffer_size_(); |   size_t buffer_size = this->get_buffer_size_(); | ||||||
|   size_t dma_buffer_size = (buffer_size * 8) + (2 * 64); |   size_t dma_buffer_size = (buffer_size * 8) + (2 * 64); | ||||||
|  |  | ||||||
|   ExternalRAMAllocator<uint8_t> allocator(ExternalRAMAllocator<uint8_t>::ALLOW_FAILURE); |   RAMAllocator<uint8_t> allocator; | ||||||
|   this->buf_ = allocator.allocate(buffer_size); |   this->buf_ = allocator.allocate(buffer_size); | ||||||
|   if (this->buf_ == nullptr) { |   if (this->buf_ == nullptr) { | ||||||
|     ESP_LOGE(TAG, "Cannot allocate LED buffer!"); |     ESP_LOGE(TAG, "Cannot allocate LED buffer!"); | ||||||
|   | |||||||
| @@ -50,7 +50,7 @@ void BH1750Sensor::read_lx_(BH1750Mode mode, uint8_t mtreg, const std::function< | |||||||
|   // turn on (after one-shot sensor automatically powers down) |   // turn on (after one-shot sensor automatically powers down) | ||||||
|   uint8_t turn_on = BH1750_COMMAND_POWER_ON; |   uint8_t turn_on = BH1750_COMMAND_POWER_ON; | ||||||
|   if (this->write(&turn_on, 1) != i2c::ERROR_OK) { |   if (this->write(&turn_on, 1) != i2c::ERROR_OK) { | ||||||
|     ESP_LOGW(TAG, "Turning on BH1750 failed"); |     ESP_LOGW(TAG, "Power on failed"); | ||||||
|     f(NAN); |     f(NAN); | ||||||
|     return; |     return; | ||||||
|   } |   } | ||||||
| @@ -60,7 +60,7 @@ void BH1750Sensor::read_lx_(BH1750Mode mode, uint8_t mtreg, const std::function< | |||||||
|     uint8_t mtreg_hi = BH1750_COMMAND_MT_REG_HI | ((mtreg >> 5) & 0b111); |     uint8_t mtreg_hi = BH1750_COMMAND_MT_REG_HI | ((mtreg >> 5) & 0b111); | ||||||
|     uint8_t mtreg_lo = BH1750_COMMAND_MT_REG_LO | ((mtreg >> 0) & 0b11111); |     uint8_t mtreg_lo = BH1750_COMMAND_MT_REG_LO | ((mtreg >> 0) & 0b11111); | ||||||
|     if (this->write(&mtreg_hi, 1) != i2c::ERROR_OK || this->write(&mtreg_lo, 1) != i2c::ERROR_OK) { |     if (this->write(&mtreg_hi, 1) != i2c::ERROR_OK || this->write(&mtreg_lo, 1) != i2c::ERROR_OK) { | ||||||
|       ESP_LOGW(TAG, "Setting measurement time for BH1750 failed"); |       ESP_LOGW(TAG, "Set measurement time failed"); | ||||||
|       active_mtreg_ = 0; |       active_mtreg_ = 0; | ||||||
|       f(NAN); |       f(NAN); | ||||||
|       return; |       return; | ||||||
| @@ -88,7 +88,7 @@ void BH1750Sensor::read_lx_(BH1750Mode mode, uint8_t mtreg, const std::function< | |||||||
|       return; |       return; | ||||||
|   } |   } | ||||||
|   if (this->write(&cmd, 1) != i2c::ERROR_OK) { |   if (this->write(&cmd, 1) != i2c::ERROR_OK) { | ||||||
|     ESP_LOGW(TAG, "Starting measurement for BH1750 failed"); |     ESP_LOGW(TAG, "Start measurement failed"); | ||||||
|     f(NAN); |     f(NAN); | ||||||
|     return; |     return; | ||||||
|   } |   } | ||||||
| @@ -99,7 +99,7 @@ void BH1750Sensor::read_lx_(BH1750Mode mode, uint8_t mtreg, const std::function< | |||||||
|   this->set_timeout("read", meas_time, [this, mode, mtreg, f]() { |   this->set_timeout("read", meas_time, [this, mode, mtreg, f]() { | ||||||
|     uint16_t raw_value; |     uint16_t raw_value; | ||||||
|     if (this->read(reinterpret_cast<uint8_t *>(&raw_value), 2) != i2c::ERROR_OK) { |     if (this->read(reinterpret_cast<uint8_t *>(&raw_value), 2) != i2c::ERROR_OK) { | ||||||
|       ESP_LOGW(TAG, "Reading BH1750 data failed"); |       ESP_LOGW(TAG, "Read data failed"); | ||||||
|       f(NAN); |       f(NAN); | ||||||
|       return; |       return; | ||||||
|     } |     } | ||||||
| @@ -156,7 +156,7 @@ void BH1750Sensor::update() { | |||||||
|         this->publish_state(NAN); |         this->publish_state(NAN); | ||||||
|         return; |         return; | ||||||
|       } |       } | ||||||
|       ESP_LOGD(TAG, "'%s': Got illuminance=%.1flx", this->get_name().c_str(), val); |       ESP_LOGD(TAG, "'%s': Illuminance=%.1flx", this->get_name().c_str(), val); | ||||||
|       this->status_clear_warning(); |       this->status_clear_warning(); | ||||||
|       this->publish_state(val); |       this->publish_state(val); | ||||||
|     }); |     }); | ||||||
|   | |||||||
| @@ -1,7 +1,10 @@ | |||||||
|  | from logging import getLogger | ||||||
|  |  | ||||||
| from esphome import automation, core | from esphome import automation, core | ||||||
| from esphome.automation import Condition, maybe_simple_id | from esphome.automation import Condition, maybe_simple_id | ||||||
| import esphome.codegen as cg | import esphome.codegen as cg | ||||||
| from esphome.components import mqtt, web_server | from esphome.components import mqtt, web_server | ||||||
|  | from esphome.components.const import CONF_ON_STATE_CHANGE | ||||||
| import esphome.config_validation as cv | import esphome.config_validation as cv | ||||||
| from esphome.const import ( | from esphome.const import ( | ||||||
|     CONF_DELAY, |     CONF_DELAY, | ||||||
| @@ -57,8 +60,8 @@ from esphome.const import ( | |||||||
|     DEVICE_CLASS_WINDOW, |     DEVICE_CLASS_WINDOW, | ||||||
| ) | ) | ||||||
| from esphome.core import CORE, coroutine_with_priority | from esphome.core import CORE, coroutine_with_priority | ||||||
|  | from esphome.core.entity_helpers import entity_duplicate_validator, setup_entity | ||||||
| from esphome.cpp_generator import MockObjClass | from esphome.cpp_generator import MockObjClass | ||||||
| from esphome.cpp_helpers import setup_entity |  | ||||||
| from esphome.util import Registry | from esphome.util import Registry | ||||||
|  |  | ||||||
| CODEOWNERS = ["@esphome/core"] | CODEOWNERS = ["@esphome/core"] | ||||||
| @@ -98,6 +101,7 @@ IS_PLATFORM_COMPONENT = True | |||||||
|  |  | ||||||
| CONF_TIME_OFF = "time_off" | CONF_TIME_OFF = "time_off" | ||||||
| CONF_TIME_ON = "time_on" | CONF_TIME_ON = "time_on" | ||||||
|  | CONF_TRIGGER_ON_INITIAL_STATE = "trigger_on_initial_state" | ||||||
|  |  | ||||||
| DEFAULT_DELAY = "1s" | DEFAULT_DELAY = "1s" | ||||||
| DEFAULT_TIME_OFF = "100ms" | DEFAULT_TIME_OFF = "100ms" | ||||||
| @@ -127,15 +131,24 @@ MultiClickTriggerEvent = binary_sensor_ns.struct("MultiClickTriggerEvent") | |||||||
| StateTrigger = binary_sensor_ns.class_( | StateTrigger = binary_sensor_ns.class_( | ||||||
|     "StateTrigger", automation.Trigger.template(bool) |     "StateTrigger", automation.Trigger.template(bool) | ||||||
| ) | ) | ||||||
|  | StateChangeTrigger = binary_sensor_ns.class_( | ||||||
|  |     "StateChangeTrigger", | ||||||
|  |     automation.Trigger.template(cg.optional.template(bool), cg.optional.template(bool)), | ||||||
|  | ) | ||||||
|  |  | ||||||
| BinarySensorPublishAction = binary_sensor_ns.class_( | BinarySensorPublishAction = binary_sensor_ns.class_( | ||||||
|     "BinarySensorPublishAction", automation.Action |     "BinarySensorPublishAction", automation.Action | ||||||
| ) | ) | ||||||
|  | BinarySensorInvalidateAction = binary_sensor_ns.class_( | ||||||
|  |     "BinarySensorInvalidateAction", automation.Action | ||||||
|  | ) | ||||||
|  |  | ||||||
| # Condition | # Condition | ||||||
| BinarySensorCondition = binary_sensor_ns.class_("BinarySensorCondition", Condition) | BinarySensorCondition = binary_sensor_ns.class_("BinarySensorCondition", Condition) | ||||||
|  |  | ||||||
| # Filters | # Filters | ||||||
| Filter = binary_sensor_ns.class_("Filter") | Filter = binary_sensor_ns.class_("Filter") | ||||||
|  | TimeoutFilter = binary_sensor_ns.class_("TimeoutFilter", Filter, cg.Component) | ||||||
| DelayedOnOffFilter = binary_sensor_ns.class_("DelayedOnOffFilter", Filter, cg.Component) | DelayedOnOffFilter = binary_sensor_ns.class_("DelayedOnOffFilter", Filter, cg.Component) | ||||||
| DelayedOnFilter = binary_sensor_ns.class_("DelayedOnFilter", Filter, cg.Component) | DelayedOnFilter = binary_sensor_ns.class_("DelayedOnFilter", Filter, cg.Component) | ||||||
| DelayedOffFilter = binary_sensor_ns.class_("DelayedOffFilter", Filter, cg.Component) | DelayedOffFilter = binary_sensor_ns.class_("DelayedOffFilter", Filter, cg.Component) | ||||||
| @@ -144,6 +157,8 @@ AutorepeatFilter = binary_sensor_ns.class_("AutorepeatFilter", Filter, cg.Compon | |||||||
| LambdaFilter = binary_sensor_ns.class_("LambdaFilter", Filter) | LambdaFilter = binary_sensor_ns.class_("LambdaFilter", Filter) | ||||||
| SettleFilter = binary_sensor_ns.class_("SettleFilter", Filter, cg.Component) | SettleFilter = binary_sensor_ns.class_("SettleFilter", Filter, cg.Component) | ||||||
|  |  | ||||||
|  | _LOGGER = getLogger(__name__) | ||||||
|  |  | ||||||
| FILTER_REGISTRY = Registry() | FILTER_REGISTRY = Registry() | ||||||
| validate_filters = cv.validate_registry("filter", FILTER_REGISTRY) | validate_filters = cv.validate_registry("filter", FILTER_REGISTRY) | ||||||
|  |  | ||||||
| @@ -157,6 +172,19 @@ async def invert_filter_to_code(config, filter_id): | |||||||
|     return cg.new_Pvariable(filter_id) |     return cg.new_Pvariable(filter_id) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | @register_filter( | ||||||
|  |     "timeout", | ||||||
|  |     TimeoutFilter, | ||||||
|  |     cv.templatable(cv.positive_time_period_milliseconds), | ||||||
|  | ) | ||||||
|  | async def timeout_filter_to_code(config, filter_id): | ||||||
|  |     var = cg.new_Pvariable(filter_id) | ||||||
|  |     await cg.register_component(var, {}) | ||||||
|  |     template_ = await cg.templatable(config, [], cg.uint32) | ||||||
|  |     cg.add(var.set_timeout_value(template_)) | ||||||
|  |     return var | ||||||
|  |  | ||||||
|  |  | ||||||
| @register_filter( | @register_filter( | ||||||
|     "delayed_on_off", |     "delayed_on_off", | ||||||
|     DelayedOnOffFilter, |     DelayedOnOffFilter, | ||||||
| @@ -386,6 +414,14 @@ def validate_click_timing(value): | |||||||
|     return value |     return value | ||||||
|  |  | ||||||
|  |  | ||||||
|  | def validate_publish_initial_state(value): | ||||||
|  |     value = cv.boolean(value) | ||||||
|  |     _LOGGER.warning( | ||||||
|  |         "The 'publish_initial_state' option has been replaced by 'trigger_on_initial_state' and will be removed in a future release" | ||||||
|  |     ) | ||||||
|  |     return value | ||||||
|  |  | ||||||
|  |  | ||||||
| _BINARY_SENSOR_SCHEMA = ( | _BINARY_SENSOR_SCHEMA = ( | ||||||
|     cv.ENTITY_BASE_SCHEMA.extend(web_server.WEBSERVER_SORTING_SCHEMA) |     cv.ENTITY_BASE_SCHEMA.extend(web_server.WEBSERVER_SORTING_SCHEMA) | ||||||
|     .extend(cv.MQTT_COMPONENT_SCHEMA) |     .extend(cv.MQTT_COMPONENT_SCHEMA) | ||||||
| @@ -395,7 +431,12 @@ _BINARY_SENSOR_SCHEMA = ( | |||||||
|             cv.OnlyWith(CONF_MQTT_ID, "mqtt"): cv.declare_id( |             cv.OnlyWith(CONF_MQTT_ID, "mqtt"): cv.declare_id( | ||||||
|                 mqtt.MQTTBinarySensorComponent |                 mqtt.MQTTBinarySensorComponent | ||||||
|             ), |             ), | ||||||
|             cv.Optional(CONF_PUBLISH_INITIAL_STATE): cv.boolean, |             cv.Exclusive( | ||||||
|  |                 CONF_PUBLISH_INITIAL_STATE, CONF_TRIGGER_ON_INITIAL_STATE | ||||||
|  |             ): validate_publish_initial_state, | ||||||
|  |             cv.Exclusive( | ||||||
|  |                 CONF_TRIGGER_ON_INITIAL_STATE, CONF_TRIGGER_ON_INITIAL_STATE | ||||||
|  |             ): cv.boolean, | ||||||
|             cv.Optional(CONF_DEVICE_CLASS): validate_device_class, |             cv.Optional(CONF_DEVICE_CLASS): validate_device_class, | ||||||
|             cv.Optional(CONF_FILTERS): validate_filters, |             cv.Optional(CONF_FILTERS): validate_filters, | ||||||
|             cv.Optional(CONF_ON_PRESS): automation.validate_automation( |             cv.Optional(CONF_ON_PRESS): automation.validate_automation( | ||||||
| @@ -454,11 +495,19 @@ _BINARY_SENSOR_SCHEMA = ( | |||||||
|                     cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(StateTrigger), |                     cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(StateTrigger), | ||||||
|                 } |                 } | ||||||
|             ), |             ), | ||||||
|  |             cv.Optional(CONF_ON_STATE_CHANGE): automation.validate_automation( | ||||||
|  |                 { | ||||||
|  |                     cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(StateChangeTrigger), | ||||||
|  |                 } | ||||||
|  |             ), | ||||||
|         } |         } | ||||||
|     ) |     ) | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | _BINARY_SENSOR_SCHEMA.add_extra(entity_duplicate_validator("binary_sensor")) | ||||||
|  |  | ||||||
|  |  | ||||||
| def binary_sensor_schema( | def binary_sensor_schema( | ||||||
|     class_: MockObjClass = cv.UNDEFINED, |     class_: MockObjClass = cv.UNDEFINED, | ||||||
|     *, |     *, | ||||||
| @@ -489,12 +538,14 @@ BINARY_SENSOR_SCHEMA.add_extra(cv.deprecated_schema_constant("binary_sensor")) | |||||||
|  |  | ||||||
|  |  | ||||||
| async def setup_binary_sensor_core_(var, config): | async def setup_binary_sensor_core_(var, config): | ||||||
|     await setup_entity(var, config) |     await setup_entity(var, config, "binary_sensor") | ||||||
|  |  | ||||||
|     if (device_class := config.get(CONF_DEVICE_CLASS)) is not None: |     if (device_class := config.get(CONF_DEVICE_CLASS)) is not None: | ||||||
|         cg.add(var.set_device_class(device_class)) |         cg.add(var.set_device_class(device_class)) | ||||||
|     if publish_initial_state := config.get(CONF_PUBLISH_INITIAL_STATE): |     trigger = config.get(CONF_TRIGGER_ON_INITIAL_STATE, False) or config.get( | ||||||
|         cg.add(var.set_publish_initial_state(publish_initial_state)) |         CONF_PUBLISH_INITIAL_STATE, False | ||||||
|  |     ) | ||||||
|  |     cg.add(var.set_trigger_on_initial_state(trigger)) | ||||||
|     if inverted := config.get(CONF_INVERTED): |     if inverted := config.get(CONF_INVERTED): | ||||||
|         cg.add(var.set_inverted(inverted)) |         cg.add(var.set_inverted(inverted)) | ||||||
|     if filters_config := config.get(CONF_FILTERS): |     if filters_config := config.get(CONF_FILTERS): | ||||||
| @@ -542,6 +593,17 @@ async def setup_binary_sensor_core_(var, config): | |||||||
|         trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var) |         trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var) | ||||||
|         await automation.build_automation(trigger, [(bool, "x")], conf) |         await automation.build_automation(trigger, [(bool, "x")], conf) | ||||||
|  |  | ||||||
|  |     for conf in config.get(CONF_ON_STATE_CHANGE, []): | ||||||
|  |         trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var) | ||||||
|  |         await automation.build_automation( | ||||||
|  |             trigger, | ||||||
|  |             [ | ||||||
|  |                 (cg.optional.template(bool), "x_previous"), | ||||||
|  |                 (cg.optional.template(bool), "x"), | ||||||
|  |             ], | ||||||
|  |             conf, | ||||||
|  |         ) | ||||||
|  |  | ||||||
|     if mqtt_id := config.get(CONF_MQTT_ID): |     if mqtt_id := config.get(CONF_MQTT_ID): | ||||||
|         mqtt_ = cg.new_Pvariable(mqtt_id, var) |         mqtt_ = cg.new_Pvariable(mqtt_id, var) | ||||||
|         await mqtt.register_mqtt_component(mqtt_, config) |         await mqtt.register_mqtt_component(mqtt_, config) | ||||||
| @@ -591,3 +653,18 @@ async def binary_sensor_is_off_to_code(config, condition_id, template_arg, args) | |||||||
| async def to_code(config): | async def to_code(config): | ||||||
|     cg.add_define("USE_BINARY_SENSOR") |     cg.add_define("USE_BINARY_SENSOR") | ||||||
|     cg.add_global(binary_sensor_ns.using) |     cg.add_global(binary_sensor_ns.using) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | @automation.register_action( | ||||||
|  |     "binary_sensor.invalidate_state", | ||||||
|  |     BinarySensorInvalidateAction, | ||||||
|  |     cv.maybe_simple_value( | ||||||
|  |         { | ||||||
|  |             cv.Required(CONF_ID): cv.use_id(BinarySensor), | ||||||
|  |         }, | ||||||
|  |         key=CONF_ID, | ||||||
|  |     ), | ||||||
|  | ) | ||||||
|  | async def binary_sensor_invalidate_state_to_code(config, action_id, template_arg, args): | ||||||
|  |     paren = await cg.get_variable(config[CONF_ID]) | ||||||
|  |     return cg.new_Pvariable(action_id, template_arg, paren) | ||||||
|   | |||||||
| @@ -96,7 +96,7 @@ class MultiClickTrigger : public Trigger<>, public Component { | |||||||
|       : parent_(parent), timing_(std::move(timing)) {} |       : parent_(parent), timing_(std::move(timing)) {} | ||||||
|  |  | ||||||
|   void setup() override { |   void setup() override { | ||||||
|     this->last_state_ = this->parent_->state; |     this->last_state_ = this->parent_->get_state_default(false); | ||||||
|     auto f = std::bind(&MultiClickTrigger::on_state_, this, std::placeholders::_1); |     auto f = std::bind(&MultiClickTrigger::on_state_, this, std::placeholders::_1); | ||||||
|     this->parent_->add_on_state_callback(f); |     this->parent_->add_on_state_callback(f); | ||||||
|   } |   } | ||||||
| @@ -130,6 +130,14 @@ class StateTrigger : public Trigger<bool> { | |||||||
|   } |   } | ||||||
| }; | }; | ||||||
|  |  | ||||||
|  | class StateChangeTrigger : public Trigger<optional<bool>, optional<bool> > { | ||||||
|  |  public: | ||||||
|  |   explicit StateChangeTrigger(BinarySensor *parent) { | ||||||
|  |     parent->add_full_state_callback( | ||||||
|  |         [this](optional<bool> old_state, optional<bool> state) { this->trigger(old_state, state); }); | ||||||
|  |   } | ||||||
|  | }; | ||||||
|  |  | ||||||
| template<typename... Ts> class BinarySensorCondition : public Condition<Ts...> { | template<typename... Ts> class BinarySensorCondition : public Condition<Ts...> { | ||||||
|  public: |  public: | ||||||
|   BinarySensorCondition(BinarySensor *parent, bool state) : parent_(parent), state_(state) {} |   BinarySensorCondition(BinarySensor *parent, bool state) : parent_(parent), state_(state) {} | ||||||
| @@ -154,5 +162,15 @@ template<typename... Ts> class BinarySensorPublishAction : public Action<Ts...> | |||||||
|   BinarySensor *sensor_; |   BinarySensor *sensor_; | ||||||
| }; | }; | ||||||
|  |  | ||||||
|  | template<typename... Ts> class BinarySensorInvalidateAction : public Action<Ts...> { | ||||||
|  |  public: | ||||||
|  |   explicit BinarySensorInvalidateAction(BinarySensor *sensor) : sensor_(sensor) {} | ||||||
|  |  | ||||||
|  |   void play(Ts... x) override { this->sensor_->invalidate_state(); } | ||||||
|  |  | ||||||
|  |  protected: | ||||||
|  |   BinarySensor *sensor_; | ||||||
|  | }; | ||||||
|  |  | ||||||
| }  // namespace binary_sensor | }  // namespace binary_sensor | ||||||
| }  // namespace esphome | }  // namespace esphome | ||||||
|   | |||||||
| @@ -7,42 +7,25 @@ namespace binary_sensor { | |||||||
|  |  | ||||||
| static const char *const TAG = "binary_sensor"; | static const char *const TAG = "binary_sensor"; | ||||||
|  |  | ||||||
| void BinarySensor::add_on_state_callback(std::function<void(bool)> &&callback) { | void BinarySensor::publish_state(bool new_state) { | ||||||
|   this->state_callback_.add(std::move(callback)); |  | ||||||
| } |  | ||||||
|  |  | ||||||
| void BinarySensor::publish_state(bool state) { |  | ||||||
|   if (!this->publish_dedup_.next(state)) |  | ||||||
|     return; |  | ||||||
|   if (this->filter_list_ == nullptr) { |   if (this->filter_list_ == nullptr) { | ||||||
|     this->send_state_internal(state, false); |     this->send_state_internal(new_state); | ||||||
|   } else { |   } else { | ||||||
|     this->filter_list_->input(state, false); |     this->filter_list_->input(new_state); | ||||||
|   } |   } | ||||||
| } | } | ||||||
| void BinarySensor::publish_initial_state(bool state) { | void BinarySensor::publish_initial_state(bool new_state) { | ||||||
|   if (!this->publish_dedup_.next(state)) |   this->invalidate_state(); | ||||||
|     return; |   this->publish_state(new_state); | ||||||
|   if (this->filter_list_ == nullptr) { | } | ||||||
|     this->send_state_internal(state, true); | void BinarySensor::send_state_internal(bool new_state) { | ||||||
|   } else { |   // copy the new state to the visible property for backwards compatibility, before any callbacks | ||||||
|     this->filter_list_->input(state, true); |   this->state = new_state; | ||||||
|  |   // Note that set_state_ de-dups and will only trigger callbacks if the state has actually changed | ||||||
|  |   if (this->set_state_(new_state)) { | ||||||
|  |     ESP_LOGD(TAG, "'%s': New state is %s", this->get_name().c_str(), ONOFF(new_state)); | ||||||
|   } |   } | ||||||
| } | } | ||||||
| void BinarySensor::send_state_internal(bool state, bool is_initial) { |  | ||||||
|   if (is_initial) { |  | ||||||
|     ESP_LOGD(TAG, "'%s': Sending initial state %s", this->get_name().c_str(), ONOFF(state)); |  | ||||||
|   } else { |  | ||||||
|     ESP_LOGD(TAG, "'%s': Sending state %s", this->get_name().c_str(), ONOFF(state)); |  | ||||||
|   } |  | ||||||
|   this->has_state_ = true; |  | ||||||
|   this->state = state; |  | ||||||
|   if (!is_initial || this->publish_initial_state_) { |  | ||||||
|     this->state_callback_.call(state); |  | ||||||
|   } |  | ||||||
| } |  | ||||||
|  |  | ||||||
| BinarySensor::BinarySensor() : state(false) {} |  | ||||||
|  |  | ||||||
| void BinarySensor::add_filter(Filter *filter) { | void BinarySensor::add_filter(Filter *filter) { | ||||||
|   filter->parent_ = this; |   filter->parent_ = this; | ||||||
| @@ -60,7 +43,6 @@ void BinarySensor::add_filters(const std::vector<Filter *> &filters) { | |||||||
|     this->add_filter(filter); |     this->add_filter(filter); | ||||||
|   } |   } | ||||||
| } | } | ||||||
| bool BinarySensor::has_state() const { return this->has_state_; } |  | ||||||
| bool BinarySensor::is_status_binary_sensor() const { return false; } | bool BinarySensor::is_status_binary_sensor() const { return false; } | ||||||
|  |  | ||||||
| }  // namespace binary_sensor | }  // namespace binary_sensor | ||||||
|   | |||||||
| @@ -1,6 +1,5 @@ | |||||||
| #pragma once | #pragma once | ||||||
|  |  | ||||||
| #include "esphome/core/component.h" |  | ||||||
| #include "esphome/core/entity_base.h" | #include "esphome/core/entity_base.h" | ||||||
| #include "esphome/core/helpers.h" | #include "esphome/core/helpers.h" | ||||||
| #include "esphome/components/binary_sensor/filter.h" | #include "esphome/components/binary_sensor/filter.h" | ||||||
| @@ -34,52 +33,39 @@ namespace binary_sensor { | |||||||
|  * The sub classes should notify the front-end of new states via the publish_state() method which |  * The sub classes should notify the front-end of new states via the publish_state() method which | ||||||
|  * handles inverted inputs for you. |  * handles inverted inputs for you. | ||||||
|  */ |  */ | ||||||
| class BinarySensor : public EntityBase, public EntityBase_DeviceClass { | class BinarySensor : public StatefulEntityBase<bool>, public EntityBase_DeviceClass { | ||||||
|  public: |  public: | ||||||
|   explicit BinarySensor(); |   explicit BinarySensor(){}; | ||||||
|  |  | ||||||
|   /** Add a callback to be notified of state changes. |  | ||||||
|    * |  | ||||||
|    * @param callback The void(bool) callback. |  | ||||||
|    */ |  | ||||||
|   void add_on_state_callback(std::function<void(bool)> &&callback); |  | ||||||
|  |  | ||||||
|   /** Publish a new state to the front-end. |   /** Publish a new state to the front-end. | ||||||
|    * |    * | ||||||
|    * @param state The new state. |    * @param new_state The new state. | ||||||
|    */ |    */ | ||||||
|   void publish_state(bool state); |   void publish_state(bool new_state); | ||||||
|  |  | ||||||
|   /** Publish the initial state, this will not make the callback manager send callbacks |   /** Publish the initial state, this will not make the callback manager send callbacks | ||||||
|    * and is meant only for the initial state on boot. |    * and is meant only for the initial state on boot. | ||||||
|    * |    * | ||||||
|    * @param state The new state. |    * @param new_state The new state. | ||||||
|    */ |    */ | ||||||
|   void publish_initial_state(bool state); |   void publish_initial_state(bool new_state); | ||||||
|  |  | ||||||
|   /// The current reported state of the binary sensor. |  | ||||||
|   bool state{false}; |  | ||||||
|  |  | ||||||
|   void add_filter(Filter *filter); |   void add_filter(Filter *filter); | ||||||
|   void add_filters(const std::vector<Filter *> &filters); |   void add_filters(const std::vector<Filter *> &filters); | ||||||
|  |  | ||||||
|   void set_publish_initial_state(bool publish_initial_state) { this->publish_initial_state_ = publish_initial_state; } |  | ||||||
|  |  | ||||||
|   // ========== INTERNAL METHODS ========== |   // ========== INTERNAL METHODS ========== | ||||||
|   // (In most use cases you won't need these) |   // (In most use cases you won't need these) | ||||||
|   void send_state_internal(bool state, bool is_initial); |   void send_state_internal(bool new_state); | ||||||
|  |  | ||||||
|   /// Return whether this binary sensor has outputted a state. |   /// Return whether this binary sensor has outputted a state. | ||||||
|   virtual bool has_state() const; |  | ||||||
|  |  | ||||||
|   virtual bool is_status_binary_sensor() const; |   virtual bool is_status_binary_sensor() const; | ||||||
|  |  | ||||||
|  |   // For backward compatibility, provide an accessible property | ||||||
|  |  | ||||||
|  |   bool state{}; | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   CallbackManager<void(bool)> state_callback_{}; |  | ||||||
|   Filter *filter_list_{nullptr}; |   Filter *filter_list_{nullptr}; | ||||||
|   bool has_state_{false}; |  | ||||||
|   bool publish_initial_state_{false}; |  | ||||||
|   Deduplicator<bool> publish_dedup_; |  | ||||||
| }; | }; | ||||||
|  |  | ||||||
| class BinarySensorInitiallyOff : public BinarySensor { | class BinarySensorInitiallyOff : public BinarySensor { | ||||||
|   | |||||||
| @@ -9,37 +9,42 @@ namespace binary_sensor { | |||||||
|  |  | ||||||
| static const char *const TAG = "sensor.filter"; | static const char *const TAG = "sensor.filter"; | ||||||
|  |  | ||||||
| void Filter::output(bool value, bool is_initial) { | void Filter::output(bool value) { | ||||||
|  |   if (this->next_ == nullptr) { | ||||||
|  |     this->parent_->send_state_internal(value); | ||||||
|  |   } else { | ||||||
|  |     this->next_->input(value); | ||||||
|  |   } | ||||||
|  | } | ||||||
|  | void Filter::input(bool value) { | ||||||
|   if (!this->dedup_.next(value)) |   if (!this->dedup_.next(value)) | ||||||
|     return; |     return; | ||||||
|  |   auto b = this->new_value(value); | ||||||
|   if (this->next_ == nullptr) { |  | ||||||
|     this->parent_->send_state_internal(value, is_initial); |  | ||||||
|   } else { |  | ||||||
|     this->next_->input(value, is_initial); |  | ||||||
|   } |  | ||||||
| } |  | ||||||
| void Filter::input(bool value, bool is_initial) { |  | ||||||
|   auto b = this->new_value(value, is_initial); |  | ||||||
|   if (b.has_value()) { |   if (b.has_value()) { | ||||||
|     this->output(*b, is_initial); |     this->output(*b); | ||||||
|   } |   } | ||||||
| } | } | ||||||
|  |  | ||||||
| optional<bool> DelayedOnOffFilter::new_value(bool value, bool is_initial) { | void TimeoutFilter::input(bool value) { | ||||||
|  |   this->set_timeout("timeout", this->timeout_delay_.value(), [this]() { this->parent_->invalidate_state(); }); | ||||||
|  |   // we do not de-dup here otherwise changes from invalid to valid state will not be output | ||||||
|  |   this->output(value); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | optional<bool> DelayedOnOffFilter::new_value(bool value) { | ||||||
|   if (value) { |   if (value) { | ||||||
|     this->set_timeout("ON_OFF", this->on_delay_.value(), [this, is_initial]() { this->output(true, is_initial); }); |     this->set_timeout("ON_OFF", this->on_delay_.value(), [this]() { this->output(true); }); | ||||||
|   } else { |   } else { | ||||||
|     this->set_timeout("ON_OFF", this->off_delay_.value(), [this, is_initial]() { this->output(false, is_initial); }); |     this->set_timeout("ON_OFF", this->off_delay_.value(), [this]() { this->output(false); }); | ||||||
|   } |   } | ||||||
|   return {}; |   return {}; | ||||||
| } | } | ||||||
|  |  | ||||||
| float DelayedOnOffFilter::get_setup_priority() const { return setup_priority::HARDWARE; } | float DelayedOnOffFilter::get_setup_priority() const { return setup_priority::HARDWARE; } | ||||||
|  |  | ||||||
| optional<bool> DelayedOnFilter::new_value(bool value, bool is_initial) { | optional<bool> DelayedOnFilter::new_value(bool value) { | ||||||
|   if (value) { |   if (value) { | ||||||
|     this->set_timeout("ON", this->delay_.value(), [this, is_initial]() { this->output(true, is_initial); }); |     this->set_timeout("ON", this->delay_.value(), [this]() { this->output(true); }); | ||||||
|     return {}; |     return {}; | ||||||
|   } else { |   } else { | ||||||
|     this->cancel_timeout("ON"); |     this->cancel_timeout("ON"); | ||||||
| @@ -49,9 +54,9 @@ optional<bool> DelayedOnFilter::new_value(bool value, bool is_initial) { | |||||||
|  |  | ||||||
| float DelayedOnFilter::get_setup_priority() const { return setup_priority::HARDWARE; } | float DelayedOnFilter::get_setup_priority() const { return setup_priority::HARDWARE; } | ||||||
|  |  | ||||||
| optional<bool> DelayedOffFilter::new_value(bool value, bool is_initial) { | optional<bool> DelayedOffFilter::new_value(bool value) { | ||||||
|   if (!value) { |   if (!value) { | ||||||
|     this->set_timeout("OFF", this->delay_.value(), [this, is_initial]() { this->output(false, is_initial); }); |     this->set_timeout("OFF", this->delay_.value(), [this]() { this->output(false); }); | ||||||
|     return {}; |     return {}; | ||||||
|   } else { |   } else { | ||||||
|     this->cancel_timeout("OFF"); |     this->cancel_timeout("OFF"); | ||||||
| @@ -61,11 +66,11 @@ optional<bool> DelayedOffFilter::new_value(bool value, bool is_initial) { | |||||||
|  |  | ||||||
| float DelayedOffFilter::get_setup_priority() const { return setup_priority::HARDWARE; } | float DelayedOffFilter::get_setup_priority() const { return setup_priority::HARDWARE; } | ||||||
|  |  | ||||||
| optional<bool> InvertFilter::new_value(bool value, bool is_initial) { return !value; } | optional<bool> InvertFilter::new_value(bool value) { return !value; } | ||||||
|  |  | ||||||
| AutorepeatFilter::AutorepeatFilter(std::vector<AutorepeatFilterTiming> timings) : timings_(std::move(timings)) {} | AutorepeatFilter::AutorepeatFilter(std::vector<AutorepeatFilterTiming> timings) : timings_(std::move(timings)) {} | ||||||
|  |  | ||||||
| optional<bool> AutorepeatFilter::new_value(bool value, bool is_initial) { | optional<bool> AutorepeatFilter::new_value(bool value) { | ||||||
|   if (value) { |   if (value) { | ||||||
|     // Ignore if already running |     // Ignore if already running | ||||||
|     if (this->active_timing_ != 0) |     if (this->active_timing_ != 0) | ||||||
| @@ -101,7 +106,7 @@ void AutorepeatFilter::next_timing_() { | |||||||
|  |  | ||||||
| void AutorepeatFilter::next_value_(bool val) { | void AutorepeatFilter::next_value_(bool val) { | ||||||
|   const AutorepeatFilterTiming &timing = this->timings_[this->active_timing_ - 2]; |   const AutorepeatFilterTiming &timing = this->timings_[this->active_timing_ - 2]; | ||||||
|   this->output(val, false);  // This is at least the second one so not initial |   this->output(val);  // This is at least the second one so not initial | ||||||
|   this->set_timeout("ON_OFF", val ? timing.time_on : timing.time_off, [this, val]() { this->next_value_(!val); }); |   this->set_timeout("ON_OFF", val ? timing.time_on : timing.time_off, [this, val]() { this->next_value_(!val); }); | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -109,18 +114,18 @@ float AutorepeatFilter::get_setup_priority() const { return setup_priority::HARD | |||||||
|  |  | ||||||
| LambdaFilter::LambdaFilter(std::function<optional<bool>(bool)> f) : f_(std::move(f)) {} | LambdaFilter::LambdaFilter(std::function<optional<bool>(bool)> f) : f_(std::move(f)) {} | ||||||
|  |  | ||||||
| optional<bool> LambdaFilter::new_value(bool value, bool is_initial) { return this->f_(value); } | optional<bool> LambdaFilter::new_value(bool value) { return this->f_(value); } | ||||||
|  |  | ||||||
| optional<bool> SettleFilter::new_value(bool value, bool is_initial) { | optional<bool> SettleFilter::new_value(bool value) { | ||||||
|   if (!this->steady_) { |   if (!this->steady_) { | ||||||
|     this->set_timeout("SETTLE", this->delay_.value(), [this, value, is_initial]() { |     this->set_timeout("SETTLE", this->delay_.value(), [this, value]() { | ||||||
|       this->steady_ = true; |       this->steady_ = true; | ||||||
|       this->output(value, is_initial); |       this->output(value); | ||||||
|     }); |     }); | ||||||
|     return {}; |     return {}; | ||||||
|   } else { |   } else { | ||||||
|     this->steady_ = false; |     this->steady_ = false; | ||||||
|     this->output(value, is_initial); |     this->output(value); | ||||||
|     this->set_timeout("SETTLE", this->delay_.value(), [this]() { this->steady_ = true; }); |     this->set_timeout("SETTLE", this->delay_.value(), [this]() { this->steady_ = true; }); | ||||||
|     return value; |     return value; | ||||||
|   } |   } | ||||||
|   | |||||||
| @@ -14,11 +14,11 @@ class BinarySensor; | |||||||
|  |  | ||||||
| class Filter { | class Filter { | ||||||
|  public: |  public: | ||||||
|   virtual optional<bool> new_value(bool value, bool is_initial) = 0; |   virtual optional<bool> new_value(bool value) = 0; | ||||||
|  |  | ||||||
|   void input(bool value, bool is_initial); |   virtual void input(bool value); | ||||||
|  |  | ||||||
|   void output(bool value, bool is_initial); |   void output(bool value); | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   friend BinarySensor; |   friend BinarySensor; | ||||||
| @@ -28,9 +28,19 @@ class Filter { | |||||||
|   Deduplicator<bool> dedup_; |   Deduplicator<bool> dedup_; | ||||||
| }; | }; | ||||||
|  |  | ||||||
|  | class TimeoutFilter : public Filter, public Component { | ||||||
|  |  public: | ||||||
|  |   optional<bool> new_value(bool value) override { return value; } | ||||||
|  |   void input(bool value) override; | ||||||
|  |   template<typename T> void set_timeout_value(T timeout) { this->timeout_delay_ = timeout; } | ||||||
|  |  | ||||||
|  |  protected: | ||||||
|  |   TemplatableValue<uint32_t> timeout_delay_{}; | ||||||
|  | }; | ||||||
|  |  | ||||||
| class DelayedOnOffFilter : public Filter, public Component { | class DelayedOnOffFilter : public Filter, public Component { | ||||||
|  public: |  public: | ||||||
|   optional<bool> new_value(bool value, bool is_initial) override; |   optional<bool> new_value(bool value) override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override; |   float get_setup_priority() const override; | ||||||
|  |  | ||||||
| @@ -44,7 +54,7 @@ class DelayedOnOffFilter : public Filter, public Component { | |||||||
|  |  | ||||||
| class DelayedOnFilter : public Filter, public Component { | class DelayedOnFilter : public Filter, public Component { | ||||||
|  public: |  public: | ||||||
|   optional<bool> new_value(bool value, bool is_initial) override; |   optional<bool> new_value(bool value) override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override; |   float get_setup_priority() const override; | ||||||
|  |  | ||||||
| @@ -56,7 +66,7 @@ class DelayedOnFilter : public Filter, public Component { | |||||||
|  |  | ||||||
| class DelayedOffFilter : public Filter, public Component { | class DelayedOffFilter : public Filter, public Component { | ||||||
|  public: |  public: | ||||||
|   optional<bool> new_value(bool value, bool is_initial) override; |   optional<bool> new_value(bool value) override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override; |   float get_setup_priority() const override; | ||||||
|  |  | ||||||
| @@ -68,7 +78,7 @@ class DelayedOffFilter : public Filter, public Component { | |||||||
|  |  | ||||||
| class InvertFilter : public Filter { | class InvertFilter : public Filter { | ||||||
|  public: |  public: | ||||||
|   optional<bool> new_value(bool value, bool is_initial) override; |   optional<bool> new_value(bool value) override; | ||||||
| }; | }; | ||||||
|  |  | ||||||
| struct AutorepeatFilterTiming { | struct AutorepeatFilterTiming { | ||||||
| @@ -86,7 +96,7 @@ class AutorepeatFilter : public Filter, public Component { | |||||||
|  public: |  public: | ||||||
|   explicit AutorepeatFilter(std::vector<AutorepeatFilterTiming> timings); |   explicit AutorepeatFilter(std::vector<AutorepeatFilterTiming> timings); | ||||||
|  |  | ||||||
|   optional<bool> new_value(bool value, bool is_initial) override; |   optional<bool> new_value(bool value) override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override; |   float get_setup_priority() const override; | ||||||
|  |  | ||||||
| @@ -102,7 +112,7 @@ class LambdaFilter : public Filter { | |||||||
|  public: |  public: | ||||||
|   explicit LambdaFilter(std::function<optional<bool>(bool)> f); |   explicit LambdaFilter(std::function<optional<bool>(bool)> f); | ||||||
|  |  | ||||||
|   optional<bool> new_value(bool value, bool is_initial) override; |   optional<bool> new_value(bool value) override; | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   std::function<optional<bool>(bool)> f_; |   std::function<optional<bool>(bool)> f_; | ||||||
| @@ -110,7 +120,7 @@ class LambdaFilter : public Filter { | |||||||
|  |  | ||||||
| class SettleFilter : public Filter, public Component { | class SettleFilter : public Filter, public Component { | ||||||
|  public: |  public: | ||||||
|   optional<bool> new_value(bool value, bool is_initial) override; |   optional<bool> new_value(bool value) override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override; |   float get_setup_priority() const override; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -16,7 +16,6 @@ class BLEBinaryOutput : public output::BinaryOutput, public BLEClientNode, publi | |||||||
|  public: |  public: | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   void loop() override {} |   void loop() override {} | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   void set_service_uuid16(uint16_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint16(uuid); } |   void set_service_uuid16(uint16_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint16(uuid); } | ||||||
|   void set_service_uuid32(uint32_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint32(uuid); } |   void set_service_uuid32(uint32_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint32(uuid); } | ||||||
|   void set_service_uuid128(uint8_t *uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_raw(uuid); } |   void set_service_uuid128(uint8_t *uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_raw(uuid); } | ||||||
|   | |||||||
| @@ -11,7 +11,11 @@ namespace ble_client { | |||||||
|  |  | ||||||
| static const char *const TAG = "ble_rssi_sensor"; | static const char *const TAG = "ble_rssi_sensor"; | ||||||
|  |  | ||||||
| void BLEClientRSSISensor::loop() {} | void BLEClientRSSISensor::loop() { | ||||||
|  |   // Parent BLEClientNode has a loop() method, but this component uses | ||||||
|  |   // polling via update() and BLE GAP callbacks so loop isn't needed | ||||||
|  |   this->disable_loop(); | ||||||
|  | } | ||||||
|  |  | ||||||
| void BLEClientRSSISensor::dump_config() { | void BLEClientRSSISensor::dump_config() { | ||||||
|   LOG_SENSOR("", "BLE Client RSSI Sensor", this); |   LOG_SENSOR("", "BLE Client RSSI Sensor", this); | ||||||
|   | |||||||
| @@ -18,7 +18,6 @@ class BLEClientRSSISensor : public sensor::Sensor, public PollingComponent, publ | |||||||
|   void loop() override; |   void loop() override; | ||||||
|   void update() override; |   void update() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   void gap_event_handler(esp_gap_ble_cb_event_t event, esp_ble_gap_cb_param_t *param) override; |   void gap_event_handler(esp_gap_ble_cb_event_t event, esp_ble_gap_cb_param_t *param) override; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -11,7 +11,11 @@ namespace ble_client { | |||||||
|  |  | ||||||
| static const char *const TAG = "ble_sensor"; | static const char *const TAG = "ble_sensor"; | ||||||
|  |  | ||||||
| void BLESensor::loop() {} | void BLESensor::loop() { | ||||||
|  |   // Parent BLEClientNode has a loop() method, but this component uses | ||||||
|  |   // polling via update() and BLE callbacks so loop isn't needed | ||||||
|  |   this->disable_loop(); | ||||||
|  | } | ||||||
|  |  | ||||||
| void BLESensor::dump_config() { | void BLESensor::dump_config() { | ||||||
|   LOG_SENSOR("", "BLE Sensor", this); |   LOG_SENSOR("", "BLE Sensor", this); | ||||||
|   | |||||||
| @@ -24,7 +24,6 @@ class BLESensor : public sensor::Sensor, public PollingComponent, public BLEClie | |||||||
|   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, |   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, | ||||||
|                            esp_ble_gattc_cb_param_t *param) override; |                            esp_ble_gattc_cb_param_t *param) override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   void set_service_uuid16(uint16_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint16(uuid); } |   void set_service_uuid16(uint16_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint16(uuid); } | ||||||
|   void set_service_uuid32(uint32_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint32(uuid); } |   void set_service_uuid32(uint32_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint32(uuid); } | ||||||
|   void set_service_uuid128(uint8_t *uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_raw(uuid); } |   void set_service_uuid128(uint8_t *uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_raw(uuid); } | ||||||
|   | |||||||
| @@ -19,7 +19,6 @@ class BLEClientSwitch : public switch_::Switch, public Component, public BLEClie | |||||||
|   void loop() override {} |   void loop() override {} | ||||||
|   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, |   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, | ||||||
|                            esp_ble_gattc_cb_param_t *param) override; |                            esp_ble_gattc_cb_param_t *param) override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   void write_state(bool state) override; |   void write_state(bool state) override; | ||||||
|   | |||||||
| @@ -14,7 +14,11 @@ static const char *const TAG = "ble_text_sensor"; | |||||||
|  |  | ||||||
| static const std::string EMPTY = ""; | static const std::string EMPTY = ""; | ||||||
|  |  | ||||||
| void BLETextSensor::loop() {} | void BLETextSensor::loop() { | ||||||
|  |   // Parent BLEClientNode has a loop() method, but this component uses | ||||||
|  |   // polling via update() and BLE callbacks so loop isn't needed | ||||||
|  |   this->disable_loop(); | ||||||
|  | } | ||||||
|  |  | ||||||
| void BLETextSensor::dump_config() { | void BLETextSensor::dump_config() { | ||||||
|   LOG_TEXT_SENSOR("", "BLE Text Sensor", this); |   LOG_TEXT_SENSOR("", "BLE Text Sensor", this); | ||||||
|   | |||||||
| @@ -20,7 +20,6 @@ class BLETextSensor : public text_sensor::TextSensor, public PollingComponent, p | |||||||
|   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, |   void gattc_event_handler(esp_gattc_cb_event_t event, esp_gatt_if_t gattc_if, | ||||||
|                            esp_ble_gattc_cb_param_t *param) override; |                            esp_ble_gattc_cb_param_t *param) override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   void set_service_uuid16(uint16_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint16(uuid); } |   void set_service_uuid16(uint16_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint16(uuid); } | ||||||
|   void set_service_uuid32(uint32_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint32(uuid); } |   void set_service_uuid32(uint32_t uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_uint32(uuid); } | ||||||
|   void set_service_uuid128(uint8_t *uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_raw(uuid); } |   void set_service_uuid128(uint8_t *uuid) { this->service_uuid_ = espbt::ESPBTUUID::from_raw(uuid); } | ||||||
|   | |||||||
| @@ -105,7 +105,6 @@ class BLEPresenceDevice : public binary_sensor::BinarySensorInitiallyOff, | |||||||
|       this->set_found_(false); |       this->set_found_(false); | ||||||
|   } |   } | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   void set_found_(bool state) { |   void set_found_(bool state) { | ||||||
|   | |||||||
| @@ -99,7 +99,6 @@ class BLERSSISensor : public sensor::Sensor, public esp32_ble_tracker::ESPBTDevi | |||||||
|     return false; |     return false; | ||||||
|   } |   } | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   enum MatchType { MATCH_BY_MAC_ADDRESS, MATCH_BY_IRK, MATCH_BY_SERVICE_UUID, MATCH_BY_IBEACON_UUID }; |   enum MatchType { MATCH_BY_MAC_ADDRESS, MATCH_BY_IRK, MATCH_BY_SERVICE_UUID, MATCH_BY_IBEACON_UUID }; | ||||||
|   | |||||||
| @@ -29,7 +29,6 @@ class BLEScanner : public text_sensor::TextSensor, public esp32_ble_tracker::ESP | |||||||
|     return true; |     return true; | ||||||
|   } |   } | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
| }; | }; | ||||||
|  |  | ||||||
| }  // namespace ble_scanner | }  // namespace ble_scanner | ||||||
|   | |||||||
| @@ -26,10 +26,17 @@ class BluetoothConnection : public esp32_ble_client::BLEClientBase { | |||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   friend class BluetoothProxy; |   friend class BluetoothProxy; | ||||||
|   bool seen_mtu_or_services_{false}; |  | ||||||
|  |  | ||||||
|   int16_t send_service_{-2}; |   // Memory optimized layout for 32-bit systems | ||||||
|  |   // Group 1: Pointers (4 bytes each, naturally aligned) | ||||||
|   BluetoothProxy *proxy_; |   BluetoothProxy *proxy_; | ||||||
|  |  | ||||||
|  |   // Group 2: 2-byte types | ||||||
|  |   int16_t send_service_{-2};  // Needs to handle negative values and service count | ||||||
|  |  | ||||||
|  |   // Group 3: 1-byte types | ||||||
|  |   bool seen_mtu_or_services_{false}; | ||||||
|  |   // 1 byte used, 1 byte padding | ||||||
| }; | }; | ||||||
|  |  | ||||||
| }  // namespace bluetooth_proxy | }  // namespace bluetooth_proxy | ||||||
|   | |||||||
| @@ -52,11 +52,21 @@ bool BluetoothProxy::parse_device(const esp32_ble_tracker::ESPBTDevice &device) | |||||||
|   return true; |   return true; | ||||||
| } | } | ||||||
|  |  | ||||||
| static constexpr size_t FLUSH_BATCH_SIZE = 8; | // Batch size for BLE advertisements to maximize WiFi efficiency | ||||||
| static std::vector<api::BluetoothLERawAdvertisement> &get_batch_buffer() { | // Each advertisement is up to 80 bytes when packaged (including protocol overhead) | ||||||
|   static std::vector<api::BluetoothLERawAdvertisement> batch_buffer; | // Most advertisements are 20-30 bytes, allowing even more to fit per packet | ||||||
|   return batch_buffer; | // 16 advertisements × 80 bytes (worst case) = 1280 bytes out of ~1320 bytes usable payload | ||||||
| } | // This achieves ~97% WiFi MTU utilization while staying under the limit | ||||||
|  | static constexpr size_t FLUSH_BATCH_SIZE = 16; | ||||||
|  |  | ||||||
|  | namespace { | ||||||
|  | // Batch buffer in anonymous namespace to avoid guard variable (saves 8 bytes) | ||||||
|  | // This is initialized at program startup before any threads | ||||||
|  | // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables) | ||||||
|  | std::vector<api::BluetoothLERawAdvertisement> batch_buffer; | ||||||
|  | }  // namespace | ||||||
|  |  | ||||||
|  | static std::vector<api::BluetoothLERawAdvertisement> &get_batch_buffer() { return batch_buffer; } | ||||||
|  |  | ||||||
| bool BluetoothProxy::parse_devices(const esp32_ble::BLEScanResult *scan_results, size_t count) { | bool BluetoothProxy::parse_devices(const esp32_ble::BLEScanResult *scan_results, size_t count) { | ||||||
|   if (!api::global_api_server->is_connected() || this->api_connection_ == nullptr || !this->raw_advertisements_) |   if (!api::global_api_server->is_connected() || this->api_connection_ == nullptr || !this->raw_advertisements_) | ||||||
| @@ -170,7 +180,7 @@ int BluetoothProxy::get_bluetooth_connections_free() { | |||||||
| void BluetoothProxy::loop() { | void BluetoothProxy::loop() { | ||||||
|   if (!api::global_api_server->is_connected() || this->api_connection_ == nullptr) { |   if (!api::global_api_server->is_connected() || this->api_connection_ == nullptr) { | ||||||
|     for (auto *connection : this->connections_) { |     for (auto *connection : this->connections_) { | ||||||
|       if (connection->get_address() != 0) { |       if (connection->get_address() != 0 && !connection->disconnect_pending()) { | ||||||
|         connection->disconnect(); |         connection->disconnect(); | ||||||
|       } |       } | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -134,11 +134,17 @@ class BluetoothProxy : public esp32_ble_tracker::ESPBTDeviceListener, public Com | |||||||
|  |  | ||||||
|   BluetoothConnection *get_connection_(uint64_t address, bool reserve); |   BluetoothConnection *get_connection_(uint64_t address, bool reserve); | ||||||
|  |  | ||||||
|   bool active_; |   // Memory optimized layout for 32-bit systems | ||||||
|  |   // Group 1: Pointers (4 bytes each, naturally aligned) | ||||||
|   std::vector<BluetoothConnection *> connections_{}; |  | ||||||
|   api::APIConnection *api_connection_{nullptr}; |   api::APIConnection *api_connection_{nullptr}; | ||||||
|  |  | ||||||
|  |   // Group 2: Container types (typically 12 bytes on 32-bit) | ||||||
|  |   std::vector<BluetoothConnection *> connections_{}; | ||||||
|  |  | ||||||
|  |   // Group 3: 1-byte types grouped together | ||||||
|  |   bool active_; | ||||||
|   bool raw_advertisements_{false}; |   bool raw_advertisements_{false}; | ||||||
|  |   // 2 bytes used, 2 bytes padding | ||||||
| }; | }; | ||||||
|  |  | ||||||
| extern BluetoothProxy *global_bluetooth_proxy;  // NOLINT(cppcoreguidelines-avoid-non-const-global-variables) | extern BluetoothProxy *global_bluetooth_proxy;  // NOLINT(cppcoreguidelines-avoid-non-const-global-variables) | ||||||
|   | |||||||
| @@ -12,8 +12,8 @@ from esphome.const import ( | |||||||
|     CONF_OVERSAMPLING, |     CONF_OVERSAMPLING, | ||||||
|     CONF_PRESSURE, |     CONF_PRESSURE, | ||||||
|     CONF_TEMPERATURE, |     CONF_TEMPERATURE, | ||||||
|     DEVICE_CLASS_HUMIDITY, |  | ||||||
|     DEVICE_CLASS_ATMOSPHERIC_PRESSURE, |     DEVICE_CLASS_ATMOSPHERIC_PRESSURE, | ||||||
|  |     DEVICE_CLASS_HUMIDITY, | ||||||
|     DEVICE_CLASS_TEMPERATURE, |     DEVICE_CLASS_TEMPERATURE, | ||||||
|     ICON_GAS_CYLINDER, |     ICON_GAS_CYLINDER, | ||||||
|     STATE_CLASS_MEASUREMENT, |     STATE_CLASS_MEASUREMENT, | ||||||
|   | |||||||
| @@ -61,8 +61,6 @@ enum IIRFilter { | |||||||
|  |  | ||||||
| class BMP581Component : public PollingComponent, public i2c::I2CDevice { | class BMP581Component : public PollingComponent, public i2c::I2CDevice { | ||||||
|  public: |  public: | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|  |  | ||||||
|   void setup() override; |   void setup() override; | ||||||
|   | |||||||
| @@ -18,8 +18,8 @@ from esphome.const import ( | |||||||
|     DEVICE_CLASS_UPDATE, |     DEVICE_CLASS_UPDATE, | ||||||
| ) | ) | ||||||
| from esphome.core import CORE, coroutine_with_priority | from esphome.core import CORE, coroutine_with_priority | ||||||
|  | from esphome.core.entity_helpers import entity_duplicate_validator, setup_entity | ||||||
| from esphome.cpp_generator import MockObjClass | from esphome.cpp_generator import MockObjClass | ||||||
| from esphome.cpp_helpers import setup_entity |  | ||||||
|  |  | ||||||
| CODEOWNERS = ["@esphome/core"] | CODEOWNERS = ["@esphome/core"] | ||||||
| IS_PLATFORM_COMPONENT = True | IS_PLATFORM_COMPONENT = True | ||||||
| @@ -61,6 +61,9 @@ _BUTTON_SCHEMA = ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | _BUTTON_SCHEMA.add_extra(entity_duplicate_validator("button")) | ||||||
|  |  | ||||||
|  |  | ||||||
| def button_schema( | def button_schema( | ||||||
|     class_: MockObjClass, |     class_: MockObjClass, | ||||||
|     *, |     *, | ||||||
| @@ -87,7 +90,7 @@ BUTTON_SCHEMA.add_extra(cv.deprecated_schema_constant("button")) | |||||||
|  |  | ||||||
|  |  | ||||||
| async def setup_button_core_(var, config): | async def setup_button_core_(var, config): | ||||||
|     await setup_entity(var, config) |     await setup_entity(var, config, "button") | ||||||
|  |  | ||||||
|     for conf in config.get(CONF_ON_PRESS, []): |     for conf in config.get(CONF_ON_PRESS, []): | ||||||
|         trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var) |         trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var) | ||||||
|   | |||||||
							
								
								
									
										1
									
								
								esphome/components/camera/__init__.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								esphome/components/camera/__init__.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | |||||||
|  | CODEOWNERS = ["@DT-art1", "@bdraco"] | ||||||
							
								
								
									
										22
									
								
								esphome/components/camera/camera.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								esphome/components/camera/camera.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,22 @@ | |||||||
|  | #include "camera.h" | ||||||
|  |  | ||||||
|  | namespace esphome { | ||||||
|  | namespace camera { | ||||||
|  |  | ||||||
|  | // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables) | ||||||
|  | Camera *Camera::global_camera = nullptr; | ||||||
|  |  | ||||||
|  | Camera::Camera() { | ||||||
|  |   if (global_camera != nullptr) { | ||||||
|  |     this->status_set_error("Multiple cameras are configured, but only one is supported."); | ||||||
|  |     this->mark_failed(); | ||||||
|  |     return; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   global_camera = this; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | Camera *Camera::instance() { return global_camera; } | ||||||
|  |  | ||||||
|  | }  // namespace camera | ||||||
|  | }  // namespace esphome | ||||||
							
								
								
									
										80
									
								
								esphome/components/camera/camera.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										80
									
								
								esphome/components/camera/camera.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,80 @@ | |||||||
|  | #pragma once | ||||||
|  |  | ||||||
|  | #include "esphome/core/automation.h" | ||||||
|  | #include "esphome/core/component.h" | ||||||
|  | #include "esphome/core/entity_base.h" | ||||||
|  | #include "esphome/core/helpers.h" | ||||||
|  |  | ||||||
|  | namespace esphome { | ||||||
|  | namespace camera { | ||||||
|  |  | ||||||
|  | /** Different sources for filtering. | ||||||
|  |  *  IDLE: Camera requests to send an image to the API. | ||||||
|  |  *  API_REQUESTER: API requests a new image. | ||||||
|  |  *  WEB_REQUESTER: ESP32 web server request an image. Ignored by API. | ||||||
|  |  */ | ||||||
|  | enum CameraRequester : uint8_t { IDLE, API_REQUESTER, WEB_REQUESTER }; | ||||||
|  |  | ||||||
|  | /** Abstract camera image base class. | ||||||
|  |  *  Encapsulates the JPEG encoded data and it is shared among | ||||||
|  |  *  all connected clients. | ||||||
|  |  */ | ||||||
|  | class CameraImage { | ||||||
|  |  public: | ||||||
|  |   virtual uint8_t *get_data_buffer() = 0; | ||||||
|  |   virtual size_t get_data_length() = 0; | ||||||
|  |   virtual bool was_requested_by(CameraRequester requester) const = 0; | ||||||
|  |   virtual ~CameraImage() {} | ||||||
|  | }; | ||||||
|  |  | ||||||
|  | /** Abstract image reader base class. | ||||||
|  |  *  Keeps track of the data offset of the camera image and | ||||||
|  |  *  how many bytes are remaining to read. When the image | ||||||
|  |  *  is returned, the shared_ptr is reset and the camera can | ||||||
|  |  *  reuse the memory of the camera image. | ||||||
|  |  */ | ||||||
|  | class CameraImageReader { | ||||||
|  |  public: | ||||||
|  |   virtual void set_image(std::shared_ptr<CameraImage> image) = 0; | ||||||
|  |   virtual size_t available() const = 0; | ||||||
|  |   virtual uint8_t *peek_data_buffer() = 0; | ||||||
|  |   virtual void consume_data(size_t consumed) = 0; | ||||||
|  |   virtual void return_image() = 0; | ||||||
|  |   virtual ~CameraImageReader() {} | ||||||
|  | }; | ||||||
|  |  | ||||||
|  | /** Abstract camera base class. Collaborates with API. | ||||||
|  |  *  1) API server starts and installs callback (add_image_callback) | ||||||
|  |  *     which is called by the camera when a new image is available. | ||||||
|  |  *  2) New API client connects and creates a new image reader (create_image_reader). | ||||||
|  |  *  3) API connection receives protobuf CameraImageRequest and calls request_image. | ||||||
|  |  *  3.a) API connection receives protobuf CameraImageRequest and calls start_stream. | ||||||
|  |  *  4) Camera implementation provides JPEG data in the CameraImage and calls callback. | ||||||
|  |  *  5) API connection sets the image in the image reader. | ||||||
|  |  *  6) API connection consumes data from the image reader and returns the image when finished. | ||||||
|  |  *  7.a) Camera captures a new image and continues with 4) until start_stream is called. | ||||||
|  |  */ | ||||||
|  | class Camera : public EntityBase, public Component { | ||||||
|  |  public: | ||||||
|  |   Camera(); | ||||||
|  |   // Camera implementation invokes callback to publish a new image. | ||||||
|  |   virtual void add_image_callback(std::function<void(std::shared_ptr<CameraImage>)> &&callback) = 0; | ||||||
|  |   /// Returns a new camera image reader that keeps track of the JPEG data in the camera image. | ||||||
|  |   virtual CameraImageReader *create_image_reader() = 0; | ||||||
|  |   // Connection, camera or web server requests one new JPEG image. | ||||||
|  |   virtual void request_image(CameraRequester requester) = 0; | ||||||
|  |   // Connection, camera or web server requests a stream of images. | ||||||
|  |   virtual void start_stream(CameraRequester requester) = 0; | ||||||
|  |   // Connection or web server stops the previously started stream. | ||||||
|  |   virtual void stop_stream(CameraRequester requester) = 0; | ||||||
|  |   virtual ~Camera() {} | ||||||
|  |   /// The singleton instance of the camera implementation. | ||||||
|  |   static Camera *instance(); | ||||||
|  |  | ||||||
|  |  protected: | ||||||
|  |   // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables) | ||||||
|  |   static Camera *global_camera; | ||||||
|  | }; | ||||||
|  |  | ||||||
|  | }  // namespace camera | ||||||
|  | }  // namespace esphome | ||||||
| @@ -1,4 +1,5 @@ | |||||||
| import re | import re | ||||||
|  |  | ||||||
| from esphome import automation | from esphome import automation | ||||||
| import esphome.codegen as cg | import esphome.codegen as cg | ||||||
| import esphome.config_validation as cv | import esphome.config_validation as cv | ||||||
|   | |||||||
| @@ -46,7 +46,6 @@ class CAP1188Component : public Component, public i2c::I2CDevice { | |||||||
|   void set_reset_pin(GPIOPin *reset_pin) { this->reset_pin_ = reset_pin; } |   void set_reset_pin(GPIOPin *reset_pin) { this->reset_pin_ = reset_pin; } | ||||||
|   void setup() override; |   void setup() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|   void loop() override; |   void loop() override; | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   | |||||||
| @@ -7,11 +7,12 @@ from esphome.const import ( | |||||||
|     PLATFORM_BK72XX, |     PLATFORM_BK72XX, | ||||||
|     PLATFORM_ESP32, |     PLATFORM_ESP32, | ||||||
|     PLATFORM_ESP8266, |     PLATFORM_ESP8266, | ||||||
|  |     PLATFORM_LN882X, | ||||||
|     PLATFORM_RTL87XX, |     PLATFORM_RTL87XX, | ||||||
| ) | ) | ||||||
| from esphome.core import CORE, coroutine_with_priority | from esphome.core import CORE, coroutine_with_priority | ||||||
|  |  | ||||||
| AUTO_LOAD = ["web_server_base"] | AUTO_LOAD = ["web_server_base", "ota.web_server"] | ||||||
| DEPENDENCIES = ["wifi"] | DEPENDENCIES = ["wifi"] | ||||||
| CODEOWNERS = ["@OttoWinter"] | CODEOWNERS = ["@OttoWinter"] | ||||||
|  |  | ||||||
| @@ -27,7 +28,15 @@ CONFIG_SCHEMA = cv.All( | |||||||
|             ), |             ), | ||||||
|         } |         } | ||||||
|     ).extend(cv.COMPONENT_SCHEMA), |     ).extend(cv.COMPONENT_SCHEMA), | ||||||
|     cv.only_on([PLATFORM_ESP32, PLATFORM_ESP8266, PLATFORM_BK72XX, PLATFORM_RTL87XX]), |     cv.only_on( | ||||||
|  |         [ | ||||||
|  |             PLATFORM_ESP32, | ||||||
|  |             PLATFORM_ESP8266, | ||||||
|  |             PLATFORM_BK72XX, | ||||||
|  |             PLATFORM_LN882X, | ||||||
|  |             PLATFORM_RTL87XX, | ||||||
|  |         ] | ||||||
|  |     ), | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  |  | ||||||
| @@ -41,6 +50,7 @@ async def to_code(config): | |||||||
|  |  | ||||||
|     if CORE.using_arduino: |     if CORE.using_arduino: | ||||||
|         if CORE.is_esp32: |         if CORE.is_esp32: | ||||||
|  |             cg.add_library("ESP32 Async UDP", None) | ||||||
|             cg.add_library("DNSServer", None) |             cg.add_library("DNSServer", None) | ||||||
|             cg.add_library("WiFi", None) |             cg.add_library("WiFi", None) | ||||||
|         if CORE.is_esp8266: |         if CORE.is_esp8266: | ||||||
|   | |||||||
| @@ -37,12 +37,16 @@ void CaptivePortal::handle_wifisave(AsyncWebServerRequest *request) { | |||||||
|   request->redirect("/?save"); |   request->redirect("/?save"); | ||||||
| } | } | ||||||
|  |  | ||||||
| void CaptivePortal::setup() {} | void CaptivePortal::setup() { | ||||||
|  | #ifndef USE_ARDUINO | ||||||
|  |   // No DNS server needed for non-Arduino frameworks | ||||||
|  |   this->disable_loop(); | ||||||
|  | #endif | ||||||
|  | } | ||||||
| void CaptivePortal::start() { | void CaptivePortal::start() { | ||||||
|   this->base_->init(); |   this->base_->init(); | ||||||
|   if (!this->initialized_) { |   if (!this->initialized_) { | ||||||
|     this->base_->add_handler(this); |     this->base_->add_handler(this); | ||||||
|     this->base_->add_ota_handler(); |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
| #ifdef USE_ARDUINO | #ifdef USE_ARDUINO | ||||||
| @@ -50,6 +54,8 @@ void CaptivePortal::start() { | |||||||
|   this->dns_server_->setErrorReplyCode(DNSReplyCode::NoError); |   this->dns_server_->setErrorReplyCode(DNSReplyCode::NoError); | ||||||
|   network::IPAddress ip = wifi::global_wifi_component->wifi_soft_ap_ip(); |   network::IPAddress ip = wifi::global_wifi_component->wifi_soft_ap_ip(); | ||||||
|   this->dns_server_->start(53, "*", ip); |   this->dns_server_->start(53, "*", ip); | ||||||
|  |   // Re-enable loop() when DNS server is started | ||||||
|  |   this->enable_loop(); | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|   this->base_->get_server()->onNotFound([this](AsyncWebServerRequest *req) { |   this->base_->get_server()->onNotFound([this](AsyncWebServerRequest *req) { | ||||||
| @@ -68,7 +74,11 @@ void CaptivePortal::start() { | |||||||
|  |  | ||||||
| void CaptivePortal::handleRequest(AsyncWebServerRequest *req) { | void CaptivePortal::handleRequest(AsyncWebServerRequest *req) { | ||||||
|   if (req->url() == "/") { |   if (req->url() == "/") { | ||||||
|  | #ifndef USE_ESP8266 | ||||||
|  |     auto *response = req->beginResponse(200, "text/html", INDEX_GZ, sizeof(INDEX_GZ)); | ||||||
|  | #else | ||||||
|     auto *response = req->beginResponse_P(200, "text/html", INDEX_GZ, sizeof(INDEX_GZ)); |     auto *response = req->beginResponse_P(200, "text/html", INDEX_GZ, sizeof(INDEX_GZ)); | ||||||
|  | #endif | ||||||
|     response->addHeader("Content-Encoding", "gzip"); |     response->addHeader("Content-Encoding", "gzip"); | ||||||
|     req->send(response); |     req->send(response); | ||||||
|     return; |     return; | ||||||
|   | |||||||
| @@ -21,8 +21,11 @@ class CaptivePortal : public AsyncWebHandler, public Component { | |||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
| #ifdef USE_ARDUINO | #ifdef USE_ARDUINO | ||||||
|   void loop() override { |   void loop() override { | ||||||
|     if (this->dns_server_ != nullptr) |     if (this->dns_server_ != nullptr) { | ||||||
|       this->dns_server_->processNextRequest(); |       this->dns_server_->processNextRequest(); | ||||||
|  |     } else { | ||||||
|  |       this->disable_loop(); | ||||||
|  |     } | ||||||
|   } |   } | ||||||
| #endif | #endif | ||||||
|   float get_setup_priority() const override; |   float get_setup_priority() const override; | ||||||
| @@ -37,7 +40,7 @@ class CaptivePortal : public AsyncWebHandler, public Component { | |||||||
| #endif | #endif | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   bool canHandle(AsyncWebServerRequest *request) override { |   bool canHandle(AsyncWebServerRequest *request) const override { | ||||||
|     if (!this->active_) |     if (!this->active_) | ||||||
|       return false; |       return false; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -25,8 +25,6 @@ class CCS811Component : public PollingComponent, public i2c::I2CDevice { | |||||||
|  |  | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|  |  | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   optional<uint8_t> read_status_() { return this->read_byte(0x00); } |   optional<uint8_t> read_status_() { return this->read_byte(0x00); } | ||||||
|   bool status_has_error_() { return this->read_status_().value_or(1) & 1; } |   bool status_has_error_() { return this->read_status_().value_or(1) & 1; } | ||||||
|   | |||||||
| @@ -48,8 +48,8 @@ from esphome.const import ( | |||||||
|     CONF_WEB_SERVER, |     CONF_WEB_SERVER, | ||||||
| ) | ) | ||||||
| from esphome.core import CORE, coroutine_with_priority | from esphome.core import CORE, coroutine_with_priority | ||||||
|  | from esphome.core.entity_helpers import entity_duplicate_validator, setup_entity | ||||||
| from esphome.cpp_generator import MockObjClass | from esphome.cpp_generator import MockObjClass | ||||||
| from esphome.cpp_helpers import setup_entity |  | ||||||
|  |  | ||||||
| IS_PLATFORM_COMPONENT = True | IS_PLATFORM_COMPONENT = True | ||||||
|  |  | ||||||
| @@ -247,6 +247,9 @@ _CLIMATE_SCHEMA = ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | _CLIMATE_SCHEMA.add_extra(entity_duplicate_validator("climate")) | ||||||
|  |  | ||||||
|  |  | ||||||
| def climate_schema( | def climate_schema( | ||||||
|     class_: MockObjClass, |     class_: MockObjClass, | ||||||
|     *, |     *, | ||||||
| @@ -273,7 +276,7 @@ CLIMATE_SCHEMA.add_extra(cv.deprecated_schema_constant("climate")) | |||||||
|  |  | ||||||
|  |  | ||||||
| async def setup_climate_core_(var, config): | async def setup_climate_core_(var, config): | ||||||
|     await setup_entity(var, config) |     await setup_entity(var, config, "climate") | ||||||
|  |  | ||||||
|     visual = config[CONF_VISUAL] |     visual = config[CONF_VISUAL] | ||||||
|     if (min_temp := visual.get(CONF_MIN_TEMPERATURE)) is not None: |     if (min_temp := visual.get(CONF_MIN_TEMPERATURE)) is not None: | ||||||
|   | |||||||
| @@ -1,10 +1,10 @@ | |||||||
| """CM1106 Sensor component for ESPHome.""" | """CM1106 Sensor component for ESPHome.""" | ||||||
|  |  | ||||||
| import esphome.codegen as cg |  | ||||||
| import esphome.config_validation as cv |  | ||||||
| from esphome import automation | from esphome import automation | ||||||
| from esphome.automation import maybe_simple_id | from esphome.automation import maybe_simple_id | ||||||
|  | import esphome.codegen as cg | ||||||
| from esphome.components import sensor, uart | from esphome.components import sensor, uart | ||||||
|  | import esphome.config_validation as cv | ||||||
| from esphome.const import ( | from esphome.const import ( | ||||||
|     CONF_CO2, |     CONF_CO2, | ||||||
|     CONF_ID, |     CONF_ID, | ||||||
|   | |||||||
| @@ -2,5 +2,7 @@ | |||||||
|  |  | ||||||
| CODEOWNERS = ["@esphome/core"] | CODEOWNERS = ["@esphome/core"] | ||||||
|  |  | ||||||
|  | CONF_BYTE_ORDER = "byte_order" | ||||||
| CONF_DRAW_ROUNDING = "draw_rounding" | CONF_DRAW_ROUNDING = "draw_rounding" | ||||||
|  | CONF_ON_STATE_CHANGE = "on_state_change" | ||||||
| CONF_REQUEST_HEADERS = "request_headers" | CONF_REQUEST_HEADERS = "request_headers" | ||||||
|   | |||||||
| @@ -11,7 +11,6 @@ class CopyBinarySensor : public binary_sensor::BinarySensor, public Component { | |||||||
|   void set_source(binary_sensor::BinarySensor *source) { source_ = source; } |   void set_source(binary_sensor::BinarySensor *source) { source_ = source; } | ||||||
|   void setup() override; |   void setup() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   binary_sensor::BinarySensor *source_; |   binary_sensor::BinarySensor *source_; | ||||||
|   | |||||||
| @@ -10,7 +10,6 @@ class CopyButton : public button::Button, public Component { | |||||||
|  public: |  public: | ||||||
|   void set_source(button::Button *source) { source_ = source; } |   void set_source(button::Button *source) { source_ = source; } | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|  protected: |  protected: | ||||||
|   void press_action() override; |   void press_action() override; | ||||||
|   | |||||||
| @@ -11,7 +11,6 @@ class CopyCover : public cover::Cover, public Component { | |||||||
|   void set_source(cover::Cover *source) { source_ = source; } |   void set_source(cover::Cover *source) { source_ = source; } | ||||||
|   void setup() override; |   void setup() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   cover::CoverTraits get_traits() override; |   cover::CoverTraits get_traits() override; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -11,7 +11,6 @@ class CopyFan : public fan::Fan, public Component { | |||||||
|   void set_source(fan::Fan *source) { source_ = source; } |   void set_source(fan::Fan *source) { source_ = source; } | ||||||
|   void setup() override; |   void setup() override; | ||||||
|   void dump_config() override; |   void dump_config() override; | ||||||
|   float get_setup_priority() const override { return setup_priority::DATA; } |  | ||||||
|  |  | ||||||
|   fan::FanTraits get_traits() override; |   fan::FanTraits get_traits() override; | ||||||
|  |  | ||||||
|   | |||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user