package net.minecraft.server.commands;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import com.mojang.brigadier.exceptions.SimpleCommandExceptionType;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.network.chat.Component;
import net.minecraft.server.MinecraftServer;

public class SaveAllCommand {
	private static final SimpleCommandExceptionType ERROR_FAILED = new SimpleCommandExceptionType(Component.translatable("commands.save.failed"));

	public static void register(CommandDispatcher<CommandSourceStack> dispatcher) {
		dispatcher.register(
			Commands.literal("save-all")
				.requires(commandSourceStack -> commandSourceStack.hasPermission(4))
				.executes(commandContext -> saveAll(commandContext.getSource(), false))
				.then(Commands.literal("flush").executes(commandContext -> saveAll(commandContext.getSource(), true)))
		);
	}

	private static int saveAll(CommandSourceStack source, boolean flush) throws CommandSyntaxException {
		source.sendSuccess(() -> Component.translatable("commands.save.saving"), false);
		MinecraftServer minecraftServer = source.getServer();
		boolean bl = minecraftServer.saveEverything(true, flush, true);
		if (!bl) {
			throw ERROR_FAILED.create();
		} else {
			source.sendSuccess(() -> Component.translatable("commands.save.success"), true);
			return 1;
		}
	}
}