.functions

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

# Change working directory to the top-most Finder window location
function cdf() { # short for `cdfinder`
  cd "$(osascript -e 'tell app "Finder" to POSIX path of (insertion location as alias)')";
}


# 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')";
}

# Run `dig` and display the most useful info
function digga() {
  dig +nocmd "$1" any +multiline +noall +answer;
}

# UTF-8-encode a string of Unicode symbols
function escape() {
  printf "\\\x%s" $(printf "$@" | xxd -p -c1 -u);
  # print a newline unless we’re piping the output to another program
  if [ -t 1 ]; then
    echo ""; # newline
  fi;
}

# Decode \x{ABCD}-style Unicode escape sequences
function unidecode() {
  perl -e "binmode(STDOUT, ':utf8'); print \"$@\"";
  # print a newline unless we’re piping the output to another program
  if [ -t 1 ]; then
    echo ""; # newline
  fi;
}

# Get a character’s Unicode code point
function codepoint() {
  perl -e "use utf8; print sprintf('U+%04X', ord(\"$@\"))";
  # print a newline unless we’re piping the output to another program
  if [ -t 1 ]; then
    echo ""; # newline
  fi;
}

# `vs` with no arguments opens the current directory in Visual Studio Code, otherwise
# opens the given location
function vs() {
  if [ $# -eq 0 ]; then
    code .;
  else
    code "$@";
  fi;
}

# `z` with no arguments opens the current directory in Zed, otherwise
# opens the given location
function z() {
  if [ $# -eq 0 ]; then
    zed .;
  else
    zed "$@";
  fi;
}

# `v` with no arguments opens the current directory in Vim, otherwise opens the
# given location
function v() {
  if [ $# -eq 0 ]; then
    vim .;
  else
    vim "$@";
  fi;
}

# `o` with no arguments opens the current directory, otherwise opens the given
# location
function o() {
  if [ $# -eq 0 ]; then
    open .;
  else
    open "$@";
  fi;
}

# `tre` is a shorthand for `tree` with hidden files and color enabled, ignoring
# the `.git` directory, listing directories first. The output gets piped into
# `less` with options to preserve color and line numbers, unless the output is
# small enough for one screen.
function tre() {
  tree -aC -I '.git|node_modules|bower_components' --dirsfirst "$@" | less -FRNX;
}

# Toggles dark mode
function dark() {
  osascript -e 'tell application "System Events" to tell appearance preferences to set dark mode to not dark mode'
}

# Dependabot merge all PRs
function dependabot-merge-all() {
  for i in $(gh pr list --limit 100 --json number --json author --jq '.[] | select(.author.login == "app/dependabot") | .number'); do
    echo "Merging PR #$i"
    gh pr review $i --approve
    gh pr merge $i --squash --delete-branch

    # Capture the exit status immediately after the merge command
    merge_status=$?

    if [ $merge_status -ne 0 ]; then
      gh pr comment $i --body "@dependabot recreate"
      echo "Failed to merge PR #$i, will recreate"
    else
      echo "Merged PR #$i"
    fi
  done
}

# Docker machine
machine () {
    DOCKER_MACHINE_NAME="$1"
    if [ -z "$DOCKER_MACHINE_NAME" ]
        then DOCKER_MACHINE_NAME=default
    fi
    eval `docker-machine env $DOCKER_MACHINE_NAME`
}
create-machine () {
    DOCKER_MACHINE_NAME="$1"
    if [ -z "$DOCKER_MACHINE_NAME" ]
        then DOCKER_MACHINE_NAME=default
    fi

    if [ ! -f ~/.ssh/config ]; then
      touch ~/.ssh/config
      chmod 600 ~/.ssh/config
    fi

    if ! grep -q "StrictHostKeyChecking no" ~/.ssh/config; then
      echo "MATCH exec \"test -d ~/.docker/machine/machines/%h\"
        StrictHostKeyChecking no
        UserKnownHostsFile /dev/null
        IdentityFile ~/.docker/machine/machines/%h/id_rsa
        User root" >> ~/.ssh/config
    fi

    sudo -v # login now for later use
    docker-machine create $DOCKER_MACHINE_NAME
    eval `docker-machine env $DOCKER_MACHINE_NAME`
    sudo hosts remove --force $DOCKER_MACHINE_NAME
    sudo hosts add `docker-machine ip $DOCKER_MACHINE_NAME` \
        $DOCKER_MACHINE_NAME "docker machine host"
}
sync-machine () {
    if [ ! -z "$1" ]; then machine "$1"; fi
    ssh $DOCKER_MACHINE_NAME "mkdir -p $PWD && chmod 777 $PWD"
    # add additional commands for your setup
    mutagen sync create --sync-mode two-way-resolved \
        --label "machine=$DOCKER_MACHINE_NAME" \
        --name "$(basename $PWD)" \
        --ignore-vcs \
        $PWD $DOCKER_MACHINE_NAME:$PWD
}
drop-machine () {
    if [ ! -z "$1" ]; then machine "$1"; fi
    sudo -v # login now for later use
    docker-machine rm $DOCKER_MACHINE_NAME
    mutagen sync terminate --label-selector "machine=$DOCKER_MACHINE_NAME"
    sudo hosts remove --force $DOCKER_MACHINE_NAME
}
create-machine-here () {
    create-machine "$1"
    sync-machine
}