I'm impressed by this and like it a lot.
If you don't mind a 'drive-by' feature request: it would be fantastic if you were to support the PlantUML state DSL:
http://plantuml.com/state-diagram
That would push me from "interesting proof of concept" to figuring out how to use it in projects.