-
Notifications
You must be signed in to change notification settings - Fork 0
/
.bashrc
319 lines (269 loc) · 10.3 KB
/
.bashrc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
if [ -f "/etc/profile.d/modules.sh" ]; then
source "/etc/profile.d/modules.sh"
module load slurm
fi
export LANG=${LANG:-"en_US.UTF-8"}
export LC_ALL=${LC_ALL:-"en_US.UTF-8"}
export LC_CTYPE=${LC_CTYPE:-"en_US.UTF-8"}
export XDG_DATA_HOME=${XDG_DATA_HOME:-"$HOME/.local/share"}
export XDG_CONFIG_HOME=${XDG_CONFIG_HOME:-"$HOME/.config"}
export XDG_STATE_HOME=${XDG_STATE_HOME:-"$HOME/.local/state"}
export XDG_CACHE_HOME=${XDG_CACHE_HOME:-"$HOME/.cache"}
export XDG_DATA_DIRS=${XDG_DATA_DIRS:-"/usr/local/share/:/usr/share"}
export XDG_CONFIG_DIRS=${XDG_CONFIG_DIRS:-"/etc/xdg"}
export XDG_RUNTIME_DIR=${XDG_RUNTIME_DIR:-"/run/user/$(id -u)"}
# non-standard variable
export XDG_PREFIX_HOME="${HOME}/.local"
export ZSH="$HOME/.oh-my-zsh"
export USER=$USERNAME
prepend_to_env_var() {
local env_var_name="$1"
shift
local args=("$@")
if [ -z "${!env_var_name}" ]; then
export "$env_var_name"=""
fi
for ((i=${#args[@]}-1; i>=0; i--)); do
dir="${args[i]}"
if [ -d "$dir" ] && ! [[ ":${!env_var_name}:" =~ :$dir: ]]; then
if [ -z "${!env_var_name}" ]; then
eval "export $env_var_name=\"$dir\""
else
eval "export $env_var_name=\"$dir:\${$env_var_name}\""
fi
fi
done
}
append_to_env_var() {
local env_var_name="$1"
shift
local args=("$@")
if [ -z "${!env_var_name}" ]; then
export "$env_var_name"=""
fi
for dir in "${args[@]}"; do
if [ -d "$dir" ] && ! [[ ":${!env_var_name}:" =~ :$dir: ]]; then
if [ -z "${!env_var_name}" ]; then
eval "export $env_var_name=\"$dir\""
else
eval "export $env_var_name=\"\${$env_var_name}:$dir\""
fi
fi
done
}
remove_from_env_var() {
local env_var_name="$1"
local path_to_remove="$2"
local current_value="${!env_var_name}"
current_value="${current_value//:$path_to_remove:/:}"
current_value="${current_value//:$path_to_remove/}"
current_value="${current_value//$path_to_remove:/}"
eval "export $env_var_name=\"$current_value\""
}
prepend_to_env_var PATH "$HOME/.local/bin" "/usr/local/bin"
prepend_to_env_var LD_LIBRARY_PATH "$HOME/.local/lib" "/usr/local/lib"
prepend_to_env_var MANPATH "$HOME/.local/man" "/usr/local/man"
alias python="python3"
alias lg="lazygit"
if [[ $(uname -r | grep 'WSL2') ]]; then
local host_ip=$(cat /etc/resolv.conf | grep '^nameserver' | cut -d ' ' -f 2)
export http_proxy=${http_proxy:-"${host_ip}:1080"}
export https_proxy=${https_proxy:-"${host_ip}:1080"}
else
export http_proxy=${http_proxy:-"http://127.0.0.1:1080"}
export https_proxy=${https_proxy:-"http://127.0.0.1:1080"}
fi
export no_proxy=${no_proxy:-"localhost,.hkust-gz.edu.cn"}
export HTTP_PROXY=${HTTP_PROXY:-${http_proxy}}
export HTTPS_PROXY=${HTTPS_PROXY:-${https_proxy}}
export NO_PROXY=${NO_PROXY:-${no_proxy}}
broadcast_proxies() {
if [ -z "${http_proxy}" ]; then
git config --global --unset http.proxy
else
git config --global http.proxy ${http_proxy}
fi
if [ -z "${https_proxy}" ]; then
git config --global --unset https.proxy
else
git config --global https.proxy ${https_proxy}
fi
}
broadcast_proxies
# ~/.bashrc: executed by bash(1) for non-login shells.
# see /usr/share/doc/bash/examples/startup-files (in the package bash-doc)
# for examples
# If not running interactively, don't do anything
case $- in
*i*) ;;
*) return;;
esac
# don't put duplicate lines or lines starting with space in the history.
# See bash(1) for more options
HISTCONTROL=ignoreboth
# append to the history file, don't overwrite it
shopt -s histappend
# for setting history length see HISTSIZE and HISTFILESIZE in bash(1)
HISTSIZE=1000
HISTFILESIZE=2000
# check the window size after each command and, if necessary,
# update the values of LINES and COLUMNS.
shopt -s checkwinsize
# If set, the pattern "**" used in a pathname expansion context will
# match all files and zero or more directories and subdirectories.
#shopt -s globstar
# make less more friendly for non-text input files, see lesspipe(1)
[ -x /usr/bin/lesspipe ] && eval "$(SHELL=/bin/sh lesspipe)"
# set variable identifying the chroot you work in (used in the prompt below)
if [ -z "${debian_chroot:-}" ] && [ -r /etc/debian_chroot ]; then
debian_chroot=$(cat /etc/debian_chroot)
fi
# set a fancy prompt (non-color, unless we know we "want" color)
case "$TERM" in
xterm-color|*-256color) color_prompt=yes;;
esac
# uncomment for a colored prompt, if the terminal has the capability; turned
# off by default to not distract the user: the focus in a terminal window
# should be on the output of commands, not on the prompt
force_color_prompt=yes
if [ -n "$force_color_prompt" ]; then
if [ -x /usr/bin/tput ] && tput setaf 1 >&/dev/null; then
# We have color support; assume it's compliant with Ecma-48
# (ISO/IEC-6429). (Lack of such support is extremely rare, and such
# a case would tend to support setf rather than setaf.)
color_prompt=yes
else
color_prompt=
fi
fi
if [ "$color_prompt" = yes ]; then
PS1='${debian_chroot:+($debian_chroot)}\[\033[01;32m\]\u@\h\[\033[00m\]:\[\033[01;34m\]\w\[\033[00m\]\$ '
else
PS1='${debian_chroot:+($debian_chroot)}\u@\h:\w\$ '
fi
unset color_prompt force_color_prompt
# If this is an xterm set the title to user@host:dir
case "$TERM" in
xterm*|rxvt*)
PS1="\[\e]0;${debian_chroot:+($debian_chroot)}\u@\h: \w\a\]$PS1"
;;
*)
;;
esac
# enable color support of ls and also add handy aliases
if [ -x /usr/bin/dircolors ]; then
test -r ~/.dircolors && eval "$(dircolors -b ~/.dircolors)" || eval "$(dircolors -b)"
alias ls='ls --color=auto'
#alias dir='dir --color=auto'
#alias vdir='vdir --color=auto'
alias grep='grep --color=auto'
alias fgrep='fgrep --color=auto'
alias egrep='egrep --color=auto'
fi
# colored GCC warnings and errors
#export GCC_COLORS='error=01;31:warning=01;35:note=01;36:caret=01;32:locus=01:quote=01'
# some more ls aliases
alias ll='ls -alF'
alias la='ls -A'
alias l='ls -CF'
# Add an "alert" alias for long running commands. Use like so:
# sleep 10; alert
alias alert='notify-send --urgency=low -i "$([ $? = 0 ] && echo terminal || echo error)" "$(history|tail -n1|sed -e '\''s/^\s*[0-9]\+\s*//;s/[;&|]\s*alert$//'\'')"'
# Alias definitions.
# You may want to put all your additions into a separate file like
# ~/.bash_aliases, instead of adding them here directly.
# See /usr/share/doc/bash-doc/examples in the bash-doc package.
if [ -f ~/.bash_aliases ]; then
. ~/.bash_aliases
fi
# enable programmable completion features (you don't need to enable
# this, if it's already enabled in /etc/bash.bashrc and /etc/profile
# sources /etc/bash.bashrc).
if ! shopt -oq posix; then
if [ -f /usr/share/bash-completion/bash_completion ]; then
. /usr/share/bash-completion/bash_completion
elif [ -f /etc/bash_completion ]; then
. /etc/bash_completion
fi
fi
# Preferred editor
if command -v nvim &>/dev/null; then
export SUDO_EDITOR='nvim'
export EDITOR='nvim'
elif command -v vim &>/dev/null; then
export SUDO_EDITOR='vim'
export EDITOR='vim'
elif command -v vi &>/dev/null; then
export SUDO_EDITOR='vi'
export EDITOR='vi'
fi
# Compilation
export ARCHFLAGS="-arch $(uname -m)"
export NUMCPUS=`grep -c '^processor' /proc/cpuinfo`
alias pmake='time nice make -j${NUMCPUS} --load-average=${NUMCPUS}'
alias nvimconfig="${EDITOR} ${XDG_CONFIG_HOME}/nvim"
alias tmuxconfig="${EDITOR} ${XDG_CONFIG_HOME}/tmux"
export NVM_DIR="${XDG_CONFIG_HOME}/nvm"
[ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh" # This loads nvm
eval "$(starship init bash)"
# <<< personal conda initialization, not need to `conda init zsh` <<<
# I personally download miniconda3 to "${XDG_PREFIX_HOME}/miniconda3"
__conda_setup="$("${XDG_PREFIX_HOME}/miniconda3/bin/conda" 'shell.zsh' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
eval "$__conda_setup"
else
if [ -f "${XDG_PREFIX_HOME}/miniconda3/etc/profile.d/conda.sh" ]; then
. "${XDG_PREFIX_HOME}/miniconda3/etc/profile.d/conda.sh"
else
export PATH="${XDG_PREFIX_HOME}/miniconda3/bin:$PATH"
fi
fi
unset __conda_setup
# <<< personal conda initialization <<<
gpu_driver_path="/usr/lib/modules/$(uname -r)/kernel/drivers/video/nvidia.ko"
print_sys_info() {
echo
echo "$(whoami) @ $(hostname) @ $(hostname -I | awk '{ print $1; }')"
echo
echo "OS: $(uname -sr)"
echo "Distro: $(cat /etc/os-release | grep ^'PRETTY_NAME' | grep -oP '"\K[^"]+(?=")')"
echo "CPU Device: $(cat /proc/cpuinfo | grep ^'model name' | sed -n '1p' | grep -oP '(?<=: ).*')"
echo "CPU Usage: $((100-$(vmstat 1 2 | tail -1 | awk '{print $15}')))%"
if [ ! -f ${gpu_driver_path} ]; then
echo "NVIDIA GPU Driver not found."
else
echo "NVIDIA GPU Driver Version $(modinfo ${gpu_driver_path} | grep ^version | grep -oP '(?<=: ).*')"
if [ ! -x "$(command -v nvidia-smi)" ]; then
echo "command nvidia-smi not found."
else
echo "GPU Device: $(nvidia-smi -L | sed 's/([^)]*)//g')"
fi
fi
echo "Avaiable Memory: $(free -mh | grep ^Mem | awk '{ print $7; }') / $(free -mh | grep ^Mem | awk '{ print $2; }')"
echo
}
# print_sys_info
check_version() {
local cli="${1}"
local command_to_print_version="${2}"
if command -v "$cli" &>/dev/null; then
eval "$command_to_print_version"
else
echo "$cli not found" >&2
fi
}
check_version_all() {
check_version "zsh" "echo -e \"zsh\tv$(zsh --version | awk '{ print $2; }')\""
check_version "tmux" "echo -e \"tmux\tv$(tmux -V | awk '{ print $2; }')\""
check_version "nvim" "echo -e \"nvim\t$(nvim --version | sed -n '1p' | head -n 1 | awk '{ print $2; }')\""
check_version "python" "echo -e \"python\tv$(python --version | awk '{ print $2; }')\""
check_version "git" "echo -e \"git\tv$(git --version | awk '{ print $3; }')\""
if [[ ! $(uname -r | grep 'WSL2') ]]; then
check_version "docker" "echo -e \"docker\tv$(docker --version | awk '{ print $3; }' | sed 's/.$//')\""
fi
if [ -f ${gpu_driver_path} ]; then
check_version "nvcc " "echo -e \"nvcc\tv$(nvcc --version | sed -n '4p' | awk '{ print $5; }' | sed 's/.$//')\""
fi
check_version "conda" "echo -e \"conda\tv$(conda --version | awk '{ print $2; }')\""
echo
}