shell: Add support for ZSH

* Add ZSH support in init_env
* Move references to BASH_SOURCE into init_env, since that is not
available in the same form under zsh
* Avoid bashisms in [ ] conditions (use the recommended [[ ]] instead)
* Avoid ${FOO^^} bashism
* general cleanup of the script