I created block marking extension for VS Code, how can I access to parsed tokens of code in a file?

Question

Explanation DEMO with react.js: https://youtu.be/qrEKSe8VkYQ

The extension itself: https://youtu.be/ZmvLgIDJNg8

Please help, I\'m trying to finish creating an exte