fix: Check if next is title

techmatters · Sep 26, 2024 · 1bfbd0e · 1bfbd0e
1 parent 45cfaaf
commit 1bfbd0e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/storyMap/components/StoryMap.js b/src/storyMap/components/StoryMap.js
@@ -207,14 +207,16 @@ const getTransition = ({ config, id, direction }) => {
     return {
       transition: config.titleTransition,
       index: -1,
+      nextTransition: config.chapters[0],
     };
   }
   const chapterIndex = config.chapters.findIndex(chapter => chapter.id === id);
   const chapter = config.chapters[chapterIndex];
-  const next =
-    direction === 'up'
-      ? _.get(`chapters[${chapterIndex - 1}]`, config)
-      : _.get(`chapters[${chapterIndex + 1}]`, config);
+  const nextIndex = direction === 'up' ? chapterIndex - 1 : chapterIndex + 1;
+  const nextIsTitle = nextIndex === -1;
+  const next = nextIsTitle
+    ? config.titleTransition
+    : _.get(`chapters[${nextIndex}]`, config);
   return {
     transition: chapter,
     nextTransition: next,