I googled quickly and unfortunately, didn’t find anything. But I did stumble on this page: https://code.visualstudio.com/api/language-extensions/syntax-highlight-guide
Looks like it’s very possible to do this type of coloring. But you may have to dig for an extension that does it. Or build one yourself?