102 lines
3.8 KiB
Java
102 lines
3.8 KiB
Java
package ru.betterend.mixin.client;
|
|
|
|
import java.util.List;
|
|
|
|
import org.spongepowered.asm.mixin.Mixin;
|
|
import org.spongepowered.asm.mixin.Shadow;
|
|
import org.spongepowered.asm.mixin.injection.At;
|
|
import org.spongepowered.asm.mixin.injection.Inject;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
|
|
|
|
import com.google.common.collect.Lists;
|
|
|
|
import net.minecraft.client.gui.screen.ingame.AnvilScreen;
|
|
import net.minecraft.client.gui.screen.ingame.ForgingScreen;
|
|
import net.minecraft.client.gui.widget.AbstractButtonWidget;
|
|
import net.minecraft.client.gui.widget.ButtonWidget;
|
|
import net.minecraft.client.gui.widget.TextFieldWidget;
|
|
import net.minecraft.client.util.math.MatrixStack;
|
|
import net.minecraft.entity.player.PlayerInventory;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraft.screen.AnvilScreenHandler;
|
|
import net.minecraft.screen.ScreenHandler;
|
|
import net.minecraft.text.LiteralText;
|
|
import net.minecraft.text.Text;
|
|
import net.minecraft.util.Identifier;
|
|
|
|
import ru.betterend.interfaces.AnvilScreenHandlerExtended;
|
|
import ru.betterend.recipe.builders.AnvilSmithingRecipe;
|
|
|
|
@Mixin(AnvilScreen.class)
|
|
public class AnvilScreenMixin extends ForgingScreen<AnvilScreenHandler> {
|
|
|
|
@Shadow
|
|
private TextFieldWidget nameField;
|
|
|
|
private List<AbstractButtonWidget> be_buttons = Lists.newArrayList();
|
|
|
|
public AnvilScreenMixin(AnvilScreenHandler handler, PlayerInventory playerInventory, Text title,
|
|
Identifier texture) {
|
|
super(handler, playerInventory, title, texture);
|
|
}
|
|
|
|
@Inject(method = "setup", at = @At("TAIL"))
|
|
protected void setup(CallbackInfo info) {
|
|
this.be_buttons.clear();
|
|
int x = (this.width - this.backgroundWidth) / 2;
|
|
int y = (this.height - this.backgroundHeight) / 2;
|
|
this.be_buttons.add(new ButtonWidget(x + 8, y + 45, 15, 20, new LiteralText("<"), (b) -> be_previousRecipe()));
|
|
this.be_buttons.add(new ButtonWidget(x + 154, y + 45, 15, 20, new LiteralText(">"), (b) -> be_nextRecipe()));
|
|
}
|
|
|
|
@Inject(method = "renderForeground", at = @At("TAIL"))
|
|
protected void renderForeground(MatrixStack matrices, int mouseX, int mouseY, float delta, CallbackInfo info) {
|
|
AnvilScreenHandlerExtended handler = AnvilScreenHandlerExtended.class.cast(this.handler);
|
|
if (handler.be_getRecipes().size() > 1) {
|
|
this.be_buttons.forEach(button -> button.render(matrices, mouseX, mouseY, delta));
|
|
}
|
|
}
|
|
|
|
@Inject(method = "onSlotUpdate", at = @At("HEAD"), cancellable = true)
|
|
public void onSlotUpdate(ScreenHandler handler, int slotId, ItemStack stack, CallbackInfo info) {
|
|
AnvilScreenHandlerExtended anvilHandler = AnvilScreenHandlerExtended.class.cast(this.handler);
|
|
if (anvilHandler.be_getCurrentRecipe() != null) {
|
|
this.nameField.setText("");
|
|
this.nameField.setEditable(false);
|
|
this.setFocused(null);
|
|
info.cancel();
|
|
}
|
|
}
|
|
|
|
private void be_nextRecipe() {
|
|
AnvilScreenHandlerExtended handler = AnvilScreenHandlerExtended.class.cast(this.handler);
|
|
List<AnvilSmithingRecipe> recipes = handler.be_getRecipes();
|
|
AnvilSmithingRecipe current = handler.be_getCurrentRecipe();
|
|
int i = recipes.indexOf(current) + 1;
|
|
if (i == recipes.size()) {
|
|
i = 0;
|
|
}
|
|
handler.be_updateCurrentRecipe(recipes.get(i));
|
|
}
|
|
|
|
private void be_previousRecipe() {
|
|
AnvilScreenHandlerExtended handler = AnvilScreenHandlerExtended.class.cast(this.handler);
|
|
List<AnvilSmithingRecipe> recipes = handler.be_getRecipes();
|
|
AnvilSmithingRecipe current = handler.be_getCurrentRecipe();
|
|
int i = recipes.indexOf(current) - 1;
|
|
if (i == 0) {
|
|
i = recipes.size() - 1;
|
|
}
|
|
handler.be_updateCurrentRecipe(recipes.get(i));
|
|
}
|
|
|
|
@Override
|
|
public boolean mouseClicked(double mouseX, double mouseY, int button) {
|
|
for (AbstractButtonWidget elem : be_buttons) {
|
|
if (elem.mouseClicked(mouseX, mouseY, button)) {
|
|
return true;
|
|
}
|
|
}
|
|
return super.mouseClicked(mouseX, mouseY, button);
|
|
}
|
|
}
|