Integrate 14 day confirmation pixel

Author: karlenDimla

pir/pir-impl/src/main/java/com/duckduckgo/pir/impl/pixels/OptOutConfirmationReporter.kt

@@ -51,29 +51,55 @@ class RealOptOutConfirmationReporter @Inject constructor(
 
             if (activeBrokers.isEmpty() || allValidRequestedOptOutJobs.isEmpty()) return@withContext
 
-            attemptFire7dayPixel(allValidRequestedOptOutJobs, activeBrokers)
+            allValidRequestedOptOutJobs.also {
+                it.attemptFirePixelForConfirmationDay(
+                    activeBrokers,
+                    7,
+                    { jobRecord -> jobRecord.confirmation7dayReportSentDateMs == 0L },
+                    { brokerUrl -> pixelSender.reportBrokerOptOutConfirmed7Days(brokerUrl) },
+                    { brokerUrl -> pixelSender.reportBrokerOptOutUnconfirmed7Days(brokerUrl) },
+                    { jobRecord, now ->
+                        pirSchedulingRepository.markOptOutDay7ConfirmationPixelSent(jobRecord.extractedProfileId, now)
+                    },
+                )
+
+                it.attemptFirePixelForConfirmationDay(
+                    activeBrokers,
+                    14,
+                    { jobRecord -> jobRecord.confirmation14dayReportSentDateMs == 0L },
+                    { brokerUrl -> pixelSender.reportBrokerOptOutConfirmed14Days(brokerUrl) },
+                    { brokerUrl -> pixelSender.reportBrokerOptOutUnconfirmed14Days(brokerUrl) },
+                    { jobRecord, now ->
+                        pirSchedulingRepository.markOptOutDay14ConfirmationPixelSent(jobRecord.extractedProfileId, now)
+                    },
+                )
+            }
         }
     }
 
-    private suspend fun attemptFire7dayPixel(
-        allValidRequestedOptOutJobs: List<OptOutJobRecord>,
+    private suspend fun List<OptOutJobRecord>.attemptFirePixelForConfirmationDay(
         activeBrokers: Map<String, Broker>,
+        confirmationDay: Long,
+        jobRecordFilter: (OptOutJobRecord) -> Boolean,
+        emitConfirmPixel: (String) -> Unit,
+        emitUnconfirmPixel: (String) -> Unit,
+        markOptOutJobRecordReporting: suspend (OptOutJobRecord, Long) -> Unit,
     ) {
         val now = currentTimeProvider.currentTimeMillis()
-        val optOutsForSevenDayPixel = allValidRequestedOptOutJobs.filter {
-            it.daysPassedSinceSubmission(now, 7) && it.confirmation7dayReportSentDateMs == 0L
+        val optOutsForSevenDayPixel = this.filter {
+            it.daysPassedSinceSubmission(now, confirmationDay) && jobRecordFilter(it)
         }
 
         optOutsForSevenDayPixel.forEach { optOutJobRecord ->
             val brokerUrl = activeBrokers[optOutJobRecord.brokerName]?.url ?: return@forEach
 
             if (optOutJobRecord.status == REMOVED) {
-                pixelSender.reportBrokerOptOutConfirmed7Days(brokerUrl)
+                emitConfirmPixel(brokerUrl)
             } else {
-                pixelSender.reportBrokerOptOutUnconfirmed7Days(brokerUrl)
+                emitUnconfirmPixel(brokerUrl)
             }
 
-            pirSchedulingRepository.markOptOutDay7ConfirmationPixelSent(optOutJobRecord.extractedProfileId, now)
+            markOptOutJobRecordReporting(optOutJobRecord, now)
         }
     }
 

pir/pir-impl/src/main/java/com/duckduckgo/pir/impl/store/PirSchedulingRepository.kt

@@ -130,6 +130,11 @@ interface PirSchedulingRepository {
         extractedProfileId: Long,
         timestampMs: Long,
     )
+
+    suspend fun markOptOutDay14ConfirmationPixelSent(
+        extractedProfileId: Long,
+        timestampMs: Long,
+    )
 }
 
 @ContributesBinding(
@@ -334,6 +339,15 @@ class RealPirSchedulingRepository @Inject constructor(
         }
     }
 
+    override suspend fun markOptOutDay14ConfirmationPixelSent(
+        extractedProfileId: Long,
+        timestampMs: Long,
+    ) {
+        withContext(dispatcherProvider.io()) {
+            jobSchedulingDao()?.update14DayConfirmationReportSentDate(extractedProfileId, timestampMs)
+        }
+    }
+
     private fun ScanJobRecordEntity.toRecord(): ScanJobRecord =
         ScanJobRecord(
             brokerName = this.brokerName,

pir/pir-impl/src/main/java/com/duckduckgo/pir/impl/store/db/JobSchedulingDao.kt

@@ -130,4 +130,16 @@ interface JobSchedulingDao {
         extractedProfileId: Long,
         newDate: Long,
     )
+
+    @Query(
+        """
+    UPDATE pir_optout_job_record
+    SET reporting_fourteenDayConfirmationReportSentDateMs = :newDate
+    WHERE extractedProfileId = :extractedProfileId
+    """,
+    )
+    fun update14DayConfirmationReportSentDate(
+        extractedProfileId: Long,
+        newDate: Long,
+    )
 }