vscode does not provide token information through an api the way that atom does.
Potential solution: Use https://github.com/Microsoft/vscode-textmate