The current functionality to align the code takes a lot of effort and time to use(Right click -> Go to align and only after that we can click again to select an option) - three steps.
What I suggest is a simpler or default auto align to format all the code from a flow.
This should be a simple option to access by mouse or keyboard shortcut(Ctrl + Shift + F).
Example of the golang implementation https://blog.golang.org/go-fmt-your-code.
The auto align rules can be defined by you :)