I ended up basing it on their yaml mode, just modified it to tokenize more and to track context