package net.minecraft.commands.execution;

import com.mojang.brigadier.Command;
import com.mojang.brigadier.context.CommandContext;
import com.mojang.brigadier.context.ContextChain;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import net.minecraft.commands.ExecutionCommandSource;
import org.jspecify.annotations.Nullable;

public interface CustomCommandExecutor<T> {
	void run(T sender, ContextChain<T> currentStep, ChainModifiers modifiers, ExecutionControl<T> output);

	interface CommandAdapter<T> extends CustomCommandExecutor<T>, Command<T> {
		@Override
		default int run(final CommandContext<T> context) throws CommandSyntaxException {
			throw new UnsupportedOperationException("This function should not run");
		}
	}

	abstract class WithErrorHandling<T extends ExecutionCommandSource<T>> implements CustomCommandExecutor<T> {
		public final void run(final T sender, final ContextChain<T> currentStep, final ChainModifiers modifiers, final ExecutionControl<T> output) {
			try {
				this.runGuarded(sender, currentStep, modifiers, output);
			} catch (CommandSyntaxException e) {
				this.onError(e, sender, modifiers, output.tracer());
				sender.callback().onFailure();
			}
		}

		protected void onError(final CommandSyntaxException e, final T sender, final ChainModifiers modifiers, final @Nullable TraceCallbacks tracer) {
			sender.handleError(e, modifiers.isForked(), tracer);
		}

		protected abstract void runGuarded(T sender, ContextChain<T> currentStep, ChainModifiers modifiers, ExecutionControl<T> output) throws CommandSyntaxException;
	}
}