|
1 | 1 | package ch.njol.skript.expressions; |
2 | 2 |
|
3 | | -import java.util.ArrayList; |
4 | | -import java.util.List; |
5 | | - |
6 | | -import org.bukkit.entity.Player; |
7 | | -import org.bukkit.event.Event; |
8 | | -import org.bukkit.inventory.Inventory; |
9 | | -import org.bukkit.inventory.InventoryHolder; |
10 | | -import org.jetbrains.annotations.Nullable; |
11 | | - |
12 | 3 | import ch.njol.skript.Skript; |
13 | 4 | import ch.njol.skript.doc.Description; |
14 | 5 | import ch.njol.skript.doc.Examples; |
|
19 | 10 | import ch.njol.skript.lang.SkriptParser.ParseResult; |
20 | 11 | import ch.njol.skript.lang.util.SimpleExpression; |
21 | 12 | import ch.njol.util.Kleenean; |
| 13 | +import org.bukkit.entity.HumanEntity; |
| 14 | +import org.bukkit.entity.Player; |
| 15 | +import org.bukkit.event.Event; |
| 16 | +import org.bukkit.inventory.Inventory; |
| 17 | +import org.bukkit.inventory.InventoryHolder; |
| 18 | +import org.jetbrains.annotations.Nullable; |
| 19 | + |
| 20 | +import java.lang.reflect.Array; |
| 21 | +import java.util.ArrayList; |
| 22 | +import java.util.List; |
22 | 23 |
|
23 | 24 | @Name("Inventory Holder/Viewers/Rows/Slots") |
24 | 25 | @Description({"Gets the amount of rows/slots, viewers and holder of an inventory.", |
@@ -50,41 +51,42 @@ public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelaye |
50 | 51 | } |
51 | 52 |
|
52 | 53 | @Override |
53 | | - protected Object[] get(Event e) { |
54 | | - Inventory[] inventories = this.inventories.getArray(e); |
55 | | - List<Object> objects = new ArrayList<>(); |
| 54 | + protected Object[] get(Event event) { |
| 55 | + Inventory[] inventories = this.inventories.getArray(event); |
56 | 56 | switch (type) { |
57 | 57 | case HOLDER: |
| 58 | + List<InventoryHolder> holders = new ArrayList<>(); |
58 | 59 | for (Inventory inventory : inventories) { |
59 | 60 | InventoryHolder holder = inventory.getHolder(); |
60 | 61 | if (holder != null) |
61 | | - objects.add(holder); |
| 62 | + holders.add(holder); |
62 | 63 | } |
63 | | - break; |
| 64 | + return holders.toArray(new InventoryHolder[0]); |
64 | 65 | case ROWS: |
| 66 | + List<Number> rows = new ArrayList<>(); |
65 | 67 | for (Inventory inventory : inventories) { |
66 | 68 | int size = inventory.getSize(); |
67 | 69 | if (size < 9) // Hoppers have a size of 5, we don't want to return 0 |
68 | | - objects.add(1); |
| 70 | + rows.add(1); |
69 | 71 | else |
70 | | - objects.add(size / 9); |
| 72 | + rows.add(size / 9); |
71 | 73 | } |
72 | | - break; |
| 74 | + return rows.toArray(new Number[0]); |
73 | 75 | case SLOTS: |
| 76 | + List<Number> sizes = new ArrayList<>(); |
74 | 77 | for (Inventory inventory : inventories) { |
75 | | - objects.add(inventory.getSize()); |
| 78 | + sizes.add(inventory.getSize()); |
76 | 79 | } |
77 | | - break; |
| 80 | + return sizes.toArray(new Number[0]); |
78 | 81 | case VIEWERS: |
| 82 | + List<HumanEntity> viewers = new ArrayList<>(); |
79 | 83 | for (Inventory inventory : inventories) { |
80 | | - objects.addAll(inventory.getViewers()); |
| 84 | + viewers.addAll(inventory.getViewers()); |
81 | 85 | } |
82 | | - break; |
| 86 | + return viewers.stream().filter(viewer -> viewer instanceof Player).toArray(Player[]::new); |
83 | 87 | default: |
84 | | - return new Object[0]; |
| 88 | + return (Object[]) Array.newInstance(getReturnType(), 0); |
85 | 89 | } |
86 | | - return objects.toArray(new Object[0]); |
87 | | - |
88 | 90 | } |
89 | 91 |
|
90 | 92 | @Override |
|
0 commit comments