I am trying to do the same thing. I was able to modify the control template pretty easily.
However, I'm having a problem wiring up the command. It looks like normally one would find the button in OnApplyTemplate and then wire up a click event. This is code that I've tried but it isn't getting me anywhere.
Any help would be great! Thanks,
-Marshall