X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FPageToadletRegistry.kt;fp=src%2Fmain%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FPageToadletRegistry.kt;h=70105bca34f8518ff91d60fcbebd8d4a6e702da9;hb=33fe718540dfcc498c06c44c957ccb76bee57eae;hp=97e6cfe9aab7a885da330aaea6fbf395107266e9;hpb=7ab07c330617f5396807cee518a119451a694683;p=Sone.git

diff --git a/src/main/kotlin/net/pterodactylus/sone/web/PageToadletRegistry.kt b/src/main/kotlin/net/pterodactylus/sone/web/PageToadletRegistry.kt
index 97e6cfe..70105bc 100644
--- a/src/main/kotlin/net/pterodactylus/sone/web/PageToadletRegistry.kt
+++ b/src/main/kotlin/net/pterodactylus/sone/web/PageToadletRegistry.kt
@@ -18,6 +18,7 @@ class PageToadletRegistry @Inject constructor(
 ) {
 
 	private val pages = mutableListOf<Page<FreenetRequest>>()
+	private val debugPages = mutableListOf<Page<FreenetRequest>>()
 	private val registeredToadlets = mutableListOf<PageToadlet>()
 	private val registered = AtomicBoolean(false)
 
@@ -26,13 +27,18 @@ class PageToadletRegistry @Inject constructor(
 		pages += page
 	}
 
+	fun addDebugPage(page: Page<FreenetRequest>) {
+		if (registered.get()) throw IllegalStateException()
+		debugPages += page
+	}
+
 	fun registerToadlets() {
 		registered.set(true)
 		pageMaker.addNavigationCategory("/Sone/index.html", soneMenuName, "$soneMenu.Tooltip", sonePlugin)
 		addPages()
 	}
 
-	private fun addPages() =
+	private fun addPages(pages: List<Page<FreenetRequest>> = this.pages) =
 			pages
 					.map { pageToadletFactory.createPageToadlet(it) }
 					.onEach(registeredToadlets::plusAssign)
@@ -55,4 +61,7 @@ class PageToadletRegistry @Inject constructor(
 		registeredToadlets.forEach(toadletContainer::unregister)
 	}
 
+	fun activateDebugMode() =
+			addPages(debugPages)
+
 }