media_player add off on capability

aioesphomeapi/api.proto

@@ -1297,13 +1297,26 @@ enum MediaPlayerState {
   MEDIA_PLAYER_STATE_IDLE = 1;
   MEDIA_PLAYER_STATE_PLAYING = 2;
   MEDIA_PLAYER_STATE_PAUSED = 3;
+  MEDIA_PLAYER_STATE_ANNOUNCING = 4;
+  MEDIA_PLAYER_STATE_OFF = 5;
+  MEDIA_PLAYER_STATE_ON = 6;
+
 }
 enum MediaPlayerCommand {
   MEDIA_PLAYER_COMMAND_PLAY = 0;
   MEDIA_PLAYER_COMMAND_PAUSE = 1;
   MEDIA_PLAYER_COMMAND_STOP = 2;
   MEDIA_PLAYER_COMMAND_MUTE = 3;
   MEDIA_PLAYER_COMMAND_UNMUTE = 4;
+  MEDIA_PLAYER_COMMAND_TOGGLE = 5;
+  MEDIA_PLAYER_COMMAND_VOLUME_UP = 6;
+  MEDIA_PLAYER_COMMAND_VOLUME_DOWN = 7;
+  MEDIA_PLAYER_COMMAND_ENQUEUE = 8;
+  MEDIA_PLAYER_COMMAND_REPEAT_ONE = 9;
+  MEDIA_PLAYER_COMMAND_REPEAT_OFF = 10;
+  MEDIA_PLAYER_COMMAND_CLEAR_PLAYLIST = 11;
+  MEDIA_PLAYER_COMMAND_TURN_ON = 12;
+  MEDIA_PLAYER_COMMAND_TURN_OFF = 13;
 }
 enum MediaPlayerFormatPurpose {
   MEDIA_PLAYER_FORMAT_PURPOSE_DEFAULT = 0;
@@ -1338,6 +1351,8 @@ message ListEntitiesMediaPlayerResponse {
   repeated MediaPlayerSupportedFormat supported_formats = 9;
 
   uint32 device_id = 10 [(field_ifdef) = "USE_DEVICES"];
+
+  uint32 feature_flags = 11;
 }
 message MediaPlayerStateResponse {
   option (id) = 64;

aioesphomeapi/model.py

@@ -873,11 +873,42 @@ class ValveState(EntityState):
 
 
 # ==================== MEDIA PLAYER ====================
+class MediaPlayerEntityFeature(enum.IntFlag):
+    """Supported features of the media player entity."""
+
+    PAUSE = 1 << 0
+    SEEK = 1 << 1
+    VOLUME_SET = 1 << 2
+    VOLUME_MUTE = 1 << 3
+    PREVIOUS_TRACK = 1 << 4
+    NEXT_TRACK = 1 << 5
+
+    TURN_ON = 1 << 7
+    TURN_OFF = 1 << 8
+    PLAY_MEDIA = 1 << 9
+    VOLUME_STEP = 1 << 10
+    SELECT_SOURCE = 1 << 11
+    STOP = 1 << 12
+    CLEAR_PLAYLIST = 1 << 13
+    PLAY = 1 << 14
+    SHUFFLE_SET = 1 << 15
+    SELECT_SOUND_MODE = 1 << 16
+    BROWSE_MEDIA = 1 << 17
+    REPEAT_SET = 1 << 18
+    GROUPING = 1 << 19
+    MEDIA_ANNOUNCE = 1 << 20
+    MEDIA_ENQUEUE = 1 << 21
+    SEARCH_MEDIA = 1 << 22
+
+
 class MediaPlayerState(APIIntEnum):
     NONE = 0
     IDLE = 1
     PLAYING = 2
     PAUSED = 3
+    ANNOUNCING = 4
+    OFF = 5
+    ON = 6
 
 
 class MediaPlayerCommand(APIIntEnum):
@@ -886,6 +917,15 @@ class MediaPlayerCommand(APIIntEnum):
     STOP = 2
     MUTE = 3
     UNMUTE = 4
+    TOGGLE = 5
+    VOLUME_UP = 6
+    VOLUME_DOWN = 7
+    ENQUEUE = 8
+    REPEAT_ONE = 9
+    REPEAT_OFF = 10
+    CLEAR_PLAYLIST = 11
+    TURN_ON = 12
+    TURN_OFF = 13
 
 
 class MediaPlayerFormatPurpose(APIIntEnum):
@@ -922,6 +962,24 @@ class MediaPlayerInfo(EntityInfo):
     supported_formats: list[MediaPlayerSupportedFormat] = converter_field(
         default_factory=list, converter=MediaPlayerSupportedFormat.convert_list
     )
+    feature_flags: int = 0
+
+    def feature_flags_compat(self, api_version: APIVersion) -> int:
+        if api_version < APIVersion(1, 11):
+            flags = (
+                MediaPlayerEntityFeature.PLAY_MEDIA
+                | MediaPlayerEntityFeature.BROWSE_MEDIA
+                | MediaPlayerEntityFeature.STOP
+                | MediaPlayerEntityFeature.VOLUME_SET
+                | MediaPlayerEntityFeature.VOLUME_MUTE
+                | MediaPlayerEntityFeature.MEDIA_ANNOUNCE
+            )
+            if self.supports_pause:
+                flags |= MediaPlayerEntityFeature.PAUSE | MediaPlayerEntityFeature.PLAY
+
+            return flags
+
+        return self.feature_flags
 
 
 @_frozen_dataclass_decorator

tests/test_model.py

@@ -103,6 +103,7 @@
     LightState,
     LockEntityState,
     LockInfo,
+    MediaPlayerEntityFeature,
     MediaPlayerEntityState,
     MediaPlayerInfo,
     NoiseEncryptionSetKeyResponse as NoiseEncryptionSetKeyResponseModel,
@@ -792,6 +793,7 @@ def test_media_player_supported_format_convert_list() -> None:
                     "sample_bytes": 2,
                 }
             ],
