Add track events and hook in the upgrade link tapped

wordpress-mobile · Jun 13, 2024 · a1a1063 · a1a1063
1 parent 788d6f6
commit a1a1063
Showing 1 changed file with 37 additions and 11 deletions.
diff --git a/WordPress/src/main/java/org/wordpress/android/ui/voicetocontent/VoiceToContentViewModel.kt b/WordPress/src/main/java/org/wordpress/android/ui/voicetocontent/VoiceToContentViewModel.kt
@@ -13,24 +13,24 @@ import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.launch
 import org.wordpress.android.R
+import org.wordpress.android.analytics.AnalyticsTracker.Stat
 import org.wordpress.android.fluxc.model.jetpackai.JetpackAIAssistantFeature
 import org.wordpress.android.modules.UI_THREAD
 import org.wordpress.android.ui.mysite.SelectedSiteRepository
-import org.wordpress.android.util.audio.IAudioRecorder
-import org.wordpress.android.viewmodel.ContextProvider
-import org.wordpress.android.viewmodel.ScopedViewModel
-import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.INITIALIZING
-import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.READY_TO_RECORD
-import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.RECORDING
 import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.ERROR
 import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.INELIGIBLE_FOR_FEATURE
+import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.INITIALIZING
 import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.PROCESSING
-
+import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.READY_TO_RECORD
+import org.wordpress.android.ui.voicetocontent.VoiceToContentUIStateType.RECORDING
+import org.wordpress.android.util.audio.IAudioRecorder
+import org.wordpress.android.util.audio.IAudioRecorder.AudioRecorderResult.Error
+import org.wordpress.android.util.audio.IAudioRecorder.AudioRecorderResult.Success
+import org.wordpress.android.viewmodel.ContextProvider
+import org.wordpress.android.viewmodel.ScopedViewModel
 import java.io.File
 import javax.inject.Inject
 import javax.inject.Named
-import org.wordpress.android.util.audio.IAudioRecorder.AudioRecorderResult.Success
-import org.wordpress.android.util.audio.IAudioRecorder.AudioRecorderResult.Error
 
 @HiltViewModel
 class VoiceToContentViewModel @Inject constructor(
@@ -41,7 +41,7 @@ class VoiceToContentViewModel @Inject constructor(
     private val recordingUseCase: RecordingUseCase,
     private val contextProvider: ContextProvider,
     private val prepareVoiceToContentUseCase: PrepareVoiceToContentUseCase,
-    private val logger: VoiceToContentLogger
+    private val logger: VoiceToContentTelemetry
 ) : ScopedViewModel(mainDispatcher) {
     private val _requestPermission = MutableLiveData<Unit>()
     val requestPermission = _requestPermission as LiveData<Unit>
@@ -52,6 +52,11 @@ class VoiceToContentViewModel @Inject constructor(
     private val _amplitudes = MutableLiveData<List<Float>>()
     val amplitudes: LiveData<List<Float>> get() = _amplitudes
 
+    private val _onIneligibleForVoiceToContent = MutableLiveData<String>()
+    val onIneligibleForVoiceToContent = _onIneligibleForVoiceToContent as LiveData<String>
+
+    private var isStarted = false
+
     private val _state = MutableStateFlow(VoiceToContentUiState(
         uiStateType = INITIALIZING,
         header = HeaderUIModel(
@@ -79,6 +84,10 @@ class VoiceToContentViewModel @Inject constructor(
         val site = selectedSiteRepository.getSelectedSite()
         if (site == null || !isVoiceToContentEnabled()) return
 
+        if (!isStarted) {
+            logger.track(Stat.VOICE_TO_CONTENT_SHEET_SHOWN)
+        }
+
         viewModelScope.launch {
             when (val result = prepareVoiceToContentUseCase.execute(site)) {
                 is PrepareVoiceToContentResult.Success -> {
@@ -90,6 +99,8 @@ class VoiceToContentViewModel @Inject constructor(
                 }
             }
         }
+
+        isStarted = true
     }
 
     // Recording
@@ -167,6 +178,7 @@ class VoiceToContentViewModel @Inject constructor(
 
     // Permissions
     private fun onRequestPermission() {
+        logger.track(Stat.VOICE_TO_CONTENT_BUTTON_START_RECORDING_TAPPED)
         _requestPermission.postValue(Unit)
     }
 
@@ -185,14 +197,17 @@ class VoiceToContentViewModel @Inject constructor(
 
     // user actions
     private fun onMicTap() {
+        logger.track(Stat.VOICE_TO_CONTENT_BUTTON_START_RECORDING_TAPPED)
         startRecording()
     }
 
     private fun onStopTap() {
+        logger.track(Stat.VOICE_TO_CONTENT_BUTTON_DONE_TAPPED)
         stopRecording()
     }
 
     private fun onClose() {
+        logger.track(Stat.VOICE_TO_CONTENT_BUTTON_CLOSE_TAPPED)
         _dismiss.postValue(Unit)
     }
 
@@ -201,6 +216,13 @@ class VoiceToContentViewModel @Inject constructor(
         start()
     }
 
+    private fun onLinkTap(url: String?) {
+        logger.track(Stat.VOICE_TO_CONTENT_BUTTON_UPGRADE_TAPPED)
+        url?.let {
+            _onIneligibleForVoiceToContent.postValue(it)
+        }
+    }
+
     // transitions
     private fun transitionToInitializing() {
         _state.value = VoiceToContentUiState(
@@ -222,6 +244,9 @@ class VoiceToContentViewModel @Inject constructor(
 
     private fun transitionToReadyToRecordOrIneligibleForFeature(model: JetpackAIAssistantFeature) {
         val isEligibleForFeature = voiceToContentFeatureUtils.isEligibleForVoiceToContent(model)
+        if (!isEligibleForFeature) {
+            logger.track(Stat.VOICE_TO_CONTENT_BUTTON_RECORDING_LIMIT_REACHED)
+        }
         val requestsAvailable = voiceToContentFeatureUtils.getRequestLimit(model)
         val currentState = _state.value
         _state.value = currentState.copy(
@@ -236,7 +261,8 @@ class VoiceToContentViewModel @Inject constructor(
                 onMicTap = ::onMicTap,
                 onRequestPermission = ::onRequestPermission,
                 hasPermission = hasAllPermissionsForRecording(),
-                upgradeUrl = model.upgradeUrl
+                upgradeUrl = model.upgradeUrl,
+                onLinkTap = ::onLinkTap
             )
         )
     }