From 350ad72c793f356a32da89af327899efe5870301 Mon Sep 17 00:00:00 2001
From: Toni Sevener <tsevener@wikimedia.org>
Date: Wed, 28 Aug 2024 17:28:31 -0500
Subject: [PATCH] Add additional metadata when sending group assignment event

---
 .../Code/ArticleViewController+Editing.swift  |  6 ++++-
 Wikipedia/Code/EditInteractionFunnel.swift    | 25 +++++++++++++++----
 Wikipedia/Code/ExploreViewController.swift    |  7 ++++--
 3 files changed, 30 insertions(+), 8 deletions(-)

diff --git a/Wikipedia/Code/ArticleViewController+Editing.swift b/Wikipedia/Code/ArticleViewController+Editing.swift
index 13d4d8bca9f..db42f135ee4 100644
--- a/Wikipedia/Code/ArticleViewController+Editing.swift
+++ b/Wikipedia/Code/ArticleViewController+Editing.swift
@@ -296,7 +296,11 @@ extension ArticleViewController: EditorViewControllerDelegate {
         
         do {
             try dataController.assignArticleEditorExperiment(isLoggedIn: isLoggedIn, project: project)
-            EditInteractionFunnel.shared.logAltTextDidAssignArticleEditorGroup(project: WikimediaProject(wmfProject: project))
+            
+            if let user = dataStore.authenticationManager.getLoggedInUserCache(for: articleURL) {
+                EditInteractionFunnel.shared.logAltTextDidAssignArticleEditorGroup(username:user.name, userEditCount: user.editCount, articleTitle: articleTitle, image: filename, registrationDate: user.registrationDateString, project: WikimediaProject(wmfProject: project))
+            }
+            
         } catch let error {
             DDLogWarn("Error assigning alt text article editor experiment: \(error)")
         }
diff --git a/Wikipedia/Code/EditInteractionFunnel.swift b/Wikipedia/Code/EditInteractionFunnel.swift
index a04c4467ca4..e9e8b4dac0b 100644
--- a/Wikipedia/Code/EditInteractionFunnel.swift
+++ b/Wikipedia/Code/EditInteractionFunnel.swift
@@ -277,13 +277,20 @@ final class EditInteractionFunnel {
     
     // MARK: Alt-Text-Experiment
     
-    func logAltTextDidAssignImageRecsGroup(project: WikimediaProject) {
-        
+    func logAltTextDidAssignImageRecsGroup(username: String, userEditCount: UInt64, articleTitle: String, image: String, registrationDate: String?, project: WikimediaProject) {
         guard let group = WMFAltTextDataController.shared?.assignedAltTextImageRecommendationsGroupForLogging() else {
             return
         }
         
-        var actionData: [String: String] = [:]
+        var actionData = ["article_title": articleTitle,
+                          "image": image,
+                          "username": username,
+                          "event_user_revision_count": String(userEditCount)]
+        
+        if let registrationDate {
+            actionData["user_create_date"] = registrationDate
+        }
+        
         switch group {
         case "A":
             actionData["exp_b_group"] = "a"
@@ -296,13 +303,21 @@ final class EditInteractionFunnel {
         logEvent(activeInterface: .altTextEditingOnboarding, action: .groupAssignment, actionData: actionData, project: project)
     }
     
-    func logAltTextDidAssignArticleEditorGroup(project: WikimediaProject) {
+    func logAltTextDidAssignArticleEditorGroup(username: String, userEditCount: UInt64, articleTitle: String, image: String, registrationDate: String?, project: WikimediaProject) {
         
         guard let group = WMFAltTextDataController.shared?.assignedAltTextArticleEditorGroupForLogging() else {
             return
         }
         
-        var actionData: [String: String] = [:]
+        var actionData = ["article_title": articleTitle,
+                          "image": image,
+                          "username": username,
+                          "event_user_revision_count": String(userEditCount)]
+        
+        if let registrationDate {
+            actionData["user_create_date"] = registrationDate
+        }
+        
         switch group {
         case "C":
             actionData["exp_c_group"] = "c"
diff --git a/Wikipedia/Code/ExploreViewController.swift b/Wikipedia/Code/ExploreViewController.swift
index e8ec5123bd2..9ed8e477cd8 100644
--- a/Wikipedia/Code/ExploreViewController.swift
+++ b/Wikipedia/Code/ExploreViewController.swift
@@ -1536,13 +1536,16 @@ extension ExploreViewController: WMFImageRecommendationsLoggingDelegate {
     
     func logAltTextExperimentDidAssignGroup() {
         
-        guard let imageRecommendationsViewModel else {
+        guard let imageRecommendationsViewModel,
+              let lastRecommendation = imageRecommendationsViewModel.lastRecommendation,
+           let siteURL = dataStore.languageLinkController.appLanguage?.siteURL,
+           let user = dataStore.authenticationManager.getLoggedInUserCache(for: siteURL) else {
             return
         }
         
         let project = WikimediaProject(wmfProject: imageRecommendationsViewModel.project)
         
-        EditInteractionFunnel.shared.logAltTextDidAssignImageRecsGroup(project: project)
+        EditInteractionFunnel.shared.logAltTextDidAssignImageRecsGroup(username:user.name, userEditCount: user.editCount, articleTitle: lastRecommendation.title, image: lastRecommendation.imageData.filename, registrationDate: user.registrationDateString, project: WikimediaProject(wmfProject: imageRecommendationsViewModel.project))
     }
 
     func logOnboardingDidTapPrimaryButton() {