ADVANCED: Swappable spells 🪄 (Arcana)

Hiiyaa! This question is for a datapack I'm working on called Arcana (not my main project It Spreads), where I want to use the new item components to create wands with spell slots you can swap out and switch between. I have a few ideas for how to do this, but I'm interested to see what other people can come up with. Basically, I want to store spells in the custom data of a Wand item, and use the consumable component to give the spells custom cooldowns, casting times, and animations. The biggest problems currently are getting the item back after consuming it, updating the spell when it is swapped, adding and removing spells from the wand, and displaying the selected spell (which changes while shift-clicking). It also needs to work in mainhand and offhand.