git psh (push -u -> psh): Push and set upstream if needed

git-psh

#!/usr/bin/env bash
red="$(tput setaf 1)"
green="$(tput setaf 2)"
cyan="$(tput setaf 6)"
reset="$(tput init)"

branch="$(git rev-parse --abbrev-ref HEAD)"
[[ $? == 0 ]] || { echo "$red"'git `psh` cannot work without a branch'"$reset"; exit 1;}
upstream="$(git rev-parse --abbrev-ref "$branch"'@{u}' 2>/dev/null)"
if [[ $? == 0 ]]; then
    echo "${green}Pushing to $upstream...${reset}"
    git push
else
    echo "${cyan}Pushing to origin/$branch and setting as upstream...${reset}"
    git push -u origin "$branch"
fi