Would also be cool if you could add TreeSitter support too so that different parts of the grammar are colored differently.