This mod disables cooldown and brings back the swing hand animation. Still work with shields.

Also, it adds a gamerule to get the mod enable or not anytime you want :

Preview