Restructure commands

build.gradle.kts

@@ -25,8 +25,11 @@ loom {
     }
     runs {
         named("client") {
+            /*
             vmArg("-XX:+AllowEnhancedClassRedefinition")
             vmArg("-XX:HotswapAgent=fatjar")
+
+             */
         }
     }
 }

src/main/kotlin/moe/nea/notenoughupdates/commands/dsl.kt

@@ -5,13 +5,19 @@ import com.mojang.brigadier.builder.ArgumentBuilder
 import com.mojang.brigadier.builder.LiteralArgumentBuilder
 import com.mojang.brigadier.builder.RequiredArgumentBuilder
 import com.mojang.brigadier.context.CommandContext
-import moe.nea.notenoughupdates.util.iterate
-import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource
 import java.lang.reflect.ParameterizedType
+import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource
+import moe.nea.notenoughupdates.util.iterate
 
 
 typealias DefaultSource = FabricClientCommandSource
 
+
+inline val <T : CommandContext<*>> T.context get() = this
+operator fun <T : Any, C : CommandContext<*>> C.get(arg: TypeSafeArg<T>): T {
+    return arg.get(this)
+}
+
 fun literal(
     name: String,
     block: LiteralArgumentBuilder<DefaultSource>.() -> Unit

src/main/kotlin/moe/nea/notenoughupdates/commands/rome.kt

@@ -12,6 +12,7 @@ import moe.nea.notenoughupdates.util.ScreenUtil.setScreenLater
 
 
 fun neuCommand() = literal("neu") {
+    thenLiteral("repo") {
         thenLiteral("reload") {
             thenLiteral("fetch") {
                 thenExecute {
@@ -24,7 +25,6 @@ fun neuCommand() = literal("neu") {
                 RepoManager.reload()
             }
         }
-    thenLiteral("repo") {
         thenExecute {
             setScreenLater(CottonClientScreen(repoGui()))
         }