More work on cleaning up TODOs (#997)

* wip

* wip

* Refactor lint tool

* Add a lint to validate that TODOs have context

* cleanup lint some more

* fix codegen failure & remove unused arguments