checkbiomecommand.java

public class StructureBiomeCheckCommand {
    public static void createCommand(CommandDispatcher<CommandSourceStack> dispatcher, CommandBuildContext buildContext) {
        String commandString = "checkvalidstructurebiomes";

        LiteralCommandNode<CommandSourceStack> source = dispatcher.register(Commands.literal(commandString)
            .requires((permission) -> permission.hasPermission(2))
            .executes(cs -> {
                checkBiomesForStructures(cs);
                return 1;
            }));

        dispatcher.register(Commands.literal(commandString).redirect(source));
    }

    private static void checkBiomesForStructures(CommandContext<CommandSourceStack> cs) {
        ServerLevel level = cs.getSource().getLevel();
        Registry<Structure> structureRegistry = level.registryAccess().registryOrThrow(Registries.STRUCTURE);
        StringBuilder stringBuilder = new StringBuilder();
        Comparator<Map.Entry<ResourceKey<Structure>, Structure>> compareByNamespace = Comparator.comparing(e -> e.getKey().location().getNamespace());
        Comparator<Map.Entry<ResourceKey<Structure>, Structure>> compareByPath = Comparator.comparing(e -> e.getKey().location().getPath());

        LinkedHashMap<String, LinkedHashMap<String, List<ResourceLocation>>> structureInfoMap = new LinkedHashMap<>();

        // Collect data
        String currentMod = "";
        for (Map.Entry<ResourceKey<Structure>, Structure> entry : structureRegistry.entrySet().stream().sorted(compareByNamespace.thenComparing(compareByPath)).toList()) {
            String modid = entry.getKey().location().getNamespace();
            if (!currentMod.equals(modid)) {
                currentMod = modid;
                structureInfoMap.put(currentMod, new LinkedHashMap<>());
            }

            LinkedHashMap<String, List<ResourceLocation>> structureBiomesMap = structureInfoMap.get(currentMod);
            String structurePath = entry.getKey().location().getPath();
            if (!structureBiomesMap.containsKey(structurePath)) {
                structureBiomesMap.put(structurePath, new ArrayList<>());
            }

            List<ResourceLocation> biomesRL = structureBiomesMap.get(structurePath);
            for (Holder<Biome> biomeHolder : entry.getValue().biomes()) {
                biomesRL.add(biomeHolder.getKey().location());
            }
        }

        // Print data
        for (Map.Entry<String, LinkedHashMap<String, List<ResourceLocation>>> modidStructureDataMap : structureInfoMap.entrySet()) {

            stringBuilder
                    .append("\n\n\n- ")
                    .append(modidStructureDataMap.getKey())
                    .append(": ")
                    .append(modidStructureDataMap.getValue().keySet().size())
                    .append(" structures");

            LinkedHashMap<String, List<ResourceLocation>> structureBiomesMap = modidStructureDataMap.getValue();
            int currentStructureCount = 0;
            for (Map.Entry<String, List<ResourceLocation>> structureBiomesEntry : structureBiomesMap.entrySet()) {

                boolean onlyVanilla = true;
                for (ResourceLocation biome : structureBiomesEntry.getValue()) {
                    if (!biome.getNamespace().equals("minecraft")) {
                        onlyVanilla = false;
                        break;
                    }
                }

                if (onlyVanilla && !structureBiomesEntry.getValue().isEmpty()) {
                    currentStructureCount++;
                    stringBuilder
                            .append("\n\n  -")
                            .append(currentStructureCount)
                            .append(". ")
                            .append(structureBiomesEntry.getKey());

                    for (ResourceLocation biome : structureBiomesEntry.getValue()) {
                        stringBuilder
                                .append("\n      - ")
                                .append(biome.toString());
                    }
                }
            }
        }

        stringBuilder.append("\n\n\n");
        CommandStructuresMain.LOGGER.info(stringBuilder);
        cs.getSource().sendSuccess(() -> Component.literal("Done! Printed results to log."), false);
    }
}