♻️ Add first stab at async freenet interface

[Sone.git] / src / main / kotlin / net / pterodactylus / sone / freenet / AsyncFreenetInterface.kt

diff --git a/src/main/kotlin/net/pterodactylus/sone/freenet/AsyncFreenetInterface.kt b/src/main/kotlin/net/pterodactylus/sone/freenet/AsyncFreenetInterface.kt
new file mode 100644 (file)
index 0000000..8219d7e
--- /dev/null
+++ b/src/main/kotlin/net/pterodactylus/sone/freenet/AsyncFreenetInterface.kt
@@ -0,0 +1,27 @@
+package net.pterodactylus.sone.freenet
+
+import freenet.client.*
+import freenet.keys.*
+import kotlinx.coroutines.*
+import net.pterodactylus.sone.core.*
+
+class AsyncFreenetInterface(private val freenetClient: FreenetClient) {
+
+       suspend fun fetchUri(freenetUri: FreenetURI): Fetched {
+               var currentUri = freenetUri
+               var result: FetchResult? = null
+               while (result == null) {
+                       try {
+                               result = withContext(Dispatchers.Default) { freenetClient.fetch(currentUri) }
+                       } catch (fetchException: FetchException) {
+                               if (fetchException.mode == FetchException.FetchExceptionMode.PERMANENT_REDIRECT) {
+                                       currentUri = fetchException.newURI
+                                       continue
+                               } else
+                                       throw fetchException
+                       }
+               }
+               return Fetched(currentUri, result)
+       }
+
+}