Visual Editor – Colorize Shortcodes

Managed to get a working solution.

What you need:

TinyMCE Advanced plugin

What you should do:

Customize the Visual Blocks Plugin that comes with the TinyMCE plugin and add css classes for each element you would to highlight (just loop through all p elements and parse the innerHTML and check if it contains your specific pattern).

Finally it looks something like this:
enter image description here