bug: EOTSSegment does not forward the weapon used to damage it to the controller (breaks weapon-enchant effects like Treasure Hunter)

[MarekWojt]

Forge/Neoforge Version

neoforge-21.1.226

Deep Aether Version

1.21.1-1.1.5.1

Terrablender Version

4.1.0.8

Aether Version

1.5.10

Log Files

N/A

Bug Report

The problem
I'm currently working on support for Spectrum's mob heads for Aether and Deep Aether. While doing this, I realized the head of the Eye of the Storm does not drop at all.

The problem seems to be that when the damage is forwarded from the segments to the controller, it does not relay the weapon that has been used to damage the segment. The mechanic only works for weapons with the Treasure Hunter enchantment, so since the weapon doesn't get passed, the enchantment doesn't work.

If my reasoning is correct, this bug would also break other mechanics depending on enchants.

Proposed solution
My proposed solution would be to relay the weapon that has been used to the controller, so this is correctly applied.