.functions

# vim:ft=sh

ios_runtime() {
    xcrun simctl list -j runtimes ios | jq -r '.[] | sort_by(.identifier)[0].identifier'
}

ios_devices() {
    xcrun simctl list -j devices ios | jq ".devices.\"$(ios_runtime)\"" | jq 'sort_by(.deviceTypeIdentifier)'
}

ios_devices_booted() {
    ios_devices | jq '[.[] | select(.state == "Booted")]'
}

ios_devices_shutdown() {
    ios_devices | jq '[.[] | select(.state == "Shutdown")]'
}

ios_devices_get_id() {
    booted=$(ios_devices_booted)
    if [ "$booted" != "[]" ]; then
        echo $booted | jq -r '.[0].udid'
    else
        device_id=$(ios_devices_shutdown | jq -r '.[0].udid')
        xcrun simctl boot $device_id
        echo $device_id
    fi
}


journal_pass() {
    pass edit Journal/$(date +%Y/%m/%d)/entry.txt
}


fix_bluetooth() {
    pids="$( ps -A | grep -iE '(blued|coreaudiod|bluetoothaudiod)' | grep -v grep | cut -c 1-90 | cut -c 1-6 )"
    echo sudo renice -10 $pids
    sudo renice -10 $pids

}
update-rust-analyzer() {
    pushd .
    cd ~/Downloads/
    gh release -R rust-lang/rust-analyzer download -p '*-aarch64-apple-darwin.gz'
    gunzip rust-analyzer-aarch64-apple-darwin.gz
    chmod +x rust-analyzer-aarch64-apple-darwin
    ./rust-analyzer-aarch64-apple-darwin --version
    mv ./rust-analyzer-aarch64-apple-darwin ~/.cargo/bin/rust-analyzer
    popd
}

random_words() {
    WORDFILE="/usr/share/dict/words"
    NUMWORDS=$1
    #Number of lines in $WORDFILE
    tL=`awk 'NF!=0 {++c} END {print c}' $WORDFILE`
    for i in `seq $NUMWORDS`
    do
        rnum=$((RANDOM%$tL+1))
        echo -n $(sed -n "$rnum p" $WORDFILE)' '
    done
}

ssh_tar() {
    tar -pcz $1 | ssh $2 'tar -xz'
}

ssh_untar() {
    ssh $1 "tar -pcz $2" | tar -pxz
}

youtube_play() {
    local_video_file=/tmp/video.flv
    fifo_loc=/tmp/fifo
    rm $local_video_file

    if [[ ! -p $fifo_loc ]]; then
        mkfifo $fifo_loc
    fi

    youtube-dl --no-part --max-quality 5 -o $local_video_file $1 > $fifo_loc & disown

    counter=0

    while ! grep '\[download\]' $fifo_loc; do

        sleep 1
        counter=$(($counter + 1))

        if [ $counter -ge 10 ]; then
            exit 1
        fi
    done

    mplayer -vo null $local_video_file
    rm $fifo_loc
}

