fix: prepare sabr cache misses

Author: Priveetee

src/main/kotlin/dev/typetype/server/services/SabrSessionPump.kt

@@ -37,7 +37,7 @@ internal class SabrSessionPump {
                 return@withLock segment
             }
             if (holder.session.isBeyondEnd(request)) return@withLock null
-            if (!request.isInitializationSegment) holder.session.prepareForForwardJump(request)
+            holder.session.prepareForRequestedSegment(request)
             runCatching { holder.session.fetchSegment(request, localization) }
                 .getOrNull()
                 ?.also { holder.markServed(it) }
@@ -93,10 +93,7 @@ internal class SabrSessionPump {
 
     private fun YoutubeSabrSession.prepareForRequestedSegment(request: SabrSegmentRequest) {
         if (request.isInitializationSegment) return
-        val max = streamState.getMaxSegment(request.format)
-        when {
-            request.sequenceNumber <= max -> prepareForRewind(request)
-            request.sequenceNumber > max + 1 -> prepareForForwardJump(request)
-        }
+        prepareForRewind(request)
+        prepareForForwardJump(request)
     }
 }

src/test/kotlin/dev/typetype/server/services/SabrSessionPumpTest.kt

@@ -30,6 +30,7 @@ class SabrSessionPumpTest {
         val segment = mediaSegment(startMs, durationMs)
         every { session.getCachedSegment(request) } returns null
         every { session.isBeyondEnd(request) } returns false
+        every { session.prepareForRewind(request) } returns Unit
         every { session.prepareForForwardJump(request) } returns Unit
         every { session.fetchSegment(request, any<Localization>()) } returns segment
         every { session.setPlayHeadMs(startMs + durationMs) } returns Unit
@@ -45,6 +46,7 @@ class SabrSessionPumpTest {
         val fetched = SabrSessionPump().fetchSegment(holder, request)
 
         assertSame(segment, fetched)
+        verify { session.prepareForRewind(request) }
         verify { session.prepareForForwardJump(request) }
         verify { session.fetchSegment(request, any<Localization>()) }
         verify { session.setPlayHeadMs(startMs + durationMs) }