I do something similar for our spindle tool - but I hate what we do. The only thing that makes it palatable is that we then put the tool into the artifact cache so the hacks only happen once per cache-key. We have to subvert the task graph (get compile to run for a certain target before codegen) so we are probably just stuck with it while using the v1 engine.