From 5a3ddd570df5d9e7fbbd7107da48bfa8418d05bf Mon Sep 17 00:00:00 2001
From: Danielle McLean <dani@00dani.me>
Date: Mon, 27 Nov 2023 16:52:00 +1100
Subject: [PATCH] Fix some types in cocoa.py so they work with the stubs

---
 src/mpd_now_playable/cocoa.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/mpd_now_playable/cocoa.py b/src/mpd_now_playable/cocoa.py
index d20c13f..8554067 100644
--- a/src/mpd_now_playable/cocoa.py
+++ b/src/mpd_now_playable/cocoa.py
@@ -58,11 +58,12 @@ def ns_image_to_media_item_artwork(img: NSImage) -> MPMediaItemArtwork:
 
 
 def playback_state_to_cocoa(state: PlaybackState) -> MPMusicPlaybackState:
-	return {
+	mapping: dict[PlaybackState, MPMusicPlaybackState] = {
 		PlaybackState.play: MPMusicPlaybackStatePlaying,
 		PlaybackState.pause: MPMusicPlaybackStatePaused,
 		PlaybackState.stop: MPMusicPlaybackStateStopped,
-	}[state]
+	}
+	return mapping[state]
 
 
 def song_to_media_item(song: Song) -> NSMutableDictionary:
@@ -139,7 +140,7 @@ class CocoaNowPlaying:
 
 	def _create_handler(
 		self, player: Callable[[], Coroutine[None, None, PlaybackState | None]]
-	) -> Callable[[MPRemoteCommandEvent], None]:
+	) -> Callable[[MPRemoteCommandEvent], MPRemoteCommandHandlerStatus]:
 		async def invoke_music_player() -> None:
 			result = await player()
 			if result: