fix error in zsh environment

When I 'source SOURCEME' in zsh environment,  Zsh doesn't recoginze variable BASH_SOURCE. So I write some judgment statements to fix it.
Hope you can accept this pull request.