Upstream/Citizens2
1
0
Fork 0
mirror of https://github.com/CitizensDev/Citizens2.git synced 2025-03-14 15:49:24 +01:00
Code Issues Projects Releases Wiki Activity

More item tweaks

Browse Source
This commit is contained in:
fullwall 2023-06-19 02:27:13 +08:00
parent ad442e4bd3
commit c001b65988
2 changed files with 25 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main/src/main/java/net/citizensnpcs/Citizens.java
View File

@ -348,15 +348,17 @@ public class Citizens extends JavaPlugin implements CitizensPlugin {
}
public void onDependentPluginDisable() {
storeNPCs(false);
saveOnDisable = false;
if (enabled) {
storeNPCs(false);
saveOnDisable = false;
}
}
@Override
public void onDisable() {
if (!enabled) {
if (!enabled)
return;
}
Bukkit.getPluginManager().callEvent(new CitizensDisableEvent());
Editor.leaveAll();
despawnNPCs(saveOnDisable);
@ -388,7 +390,8 @@ public class Citizens extends JavaPlugin implements CitizensPlugin {
e.printStackTrace();
}
Messaging.severeTr(Messages.CITIZENS_INCOMPATIBLE, getDescription().getVersion(), mcVersion);
enabled = true;
NMS.shutdown();
CitizensAPI.shutdown();
Bukkit.getPluginManager().disablePlugin(this);
return;
}
@ -626,6 +629,7 @@ public class Citizens extends JavaPlugin implements CitizensPlugin {
}
}
saves.loadInto(npcRegistry);
shops.load();

32
main/src/main/java/net/citizensnpcs/trait/shop/ItemAction.java
View File

@ -2,7 +2,6 @@ package net.citizensnpcs.trait.shop;
import java.util.Arrays;
import java.util.List;
import java.util.function.BiFunction;
import java.util.function.Consumer;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
@ -52,7 +51,7 @@ public class ItemAction extends NPCShopAction {
this.items = items;
}
private boolean containsItems(Inventory source, int repeats, BiFunction<ItemStack, Integer, ItemStack> filter) {
private boolean containsItems(Inventory source, int repeats, boolean modify) {
List<Integer> req = items.stream().map(i -> i.getAmount() * repeats).collect(Collectors.toList());
ItemStack[] contents = source.getContents();
for (int i = 0; i < contents.length; i++) {
@ -61,6 +60,7 @@ public class ItemAction extends NPCShopAction {
continue;
if (tooDamaged(toMatch))
continue;
toMatch = toMatch.clone();
for (int j = 0; j < items.size(); j++) {
if (toMatch == null)
break;
@ -70,14 +70,21 @@ public class ItemAction extends NPCShopAction {
int remaining = req.get(j);
int taken = toMatch.getAmount() > remaining ? remaining : toMatch.getAmount();
ItemStack res = filter.apply(toMatch, taken);
if (res == null) {
source.clear(i);
if (toMatch.getAmount() == taken) {
toMatch = null;
} else {
source.setItem(i, res);
toMatch.setAmount(toMatch.getAmount() - taken);
}
if (modify) {
if (toMatch == null) {
source.clear(i);
} else {
source.setItem(i, toMatch.clone());
}
}
req.set(j, remaining - taken);
toMatch = res;
}
}
return req.stream().collect(Collectors.summingInt(n -> n)) <= 0;
@ -196,16 +203,9 @@ public class ItemAction extends NPCShopAction {
return Transaction.fail();
Inventory source = ((InventoryHolder) entity).getInventory();
return Transaction.create(() -> {
return containsItems(source, repeats, (stack, taken) -> stack);
return containsItems(source, repeats, false);
}, () -> {
containsItems(source, repeats, (stack, taken) -> {
if (stack.getAmount() == taken) {
return null;
} else {
stack.setAmount(stack.getAmount() - taken);
return stack;
}
});
containsItems(source, repeats, true);
}, () -> {
source.addItem(items.stream().map(ItemStack::clone).toArray(ItemStack[]::new));
});