Yeah, you would just have the game unequip your weapon, check to see what weapon it is, and upgrade it. It could be based on items you have, how far along in the game you are, or just raw GP.

Of course, I have no idea what the upgrade system is like for those games, so I don't know exactly what you're looking for.. can't code without specifics.