Fix:Android sleep timer end of chapter use next chapter if within 2s of current chapter ending when reset #260

android/app/src/main/java/com/audiobookshelf/app/data/PlaybackSession.kt

@@ -78,6 +78,17 @@ class PlaybackSession(
     return audioTracks.size - 1
   }
 
+  @JsonIgnore
+  fun getNextTrackIndex():Int {
+    for (i in 0 until audioTracks.size) {
+      val track = audioTracks[i]
+      if (currentTimeMs < track.startOffsetMs) {
+        return i
+      }
+    }
+    return audioTracks.size - 1
+  }
+
   @JsonIgnore
   fun getChapterForTime(time:Long):BookChapter? {
     if (chapters.isEmpty()) return null
@@ -85,11 +96,23 @@ class PlaybackSession(
   }
 
   @JsonIgnore
-  fun getCurrentTrackEndTime():Long? {
+  fun getCurrentTrackEndTime():Long {
     val currentTrack = audioTracks[this.getCurrentTrackIndex()]
     return currentTrack.startOffsetMs + currentTrack.durationMs
   }
 
+  @JsonIgnore
+  fun getNextChapterForTime(time:Long):BookChapter? {
+    if (chapters.isEmpty()) return null
+    return chapters.find { time < it.startMs } // First chapter where start time is > then time
+  }
+
+  @JsonIgnore
+  fun getNextTrackEndTime():Long {
+    val currentTrack = audioTracks[this.getNextTrackIndex()]
+    return currentTrack.startOffsetMs + currentTrack.durationMs
+  }
+
   @JsonIgnore
   fun getCurrentTrackTimeMs():Long {
     val currentTrack = audioTracks[this.getCurrentTrackIndex()]

android/app/src/main/java/com/audiobookshelf/app/managers/SleepTimerManager.kt

@@ -160,7 +160,40 @@ class SleepTimerManager constructor(private val playerNotificationService: Playe
     }
   }
 
-  fun checkShouldResetSleepTimer() {
+  // Get the chapter end time for use in End of Chapter timers
+  //   if less than 2s remain in chapter then use the next chapter
+  private fun getChapterEndTime():Long? {
+    val currentChapterEndTimeMs = playerNotificationService.getEndTimeOfChapterOrTrack()
+    if (currentChapterEndTimeMs == null) {
+      Log.e(tag, "Getting chapter sleep timer end of chapter/track but there is no current session")
+      return null
+    }
+
+    val timeLeftInChapter = currentChapterEndTimeMs - getCurrentTime()
+    return if (timeLeftInChapter < 2000L) {
+      Log.i(tag, "Getting chapter sleep timer time and current chapter has less than 2s remaining")
+      val nextChapterEndTimeMs = playerNotificationService.getEndTimeOfNextChapterOrTrack()
+      if (nextChapterEndTimeMs == null || currentChapterEndTimeMs == nextChapterEndTimeMs) {
+        Log.e(tag, "Invalid next chapter time. No current session or equal to current chapter. $nextChapterEndTimeMs")
+        null
+      } else {
+        nextChapterEndTimeMs
+      }
+    } else {
+      currentChapterEndTimeMs
+    }
+  }
+
+  private fun resetChapterTimer() {
+    this.getChapterEndTime()?.let { chapterEndTime ->
+      Log.d(tag, "Resetting stopped sleep timer to end of chapter $chapterEndTime")
+      vibrate()
+      setSleepTimer(chapterEndTime, true)
+      play()
+    }
+  }
+
+  private fun checkShouldResetSleepTimer() {
     if (!sleepTimerRunning) {
       if (sleepTimerFinishedAt <= 0L) return
 
@@ -175,15 +208,10 @@ class SleepTimerManager constructor(private val playerNotificationService: Playe
       // Set sleep timer
       //   When sleepTimerLength is 0 then use end of chapter/track time
       if (sleepTimerLength == 0L) {
-        val currentChapterEndTimeMs = playerNotificationService.getEndTimeOfChapterOrTrack()
+        Log.d(tag, "Resetting stopped chapter sleep timer")
-        if (currentChapterEndTimeMs == null) {
+        resetChapterTimer()
-          Log.e(tag, "Checking reset sleep timer to end of chapter/track but there is no current session")
-        } else {
-          vibrate()
-          setSleepTimer(currentChapterEndTimeMs, true)
-          play()
-        }
       } else {
+        Log.d(tag, "Resetting stopped sleep timer to length $sleepTimerLength")
         vibrate()
         setSleepTimer(sleepTimerLength, false)
         play()
@@ -193,8 +221,19 @@ class SleepTimerManager constructor(private val playerNotificationService: Playe
 
     // Does not apply to chapter sleep timers and timer must be running for at least 3 seconds
     if (sleepTimerLength > 0L && sleepTimerElapsed > 3000L) {
+      Log.d(tag, "Resetting running sleep timer to length $sleepTimerLength")
       vibrate()
       setSleepTimer(sleepTimerLength, false)
+    } else if (sleepTimerLength == 0L) {
+      // When navigating to previous chapters make sure this is still the end of the current chapter
+      this.getChapterEndTime()?.let { chapterEndTime ->
+        if (chapterEndTime != sleepTimerEndTime) {
+          Log.d(tag, "Resetting chapter sleep timer to end of chapter $chapterEndTime from $sleepTimerEndTime")
+          vibrate()
+          setSleepTimer(chapterEndTime, true)
+          play()
+        }
+      }
     }
   }
 
@@ -279,11 +318,11 @@ class SleepTimerManager constructor(private val playerNotificationService: Playe
         // Set sleep timer
         //   When sleepTimerLength is 0 then use end of chapter/track time
         if (deviceSettings.sleepTimerLength == 0L) {
-          val currentChapterEndTimeMs = playerNotificationService.getEndTimeOfChapterOrTrack()
+          val chapterEndTimeMs = this.getChapterEndTime()
-          if (currentChapterEndTimeMs == null) {
+          if (chapterEndTimeMs == null) {
             Log.e(tag, "Setting auto sleep timer to end of chapter/track but there is no current session")
           } else {
-            setSleepTimer(currentChapterEndTimeMs, true)
+            setSleepTimer(chapterEndTimeMs, true)
           }
         } else {
           setSleepTimer(deviceSettings.sleepTimerLength, false)

android/app/src/main/java/com/audiobookshelf/app/player/PlayerNotificationService.kt

@@ -648,6 +648,14 @@ class PlayerNotificationService : MediaBrowserServiceCompat()  {
     return getCurrentBookChapter()?.endMs ?: currentPlaybackSession?.getCurrentTrackEndTime()
   }
 
+  private fun getNextBookChapter():BookChapter? {
+    return currentPlaybackSession?.getNextChapterForTime(this.getCurrentTime())
+  }
+
+  fun getEndTimeOfNextChapterOrTrack():Long? {
+    return getNextBookChapter()?.endMs ?: currentPlaybackSession?.getNextTrackEndTime()
+  }
+
   // Called from PlayerListener play event
   // check with server if progress has updated since last play and sync progress update
   fun checkCurrentSessionProgress(seekBackTime:Long):Boolean {