Fragment for displaying media preview in cameraX sample

MediaViewerFragment.kt

class MediaViewerFragment : Fragment(R.layout.fragment_media_viewer) {

    private val args: MediaViewerFragmentArgs by navArgs()

    override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
        super.onViewCreated(view, savedInstanceState)

        btnBack.setOnClickListener { findNavController().popBackStack() }

        val mediaUri = args.mediaUri
        if (mediaUri.path?.endsWith(VIDEO_FILE_EXTENSION, true) == true) {
            val mediaController = MediaController(requireContext())
            videoView.setMediaController(mediaController)
            videoView.setVideoURI(mediaUri)
            videoView.seekTo(1)
            videoView.visibility = VISIBLE
            photoView.visibility = GONE
        } else {
            photoView.setImageURI(mediaUri)
            photoView.visibility = VISIBLE
            videoView.visibility = GONE
        }
    }
}