Refactor attrs handling logic in code mods

[at-nathan]

As discussed here: https://github.com/atlassian-labs/compiled/pull/1046#discussion_r787295960

The flow in which we code mod components with attrs is not kosher with the rest of code mod logic.