22declare(strict_types=1);
24namespace pocketmine\item\enchantment;
30use pocketmine\utils\SingletonTrait;
31use
function array_filter;
32use
function array_values;
34use
function spl_object_id;
43 private array $enchantments = [];
46 private array $primaryItemTags = [];
49 private array $secondaryItemTags = [];
51 private function __construct(){
52 $this->
register(Enchantments::PROTECTION(), [Tags::ARMOR], []);
53 $this->
register(Enchantments::FIRE_PROTECTION(), [Tags::ARMOR], []);
54 $this->
register(Enchantments::FEATHER_FALLING(), [Tags::BOOTS], []);
55 $this->
register(Enchantments::BLAST_PROTECTION(), [Tags::ARMOR], []);
56 $this->
register(Enchantments::PROJECTILE_PROTECTION(), [Tags::ARMOR], []);
57 $this->
register(Enchantments::THORNS(), [Tags::CHESTPLATE], [Tags::HELMET, Tags::LEGGINGS, Tags::BOOTS]);
58 $this->
register(Enchantments::RESPIRATION(), [Tags::HELMET], []);
59 $this->
register(Enchantments::SHARPNESS(), [Tags::SWORD, Tags::AXE], []);
60 $this->
register(Enchantments::KNOCKBACK(), [Tags::SWORD], []);
61 $this->
register(Enchantments::FIRE_ASPECT(), [Tags::SWORD], []);
62 $this->
register(Enchantments::EFFICIENCY(), [Tags::BLOCK_TOOLS], [Tags::SHEARS]);
63 $this->
register(Enchantments::FORTUNE(), [Tags::BLOCK_TOOLS], []);
64 $this->
register(Enchantments::SILK_TOUCH(), [Tags::BLOCK_TOOLS], [Tags::SHEARS]);
66 Enchantments::UNBREAKING(),
67 [Tags::ARMOR, Tags::WEAPONS, Tags::FISHING_ROD],
68 [Tags::SHEARS, Tags::FLINT_AND_STEEL, Tags::SHIELD, Tags::CARROT_ON_STICK, Tags::ELYTRA, Tags::BRUSH]
70 $this->
register(Enchantments::POWER(), [Tags::BOW], []);
71 $this->
register(Enchantments::PUNCH(), [Tags::BOW], []);
72 $this->
register(Enchantments::FLAME(), [Tags::BOW], []);
73 $this->
register(Enchantments::INFINITY(), [Tags::BOW], []);
75 Enchantments::MENDING(),
77 [Tags::ARMOR, Tags::WEAPONS, Tags::FISHING_ROD,
78 Tags::SHEARS, Tags::FLINT_AND_STEEL, Tags::SHIELD, Tags::CARROT_ON_STICK, Tags::ELYTRA, Tags::BRUSH]
80 $this->
register(Enchantments::VANISHING(), [], [Tags::ALL]);
81 $this->
register(Enchantments::SWIFT_SNEAK(), [], [Tags::LEGGINGS]);
88 public function register(
Enchantment $enchantment, array $primaryItemTags, array $secondaryItemTags) :
void{
89 $this->enchantments[spl_object_id($enchantment)] = $enchantment;
94 public function unregister(
Enchantment $enchantment) : void{
95 unset($this->enchantments[spl_object_id($enchantment)]);
96 unset($this->primaryItemTags[spl_object_id($enchantment)]);
97 unset($this->secondaryItemTags[spl_object_id($enchantment)]);
100 public function unregisterAll() : void{
101 $this->enchantments = [];
102 $this->primaryItemTags = [];
103 $this->secondaryItemTags = [];
106 public function isRegistered(Enchantment $enchantment) : bool{
107 return isset($this->enchantments[spl_object_id($enchantment)]);
120 return $this->primaryItemTags[spl_object_id($enchantment)] ?? [];
127 if(!$this->isRegistered($enchantment)){
128 throw new \LogicException(
"Cannot set primary item tags for non-registered enchantment");
130 $this->primaryItemTags[spl_object_id($enchantment)] = array_values($tags);
143 return $this->secondaryItemTags[spl_object_id($enchantment)] ?? [];
150 if(!$this->isRegistered($enchantment)){
151 throw new \LogicException(
"Cannot set secondary item tags for non-registered enchantment");
153 $this->secondaryItemTags[spl_object_id($enchantment)] = array_values($tags);
162 $itemTags = $item->getEnchantmentTags();
163 if(count($itemTags) === 0 || $item->hasEnchantments()){
169 fn(
Enchantment $e) => TagRegistry::getInstance()->isTagArrayIntersection($this->getPrimaryItemTags($e), $itemTags)
182 if(count($item->getEnchantmentTags()) === 0){
188 fn(
Enchantment $enchantment) => $this->isAvailableForItem($enchantment, $item)
198 $itemTags = $item->getEnchantmentTags();
199 $tagRegistry = TagRegistry::getInstance();
201 return $tagRegistry->isTagArrayIntersection($this->getPrimaryItemTags($enchantment), $itemTags) ||
202 $tagRegistry->isTagArrayIntersection($this->getSecondaryItemTags($enchantment), $itemTags);
209 return $this->enchantments;
setSecondaryItemTags(Enchantment $enchantment, array $tags)
isAvailableForItem(Enchantment $enchantment, Item $item)
getPrimaryEnchantmentsForItem(Item $item)
getPrimaryItemTags(Enchantment $enchantment)
getSecondaryItemTags(Enchantment $enchantment)
getAllEnchantmentsForItem(Item $item)
setPrimaryItemTags(Enchantment $enchantment, array $tags)