Making Tool/Weapon that acts like a trident

Hey all! I've been trying to make a tool/weapon for a mod that acts sort of like a trident in the way that you can throw it to attack and will return to you if you have the enchantment/if you program it to do so.

sorry if this is obvious, i'm a little smooth brain lmao. thanks in advance!