feat(drive): In-app pdf viewer (WPB-28358)

app/build.gradle.kts

@@ -216,6 +216,7 @@ dependencies {
     implementationWithCoverage(projects.core.media)
     implementationWithCoverage(projects.core.videoPlayer)
     implementationWithCoverage(projects.core.audioPlayer)
+    implementationWithCoverage(projects.core.pdfViewer)
     implementationWithCoverage(projects.core.notification)
     implementationWithCoverage(projects.core.navigation)
     implementationWithCoverage(projects.core.search)

app/src/main/kotlin/com/wire/android/di/accountScoped/CellsModule.kt

@@ -20,6 +20,7 @@ package com.wire.android.di.accountScoped
 import com.wire.android.di.CurrentAccount
 import com.wire.android.di.KaliumCoreLogic
 import com.wire.android.ui.home.conversations.model.messagetypes.multipart.CellAssetRefreshHelper
+import com.wire.android.pdfviewer.PdfRemoteLoader
 import com.wire.kalium.cells.CellsScope
 import com.wire.kalium.cells.domain.CellUploadManager
 import com.wire.kalium.cells.domain.usecase.AddAttachmentDraftUseCase
@@ -71,11 +72,14 @@ import com.wire.kalium.cells.domain.usecase.versioning.GetNodeVersionsUseCase
 import com.wire.kalium.cells.domain.usecase.versioning.RestoreNodeVersionUseCase
 import com.wire.kalium.cells.paginatedConversationsFlowUseCase
 import com.wire.kalium.cells.paginatedFilesFlowUseCase
+import com.wire.kalium.common.functional.fold
 import com.wire.kalium.logic.CoreLogic
 import com.wire.kalium.logic.data.user.UserId
 import com.wire.kalium.logic.featureFlags.KaliumConfigs
 import dev.zacsweers.metro.BindingContainer
 import dev.zacsweers.metro.Provides
+import java.io.IOException
+import okio.Path.Companion.toOkioPath
 
 @Suppress("TooManyFunctions")
 @BindingContainer
@@ -249,4 +253,20 @@ class CellsModule {
 
     @Provides
     fun provideGetUserNamesUseCase(cellsScope: CellsScope): GetUserNameUseCase = cellsScope.getUserName
+
+    @Provides
+    fun providePdfRemoteLoader(download: DownloadCellFileUseCase): PdfRemoteLoader =
+        PdfRemoteLoader { assetId, remotePath, conversationId, assetSize, outFile ->
+            download(
+                assetId = assetId,
+                conversationId = conversationId,
+                outFilePath = outFile.toPath().toOkioPath(),
+                assetSize = assetSize,
+                remoteFilePath = remotePath,
+                onProgressUpdate = {},
+            ).fold(
+                { failure -> Result.failure(IOException("PDF download failed: $failure")) },
+                { Result.success(Unit) },
+            )
+        }
 }

app/src/main/kotlin/com/wire/android/di/metro/AppSessionViewModelGraph.kt

@@ -17,7 +17,6 @@
  */
 package com.wire.android.di.metro
 
-import com.wire.android.audioplayer.AudioPlayerManualViewModelFactoryMetroBindings
 import com.wire.android.di.CurrentAccount
 import com.wire.android.di.ImageLoadingModule
 import com.wire.android.di.SessionModule
@@ -41,6 +40,7 @@ import com.wire.android.feature.cells.ui.CellsMetroViewModelBindings
 import com.wire.android.feature.meetings.ui.MeetingsManualViewModelFactoryMetroBindings
 import com.wire.android.feature.meetings.ui.MeetingsMetroViewModelBindings
 import com.wire.android.feature.sketch.SketchMetroViewModelBindings
+import com.wire.android.pdfviewer.PdfViewerManualViewModelFactoryMetroBindings
 import com.wire.android.search.SearchManualViewModelFactoryMetroBindings
 import com.wire.android.ui.authentication.AuthenticationViewModelGraph
 import com.wire.android.ui.calling.CallingManualViewModelFactoryMetroBindings
@@ -60,6 +60,7 @@ import com.wire.android.ui.home.conversations.ConversationSearchFolderMetroViewM
 import com.wire.android.ui.home.settings.SettingsManualViewModelFactoryMetroBindings
 import com.wire.android.util.ui.WireSessionImageLoader
 import com.wire.android.videoplayer.VideoPlayerManualViewModelFactoryMetroBindings
+import com.wire.android.audioplayer.AudioPlayerManualViewModelFactoryMetroBindings
 import com.wire.kalium.logic.data.user.UserId
 import dev.zacsweers.metro.AppScope
 import dev.zacsweers.metro.ContributesTo
@@ -113,6 +114,7 @@ annotation class MetroSessionScope
         SearchManualViewModelFactoryMetroBindings::class,
         VideoPlayerManualViewModelFactoryMetroBindings::class,
         AudioPlayerManualViewModelFactoryMetroBindings::class,
+        PdfViewerManualViewModelFactoryMetroBindings::class,
         ImageLoadingModule::class,
     ]
 )

