Open preferences in atom and find/install language-babel.
Open the settings of language-babel.
Add /* sql */:source.sql to the Javascript tagged template literal grammer extensions.
Now write your queries like this /* sql */`SELECT * FROM users`. Make sure the query is in backticks `.
Enjoy the pretty queries!