-
Notifications
You must be signed in to change notification settings - Fork 0
/
.personal.bashrc
1551 lines (1365 loc) · 43.5 KB
/
.personal.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
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#!/usr/bin/env false bash
has_panfs=0
mount | grep -q "type panfs " && has_panfs=1
# Something to do with panfs and __git_ps1
if [ "${BASH_VERSINFO[0]}" -ge "4" ]; then
backup_aliases[git]="$(alias git 2> /dev/null || :)"
else
backup_aliases__git="$(alias git 2> /dev/null || :)"
fi
unalias git &> /dev/null
umask 0022
# If not running interactively, don't do anything
case $- in
*i*) ;;
*) return;;
esac
###########################
### Env var and options ###
###########################
# colored GCC warnings and errors
export GCC_COLORS='error=01;31:warning=01;35:note=01;36:caret=01;32:locus=01:quote=01'
# 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
# Make LESS always use ANSI (raw mode)
# https://askubuntu.com/a/1156810
export LESS=FRX
# There is still a possibility this will mess up another application in the future
# at which point, I should use one of these?
# alias less="less -FRX"
# alias less="env LESS=FRX less"
# check the window size after each command and, if necessary,
# update the values of LINES and COLUMNS.
shopt -s checkwinsize
if [ "${BASH_VERSINFO[0]}" -gt "3" ]; then
# If set, the pattern "**" used in a pathname expansion context will
# match all files and zero or more directories and subdirectories.
shopt -s globstar
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
if [ -n "${DISPLAY+set}" ]; then
_ORIGINAL_DISPLAY="${DISPLAY}"
fi
if [ -f ~/.displayrc ]; then
source ~/.displayrc
if [ -z "$(xauth -n list "${DISPLAY}")" ] || ! timeout 1 xhost &> /dev/null; then
echo "Stale x11 detected, removing ~/.displayrc"
rm -f ~/.displayrc
unset DISPLAY
if [ -n "${_ORIGINAL_DISPLAY+set}" ]; then
DISPLAY="${_ORIGINAL_DISPLAY}"
fi
fi
fi
unset _ORIGINAL_DISPLAY
if [[ "${DISPLAY-}" =~ ^:[0-9]+$ ]]; then
X_WORKING=1 #don't check :0
elif command -v xhost &> /dev/null; then
if timeout 1 xhost &> /dev/null; then
X_WORKING=1
else
X_WORKING=0
fi
elif timeout 1 xset &> /dev/null; then
X_WORKING=1
else
X_WORKING=0
fi
export X_WORKING
# .docker/config
if [ "${OS-}" = "Windows_NT" ]; then
export DOCKER_CONFIG=~/.docker/windows
elif [ -n "${WSL_DISTRO_NAME+set}" ]; then
# export DOCKER_CONFIG=~/.docker/wsl
# WSL has special integration, it's just easier to leave it here for now
export DOCKER_CONFIG=~/.docker
elif [[ ${OSTYPE-} = darwin* ]]; then
export DOCKER_CONFIG=~/.docker/macos
else
export DOCKER_CONFIG=~/.docker/linux
fi
if [ "${X_WORKING}" = "1" ]; then
if [[ "${DISPLAY-}" =~ ^:[0-9]+$ ]]; then
function edit()
{
code -n -w "${@}"
}
export EDITOR=code
else
function edit()
{
vim "${@}"
}
export EDITOR=vim
fi
fi
# add_element_post
if [ -z "${PRE_VSI_COMMON_PATH+set}" ]; then
PRE_VSI_COMMON_PATH=${PATH}
fi
export PATH=$(
# I don't want this sourced in my bash environment, too noisy
source ~/.dot/external/dot_core/external/vsi_common/linux/elements.bsh
IFS=: add_element_post PATH ~/bin
if [ -d ~/.cargo/bin ]; then
IFS=: add_element_post PATH ~/.cargo/bin
fi
IFS=: add_element_post PATH "$(echo ~/.dot/external/dot_core/external/vsi_common/linux)"
echo "${PATH}"
)
#IFS=: add_element_pre PYTHONPATH /home/andy/tools/
#IFS=: add_element_pre PYTHONPATH /usr/local/lib64/python2.7/site-packages
export PYTHONSTARTUP=~/.pyrc
#export PS1='\[\e[40;93m\]\w\[\e[0m\]\n[\u@\h \W]$ '
################
### bashrc.d ###
################
if [ -d ~/.bashrc.d ]; then
for bashrc_file in ~/.bashrc.d/*; do
if [ -r "${bashrc_file}" ]; then
source "${bashrc_file}"
fi
done
fi
unset bashrc_file
################
### Bindings ###
################
# To discover bind key codes:
# bind -p | grep quote
# "\C-q": quoted-insert # Ctrl+q, doesn't always work (e.g. cygwin)
# "\C-v": quoted-insert # Ctrl+v, doesn't work on Windows Terminal, but does work in Command prompt
# "\e[2~": quoted-insert # Works in Windows Terminal!
# ^[[H - "^[" is "\e" (Alt), so "\e[H"
# ^? - "^" is "\C-" (Ctrl), so "\C-?"
# ?+Left/Right
bind '"\e[5C": forward-word'
bind '"\e[5D": backward-word'
# Ctrl+Left/Right
bind '"\e[1;5C": forward-word'
bind '"\e[1;5D": backward-word'
# bind '"\e[1~": beginning-of-line' # Home
# bind '"\e[4~": end-of-line' # End
#################
### My Prompt ###
#################
source ~/.git-prompt
#PS1='\[\e[40;93m\]\w\[\e[0m\]\n[\u@\h \W]$ '
#PROMPT_COMMAND='__git_ps1 "\u@\h:\w" "\\\$ "'
GIT_PS1_SHOWCOLORHINTS=1
if [ "${OS-}" != "Windows_NT" ]; then
# Too slow on windows, CreateProcess is too slow
GIT_PS1_SHOWUNTRACKEDFILES=1
GIT_PS1_SHOWUPSTREAM=verbose
GIT_PS1_SHOWSTASHSTATE=1
GIT_PS1_SHOWDIRTYSTATE=1
fi
# If one of the HPC machines
if [ -n "${BC_HOST+set}" ]; then
# disable this for speed
unset GIT_PS1_SHOWDIRTYSTATE
fi
alias quick_git_ps1='unset GIT_PS1_SHOWDIRTYSTATE'
if command -v tput &>/dev/null && [[ $(tput colors) -ge 256 ]]; then
hostcolor=$(echo ${BC_HOST:-$(hostname)} | cksum | awk '{print $1}')
if [ "${BC_HOST-}" = "gaffney" ]; then
hostcolor=$(( hostcolor + 10 ))
fi
# 24-231
hostcolor=$(( hostcolor % 207 + 24 ))
# I'm not 100% sure I understand this ansi code, especially the [] on the
# outside, but this is passed to PROMPT_COMMAND and evaluated later, hence no
# literal $''. Using the literal causes issues with chars being left behind.
hostcolor='\[\e[38;5;'$hostcolor'm\]'
fi
# Modifies the title and inserts the tty name after the hostname in ()
function my_vte_prompt_command()
{
local vte
local tty_name
declare -f -F __vte_prompt_command > /dev/null #In case not using gnome-terminal
if type -p __vte_prompt_command; then
if command -v tty &> /dev/null; then
vte="$(__vte_prompt_command)"
tty_name="$(tty)"
tty_name="${tty_name#/*/}"
# Use the :, since I know it is between the hostname and directory. This
# is easier than rewriting __vte_prompt_command myself
printf '%s' "${vte/:/(${tty_name}):}"
else
__vte_prompt_command
fi
fi
}
if [ -e /.dockerenv ]; then
SINGULARITY_NAME=docker
fi
OS_PROMPT=
if [ -n "${MSYSTEM+set}" ]; then
OS_PROMPT=" \[\e[35m\]${MSYSTEM}\[\e[0m\]"
elif [ -n "${WSL_INTEROP+set}" ]; then
OS_PROMPT=" \[\e[33m\]${WSL_DISTRO_NAME}\[\e[0m\]"
elif [ -n "${WSL_DISTRO_NAME+set}" ]; then
OS_PROMPT=" \[\e[36m\]${WSL_DISTRO_NAME}\[\e[0m\]"
elif [ "${OS-}" = "Windows_NT" ]; then
OS_PROMPT=" \[\e[34m\]Windows\[\e[0m\]"
elif [[ ${OSTYPE-} = darwin* ]]; then
OS_PROMPT=" \[\e[31m\]macOS\[\e[0m\]"
else
OS_PROMPT=" \[\e[32m\]Linux\[\e[0m\]"
fi
source ~/.dot/external/dot_core/external/vsi_common/linux/time_tools.bsh
get_time_nanoseconds >& /dev/null # preload optimization
PROMPT_COMMAND='__ps1_rv=${__vsc_status-$?};
__ps1_time=$(get_time_nanoseconds);
__git_ps1 "\[\e[40;93m\]\w\[\e[0m\]\n'\
'$(for x in ~/.prompt_command/*; do'\
' if [ -r "${x}" ]; then'\
' source "${x}" 0;'\
' fi;'\
' done)'\
'${SINGULARITY_NAME+\[\e[30;41m\]{${SINGULARITY_NAME}\}\[\e[0m\] }'\
'${VIRTUAL_ENV+($(basename "${VIRTUAL_ENV}")) }$(printf -v x "%*s" ${SHLVL}; echo -n "${x// /[}")\u@'"${hostcolor-}"'${BC_HOST+${BC_HOST}(}\h${BC_HOST+)}\[\e[0m\] $('\
'if [ "$__ps1_rv" != "0" ]; then'\
' echo "\[\e[41m\]${__ps1_rv}\[\e[0m\]";'\
'else'\
' echo "${__ps1_rv}";'\
'fi)'"${OS_PROMPT}"' \W" "]$ ";
__ps1_time=$(($(get_time_nanoseconds)-__ps1_time+500000));
__ps1_time="${__ps1_time::${#__ps1_time}-6}";
__ps1_time_limit=250
if [ "${OS-}" = "Windows_NT" ]; then
__ps1_time_limit=500
fi
if [ "${__ps1_time}" -gt "${__ps1_time_limit}" ]; then
echo "Git took ${__ps1_time}ms, this is longer than expected." >&2
echo "Consider calling: quick_git_ps1 or set DISABLE_GIT_PROMPT=1" >& 2
fi
unset __ps1_time_limit __ps1_time __ps1_rv
touch ~/.last_ran_command
my_vte_prompt_command'
# PROMPT_COMMAND+=$'; printf "\033]0;%s@%s(%s):%s\033\\\\" "${USER}" "${HOSTNAME%%.*}" -1 "${PWD}"'
# To add a custom terminal title, add '; printf "\033]0;CUSTOM TITLE\007"' to
# the *end* of PROMPT_COMMAND (After vte_command, which sets it too). You can
# also add $'\e]0;CUSTOM TITLE\007' or $'\e]0;CUSTOM TITLE\e\\' to PS1 too...
# (\e and \033 and \x1b are the same), only if TERM is xterm*|rxvt*)
# To disable git part of prompt, set DISABLE_GIT_PROMPT=1
# A great way to time the PROMPT_COMMAND is to change the last line like so
# 'fi) \W" "] $(date +%s.%N) $ "; my_vte_prompt_command'
# and just run "date +%s.%N" and take the diff
unset hostcolor OS_PROMPT
alias no_prompt='unset PROMPT_COMMAND; PS1="$ "'
function parent_find()
{
local OLDPWD
local previous_pwd
# Speed improvement: if it's right there, echo it out right away
if [ -e "${1}" ]; then
return 0
fi
\pushd . > /dev/null
# Search for the file until some match is found
while [ "${PWD}" != "${previous_pwd-}" ]; do
if [ -e "${1}" ]; then
\popd > /dev/null
return 0
fi
previous_pwd="${PWD}"
\cd ..
done
\popd > /dev/null
return 1
}
# This was so efficient, I took off the has_panfs gate.
eval "$(declare -f __git_ps1 | sed $'1c\\\n__orig_git_ps1()\n')"
function __git_ps1()
{
if [ "${DISABLE_GIT_PROMPT-}" != "1" ] && (parent_find .git &> /dev/null); then
unalias git &> /dev/null
__orig_git_ps1 "${1}" "${2}"
if [ "${BASH_VERSINFO[0]}" -ge "4" ]; then
eval "${backup_aliases[git]-}"
else
eval "${backup_aliases__git-}"
fi
else
PS1="$1$2"
fi
}
# Have to set git aliase AFTER __git_ps1, or else the alias is hardcoded into
# the function
if [ "${BASH_VERSINFO[0]}" -ge "4" ]; then
eval "${backup_aliases[git]-}"
else
eval "${backup_aliases__git-}"
fi
# export TERM=linux
# export TERM=xterm-256color
#Apparently its bad to mess with the TERM variable, for example it fucks up less and changed the End key from doing G to F
# Stop virtual env activate script from messing with PS1, I use PROMPT_COMMAND
# anyways
VIRTUAL_ENV_DISABLE_PROMPT=1
#gnome-terminal --save-config ~/.gnome-termainl-session
#############################
### Host Specific Section ###
#############################
# WSL
if [ "${BASH_VERSINFO[0]}" -ge "5" -a -n "${WSL_INTEROP+set}" ]; then
function _custom_initial_word_complete()
{
if [ "${2-}" != "" ]; then
if [ "${2::3}" == "wor" ]; then
COMPREPLY=($(compgen -c "${2}" | \grep -v workfolderssvc))
else
COMPREPLY=($(compgen -c "${2}"))
fi
fi
}
complete -I -F _custom_initial_word_complete
fi
# WSL
if [ -n "${WSL_DISTRO_NAME+set}" ]; then
function mount_self()
{
mkdir -p "/mnt/wsl/${WSL_DISTRO_NAME}"
if sudo -n mount --bind / "/mnt/wsl/${WSL_DISTRO_NAME}/"; then
echo "Mounted to /mnt/wsl/${WSL_DISTRO_NAME}/" >&2
else
(
source ~/.dot/external/dot_core/external/vsi_common/linux/quotemire
echo "Unable to mount. Make sure the following is in your sudoers file:" >&2
quotemire "sudo bash -c" "echo 'ALL ALL=(ALL) NOPASSWD: /usr/bin/mount --bind / $(printf %q /mnt/wsl/${WSL_DISTRO_NAME}/)' > /etc/sudoers.d/wsl_mount_self" >&2
)
fi
}
fi
# Deprecated: https://github.com/microsoft/WSL/issues/4246#issuecomment-1778060287
# WSL2 fix
# if [ -n "${WSL_INTEROP+set}" ]; then
# if [ "${HOSTNAME-}" = "kaku" ]; then
# if [ "${WSL_DISTRO_NAME}" != "Ubuntu-20.04" ]; then
# if ! /mnt/c/Windows/System32/wsl.exe --cd / -d Ubuntu-20.04 service wsl-vpnkit status &>/dev/null; then
# /mnt/c/Windows/System32/wsl.exe --cd / -d Ubuntu-20.04 --user root service wsl-vpnkit start
# fi
# fi
# fi
# fi
#############################
### Functions and Aliases ###
#############################
## Docker ##
function dc_images_parent_id(){
/opt/venvs/docker/bin/python -c "from __future__ import print_function; import docker; c=docker.client.from_env();imgs=c.images.list(all=True);ids=[(i.attrs['ParentId'], i.id) for i in imgs]; [print('{} {}'.format(i[0].replace('sha256:', ''), i[1].replace('sha256:', ''))) for i in ids]"
}
function dc_children(){
dc_children_q $1 " "
}
function dc_children_q(){
if [ "$2" != "" ]; then
local depth=-$2
fi
local cache=$3
if [ "$cache" == "" ]; then
local cache_use=`mktemp`
dc_images_parent_id > "${cache_use}"
local dockid=$(docker inspect -f '{{.Id}}' $1 | sed 's|sha256:||')
else
local cache_use="${cache}"
local dockid=$1
fi
if (( ${#dockid} < 64 )); then return; fi
while read line; do
local newid=`echo $line | awk '{print $2}'`
echo ${depth}${newid}
dc_children_q "${newid}" "${depth}" "${cache_use}"
done < <(grep "^${dockid}" "${cache_use}")
if [ "${cache}" == "" ]; then
\rm "${cache_use}"
fi
}
# Find out which docker is responsible for host pid
function dc_pid(){
local ds=($(docker ps --format '{{.ID}}' | xargs docker inspect -f '{{.State.Pid}} {{.Id}}'))
local ppid=$1
while (( ${ppid} != 1 )); do
for x in $(seq 0 2 ${#ds[@]}); do
if [ "${ds[$x]}" == "${ppid}" ]; then
echo ${ds[$(($x+1))]}
return
fi
done
ppid=$(ps -ho ppid $ppid)
done
}
# List all containers (running and not-running) that are bound to a particular
# volume (from docker volume ls). This is useful for trying to remove a volume,
# but a stray container has it mounted
function dc_find_volume()
{
local volumes=($(docker inspect --format '{{$x:=.Name}} {{range .Mounts}} {{if .Name}} {{$x}}@{{ .Name }} {{end}} {{end}}' $(docker ps -aq)))
local x
for x in "${volumes[@]}"; do
if [[ ${x} =~ .*@$1 ]]; then
echo ${x%@*}
fi
done
}
function dc_volume()
{
docker volume rm $(docker volume ls | \grep -E '^local +[a-f0-9]{64}$' | awk '{print $2}')
}
# Removes an image and all its children. Will probably fail if containers are
# left behind using any of the images.
function dc_rmi(){
docker rmi $(dc_children_q $1 | tac)
docker rmi $1
}
function docker_list-tags()
{
curl -sL "https://hub.docker.com/v1/repositories/${1}/tags" | jq -r '.[].name'
}
function _de_cleanup()
{
local nullglob=1
if \shopt -q nullglob; then
nullglob=0
fi
for f in ~/.ssh/docker_*; do
if ! \fuser "${f}" >& /dev/null; then
\rm "${f}"
fi
done
if [ "${nullglob}" = "1" ]; then
\shopt -u nullglob
fi
}
#**
# .. function:: de_activate
#
# Docker environment activate - connected to remote docker server using ssh tunneling. This does **not** employ the *insecure* method of exposing the docker daemon to an open TCP port.
#
# :Arguments: ``$1``... - Arguments to ``ssh`` command, machine name and options
#
# :Parameters: [``DE_OLDER``] - Set to 1 if using ssh older than 6.7. When you use this flag, the server needs to have ``socat`` installed.
#
# .. rubric:: Example
#
# .. code-block:: bash
#
# de_activate username@server -p 1234
#
# .. seealso::
# :func:`de_reverse_activate`
#**
function de_activate()
{
local docker_host
local ssh_args
if [ "$#" == "0" ]; then
echo "usage: $0 <ssh flags>"
echo " Set DE_OLDER to 1 when using openssh version before 6.7"
return 1
fi
_de_ssh_args=("${@}")
_OLD_DOCKER_HOST="${DOCKER_HOST-}"
ssh_args=(-o ControlPath=~/.ssh/%C -o ControlMaster=auto -o ControlPersist=yes)
_de_cleanup
if [ "${DE_OLDER-}" = "1" ]; then
ssh -n -L 2375:localhost:2375 "${@}" \
'socat TCP-LISTEN:2375,fork,bind=localhost UNIX-CONNECT:/var/run/docker.sock&
pid=$!
trap "kill $pid" 0
while printf \\0; do
sleep 5
done' &
docker_host="tcp://localhost:2375"
else
# Requires opensshd 6.7 or newer... THANKS CentOS! :(
_de_socket="$(mktemp -u -d ~/.ssh/docker_XXXXXXXX)"
ssh_args+=(-fTN -L "${_de_socket}":/var/run/docker.sock "${@}")
docker_host="unix://${_de_socket}"
if ssh -O check "${ssh_args[@]}" >&/dev/null; then
# I can't tell when a command fails. Oh well
ssh -o ServerAliveInterval=60 "${ssh_args[@]}"&
else
ssh -o ServerAliveInterval=60 "${ssh_args[@]}" || return $?
fi
fi
export DOCKER_HOST="${docker_host}"
#This will not always be correct, but oh well
VIRTUAL_ENV="de $1"
echo "VIRTUAL_ENV=\"${VIRTUAL_ENV}\""
echo "export DOCKER_HOST=\"${DOCKER_HOST}\""
function de_deactivate()
{
ssh -o ControlPath=~/.ssh/%C -O exit "${_de_ssh_args[@]}" # || return $?
if [ "${_OLD_DOCKER_HOST-}" == "" ]; then
unset DOCKER_HOST
else
export DOCKER_HOST="${_OLD_DOCKER_HOST}"
fi
# Newer way leaves a stray socket behind... which if you close without
# deactivating will result in a "socket" leak
if [ "${_de_socket+set}" == "set" ]; then
\rm "${_de_socket}"
unset _de_socket
fi
_de_cleanup
unset _de_ssh_args VIRTUAL_ENV
unset -f de_deactivate
}
}
#**
# .. function:: de_reverse_activate
#
# Not as good as :func:`de_activate`, but needed on systems where you can't directly ssh into a docker enabled user or for some reason on Synology this is needed too.
#
# :Arguments: [``$1``] - Name of computer connecting back on. Optional if 2... is only one argument, and defaults to $(hostname)
# ``$2``... - Arguments to ``ssh`` command, machine name and options
#
# :Parameters: [``CHANGE_USER``] - The ssh-ed user needs to ``sudo`` to another user, add that command using the ``CHANGE_USER`` environment variable. With something like ``sudo su - root -c``, the ``-c`` needs the next ``ssh`` command to be passed in as one argument to the ``su`` command, use ``CHANGE_SINGLE`` too to make this work.
# [``CHANGE_SINGLE``] - Set to one for a more complicated scenario for chaning user, such as ``su -c`` where the ``-c`` take only one single argument.
# .. rubric:: Example
#
# .. code-block:: bash
#
# de_reverse_activate my_username@my_hostname username@server -p 1234
# # or
# CHANGE_SINGLE="1" CHANGE_USER="sudo su - root -c" de_reverse_activate my_username@my_hostname username@server -p 1234
# # or
# CHANGE_USER="sudo" de_reverse_activate my_username@my_hostname username@server -p 1234
#
# .. seealso::
# :func:`de_reverse_activate`
#**
function de_reverse_activate()
{
if [ "$#" = "1" ]; then
de_reverse_activate $(hostname) "$1"
fi
local phone_home="${1}"
local args=()
shift 1
local de_socket="$(mktemp -u -d ~/".ssh/docker_XXXXXXXX")"
args[0]="$(~/.dot/external/dot_core/external/vsi_common/linux/print_command ssh -t "${@}")"
args[1]="ssh -t -R '${de_socket}:/var/run/docker.sock' ${phone_home}"
args[2]="env 'DOCKER_HOST=unix://${de_socket}' 'DISPLAY=${DISPLAY}' bash -c"
args[3]="cd '$(pwd)'; exec bash"
if [ -n "${CHANGE_USER+set}" ]; then
if [ "${CHANGE_SINGLE-}" = "1" ]; then
args=("${args[0]}" "${CHANGE_USER}" "${args[@]:1}")
else
args[1]="${CHANGE_USER} ${args[1]}"
fi
fi
eval "$(~/.dot/external/dot_core/external/vsi_common/linux/quotemire "${args[@]}")"
}
#**
# .. function:: dcp2v
#
# Docker copy files to a docker volume
#
# :Arguments: ``$1`` - Volume name
# ``$2``... - Filenames
# .. seealso::
# :func:`dcpfv`
#**
function dcp2v()
{
local volume="$1"
shift 1
tar zc "${@}" | docker run -i --rm -v "${volume}":/cp -w /cp alpine tar zx
}
#**
# .. function:: dcpfv
#
# Copy files from a docker volume
#
# :Arguments: ``$1`` - Volume name
# ``$2``... - Filenames
#
# .. note::
#
# Does not handle spaces correctly. See :func:`dcpfv2`
#
# .. seealso::
# :func:`dcpfv2`
#**
function dcpfv()
{
local volume="$1"
shift 1
local args=("${@}")
docker run --rm -e IFS=t \
-v "${volume}":/cp:ro -w /cp \
debian bash -xvc "$(declare -p args);"' eval tar zc "${args[@]}"' | tar zx
}
#**
# .. function:: dcpfv2
#
# Version of :func:`dcpfv` that supports spaces
#
# :Arguments: ``$1`` - Volume name
# ``$2``... - Filenames
# .. seealso::
# :func:`dcpfv`
#**
function dcpfv2()
{
local volume="$1"
shift 1
local args=("${@}")
docker run --rm -e IFS=$'\t' \
-v "${volume}":/cp:ro -w /cp \
debian bash -c "$(declare -p args);"' tar zc "${args[@]}"' | tar zx
}
alias dc_images_list='docker images -a -q | uniq'
# Get an anonymous token. $1 = repo (no tag)
function docker_login()
{
TOKEN=$(curl "https://auth.docker.io/token?service=registry.docker.io&scope=repository:${1}:pull" | jq -r .token)
}
function pad_base64()
{
local data="${1-$(cat -)}"
case $((${#data} % 4)) in
2)
data+="=="
;;
3)
data+="="
;;
esac
echo -n "${data}"
}
function jwt()
{
local IFS=.
local tokens=(${1-$(cat -)})
# Break apart the main components
local header=$(pad_base64 <<< "${tokens[0]}" | base64 -d)
local payload=$(pad_base64 <<< "${tokens[1]}" | base64 -d)
local signature=$(pad_base64 <<< "${tokens[2]}" | tr _- /+) # why was /+ replaced with _-?
# Print
jq <<< "${header}"
jq <<< "${payload}"
########################
# The rest assumes RS256
########################
local pub_cert=$(jq -r '.x5c[0]' <<< "${header}" | pad_base64)
pub_cert=$'-----BEGIN CERTIFICATE-----\n'"${pub_cert}"$'\n-----END CERTIFICATE-----'
local pub_key=$(openssl x509 -pubkey -noout -inform PEM <<< "${pub_cert}")
# Verify signature
echo -n "${tokens[0]}.${tokens[1]}" | openssl dgst -sha256 -verify <(echo -n "${pub_key}") -signature <(base64 -d <<< "${signature}")
# # Useless information
# openssl x509 -noout -text -inform PEM <<< "${pub_cert}"
}
#**
# .. function:: docker_top_with_nspid
#
# Insert a NSPID column before the PID with the container's namespace pid in it
#
# :Arguments: ``$1``... - Docker's arguments include ``top``
#
# .. note::
#
# Should handle cases that even ``docker`` doesn't handle yet, like ``CMD`` coming before ``pid``. But will not handle no-header, but neither does ``docker``.
#**
function docker_top_with_nspid()
{
local IFS=$'\n'
local top_lines=($(command docker "${@}"))
IFS=$' \t\n'
# _Docker_ requires the PID field, so it's always there, and must have header, so no-header option is not possible
# https://github.com/docker/docker-ce/blob/44a430f4c43e61c95d4e9e9fd6a0573fa113a119/components/engine/daemon/top_unix.go#L81
# Figure out PID offset
local substr_index
# If PID is first
if [[ ${top_lines[0]} =~ ^PID( |$) ]]; then
substr_index=0
# If PID is last
elif [[ ${top_lines[0]} = *\ PID ]]; then
substr_index=$((${#top_lines[0]}-3))
else
# if it's not first or last, then it is surrounded by spaces
substr_index=${top_lines[0]% PID *}
substr_index=$((${#substr_index}+1))
fi
local index
# Print header
echo "${top_lines[0]:0:${substr_index}}$(printf "%-20s" NSPID)${top_lines[0]:${substr_index}:${#top_lines[0]}}"
for (( index=1; index<${#top_lines[@]}; index++ )); do
top_line=(${top_lines[index]:${substr_index}:${#top_lines[index]}})
# figure out nspid
nspid=$(cat /proc/${top_line[0]}/status | sed -n 's|NSpid:[ \t]*[0-9]*[ \t]*||p')
echo "${top_lines[index]:0:${substr_index}}$(printf "%-20s" "${nspid}")${top_lines[index]:${substr_index}:${#top_lines[index]}}"
done
}
function docker_fun()
{
local args=("${@}")
while (( ${#} )); do
case "${1}" in
-H|-l|\
--config|--context|--host|--log-level|--tlscacert|--tlscert|--tlskey)
shift 1
shift 1
;;
-*|image)
shift 1
;;
top)
docker_top_with_nspid "${args[@]}"
return
;;
images|ls)
command docker "${args[@]}" | sed -E 's|^([^ ]*) ( *)([^ ]*)|\1:\3\2|'
return
;;
*)
break
;;
esac
done
command docker "${args[@]}"
return
}
alias docker=docker_fun
## SSH ##
alias forcessh='ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no'
alias ssh2="ssh -o ControlPath=none"
alias sshki="ssh2 -o PreferredAuthentications=keyboard-interactive"
alias scp2="scp -O -o ControlPath=none"
alias ssh3="ssh -o 'UserKnownHostsFile=/dev/null'"
alias ssh_check='ssh -O check'
alias ssh_close='ssh -O exit'
#alias ssh_forward='ssh -O forward'
#alias ssh_unforward='ssh -O cancel'
function ssh_check_socket()
{
local pid="$(ssh -S "${1}" -O check 127.0.0.0 2>&1)"
pid="${pid#*=}"
pid="${pid%)*}"
lsof -P -i -a -p "${pid}" 2>/dev/null
}
# Disable capture mouse clicks to terminal which tmux turn on, and gets left on if ssh connection disconnects.
alias mouse_reset="echo -n $'\x1b[?1000l'"
alias buffer_reset="echo -n $'\x1b[?1049l'" # fixes scroll bar and returns to original buffer
# More codes that tmux uses, but I don't see much additional benifit
# $'\033[?1000l\033[?1002l\033[?1003l'; echo $'\033[?1006l\033[?1005l'
# $'\033[?1007l' - Fixes scroll wheel scrolls bash history, but it doesn't renabled the scroll bar still
# $'\033[?1049l' - This will return to the normal buffer and enabled scroll bar again, but you won't be able to scroll in the alternative buffer
# TL;DR, just run `reset`
function ssh_close_all()
{
find ~/.ssh/ -maxdepth 1 -type s -regextype posix-egrep -regex '.*/[0-9a-f]{40}' -print0 | \
xargs -0 -I controlpath bash -c "ssh -O exit -S controlpath foobar || rm controlpath"
}
## GPG ##
export GPG_TTY="$(tty)"
alias gpg_load_key='gpg --sign -o /dev/null /dev/null'
alias gpg_reset_agent='gpg-connect-agent reloadagent /bye'
alias gpg_keyinfo='gpg-connect-agent "keyinfo --list" /bye'
## Process ##
function ptop(){
local PGREP_ARG=$1
shift
top -p$(pgrep $PGREP_ARG | tr '\n' , | sed 's/,$//') "${@}"
}
# NAME
# ppgrep - Pgrep that returns the parent ids instead
# BUGS
# Uses ps right justify, so extra spaces are included
function ppgrep()
{
local output="$(pgrep "${@}")"
if [ "${output}" != "" ]; then
ps -o'%P' --no-header ${output}
fi
}
function pps()
{
ps -f $(pgrep "${@}")
}
## Core ##
if [[ ! ${OSTYPE} =~ darwin.* ]]; then
alias ls='ls --color=auto'
alias lsd='ls --color=auto */ -lasd'
else
alias ls='ls -G'
alias lsd='ls -G */ -lasd'
fi
# This determines if you are using a file in order to decide if you
# should recursively grep or not. This logic is not perfect. It does not appear
# to detect --file, -f[filename], -f=filename, but will detect -f filename
function grep_fun() {
local orig=("${@}")
declare -i stdin=0
declare -i pattern=0
declare -i files=0
while [[ ${#} > 0 ]]; do
# Anything after -- is automatically a file
if [ "${files}" = 1 ]; then
files=2
fi
# Patterns that are expressions and take an extra arg
if [[ $1 =~ ^-[a-eg-zA-Z]*f$|^-[a-df-zA-Z]*e$ ||
$1 = --file ||
$1 = --regexp ]]; then
shift
pattern=1
# Same, arg included
elif [[ $1 =~ ^-[a-eg-zA-Z]*f|^-[a-df-zA-Z]*e ||
$1 = --file=* ||
$1 = --regexp=* ]]; then
pattern=1
# Everything that takes an arg
elif [[ $1 =~ ^-[a-ln-zA-Z]*m|^-[a-zB-Z]*A|^-[a-zAC-Z]*B|^-[a-zABD-Z]*C|^-[a-zA-CE-Z]*D|^-[a-ce-zA-Z]*d ||
$1 = --max-count ||
$1 = --after-context ||
$1 = --before-context ||
$1 = --context ||
$1 = --devices ||
$1 = --directories ||
$1 = --color ||
$1 = --label ||
$1 = --exclude ||
$1 = --exclude-from ||
$1 = --exclude-dir ||
$1 = --include ||
$1 = --group-separator ||
$1 = --binary-files ]]; then
shift
# Special case
elif [ "$1" = "--" ]; then
files+=1
# Anything else not starting with - is a pattern/file
elif [[ $1 != -* ]]; then
# If pattern already specified, it's a file
if [ "${pattern}" = "1" ]; then
files=2
else # else it WAS a pattern!
pattern=1
fi
fi
shift
done
if (( files + pattern < 3 )); then
\grep -in --color=always ${orig[@]+"${orig[@]}"} -
else
\grep -rin --color=always ${orig[@]+"${orig[@]}"}
fi
}
#alias grep="grep -rin --color=always"
alias grep="grep_fun"
# alias myeclipse="env GIT_SSH=/usr/bin/ssh /opt/software/eclipse/luna/eclipse"
alias rm="rm -i"
# json powered grep, output looks like "grep -Hn" a little even. Not to be confused
# with gron, which produces a greppable representastion of a json document
# Note: Line number is always the last line
# $1 - jq filter (but it's really part of an [] array. That means if you used a comma
# separated pattern, each one will be a column
# $2... - filenames to parse
function jrep()
{
local filter=$1