I’d put together a patch to do something similar a while ago. I wish I’d wrote up a doc for it so I could remember the approach.