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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/*
* This file is licensed under the MIT License, part of Roughly Enough Items.
* Copyright (c) 2018, 2019, 2020, 2021, 2022, 2023 shedaniel
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package me.shedaniel.rei.plugin.test;
import dev.architectury.event.events.common.CommandRegistrationEvent;
import me.shedaniel.rei.api.client.entry.filtering.FilteringRuleTypeRegistry;
import me.shedaniel.rei.api.client.entry.filtering.base.BasicFilteringRule;
import me.shedaniel.rei.api.client.plugins.REIClientPlugin;
import me.shedaniel.rei.api.client.registry.entry.CollapsibleEntryRegistry;
import me.shedaniel.rei.api.client.registry.entry.EntryRegistry;
import me.shedaniel.rei.api.common.entry.EntryIngredient;
import me.shedaniel.rei.api.common.entry.EntryStack;
import me.shedaniel.rei.api.common.entry.type.VanillaEntryTypes;
import me.shedaniel.rei.api.common.plugins.PluginManager;
import me.shedaniel.rei.api.common.registry.ReloadStage;
import me.shedaniel.rei.api.common.util.EntryStacks;
import me.shedaniel.rei.impl.common.InternalLogger;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.commands.Commands;
import net.minecraft.commands.arguments.item.ItemArgument;
import net.minecraft.commands.arguments.item.ItemInput;
import net.minecraft.core.component.DataComponents;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.network.chat.Component;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.component.CustomData;
import org.jetbrains.annotations.TestOnly;
import java.util.Random;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
@TestOnly
@Environment(EnvType.CLIENT)
public class REITestPlugin implements REIClientPlugin {
private final Random random = new Random();
private BasicFilteringRule.MarkDirty markDirty;
public REITestPlugin() {
CommandRegistrationEvent.EVENT.register((dispatcher, registry, selection) -> {
dispatcher.register(Commands.literal("rei_test_reload_filtering")
.then(Commands.argument("item", ItemArgument.item(registry))
.executes(context -> {
BasicFilteringRule<?> basic = FilteringRuleTypeRegistry.getInstance().basic();
basic.hide(EntryStacks.of(context.getArgument("item", ItemInput.class).createItemStack(1, false)));
return 0;
}))
.executes(context -> {
if (this.markDirty != null) this.markDirty.markDirty();
return 0;
}));
});
}
@Override
public void preStage(PluginManager<REIClientPlugin> manager, ReloadStage stage) {
InternalLogger.getInstance().error("REI Test Plugin is enabled! If you see this unintentionally, please report this!");
}
@Override
public void registerEntries(EntryRegistry registry) {
if (1 + 1 == 2) return;
int times = 10;
for (Item item : BuiltInRegistries.ITEM) {
EntryStack<ItemStack> base = EntryStacks.of(item);
registry.addEntriesAfter(base, IntStream.range(0, times).mapToObj(value -> transformStack(EntryStacks.of(item))).collect(Collectors.toList()));
}
}
@Override
public void registerCollapsibleEntries(CollapsibleEntryRegistry registry) {
int i = 0;
for (Item item : BuiltInRegistries.ITEM) {
if (i++ % 10 != 0)
continue;
registry.group(BuiltInRegistries.ITEM.getKey(item), Component.literal(BuiltInRegistries.ITEM.getKey(item).toString()),
stack -> stack.getType() == VanillaEntryTypes.ITEM && stack.<ItemStack>castValue().is(item));
}
}
@Override
public void registerBasicEntryFiltering(BasicFilteringRule<?> rule) {
markDirty = rule.hide(() -> {
EntryIngredient.Builder builder = EntryIngredient.builder();
for (Item item : BuiltInRegistries.ITEM) {
if (random.nextInt() % 10 == 0) {
builder.add(EntryStacks.of(item));
}
}
return builder.build();
});
}
public EntryStack<ItemStack> transformStack(EntryStack<ItemStack> stack) {
CustomData data = stack.getValue().getOrDefault(DataComponents.CUSTOM_DATA, CustomData.EMPTY).update(tag -> {
tag.putInt("Whatever", random.nextInt(Integer.MAX_VALUE));
});
stack.getValue().set(DataComponents.CUSTOM_DATA, data);
return stack;
}
}
|