app/src/main/kotlin/com/wire/android/navigation/routes/media/MediaNavigation3Entries.kt

@@ -12,6 +12,7 @@ import com.wire.android.navigation.navigation3.WireNavigation3ResultType
 import com.wire.android.navigation.navigation3.WireNavigation3Runtime
 import com.wire.android.navigation.navigation3.wireEntry
 import com.wire.android.videoplayer.VideoPlayer
+import com.wire.android.pdfviewer.PdfViewer
 import com.wire.android.ui.home.FeatureFlagState
 import com.wire.android.ui.home.conversations.ConversationNavArgs
 import com.wire.android.ui.home.conversations.checkAssetRestrictionsViewModel
@@ -122,6 +123,17 @@ internal fun mediaNavigation3Entries(
             onNavigateBack = runtime.navigator::goBack,
         )
     }
+    wireEntry<PdfViewerRoute>(presentation = WireEntryPresentation.PopUp) { route ->
+        PdfViewer(
+            localPath = route.localPath,
+            assetId = route.assetId,
+            remotePath = route.remotePath,
+            conversationId = route.conversationId,
+            assetSize = route.assetSize,
+            fileName = route.fileName,
+            onNavigateBack = runtime.navigator::goBack,
+        )
+    }
     wireEntry<MessageDetailsRoute>(presentation = WireEntryPresentation.PopUp) { route ->
         MessageDetailsRouteScreen(messageDetailsViewModel(route.toViewModelArgs()), runtime.navigator::goBack)
     }

app/src/main/kotlin/com/wire/android/navigation/routes/media/MediaRoutes.kt

@@ -75,6 +75,21 @@ data class VideoPlayerRoute(
     companion object { const val ROUTE_ID = "app/video_player_screen" }
 }
 
