forked from colonelpanic/dotfiles
417 lines
11 KiB
Bash
417 lines
11 KiB
Bash
function path_lines {
|
||
target=${1-PATH}
|
||
if is_zsh; then
|
||
echo ${(P)target} | tr ':' '\n'
|
||
else
|
||
echo ${!target} | tr ':' '\n'
|
||
fi
|
||
}
|
||
|
||
function _add_to_front_of_path_lines {
|
||
target=${2-PATH}
|
||
echo $1
|
||
path_lines $target | grep -Ev "^$1$"
|
||
}
|
||
|
||
function _add_to_back_of_path_lines {
|
||
target=${2-PATH}
|
||
path_lines $target | grep -Ev "^$1$"
|
||
echo $1
|
||
}
|
||
|
||
function remove_trailing_colon {
|
||
sed 's|:*$||'
|
||
}
|
||
|
||
function add_to_front_of_path {
|
||
target=${2-PATH}
|
||
export $target="$(_add_to_front_of_path_lines $1 $target | tr '\n' ':' | remove_trailing_colon)"
|
||
}
|
||
|
||
function add_to_back_of_path {
|
||
target=${2-PATH}
|
||
export $target="$(_add_to_back_of_path_lines $1 $target | tr '\n' ':' | remove_trailing_colon)"
|
||
}
|
||
|
||
function split_into_vars () {
|
||
local string IFS
|
||
|
||
string="$1"
|
||
IFS="$2"
|
||
shift 2
|
||
read -r -- "$@" <<EOF
|
||
$string
|
||
EOF
|
||
}
|
||
|
||
function echo_split () {
|
||
local IFS
|
||
IFS="$2" read -rA -- arr <<EOF
|
||
$1
|
||
EOF
|
||
for i in "${arr[@]}"; do
|
||
echo $i
|
||
done
|
||
}
|
||
|
||
function shell_contains () {
|
||
local e
|
||
for e in "${@:2}"; do [[ "$e" == "$1" ]] && return 0; done
|
||
return 1
|
||
}
|
||
|
||
function dotfiles_directory() {
|
||
echo $(dirname `readlink -f ~/.zshrc | xargs dirname`)
|
||
}
|
||
|
||
function go2dotfiles() {
|
||
cd $(dotfiles_directory)
|
||
}
|
||
|
||
function update_dotfiles() {
|
||
local old_pwd=$(pwd)
|
||
go2dotfiles
|
||
git ffo
|
||
cd $old_pwd
|
||
}
|
||
|
||
function current_shell() {
|
||
which "$(ps -p $$ | tail -1 | awk '{print $NF}' | sed 's/\-//')"
|
||
}
|
||
|
||
function is_zsh() {
|
||
[ ! -z ${ZSH_VERSION+x} ]
|
||
}
|
||
|
||
function git_diff_add() {
|
||
git status --porcelain | awk '{print $2}' | xargs -I filename sh -c "git du filename && git add filename"
|
||
}
|
||
|
||
function confirm() {
|
||
# call with a prompt string or use a default
|
||
read -r -p "$1" response
|
||
case $response in
|
||
[yY][eE][sS]|[yY])
|
||
return 0
|
||
;;
|
||
*)
|
||
return 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
function get_cols() {
|
||
FS=' '
|
||
OPTIND=1
|
||
while getopts "F:" OPTCHAR; do
|
||
case $OPTCHAR in
|
||
F)
|
||
FS=$OPTARG
|
||
;;
|
||
esac
|
||
done
|
||
shift $((OPTIND-1))
|
||
awk -f "$HOME/.lib/get_cols.awk" -v "cols=$*" -v "FS=$FS"
|
||
}
|
||
|
||
function find_all_ssh_agent_sockets() {
|
||
find /tmp -type s -name agent.\* 2> /dev/null | grep '/tmp/ssh-.*/agent.*'
|
||
}
|
||
|
||
function set_ssh_agent_socket() {
|
||
export SSH_AUTH_SOCK=$(find_all_ssh_agent_sockets | tail -n 1 | awk -F: '{print $1}')
|
||
}
|
||
|
||
# Determine size of a file or total size of a directory
|
||
function fs() {
|
||
if du -b /dev/null > /dev/null 2>&1; then
|
||
local arg=-sbh
|
||
else
|
||
local arg=-sh
|
||
fi
|
||
if [[ -n "$@" ]]; then
|
||
du $arg -- "$@"
|
||
else
|
||
du $arg .[^.]* *
|
||
fi
|
||
}
|
||
|
||
# Start an HTTP server from a directory, optionally specifying the port
|
||
function server() {
|
||
local port="${1:-8000}"
|
||
sleep 1 && open "http://localhost:${port}/" &
|
||
# Set the default Content-Type to `text/plain` instead of `application/octet-stream`
|
||
# And serve everything as UTF-8 (although not technically correct, this doesn’t break anything for binary files)
|
||
python -c $'import SimpleHTTPServer;\nmap = SimpleHTTPServer.SimpleHTTPRequestHandler.extensions_map;\nmap[""] = "text/plain";\nfor key, value in map.items():\n\tmap[key] = value + ";charset=UTF-8";\nSimpleHTTPServer.test();' "$port"
|
||
}
|
||
|
||
# All the dig info
|
||
function digga() {
|
||
dig +nocmd "$1" any +multiline +noall +answer
|
||
}
|
||
|
||
function shell_stats() {
|
||
history 0 | awk '{CMD[$2]++;count++;}END { for (a in CMD)print CMD[a] " " CMD[a]/count*100 "% " a;}' | grep -v "./" | column -c3 -s " " -t | sort -nr | nl | head -n20
|
||
}
|
||
|
||
function is_ssh() {
|
||
test $SSH_CLIENT
|
||
}
|
||
|
||
function is_osx() {
|
||
case `uname` in
|
||
'Darwin')
|
||
return 0
|
||
;;
|
||
*)
|
||
return 1;
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# TODO: Remove this.
|
||
alias clipboard='oscopy'
|
||
|
||
function oscopy() {
|
||
if is_osx;
|
||
then
|
||
reattach-to-user-namespace pbcopy
|
||
else
|
||
test -n "$DISPLAY" && xclip -selection c
|
||
fi
|
||
}
|
||
|
||
function ospaste() {
|
||
if is_osx;
|
||
then
|
||
reattach-to-user-namespace pbpaste
|
||
else
|
||
xclip -o
|
||
fi
|
||
}
|
||
|
||
function git_root() {
|
||
cd `git root`
|
||
}
|
||
|
||
function git_diff_replacing() {
|
||
local original_sha='HEAD~1'
|
||
local new_sha='HEAD'
|
||
OPTIND=1
|
||
while getopts "do:n:" OPTCHAR;
|
||
do
|
||
case $OPTCHAR in
|
||
o)
|
||
original_sha="$OPTARG"
|
||
;;
|
||
n)
|
||
new_sha="$OPTARG"
|
||
;;
|
||
d)
|
||
debug="true"
|
||
esac
|
||
done
|
||
shift $((OPTIND-1))
|
||
local replaced="$1"
|
||
local replacing="$2"
|
||
local replace_sha_string='$(echo filename | sed '"s:$replaced:$replacing:g"')'
|
||
test -z $debug || echo "Diffing from $original_sha to $new_sha, replacing $replaced with $replacing"
|
||
test -z $debug || git diff $original_sha $new_sha --name-only | grep -v "$replacing"
|
||
git diff $original_sha $new_sha --name-only | grep -v "$replacing" | xargs -I filename sh -c "git diff $original_sha:filename $new_sha:"$replace_sha_string
|
||
}
|
||
|
||
function git_reset_author() {
|
||
local should_update_command=''
|
||
local update_command=''
|
||
OPTIND=1
|
||
while getopts "a:e:A:E:h" OPTCHAR;
|
||
do
|
||
case $OPTCHAR in
|
||
a)
|
||
new_author="$OPTARG";
|
||
test -n "$update_command" && update_command="$update_command"' && '
|
||
update_command="$update_command"'export GIT_AUTHOR_NAME='"'$new_author'"' && export GIT_COMMITTER_NAME='"'$new_author'"
|
||
;;
|
||
A)
|
||
author_regex="$OPTARG";
|
||
test -n "$should_update_command" && should_update_command="$should_update_command"' && '
|
||
should_update_command=$should_update_command'[[ "$GIT_AUTHOR_NAME" =~ "'"$author_regex"'" ]]'
|
||
;;
|
||
e)
|
||
new_email="$OPTARG";
|
||
test -n "$update_command" && update_command="$update_command"' && '
|
||
update_command="$update_command"'export GIT_AUTHOR_EMAIL='"'$new_email'"' && export GIT_COMMITTER_EMAIL='"'$new_email'"
|
||
;;
|
||
E)
|
||
email_regex="$OPTARG";
|
||
test -n "$should_update_command" && should_update_command="$should_update_command"' && '
|
||
should_update_command=$should_update_command'[[ "$GIT_AUTHOR_EMAIL" =~ "'"$email_regex"'" ]]'
|
||
;;
|
||
h)
|
||
echo "Usage:
|
||
-a specify the new author/committer name.
|
||
-A specify a regex that will be used to filter commits by author name.
|
||
-e specify the new author/committer email.
|
||
-E specify a regex that will be used to filter commits by author email.
|
||
-h show this help message.
|
||
"
|
||
return
|
||
;;
|
||
esac
|
||
done
|
||
local filter_branch_command="$should_update_command"' && '"$update_command"' || test true'
|
||
git filter-branch -f --env-filter $filter_branch_command -- --all
|
||
}
|
||
|
||
alias git_reset_author_to_user='git_reset_author -a "$(git config --get user.name)" -e "$(git config --get user.email)" '
|
||
|
||
function set_osx_hostname() {
|
||
local new_hostname="${1-imalison}"
|
||
sudo scutil --set ComputerName $new_hostname
|
||
sudo scutil --set HostName $new_hostname
|
||
sudo scutil --set LocalHostName $new_hostname
|
||
sudo defaults write /Library/Preferences/SystemConfiguration/com.apple.smb.server NetBIOSName -string $new_hostname
|
||
}
|
||
|
||
function pip_package_location() {
|
||
pip show $1 | grep Location | get_cols 2
|
||
}
|
||
|
||
function set_modifier_keys_on_all_keyboards() {
|
||
for vendor_product_id in $(get_keyboard_vendor_id_product_id_pairs | tr " " "-"); do
|
||
set_modifier_keys_for_vendor_product_id $vendor_product_id 0 2; echo $vendor_product_id;
|
||
done;
|
||
}
|
||
|
||
function get_keyboard_vendor_id_product_id_pairs() {
|
||
ioreg -n IOHIDKeyboard -r | grep -e 'class IOHIDKeyboard' -e VendorID\" -e Product | gawk 'BEGIN { RS = "class IOHIDKeyboard" } match($0, /VendorID. = ([0-9]*)/, arr) { printf arr[1]} match($0, /ProductID. = ([0-9]*)/, arr) { printf " %s\n", arr[1]} '
|
||
}
|
||
|
||
function git_config_string() {
|
||
git config -f $1 --list | xargs -I kv printf '-c \"%s\" ' kv
|
||
}
|
||
|
||
function talk_dirty_to_me() {
|
||
python - <<EOF
|
||
from random import randrange
|
||
import re
|
||
import urllib
|
||
|
||
def talk_dirty_to_me():
|
||
socket = urllib.urlopen("http://www.youporn.com/random/video/")
|
||
htmlSource = socket.read()
|
||
socket.close()
|
||
result = re.findall('<p class="message">((?:.|\\n)*?)</p>', htmlSource)
|
||
if len(result):
|
||
print result[randrange(len(result))]
|
||
else:
|
||
talk_dirty_to_me()
|
||
|
||
talk_dirty_to_me()
|
||
EOF
|
||
}
|
||
|
||
function dirty_talk() {
|
||
while true; do talk_dirty_to_me | tee >(cat) | say; done
|
||
}
|
||
|
||
function track_modified {
|
||
local timestamp_file="/tmp/__track_modified_timestamp__"
|
||
touch $timestamp_file
|
||
stat $timestamp_file
|
||
echo "Press any key to execute find command"
|
||
read -r key
|
||
echo "Finding..."
|
||
find $1 -cnewer "$timestamp_file"
|
||
}
|
||
|
||
function python_module_path {
|
||
python -c "import os, $1; print(os.path.dirname($1.__file__))"
|
||
}
|
||
|
||
function mu4e_directory {
|
||
if is_osx; then
|
||
echo "$(brew --prefix mu)/share/emacs/site-lisp/mu4e"
|
||
else
|
||
# TODO: make this cleaner.
|
||
echo "/usr/share/emacs/site-lisp/mu4e"
|
||
fi
|
||
}
|
||
|
||
function timestamp {
|
||
date +%s
|
||
}
|
||
|
||
function parse_timestamp {
|
||
date -d "@$(echo $1 | cut -c -10)"
|
||
}
|
||
|
||
function refresh_config {
|
||
source ~/.zshenv
|
||
source ~/.zshrc
|
||
}
|
||
|
||
function file_ends_with_newline {
|
||
[[ $(tail -c1 "$1" | wc -l) -gt 0 ]]
|
||
}
|
||
|
||
function add_authorized_key_to_host {
|
||
local command='test -e ~/.ssh/authorized_keys && [[ $(tail -c1 ~/.ssh/authorized_keys | wc -l) -gt 0 ]] || echo "\n" >> ~/.ssh/authorized_keys;'"echo $(cat ~/.ssh/id_rsa.pub) >> ~/.ssh/authorized_keys"
|
||
echo "Running:"
|
||
echo $command
|
||
ssh $1 "$command"
|
||
}
|
||
|
||
function add_ssh_key {
|
||
[[ $(tail -c1 ~/.ssh/authorized_keys | wc -l) -gt 0 ]] || echo "\n" >> ~/.ssh/authorized_keys;
|
||
echo $1 >> ~/.ssh/authorized_keys;
|
||
}
|
||
|
||
function git_free_ssh_rsync {
|
||
echo $1
|
||
rsync -avz -e "ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null" --progress --exclude=".git/" $(readlink -f "$1") $2:$1
|
||
}
|
||
|
||
function project_sync {
|
||
git_free_ssh_rsync '~/Projects/'"$1"'/' $2
|
||
}
|
||
|
||
function android_sdk_directory {
|
||
if is_osx; then
|
||
brew --prefix android-sdk
|
||
fi
|
||
}
|
||
|
||
function pkill_zsh {
|
||
ps aux | grep "$1" | grep -v grep | get_cols 2 | xargs kill -9
|
||
}
|
||
|
||
|
||
function find_by_size {
|
||
find . -type f -size +$1
|
||
}
|
||
|
||
function get_git_project_name {
|
||
# "$(basename $(git rev-parse --show-toplevel))"
|
||
basename $(git remotes | get_cols 2)
|
||
}
|
||
|
||
function add_github_remote {
|
||
local project_name="$(get_git_project_name)"
|
||
git remote add "$1" "git@github.com:$1/$project_name"
|
||
}
|
||
|
||
function define_jump_alias_with_completion {
|
||
eval "alias $1='jump_cd $2'"
|
||
compdef "_files -g '$2/*'" $1
|
||
}
|
||
|
||
function jump_cd {
|
||
if [[ $2 =~ ^/ ]]; then
|
||
cd $2
|
||
else
|
||
cd $1
|
||
cd $2
|
||
fi
|
||
}
|