122 lines
3.7 KiB
Bash
122 lines
3.7 KiB
Bash
# VCS
|
|
YS_VCS_PROMPT_PREFIX1=" %{$reset_color%}on%{$fg[blue]%} "
|
|
YS_VCS_PROMPT_PREFIX2=":%{$fg[cyan]%}"
|
|
YS_VCS_PROMPT_SUFFIX="%{$reset_color%}"
|
|
YS_VCS_PROMPT_DIRTY=" %{$fg[red]%}x"
|
|
YS_VCS_PROMPT_CLEAN=" %{$fg[green]%}o"
|
|
|
|
# Git info
|
|
git_branch_info () {
|
|
local git_branch_name=$(git branch --show-current 2&> /dev/null | xargs -I branch echo "branch")
|
|
if [[ $git_branch_name = "" ]]; then
|
|
echo ""
|
|
else
|
|
echo "${ZSH_THEME_GIT_PROMPT_PREFIX}$git_branch_name${ZSH_THEME_GIT_PROMPT_SUFFIX}"
|
|
fi
|
|
}
|
|
|
|
local git_info='$(git_branch_info)'
|
|
ZSH_THEME_GIT_PROMPT_PREFIX="${YS_VCS_PROMPT_PREFIX1}git${YS_VCS_PROMPT_PREFIX2}"
|
|
ZSH_THEME_GIT_PROMPT_SUFFIX="$YS_VCS_PROMPT_SUFFIX"
|
|
ZSH_THEME_GIT_PROMPT_DIRTY="$YS_VCS_PROMPT_DIRTY"
|
|
ZSH_THEME_GIT_PROMPT_CLEAN="$YS_VCS_PROMPT_CLEAN"
|
|
|
|
# SVN info
|
|
local svn_info='$(svn_prompt_info)'
|
|
ZSH_THEME_SVN_PROMPT_PREFIX="${YS_VCS_PROMPT_PREFIX1}svn${YS_VCS_PROMPT_PREFIX2}"
|
|
ZSH_THEME_SVN_PROMPT_SUFFIX="$YS_VCS_PROMPT_SUFFIX"
|
|
ZSH_THEME_SVN_PROMPT_DIRTY="$YS_VCS_PROMPT_DIRTY"
|
|
ZSH_THEME_SVN_PROMPT_CLEAN="$YS_VCS_PROMPT_CLEAN"
|
|
|
|
# HG info
|
|
local hg_info='$(ys_hg_prompt_info)'
|
|
ys_hg_prompt_info() {
|
|
# make sure this is a hg dir
|
|
if [ -d '.hg' ]; then
|
|
echo -n "${YS_VCS_PROMPT_PREFIX1}hg${YS_VCS_PROMPT_PREFIX2}"
|
|
echo -n $(hg branch 2>/dev/null)
|
|
if [[ "$(hg config oh-my-zsh.hide-dirty 2>/dev/null)" != "1" ]]; then
|
|
if [ -n "$(hg status 2>/dev/null)" ]; then
|
|
echo -n "$YS_VCS_PROMPT_DIRTY"
|
|
else
|
|
echo -n "$YS_VCS_PROMPT_CLEAN"
|
|
fi
|
|
fi
|
|
echo -n "$YS_VCS_PROMPT_SUFFIX"
|
|
fi
|
|
}
|
|
|
|
# Virtualenv
|
|
local venv_info='$(virtenv_prompt)'
|
|
YS_THEME_VIRTUALENV_PROMPT_PREFIX=" %{$fg[green]%}"
|
|
YS_THEME_VIRTUALENV_PROMPT_SUFFIX=" %{$reset_color%}%"
|
|
virtenv_prompt() {
|
|
[[ -n "${VIRTUAL_ENV:-}" ]] || return
|
|
echo "${YS_THEME_VIRTUALENV_PROMPT_PREFIX}${VIRTUAL_ENV:t}${YS_THEME_VIRTUALENV_PROMPT_SUFFIX}"
|
|
}
|
|
|
|
local exit_code="%(?,,| C:%{$fg[red]%}%?%{$reset_color%})"
|
|
|
|
# Prompt format:
|
|
#
|
|
# PRIVILEGES USER @ MACHINE in DIRECTORY on git:BRANCH STATE [TIME] C:LAST_EXIT_CODE
|
|
# $ COMMAND
|
|
#
|
|
# For example:
|
|
#
|
|
# % ys @ ys-mbp in ~/.oh-my-zsh on git:master x | C:0
|
|
# [21:47:42] $
|
|
|
|
PROMPT="
|
|
%{$terminfo[bold]$fg[blue]%}#%{$reset_color%} \
|
|
%(#,%{$bg[yellow]%}%{$fg[black]%}%n%{$reset_color%},%{$fg[cyan]%}%n) \
|
|
%{$reset_color%}@ \
|
|
%{$fg[green]%}%m \
|
|
%{$reset_color%}in \
|
|
%{$terminfo[bold]$fg[yellow]%}%~%{$reset_color%}\
|
|
${hg_info}\
|
|
${git_info}\
|
|
${svn_info}\
|
|
${venv_info}\
|
|
\
|
|
$exit_code
|
|
%{$terminfo[bold]$fg[red]%}%U%*%u %{$reset_color%}%U$%u %{$reset_color%}"
|
|
|
|
# source $HOME/.software/software_env.sh
|
|
function add_path_to_software_env() {
|
|
echo -e "\033[4;36mAdd [$(pwd)] to software evn\033[m"
|
|
grep $(pwd) $HOME/.software/software_env.sh
|
|
if [[ ! $? == 0 ]]; then
|
|
echo "export PATH=$(pwd):\$PATH" >> $HOME/.software/software_env.sh
|
|
fi
|
|
|
|
source $HOME/.software/software_env.sh
|
|
|
|
}
|
|
|
|
. $HOME/.software/software_env.sh
|
|
|
|
# Homebrew tsinghua mirror for macOS
|
|
function brewthu() {
|
|
if [[ $HOMEBREW_BOTTLE_DOMAIN = "" ]]; then
|
|
export HOMEBREW_API_DOMAIN="https://mirrors.tuna.tsinghua.edu.cn/homebrew-bottles/api"
|
|
export HOMEBREW_BOTTLE_DOMAIN="https://mirrors.tuna.tsinghua.edu.cn/homebrew-bottles"
|
|
export HOMEBREW_BREW_GIT_REMOTE="https://mirrors.tuna.tsinghua.edu.cn/git/homebrew/brew.git"
|
|
export HOMEBREW_CORE_GIT_REMOTE="https://mirrors.tuna.tsinghua.edu.cn/git/homebrew/homebrew-core.git"
|
|
export HOMEBREW_PIP_INDEX_URL="https://pypi.tuna.tsinghua.edu.cn/simple"
|
|
# Bottles
|
|
export HOMEBREW_BOTTLE_DOMAIN="https://mirrors.tuna.tsinghua.edu.cn/homebrew-bottles"
|
|
else
|
|
unset HOMEBREW_API_DOMAIN
|
|
unset HOMEBREW_BOTTLE_DOMAIN
|
|
unset HOMEBREW_BREW_GIT_REMOTE
|
|
unset HOMEBREW_CORE_GIT_REMOTE
|
|
unset HOMEBREW_PIP_INDEX_URL
|
|
# Bottles
|
|
unset HOMEBREW_BOTTLE_DOMAIN
|
|
fi
|
|
}
|
|
|
|
# alias
|
|
alias py=python3
|
|
alias ap2se=add_path_to_software_env |