Refactor: tools: Split one shot mode out into its own function.

Also, the connection error handling function can get split out on its
own as well to allow it to be reused in both the one shot and loop
cases.