Feature request; set button background based on selected output

I'd like to be able to set the background colour of the button based on which output is selected; if the headphones, then a red background; if the speakers, a black background.

I've allowed the action to control the background colour, but presumably that isn't a StreamController builtin function and the plugin needs to support it. Is that possible at all?