use episode after add

diff --git a/resources/js/hooks/episodes.jsx b/resources/js/hooks/episodes.jsx
index a98dad389bdc23362dd0af4a7fb24d4e26138c71..0a7a6e932ee2b4284ff42603ea418c06066ba868 100644 (file)
--- a/resources/js/hooks/episodes.jsx
+++ b/resources/js/hooks/episodes.jsx
@@ -47,12 +47,17 @@ export const EpisodesProvider = ({ children, setEpisodes }) => {
                                ? await axios.put(`/api/episodes/${values.id}`, values)
                                : await axios.post(`/api/events/${values.event_id}/add-episode`, values);
                        const newEpisode = response.data;
-                       setEpisodes(episodes => episodes.map(episode =>
-                               episode.id === newEpisode.id ? {
-                                       ...episode,
-                                       ...newEpisode,
-                               } : episode
-                       ));
+                       if (values.id) {
+                               setEpisodes(episodes => episodes.map(episode =>
+                                       episode.id === newEpisode.id ? {
+                                               ...episode,
+                                               ...newEpisode,
+                                       } : episode
+                               ));
+                       } else {
+                               // TODO: fix order
+                               setEpisodes(episodes => [...episodes, newEpisode]);
+                       }
                        setEditEpisode(episode => ({
                                ...episode,
                                ...newEpisode,