From 16511cc76d686aa1e6ddbfcae8fa6f2ef1b59129 Mon Sep 17 00:00:00 2001
From: koalasat <koalasat@satstralia.com>
Date: Fri, 1 Nov 2024 15:09:08 +0100
Subject: [PATCH] Connect to inbox relays only

---
 .../pokey/service/NotificationsService.kt     | 48 +++++++++++++------
 app/src/main/res/values/strings.xml           |  1 +
 2 files changed, 35 insertions(+), 14 deletions(-)

diff --git a/app/src/main/java/com/koalasat/pokey/service/NotificationsService.kt b/app/src/main/java/com/koalasat/pokey/service/NotificationsService.kt
index 6e90de0..0f73e0a 100644
--- a/app/src/main/java/com/koalasat/pokey/service/NotificationsService.kt
+++ b/app/src/main/java/com/koalasat/pokey/service/NotificationsService.kt
@@ -43,6 +43,8 @@ import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.SupervisorJob
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.launch
+import org.json.JSONException
+import org.json.JSONObject
 
 class NotificationsService : Service() {
     private var channelRelaysId = "RelaysConnections"
@@ -215,6 +217,7 @@ class NotificationsService : Service() {
                 TypedFilter(
                     types = COMMON_FEED_TYPES,
                     filter = SincePerRelayFilter(
+                        kinds = listOf(1, 3, 4, 6, 7, 1059, 9735),
                         tags = mapOf("p" to listOf(hexKey)),
                         since = RelayPool.getAll().associate { it.url to EOSETime(latestNotification) },
                     ),
@@ -305,13 +308,29 @@ class NotificationsService : Service() {
             if (lastCreatedRelayAt == null || lastCreatedRelayAt < event.createdAt) {
                 stopSubscription()
                 dao.deleteRelaysByKind(event.kind)
+
                 event.tags
                     .filter { it.size > 1 && (it[0] == "relay" || it[0] == "r") }
-                    .map {
-                        val entity = RelayEntity(id = 0, url = it[1], kind = event.kind, createdAt = event.createdAt)
-                        dao.insertRelay(entity)
-                        entity
+                    .forEach {
+                        var read = true
+                        if (event.kind == 10002) {
+                            try {
+                                val relaysConfig = JSONObject(event.content)
+                                val config = relaysConfig.getJSONObject(it[1])
+                                read = config.getBoolean("read")
+                            } catch (_: JSONException) {
+                                Log.d(
+                                    "Pokey",
+                                    "Public inbox relays not configured",
+                                )
+                            }
+                        }
+                        if (read) {
+                            val entity = RelayEntity(id = 0, url = it[1], kind = event.kind, createdAt = event.createdAt)
+                            dao.insertRelay(entity)
+                        }
                     }
+
                 startSubscription()
             }
         }
@@ -433,7 +452,6 @@ class NotificationsService : Service() {
     }
 
     private fun connectRelays() {
-        RelayPool.unloadRelays()
         val dao = AppDatabase.getDatabase(this@NotificationsService, Pokey.getInstance().getHexKey()).applicationDao()
         var relays = dao.getRelays()
         if (relays.isEmpty()) {
@@ -442,15 +460,17 @@ class NotificationsService : Service() {
 
         relays.forEach {
             Client.sendFilterOnlyIfDisconnected()
-            RelayPool.addRelay(
-                Relay(
-                    it.url,
-                    read = true,
-                    write = false,
-                    forceProxy = false,
-                    activeTypes = COMMON_FEED_TYPES,
-                ),
-            )
+            if (RelayPool.getRelays(it.url).isEmpty()) {
+                RelayPool.addRelay(
+                    Relay(
+                        it.url,
+                        read = true,
+                        write = false,
+                        forceProxy = false,
+                        activeTypes = COMMON_FEED_TYPES,
+                    ),
+                )
+            }
         }
     }
 }
diff --git a/app/src/main/res/values/strings.xml b/app/src/main/res/values/strings.xml
index cbb21ef..faefa80 100644
--- a/app/src/main/res/values/strings.xml
+++ b/app/src/main/res/values/strings.xml
@@ -7,6 +7,7 @@
     <string name="title_notifications">Notifications</string>
     <string name="pokey_is_running_in_background">Pokey is running in background</string>
     <string name="amber_not_found">Amber not installed</string>
+    <string name="inbox_relays_error">Your DM inbox relays are not configured</string>
     <string name="relays_connection">Relays connection</string>
     <string name="notifications">Notifications</string>
     <string name="new_mention">New mention</string>