Probably need to hook into internal functions in order to do this.

Already implemented in e65d164 (albeit with some bugs, #53 & #54).