Unicopia/src/main/java/com/minelittlepony/unicopia/command/Commands.java

package com.minelittlepony.unicopia.command;

import net.fabricmc.fabric.api.command.v2.CommandRegistrationCallback;

import com.minelittlepony.unicopia.Debug;
import com.minelittlepony.unicopia.Unicopia;

import net.fabricmc.fabric.api.command.v2.ArgumentTypeRegistry;
import net.fabricmc.loader.api.FabricLoader;
import net.minecraft.server.MinecraftServer;

public class Commands {
    @SuppressWarnings({ "deprecation", "unchecked", "rawtypes" })
    public static void bootstrap() {
        ArgumentTypeRegistry.registerArgumentType(
                Unicopia.id("enumeration"),
                EnumArgumentType.class,
                new EnumArgumentType.Serializer()
        );
        CommandRegistrationCallback.EVENT.register((dispatcher, access, environment) -> {
            RacelistCommand.register(dispatcher);
            EmoteCommand.register(dispatcher);
            if (Unicopia.getConfig().enableCheats.get() || environment.dedicated) {
                SpeciesCommand.register(dispatcher, environment);
            }
            if (Unicopia.getConfig().enableCheats.get()) {
                GravityCommand.register(dispatcher);
                DisguiseCommand.register(dispatcher, access);
                if (Debug.DEBUG_COMMANDS) {
                    TraitCommand.register(dispatcher);
                    ManaCommand.register(dispatcher);
                }
            }
        });
        Object game = FabricLoader.getInstance().getGameInstance();
        if (game instanceof MinecraftServer) {
            ((MinecraftServer)game).setFlightEnabled(true);
        }
    }
}