honestly extensions look way easy to implement and this functionality may be useful in other places on my project so i'll give that a go first -- probably end up using afterSwap inside the extension actually