PIR: Implement getDataBrokers message (#6613)

Task/Issue URL:
https://app.asana.com/1/137249556945/project/488551667048375/task/1211088335699800?focus=true

### Description
Adds proper implementation of the getDataBrokers message for PIR Web UI.

### Steps to test this PR
https://app.asana.com/1/137249556945/task/1211095110950029?focus=true

### UI changes
No UI changes

Author: landomen

pir/pir-impl/src/main/java/com/duckduckgo/pir/impl/dashboard/messaging/handlers/PirWebGetDataBrokersMessageHandler.kt

@@ -55,20 +55,37 @@ class PirWebGetDataBrokersMessageHandler @Inject constructor(
         logcat { "PIR-WEB: PirWebGetDataBrokersMessageHandler: process $jsMessage" }
 
         appCoroutineScope.launch(dispatcherProvider.io()) {
-            val brokers = repository.getAllActiveBrokerObjects()
-
             jsMessaging.sendResponse(
                 jsMessage,
                 response = PirWebMessageResponse.GetDataBrokersResponse(
-                    dataBrokers = brokers.map {
-                        PirWebMessageResponse.GetDataBrokersResponse.DataBroker(
-                            url = it.url,
-                            name = it.name,
-                            parentURL = it.parent,
-                        )
-                    },
+                    dataBrokers = getDataBrokers(),
                 ),
             )
         }
     }
+
+    private suspend fun getDataBrokers(): List<PirWebMessageResponse.GetDataBrokersResponse.DataBroker> {
+        val activeBrokers = repository.getAllActiveBrokerObjects().associateBy { it.name }
+        val mirrorSites = repository.getAllMirrorSites().filter { it.removedAt == 0L }
+        val brokerOptOutUrls = repository.getAllBrokerOptOutUrls()
+
+        val mappedBrokers = activeBrokers.values.map {
+            PirWebMessageResponse.GetDataBrokersResponse.DataBroker(
+                url = it.url,
+                name = it.name,
+                parentURL = it.parent,
+                optOutUrl = brokerOptOutUrls[it.name],
+            )
+        }
+        val mappedMirrorSites = mirrorSites.map {
+            PirWebMessageResponse.GetDataBrokersResponse.DataBroker(
+                url = it.url,
+                name = it.name,
+                parentURL = activeBrokers[it.parentSite]?.url,
+                optOutUrl = brokerOptOutUrls[it.parentSite],
+            )
+        }
+
+        return (mappedBrokers + mappedMirrorSites).distinct()
+    }
 }

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

@@ -66,6 +66,11 @@ interface PirRepository {
 
     suspend fun getAllBrokersForScan(): List<String>
 
+    /**
+     * Returns a map of broker names to their opt-out URLs.
+     */
+    suspend fun getAllBrokerOptOutUrls(): Map<String, String?>
+
     suspend fun getEtagForFilename(fileName: String): String?
 
     suspend fun updateBrokerData(
@@ -234,6 +239,12 @@ internal class RealPirRepository(
             return@withContext brokerDao.getAllBrokersNamesWithScanSteps()
         }
 
+    override suspend fun getAllBrokerOptOutUrls(): Map<String, String?> =
+        withContext(dispatcherProvider.io()) {
+            val brokerOptOuts = brokerDao.getAllBrokerOptOuts()
+            return@withContext brokerOptOuts.associate { it.brokerName to it.optOutUrl }
+        }
+
     override suspend fun getEtagForFilename(fileName: String): String =
         withContext(dispatcherProvider.io()) {
             return@withContext brokerJsonDao.getEtag(fileName)

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

@@ -48,6 +48,9 @@ interface BrokerDao {
     @Query("DELETE from pir_broker_details where name = :brokerName")
     fun deleteBroker(brokerName: String)
 
+    @Query("SELECT * FROM pir_broker_opt_out")
+    fun getAllBrokerOptOuts(): List<BrokerOptOut>
+
     @Query("SELECT * from pir_broker_mirror_sites")
     fun getAllMirrorSites(): List<MirrorSiteEntity>