Close group poller issue

libsession/src/main/java/org/session/libsession/messaging/sending_receiving/pollers/ClosedGroupPoller.kt

@@ -59,12 +59,12 @@ class ClosedGroupPoller(
         job = scope.launch(executor) {
             while (isActive) {
                 try {
-                    val swarmNodes = SnodeAPI.getSwarm(closedGroupSessionId.hexString).await().toMutableSet()
+                    val swarmNodes = SnodeAPI.fetchSwarmNodes(closedGroupSessionId.hexString).toMutableSet()
                     var currentSnode: Snode? = null
 
                     while (isActive) {
                         if (currentSnode == null) {
-                            check(swarmNodes.isNotEmpty()) { "No swarm nodes found" }
+                            check(swarmNodes.isNotEmpty()) { "No more swarm nodes found" }
                             Log.d(TAG, "No current snode, getting a new one. Remaining in pool = ${swarmNodes.size - 1}")
                             currentSnode = swarmNodes.random()
                             swarmNodes.remove(currentSnode)

libsession/src/main/java/org/session/libsession/snode/SnodeAPI.kt

@@ -312,6 +312,23 @@ object SnodeAPI {
                 database.setSwarm(publicKey, it)
             }
 
+    /**
+     * Fetch swarm nodes for the specific public key.
+     *
+     * Note: this differs from [getSwarm] in that it doesn't store the swarm nodes in the database.
+     * This always fetches from network.
+     */
+    suspend fun fetchSwarmNodes(publicKey: String): List<Snode> {
+        val randomNode = getRandomSnode().await()
+        val response = invoke(
+            method = Snode.Method.GetSwarm,
+            snode = randomNode, parameters = buildMap { this["pubKey"] = publicKey },
+            publicKey = publicKey
+        ).await()
+
+        return parseSnodes(response)
+    }
+
 
     /**
      * Build parameters required to call authenticated storage API.
@@ -706,8 +723,24 @@ object SnodeAPI {
             parameters = mapOf("requests" to requests),
             responseClass = BatchResponse::class.java,
             publicKey = publicKey
+        ).also { resp ->
+            // If there's a unsuccessful response, go through specific logic to handle
+            // potential snode errors.
+            val firstError = resp.results.firstOrNull { !it.isSuccessful }
+            if (firstError != null) {
+                handleSnodeError(
+                    statusCode = firstError.code,
+                    json = if (firstError.body.isObject) {
+                        JsonUtil.fromJson(firstError.body, Map::class.java)
+                    } else {
+                        null
+                    },
+                    snode = snode,
+                    publicKey = publicKey
                 )
             }
+        }
+    }
 
     fun getExpiries(
         messageHashes: List<String>,