+            "feature_flags": 0,
         }
     ) == MediaPlayerInfo(
         supports_pause=False,
@@ -804,9 +806,50 @@ def test_media_player_supported_format_convert_list() -> None:
                 sample_bytes=2,
             )
         ],
+        feature_flags=0,
     )
 
 
+def test_media_player_feature_flags_compat() -> None:
+    """Test feature flags compatibility across API versions"""
+    info = MediaPlayerInfo(
+        supports_pause=False,
+        supported_formats=[
+            MediaPlayerSupportedFormat(
+                format="flac",
+                sample_rate=48000,
+                num_channels=2,
+                purpose=1,
+                sample_bytes=2,
+            )
+        ],
+        feature_flags=999999,  # Different from calculated compatibility flags
+    )
+    # For API version < 1.11, should return calculated compatibility flags
+    compat_flags = info.feature_flags_compat(APIVersion(1, 10))
+    expected_compat = (
+        MediaPlayerEntityFeature.PLAY_MEDIA
+        | MediaPlayerEntityFeature.BROWSE_MEDIA
+        | MediaPlayerEntityFeature.STOP
+        | MediaPlayerEntityFeature.VOLUME_SET
+        | MediaPlayerEntityFeature.VOLUME_MUTE
+        | MediaPlayerEntityFeature.MEDIA_ANNOUNCE
+    )
+    assert compat_flags == expected_compat
+
+    # For API version >= 1.11, should return feature_flags directly
+    direct_flags = info.feature_flags_compat(APIVersion(1, 11))
+    assert direct_flags == 999999
+
+    # Test with supports_pause=True to verify PAUSE|PLAY flags are added
+    info_with_pause = MediaPlayerInfo(supports_pause=True, feature_flags=888888)
+    compat_with_pause = info_with_pause.feature_flags_compat(APIVersion(1, 10))
+    expected_with_pause = (
+        expected_compat | MediaPlayerEntityFeature.PAUSE | MediaPlayerEntityFeature.PLAY
+    )
+    assert compat_with_pause == expected_with_pause
+
+
 def test_device_info_area_field() -> None:
     """Test DeviceInfo with area field set."""
     device_info = DeviceInfo(