mirror of
https://github.com/robbyrussell/oh-my-zsh.git
synced 2026-01-07 11:04:46 +01:00
#new-plugin add a new plugin "vault-switch"
This commit is contained in:
57
plugins/vault-switch/vault-switch.plugin.zsh
Normal file
57
plugins/vault-switch/vault-switch.plugin.zsh
Normal file
@@ -0,0 +1,57 @@
|
||||
autoload -U add-zsh-hook
|
||||
add-zsh-hook precmd _restore_cache
|
||||
|
||||
FILE_CREDENTIALS="${HOME}/.vault-switch/credentials"
|
||||
|
||||
function _restore_cache(){
|
||||
[ ! -d ${HOME}/.vault-switch ] && mkdir -p ${HOME}/.vault-switch
|
||||
[ ! -f $FILE_CREDENTIALS ] && touch ${FILE_CREDENTIALS}
|
||||
source ${FILE_CREDENTIALS}
|
||||
}
|
||||
|
||||
function _get-nodes(){
|
||||
IFS=";" read -A NODES <<< ${VAULT_NODES}
|
||||
}
|
||||
|
||||
function _set-color(){
|
||||
echo "\e[1;32m$1\e[0m"
|
||||
}
|
||||
|
||||
function _list-nodes(){
|
||||
INDEX=1
|
||||
for i in ${NODES[@]}
|
||||
do
|
||||
NODE=$(echo $i | cut -d "," -f 1)
|
||||
[[ "${NODE}" == "${VAULT_SELECT_NODE}" ]] && ASTERISK="*"
|
||||
echo "${INDEX}) ${NODE} $(_set-color ${ASTERISK})"
|
||||
INDEX=$[$INDEX+1]
|
||||
unset ASTERISK
|
||||
done
|
||||
}
|
||||
|
||||
function _set-work-node(){
|
||||
if [ $1 -gt ${#NODES[@]} ]
|
||||
then
|
||||
echo "Number of node not found"
|
||||
else
|
||||
VAULT_SELECT_NODE=$(echo ${NODES[$1]} | cut -d "," -f 1)
|
||||
VAULT_ADDR=$(echo ${NODES[$1]} | cut -d "," -f 2)
|
||||
VAULT_TOKEN=$(echo ${NODES[$1]} | cut -d "," -f 3)
|
||||
VAULT_SKIP_VERIFY=$(echo ${NODES[$1]} | cut -d "," -f 4)
|
||||
|
||||
echo > ${FILE_CREDENTIALS}
|
||||
echo "export VAULT_SELECT_NODE=${VAULT_SELECT_NODE}" >> ${FILE_CREDENTIALS}
|
||||
echo "export VAULT_ADDR=${VAULT_ADDR}" >> ${FILE_CREDENTIALS}
|
||||
echo "export VAULT_TOKEN=${VAULT_TOKEN}" >> ${FILE_CREDENTIALS}
|
||||
[[ $VAULT_SKIP_VERIFY ]] && echo "export VAULT_SKIP_VERIFY=true" >> ${FILE_CREDENTIALS}
|
||||
|
||||
_list-nodes
|
||||
fi
|
||||
|
||||
}
|
||||
|
||||
function vault-switch() {
|
||||
_get-nodes
|
||||
[ ! $1 ] && _list-nodes
|
||||
[ $1 ] && _set-work-node $1
|
||||
}
|
||||
Reference in New Issue
Block a user