You access the element in the in afterGetColHeader callback in the first demo that you’ve send me https://codepen.io/js-ziggle/pen/pjpNWZ?editors=0110
It uses an index and the TH element.
Once a new color is picked by the context menu you can update the afterGetColHeader via updateSettings method.