ohmyzsh/plugins/history-sync/history-sync.plugin.zsh
2016-04-19 21:10:13 +10:00

133 lines
3.1 KiB
Bash

###
# James Fraser
# <wulfgar.pro@gmail.com>
###
autoload -U colors
colors
ZSH_HISTORY_FILE=$HOME/.zsh_history
ZSH_HISTORY_PROJ=$HOME/.zsh_history_proj
ZSH_HISTORY_FILE_ENC=$ZSH_HISTORY_PROJ/zsh_history
GIT_COMMIT_MSG="latest $(date)"
function print_git_error_msg() {
echo "$bold_color$fg[red]Fix your git repo...${reset_color}";
}
function print_gpg_encrypt_error_msg() {
echo "$bold_color$fg[red]GPG failed to encrypt history file... exiting.${reset_color}";
}
function print_gpg_decrypt_error_msg() {
echo "$bold_color$fg[red]GPG failed to decrypt history file... exiting.${reset_color}";
}
function usage() {
echo "$bold_color$fg[red]Usage: $0 [-r <string> -r <string>...]${reset_color}" 1>&2; return;
}
# Pull current master, decrypt, and merge with .zsh_history
function history_sync_pull() {
# Backup
cp -a $HOME/{.zsh_history,.zsh_history.backup}
DIR=$CWD
# Pull
cd $ZSH_HISTORY_PROJ && git pull
if [[ $? != 0 ]]; then
print_git_error_msg
cd $DIR
return
fi
# Decrypt
gpg --output zsh_history_decrypted --decrypt zsh_history
if [[ $? != 0 ]]; then
print_gpg_decrypt_error_msg
cd $DIR
return
fi
# Merge
cat $HOME/.zsh_history zsh_history_decrypted | sort -u > $HOME/.zsh_history
rm zsh_history_decrypted
cd $DIR
}
# Encrypt and push current history to master
function history_sync_push() {
# Get option recipients
local recipients=()
while getopts -r: opt; do
case "$opt" in
r)
recipients+="$OPTARG"
;;
*)
usage
return
;;
esac
done
# Encrypt
if ! [[ ${#recipients[@]} > 0 ]]; then
echo -n "Please enter GPG recipient name: "
read name
recipients+=$name
fi
ENCRYPT_CMD="gpg -v "
for r in $recipients; do
ENCRYPT_CMD+="-r \"$r\" "
done
if [[ $ENCRYPT_CMD =~ '.(-r).+.' ]]; then
ENCRYPT_CMD+="--encrypt --sign --armor --output $ZSH_HISTORY_FILE_ENC $ZSH_HISTORY_FILE"
eval ${ENCRYPT_CMD}
if [[ $? != 0 ]]; then
print_gpg_encrypt_error_msg
return
fi
echo -n "$bold_color$fg[yellow]Do you want to commit current local history file? ${reset_color}"
read commit
if [[ -n $commit ]]; then
case $commit in
[Yy]* )
DIR=$CWD
cd $ZSH_HISTORY_PROJ && git add * && git commit -am $GIT_COMMIT_MSG
echo -n "$bold_color$fg[yellow]Do you want to push to remote? ${reset_color}"
read push
if [[ -n $push ]]; then
case $push in
[Yy]* )
git push
if [[ $? != 0 ]]; then
print_git_error_msg
cd $DIR
return
fi
cd $DIR
;;
esac
fi
if [[ $? != 0 ]]; then
print_git_error_msg
cd $DIR
return
fi
;;
[Nn]* )
;;
* )
;;
esac
fi
fi
}
alias zhpl=history_sync_pull
alias zhps=history_sync_push
alias zhsync="history_sync_pull && history_sync_push"