histogram() {
    if [ $# -lt 1 ]
    then
        echo "Usage: histogram  [field] [precision]"
        return 0
    fi

    field=1
    if [ $# -gt 1 ]
    then
        field=$2
    fi

    if [ $# -gt 2 ]
    then
        precision="."
        for i in `seq 1 1 $3`
        do
            precision=$precision.
        done

        cat $1 | gsort -k $field -n -t ',' | cut -d ',' -f $field | gsed -e "s/\($precision\).*/\1/" | uniq -c
    else
        cat $1 | sort -k $field -n -t ',' | cut -d ',' -f $field | uniq -c
    fi
}

cdn() {
    num_up=$(echo $1 | grep -o "\." | wc -l)
    updot="./"
    for ((i=0;i<$((${num_up}-1));i++))
    do
        updot="$updot../"
    done
    cd $updot
}

get_cols () {
    # Usage: get_cols [-F$separator] [file_name] [column_num]
    # separator
    separator=' '
    column_list=""
    for column_num in $@; do
        [ -r $column_num ] && continue

        if echo $column_num | grep '\-F' ; then
            separator=$(echo $column_num | sed 's/-F//g' )
            continue
        fi

        [ $column_num -lt '0' ] && column_num="(NF + 1 $column_num )" #Negative indices like python's array[-1]

        [ -z $column_list ] && column_num="\$$column_num" || column_num=",\$$column_num" # Place commas appropriately.

        column_list="$column_list$column_num"
    done
    unset column_num

    # Is the first parameter a file?
    awk_string="BEGIN { FS = \"$separator\" } ; {print $column_list}"

    if [ -r $1 ]; then
        awk $awk_string $1
    else
        awk "$awk_string"
    fi

    unset column_list
    unset awk_string
}

aggregate () {
    awk "{sum += \$1} END { printf \"%d\n\", sum }"
}

git_modified () {
	git status | grep modified | get_cols -1
}

get_col () {
	for item in `awk "{print \\$$@}"`; do
		printf $item
		printf '\n'
	done
}

get_line() {
	if [ -t 0 ]; then
		head -n $1 | tail -n 1
	else
		head -n $1 $2 | tail -n 1
	fi
}

du_sorted () {
    du "$@" | sort -n | get_col 2 | xargs du -hs
}

function git_checkout_last() {
	git checkout @{-1}
}

function average () {
	if [ '0' -eq $# ]
	then
		awk '{print $1}' | awk '{s+=$1} END{print "Sum: "s, "\nNumber of lines : "NR, "\nAverage: "s/(NR)}'
	else
		awk '{print $1}' | awk '{s+=$1} END{print "Sum: "s, "\nNumber of lines : "NR, "\nAverage: "s/(NR)}' $1
	fi
}

function get_shas_that_touched() {
    git log --oneline --reverse "$@" | get_col 1
}

function echo_hash_if_exp_found_in_filename_diff() {
    [ ! -z "$(git diff $1~1 $1 -- $2 | cat | grep $3)" ] && echo $1
}

function get_shas_that_touched_with_grep() {
    get_shas_that_touched $1 | xargs -I the_hash sh -c 'echo_hash_if_exp_found_in_filename_diff the_hash $1 $2'
}

function show_interdiffs_matching_grep() {
    get_shas_that_touched_with_grep $1 $2 | xargs -I hash sh -c 'git diff hash~1 hash -- $1'
}

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}')
}

# Create a new directory and enter it
function mkd() {
    mkdir -p "$@" && cd "$@"
}

# 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
}

# Use Git’s colored diff when available
hash git &>/dev/null
if [ $? -eq 0 ]; then
    function diff() {
    git diff --no-index --color-words "$@"
    }
fi

# Create a data URL from a file
function dataurl() {
    local mimeType=$(file -b --mime-type "$1")
    if [[ $mimeType == text/* ]]; then
        mimeType="${mimeType};charset=utf-8"
    fi
    echo "data:${mimeType};base64,$(openssl base64 -in "$1" | tr -d '\n')"
}

# 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)
    python3 -m http.server "$port"
}

# Start a PHP server from a directory, optionally specifying the port
# (Requires PHP 5.4.0+.)
function phpserver() {
    local port="${1:-4000}"
    local ip=$(ipconfig getifaddr en1)
    sleep 1 && open "http://${ip}:${port}/" &
    php -S "${ip}:${port}"
}

# Compare original and gzipped file size
function gz() {
    local origsize=$(wc -c < "$1")
    local gzipsize=$(gzip -c "$1" | wc -c)
    local ratio=$(echo "$gzipsize * 100/ $origsize" | bc -l)
    printf "orig: %d bytes\n" "$origsize"
    printf "gzip: %d bytes (%2.2f%%)\n" "$gzipsize" "$ratio"
}

# Test if HTTP compression (RFC 2616 + SDCH) is enabled for a given URL.
# Send a fake UA string for sites that sniff it instead of using the Accept-Encoding header. (Looking at you, ajax.googleapis.com!)
function httpcompression() {
    encoding="$(curl -LIs -H 'User-Agent: Mozilla/5 Gecko' -H 'Accept-Encoding: gzip,deflate,compress,sdch' "$1" | grep '^Content-Encoding:')" && echo "$1 is encoded using ${encoding#* }" || echo "$1 is not using any encoding"
}

# Syntax-highlight JSON strings or files
# Usage: `json '{"foo":42}'` or `echo '{"foo":42}' | json`
function json() {
    if [ -t 0 ]; then # argument
        python -mjson.tool <<< "$*" | pygmentize -l javascript
    else # pipe
        python -mjson.tool | pygmentize -l javascript
    fi
}

# All the dig info
function digga() {
    dig +nocmd "$1" any +multiline +noall +answer
}

# Escape UTF-8 characters into their 3-byte format
function escape() {
    printf "\\\x%s" $(printf "$@" | xxd -p -c1 -u)
    echo # newline
}

# Decode \x{ABCD}-style Unicode escape sequences
function unidecode() {
    perl -e "binmode(STDOUT, ':utf8'); print \"$@\""
    echo # newline
}

# Get a character’s Unicode code point
function codepoint() {
    perl -e "use utf8; print sprintf('U+%04X', ord(\"$@\"))"
    echo # newline
}

# Add note to Notes.app (OS X 10.8)
# Usage: `note 'foo'` or `echo 'foo' | note`
function note() {
    local text
    if [ -t 0 ]; then # argument
	text="$1"
    else # pipe
	text=$(cat)
    fi
    body=$(echo "$text" | sed -E 's|$|<br>|g')
    osascript >/dev/null <<EOF
tell application "Notes"
	tell account "iCloud"
		tell folder "Notes"
			make new note with properties {name:"$text", body:"$body"}
		end tell
	end tell
end tell
EOF
}

# Add reminder to Reminders.app (OS X 10.8)
# Usage: `remind 'foo'` or `echo 'foo' | remind`
function remind() {
    local text
    if [ -t 0 ]; then
	text="$1" # argument
    else
	text=$(cat) # pipe
    fi
    osascript >/dev/null <<EOF
tell application "Reminders"
	tell the default list
		make new reminder with properties {name:"$text"}
	end tell
end tell
EOF
}

# Manually remove a downloaded app or file from the quarantine
function unquarantine() {
    for attribute in com.apple.metadata:kMDItemDownloadedDate com.apple.metadata:kMDItemWhereFroms com.apple.quarantine; do
	xattr -r -d "$attribute" "$@"
    done
}