Looks great, and it shouldn't be difficult at all to adapt for the codebuttons! Just have the actual HTML that the ubb codes resolve to in one variable (that's used to construct the formatted output) and hold the UBB-code-equivilent in another variable that actually gets submitted!
Donate to UBBDev today to help aid in Operational, Server and Script Maintenance, and Development costs.
Please also see our parent organization VNC Web Services if you're in the need of a new UBB.threads Install or Upgrade, Site/Server Migrations, or Security and Coding Services.