src/main/java/ftbsc/bscv/modules/hud/Highlighter.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

package ftbsc.bscv.modules.hud;

import com.mojang.blaze3d.matrix.MatrixStack;
import com.mojang.blaze3d.systems.RenderSystem;

import ftbsc.bscv.modules.AbstractModule;
import ftbsc.bscv.tools.Setting;
import net.minecraft.client.gui.screen.inventory.ContainerScreen;
import net.minecraft.inventory.container.Slot;
import net.minecraft.item.EnchantedBookItem;
import net.minecraft.item.ItemStack;
import net.minecraft.item.Items;
import net.minecraft.nbt.CompoundNBT;
import net.minecraft.nbt.ListNBT;
import net.minecraftforge.client.event.GuiContainerEvent;
import net.minecraftforge.common.ForgeConfigSpec;
import net.minecraftforge.eventbus.api.SubscribeEvent;
import net.minecraftforge.fml.client.gui.GuiUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.regex.Pattern;

import ftbsc.bscv.api.ILoadable;
import com.google.auto.service.AutoService;

@AutoService(ILoadable.class)
public class Highlighter extends AbstractModule {

   public final ForgeConfigSpec.ConfigValue<String> query;
   private Pattern pattern;

   public Highlighter() {
      super();

      this.pattern = Pattern.compile("");
      this.query = Setting.Str.builder()
         .fallback("")
         .name("query")
         .comment("search query")
         .callback(to -> this.pattern = Pattern.compile(to))
         .build(this);
   }

   @Override
   protected void onEnabled() {
      this.pattern = Pattern.compile(this.query.get());
   }

   private List<String> enchantments(ItemStack stack) {
      final ListNBT tags;
      if (Items.ENCHANTED_BOOK.equals(stack.getItem())) {
         tags = EnchantedBookItem.getEnchantments(stack); // special case to also search book enchants
      } else {
         tags = stack.getEnchantmentTags();
      }

      List<String> out = new ArrayList<>();
      for (int i = 0; i < tags.size(); i++) {
         CompoundNBT tag = tags.getCompound(i);
         out.add(String.format("%s %s", tag.getString("id"), tag.getInt("lvl")));
      }

      return out;
   }

   private boolean matches(ItemStack stack) {
      if (stack.isEmpty()) return false;

      String displayName = stack.getDisplayName().getString();
      if (this.pattern.matcher(displayName).find()) return true;

      if (Items.ENCHANTED_BOOK.equals(stack.getItem()) || stack.isEnchanted()) {
         for (String ench : this.enchantments(stack)) {
            if (this.pattern.matcher(ench).find()) return true;
         }
      }

      return false;
   }

   @SubscribeEvent
   public void onGuiContainerDraw(GuiContainerEvent.DrawBackground event) {
      MatrixStack matrix = event.getMatrixStack();
      ContainerScreen<?> screen = event.getGuiContainer();

      RenderSystem.enableDepthTest();
      matrix.pushPose();
      matrix.translate(screen.getGuiLeft(), screen.getGuiTop(), 0);

      for (Slot slot : screen.getMenu().slots) {
         ItemStack stack = slot.getItem();
         if (this.matches(stack)) {
            GuiUtils.drawGradientRect(
               matrix.last().pose(), 0,
               slot.x, slot.y, slot.x + 16, slot.y + 16,
               // TODO de-hardcode these colors by creating our util!
               -925194976,  // R218 G165 B 32 A200
               -927090837   // R189 G183 B107 A200
            );
         }
      }

      matrix.popPose();
   }
}