Modding isn't even needed, the Channeling enchantment for tridents does this: https://minecraft.fandom.com/wiki/Channeling