Fix IndexOutOfBounds after getRefreshKey

It's possible that on a paging source that's not in the initial
generation, we fail with an `IndexOutOfBoundsException`. This
is caused when the underlying data source undergoes a delete, which
then invalidates the paging source, which then calls
`getRefreshKey`. In the event that the the refresh key is the
very last index of the previous paging source, we would fail
with the above exception because that index does not exist
in the next generation paging source.

Closes cashapp#2434.

extensions/android-paging3/src/main/java/com/squareup/sqldelight/android/paging3/OffsetQueryPagingSource.kt

@@ -34,9 +34,13 @@ internal class OffsetQueryPagingSource<RowType : Any>(
     params: LoadParams<Long>
   ): LoadResult<Long, RowType> = withContext(dispatcher) {
     try {
-      val key = params.key ?: 0L
       transacter.transactionWithResult {
         val count = countQuery.executeAsOne()
+        val key = when (params) {
+          is LoadParams.Refresh -> params.key?.let { notNullKey -> minOf(count - 1, notNullKey) }
+          else -> params.key
+        } ?: 0L
+
         if (count != 0L && key >= count) throw IndexOutOfBoundsException()
 
         val loadSize = if (key < 0) params.loadSize + key else params.loadSize

extensions/android-paging3/src/test/java/com/squareup/sqldelight/android/paging3/OffsetQueryPagingSourceTest.kt

@@ -234,11 +234,25 @@ class OffsetQueryPagingSourceTest {
 
     runBlocking {
       assertFailsWith<IndexOutOfBoundsException> {
-        source.load(Refresh(10, 2, false))
+        source.load(PagingSource.LoadParams.Append(10, 2, false))
       }
     }
   }
 
+  @Test fun `refresh key too big gets truncated`() {
+    val source = OffsetQueryPagingSource(
+      this::query,
+      countQuery(),
+      transacter,
+      TestCoroutineDispatcher()
+    )
+
+    runBlocking {
+      val results = source.load(Refresh(10, 2, false))
+      assertEquals(listOf(9L), (results as LoadResult.Page).data)
+    }
+  }
+
   @Test fun `query invalidation invalidates paging source`() {
     val query = query(2, 0)
     val source = OffsetQueryPagingSource(