+@Serializable
+data class PdfViewerRoute(
+    override val sessionId: WireSessionId,
+    val localPath: String? = null,
+    val assetId: String? = null,
+    val remotePath: String? = null,
+    val conversationId: String? = null,
+    val assetSize: Long = 0L,
+    val fileName: String? = null,
+    override val entryId: WireNavEntryId = WireNavEntryId.random(),
+) : SessionRoute {
+    override val routeId = ROUTE_ID
+    companion object { const val ROUTE_ID = "app/pdf_viewer_screen" }
+}
+
 @Serializable
 data class MessageDetailsRoute(
     override val sessionId: WireSessionId,

app/src/main/kotlin/com/wire/android/navigation/runtime/WireNavigation3Contributions.kt

@@ -95,7 +95,7 @@ internal data class WireNavigation3ContributionCatalog(
  * app-lock overlays finish the registry. An entry type must be owned by exactly one contribution.
  */
 internal object WireNavigation3Contributions {
-    const val EXPECTED_ROUTE_REGISTRATION_COUNT: Int = 107
+    const val EXPECTED_ROUTE_REGISTRATION_COUNT: Int = 109
     const val EXPECTED_INSTALLER_COUNT: Int = 19
 
     fun create(

app/src/main/kotlin/com/wire/android/navigation/runtime/WireNavigation3ProductionActions.kt

@@ -24,6 +24,7 @@ import com.wire.android.feature.cells.navigation.CellImageViewerRoute
 import com.wire.android.feature.cells.navigation.CellsFilesArguments
 import com.wire.android.feature.cells.navigation.CellsSearchType
 import com.wire.android.feature.cells.navigation.ConversationFilesRoute
+import com.wire.android.feature.cells.navigation.PdfViewerRoute
 import com.wire.android.feature.cells.navigation.PublicLinkRoute
 import com.wire.android.feature.cells.navigation.SearchRoute
 import com.wire.android.feature.cells.navigation.VideoPlayerRoute
@@ -196,6 +197,19 @@ internal class WireNavigation3ProductionActions(
                 )
             )
         },
+        showPdfViewer = {
+            navigate(
+                PdfViewerRoute(
+                    sessionId = requireSession(),
+                    localPath = it.localPath,
+                    assetId = it.uuid,
+                    remotePath = it.remotePath,
+                    conversationId = it.conversationId,
+                    assetSize = it.size ?: 0L,
+                    fileName = it.name,
+                )
+            )
+        },
     )
     override val conversationList: ConversationListNavigationActions = ConversationListNavigationActions(
         openConversation = { openConversation(it.toProfileId()) },

app/src/main/kotlin/com/wire/android/ui/common/multipart/MultipartAttachmentUi.kt

@@ -33,6 +33,7 @@ data class MultipartAttachmentUi(
     val contentUrl: String? = null,
     val contentUrlExpiresAt: Long? = null,
     val previewUrl: String? = null,
+    val remotePath: String? = null,
     val mimeType: String,
     val assetType: AttachmentFileType,
     val assetSize: Long?,
@@ -60,6 +61,7 @@ fun CellAssetContent.toUiModel(progress: Float?, isAvailableOffline: Boolean = f
     contentUrl = this.contentUrl,
     contentUrlExpiresAt = this.contentUrlExpiresAt,
     previewUrl = this.previewUrl,
+    remotePath = this.assetPath,
     mimeType = this.mimeType,
     assetType = AttachmentFileType.fromMimeType(mimeType),
     assetSize = this.assetSize,

app/src/main/kotlin/com/wire/android/ui/home/conversations/ConversationMessageComposer.kt

@@ -59,6 +59,7 @@ internal fun ConversationMessageComposer(
     onImageFullScreenMode: (UIMessage.Regular, Boolean, String?) -> Unit,
     onVideoClick: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit,
     onAudioClick: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit,
+    onPdfClick: (localPath: String?, assetId: String?, remotePath: String?, assetSize: Long, fileName: String?) -> Unit,
     onReactionClicked: (String, String) -> Unit,
     onResetSessionClicked: (senderUserId: UserId, clientId: String?) -> Unit,
     onOpenProfile: (senderId: MessageSenderId) -> Unit,
@@ -117,6 +118,7 @@ internal fun ConversationMessageComposer(
                     onImageClicked = onImageFullScreenMode,
                     onVideoClicked = onVideoClick,
                     onAudioClicked = onAudioClick,
+                    onPdfClicked = onPdfClick,
                     onLinkClicked = onLinkClick,
                     onReplyClicked = onNavigateToReplyOriginalMessage,
                     onResetSessionClicked = onResetSessionClicked,

app/src/main/kotlin/com/wire/android/ui/home/conversations/ConversationNavigation3Entries.kt

@@ -21,17 +21,18 @@ import com.wire.android.navigation.navigation3.WireEntryProviderInstaller
 import com.wire.android.navigation.navigation3.WireNavigation3ResultType
 import com.wire.android.navigation.navigation3.WireNavigation3Runtime
 import com.wire.android.navigation.navigation3.wireEntry
-import com.wire.android.navigation.routes.media.ImagesPreviewNavigation3ResultType
 import com.wire.android.navigation.routes.media.AuthenticatedImportMediaRoute
+import com.wire.android.navigation.routes.media.ImagesPreviewNavigation3ResultType
 import com.wire.android.navigation.routes.media.ImagesPreviewResult
 import com.wire.android.navigation.routes.media.ImagesPreviewRoute
 import com.wire.android.navigation.routes.media.MediaConversationId
 import com.wire.android.navigation.routes.media.MediaGalleryNavigation3ResultType
 import com.wire.android.navigation.routes.media.MediaGalleryResult
 import com.wire.android.navigation.routes.media.MediaGalleryResultAction
 import com.wire.android.navigation.routes.media.MediaGalleryRoute
-import com.wire.android.navigation.routes.media.VideoPlayerRoute
 import com.wire.android.navigation.routes.media.MessageDetailsRoute
+import com.wire.android.navigation.routes.media.PdfViewerRoute
+import com.wire.android.navigation.routes.media.VideoPlayerRoute
 import com.wire.android.navigation.routes.media.toLegacy
 import com.wire.android.ui.calling.conversationCallViewModel
 import com.wire.android.ui.home.conversations.details.ConversationDetailsId
@@ -260,6 +261,21 @@ private fun ConversationNavigation3Entry(
             )
         }
 
+        override fun openPdfViewer(localPath: String?, assetId: String?, remotePath: String?, assetSize: Long, fileName: String?) {
+            runtime.navigator.navigate(
+                WireNavigationCommand(
+                    PdfViewerRoute(
+                        sessionId = route.sessionId,
+                        localPath = localPath,
+                        assetId = assetId,
+                        remotePath = remotePath,
+                        assetSize = assetSize,
+                        fileName = fileName,
+                    )
+                )
+            )
+        }
+
         override fun openDrawingCanvas(conversationName: String, tempWritableUri: Uri?) {
             drawingRequestId = runtime.navigateForResult(
                 DrawingCanvasRoute(

app/src/main/kotlin/com/wire/android/ui/home/conversations/ConversationRouteScreen.kt

@@ -53,6 +53,8 @@ internal interface ConversationRouteScreenNavigation {
     fun openVideoPlayer(localPath: String?, contentUrl: String?, fileName: String?)
     fun openAudioPlayer(localPath: String?, contentUrl: String?, fileName: String?)
 
+    fun openPdfViewer(localPath: String?, assetId: String?, remotePath: String?, assetSize: Long, fileName: String?)
+
     fun openDrawingCanvas(
         conversationName: String,
         tempWritableUri: Uri?,

app/src/main/kotlin/com/wire/android/ui/home/conversations/ConversationScreen.kt

@@ -439,6 +439,7 @@ internal fun ConversationScreenRouteContent(
         },
         onVideoClick = navigation::openVideoPlayer,
         onAudioClick = navigation::openAudioPlayer,
+        onPdfClick = navigation::openPdfViewer,
         onStartCall = {
             conversationCallViewModel.startCallIfPossible(conversationInfoViewModel.conversationInfoViewState.conversationType)
         },
@@ -644,6 +645,7 @@ private fun ConversationScreenContent(
     onImageFullScreenMode: (UIMessage.Regular, Boolean, String?) -> Unit,
     onVideoClick: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit,
     onAudioClick: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit,
+    onPdfClick: (localPath: String?, assetId: String?, remotePath: String?, assetSize: Long, fileName: String?) -> Unit,
     onStartCall: () -> Unit,
     onJoinCall: () -> Unit,
     onReactionClick: (messageId: String, reactionEmoji: String) -> Unit,
@@ -757,6 +759,7 @@ private fun ConversationScreenContent(
                         onImageFullScreenMode = onImageFullScreenMode,
                         onVideoClick = onVideoClick,
                         onAudioClick = onAudioClick,
+                        onPdfClick = onPdfClick,
                         onReactionClicked = onReactionClick,
                         onResetSessionClicked = onResetSessionClick,
                         onOpenProfile = onOpenProfile,
@@ -900,6 +903,7 @@ fun PreviewConversationScreen() = WireTheme {
         onImageFullScreenMode = { _, _, _ -> },
         onVideoClick = { _, _, _ -> },
         onAudioClick = { _, _, _ -> },
+        onPdfClick = { _, _, _, _, _ -> },
         onStartCall = { },
         onJoinCall = { },
         onReactionClick = { _, _ -> },

app/src/main/kotlin/com/wire/android/ui/home/conversations/messages/item/MessageClickActions.kt

@@ -31,6 +31,13 @@
     open val onImageClicked: (UIMessage.Regular, Boolean, String?) -> Unit = { _, _, _ -> }
     open val onVideoClicked: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit = { _, _, _ -> }
     open val onAudioClicked: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit = { _, _, _ -> }
+    open val onPdfClicked: (
+        localPath: String?,
+        assetId: String?,
+        remotePath: String?,
+        assetSize: Long,
+        fileName: String?
+    ) -> Unit = { _, _, _, _, _ -> }
     open val onLinkClicked: (String) -> Unit = {}
     open val onReplyClicked: (UIMessage.Regular) -> Unit = {}
     open val onResetSessionClicked: (senderUserId: UserId, clientId: String?) -> Unit = { _, _ -> }
@@ -50,6 +57,13 @@
         override val onImageClicked: (UIMessage.Regular, Boolean, String?) -> Unit = { _, _, _ -> },
         override val onVideoClicked: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit = { _, _, _ -> },
         override val onAudioClicked: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit = { _, _, _ -> },
+        override val onPdfClicked: (
+            localPath: String?,
+            assetId: String?,
+            remotePath: String?,
+            assetSize: Long,
+            fileName: String?
+        ) -> Unit = { _, _, _, _, _ -> },
         override val onLinkClicked: (String) -> Unit = {},
         override val onReplyClicked: (UIMessage.Regular) -> Unit = {},
         override val onResetSessionClicked: (senderUserId: UserId, clientId: String?) -> Unit = { _, _ -> },

app/src/main/kotlin/com/wire/android/ui/home/conversations/messages/item/MessageContentAndStatus.kt

@@ -74,6 +74,13 @@ internal fun UIMessage.Regular.MessageContentAndStatus(
     onImageClicked: (UIMessage.Regular, Boolean, String?) -> Unit,
     onVideoClicked: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit,
     onAudioClicked: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit,
+    onPdfClicked: (
+        localPath: String?,
+        assetId: String?,
+        remotePath: String?,
+        assetSize: Long,
+        fileName: String?
+    ) -> Unit,
     onProfileClicked: (senderId: MessageSenderId) -> Unit,
     onLinkClicked: (String) -> Unit,
     onReplyClicked: (UIMessage.Regular) -> Unit,
@@ -123,6 +130,7 @@ internal fun UIMessage.Regular.MessageContentAndStatus(
                 onMultipartImageClick = onMultipartImageClickable,
                 onMultipartVideoClick = onVideoClicked,
                 onMultipartAudioClick = onAudioClicked,
+                onMultipartPdfClick = onPdfClicked,
                 onOpenProfile = onProfileClicked,
                 onLinkClick = onLinkClicked,
                 onReplyClick = onReplyClickable,
@@ -173,6 +181,13 @@ private fun MessageContent(
     onMultipartImageClick: (String) -> Unit,
     onMultipartVideoClick: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit,
     onMultipartAudioClick: (localPath: String?, contentUrl: String?, fileName: String?) -> Unit,
+    onMultipartPdfClick: (
+        localPath: String?,
+        assetId: String?,
+        remotePath: String?,
+        assetSize: Long,
+        fileName: String?
+    ) -> Unit,
     onOpenProfile: (senderId: MessageSenderId) -> Unit,
     onLinkClick: (String) -> Unit,
     onReplyClick: Clickable,
@@ -468,6 +483,7 @@ private fun MessageContent(
                     onImageAttachmentClick = onMultipartImageClick,
                     onVideoAttachmentClick = onMultipartVideoClick,
                     onAudioAttachmentClick = onMultipartAudioClick,
+                    onPdfAttachmentClick = onMultipartPdfClick,
                 )
             }
 

app/src/main/kotlin/com/wire/android/ui/home/conversations/messages/item/MessageContentItem.kt

@@ -97,6 +97,7 @@ fun MessageContentItem(
                     onImageClicked = clickActions.onImageClicked,
                     onVideoClicked = clickActions.onVideoClicked,
                     onAudioClicked = clickActions.onAudioClicked,
+                    onPdfClicked = clickActions.onPdfClicked,
                     searchQuery = searchQuery,
                     accent = accent,
                     onProfileClicked = clickActions.onProfileClicked,