✨ Allow activateDebugMode() to be called multiple times

[Sone.git] / src / main / kotlin / net / pterodactylus / sone / web / PageToadletRegistry.kt

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/PageToadletRegistry.kt b/src/main/kotlin/net/pterodactylus/sone/web/PageToadletRegistry.kt
index 70105bc..651bffd 100644 (file)
--- a/src/main/kotlin/net/pterodactylus/sone/web/PageToadletRegistry.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/PageToadletRegistry.kt
@@ -21,6 +21,7 @@ class PageToadletRegistry @Inject constructor(
        private val debugPages = mutableListOf<Page<FreenetRequest>>()
        private val registeredToadlets = mutableListOf<PageToadlet>()
        private val registered = AtomicBoolean(false)
+       private val debugActivated = AtomicBoolean(false)
 
        fun addPage(page: Page<FreenetRequest>) {
                if (registered.get()) throw IllegalStateException()
@@ -61,7 +62,11 @@ class PageToadletRegistry @Inject constructor(
                registeredToadlets.forEach(toadletContainer::unregister)
        }
 
-       fun activateDebugMode() =
+       fun activateDebugMode() {
+               if (!debugActivated.get()) {
                        addPages(debugPages)
+                       debugActivated.set(true)
+               }
+       }
 
 }