2 files changed, 25 insertions, 3 deletions
diff --git a/src/main/kotlin/moe/nea/firmament/gui/config/AllConfigsGui.kt b/src/main/kotlin/moe/nea/firmament/gui/config/AllConfigsGui.kt
index 411d7bd..dd8108f 100644
--- a/src/main/kotlin/moe/nea/firmament/gui/config/AllConfigsGui.kt
+++ b/src/main/kotlin/moe/nea/firmament/gui/config/AllConfigsGui.kt
@@ -27,15 +27,17 @@ import io.github.cottonmc.cotton.gui.widget.WLabel
 import io.github.cottonmc.cotton.gui.widget.WListPanel
 import io.github.cottonmc.cotton.gui.widget.data.Insets
 import io.ktor.http.*
+import net.minecraft.client.gui.screen.Screen
 import net.minecraft.text.Text
 import moe.nea.firmament.features.FeatureManager
 import moe.nea.firmament.gui.WFixedPanel
 import moe.nea.firmament.repo.RepoManager
+import moe.nea.firmament.util.MC
 import moe.nea.firmament.util.ScreenUtil.setScreenLater
 
 object AllConfigsGui {
 
-    fun showAllGuis() {
+    fun makeScreen(parent: Screen? = null): CottonClientScreen {
         val lwgd = LightweightGuiDescription()
         var screen: CottonClientScreen? = null
         lwgd.setRootPanel(WListPanel(
@@ -56,7 +58,15 @@ object AllConfigsGui {
         }.also {
             it.setSize(10 * 18 + 14 + 16, 300)
         })
-        screen = CottonClientScreen(lwgd)
-        setScreenLater(screen)
+        screen = object :  CottonClientScreen(lwgd) {
+            override fun close() {
+                MC.screen = parent
+            }
+        }
+        return screen
+    }
+
+    fun showAllGuis() {
+        setScreenLater(makeScreen())
     }
 }
diff --git a/src/main/kotlin/moe/nea/firmament/modmenu/FirmamentModMenuPlugin.kt b/src/main/kotlin/moe/nea/firmament/modmenu/FirmamentModMenuPlugin.kt
new file mode 100644
index 0000000..44e491e
--- /dev/null
+++ b/src/main/kotlin/moe/nea/firmament/modmenu/FirmamentModMenuPlugin.kt
@@ -0,0 +1,12 @@
+package moe.nea.firmament.modmenu
+
+import com.terraformersmc.modmenu.api.ConfigScreenFactory
+import com.terraformersmc.modmenu.api.ModMenuApi
+import moe.nea.firmament.gui.config.AllConfigsGui
+
+class FirmamentModMenuPlugin : ModMenuApi {
+    override fun getModConfigScreenFactory(): ConfigScreenFactory<*> {
+        return ConfigScreenFactory { AllConfigsGui.makeScreen(it) }
+    }
+}
+