customFunctions

# vim: ft=sh

function extract() {
    if [[ -f "$1" ]]; then
        case "$1" in
            *.tar.bz2)   tar xvjf "$1"     ;;
            *.tar.gz)    tar xvzf "$1"     ;;
            *.bz2)       bunzip2 "$1"      ;;
            *.rar)       unrar x "$1"      ;;
            *.gz)        gunzip "$1"       ;;
            *.tar)       tar xvf "$1"      ;;
            *.tbz2)      tar xvjf "$1"     ;;
            *.tgz)       tar xvzf "$1"     ;;
            *.zip)       unzip "$1"        ;;
            *.Z)         uncompress "$1"   ;;
            *.7z)        7z x "$1"         ;;
            *.tar.xz)    tar xvJf "$1"     ;;
            *.xz)        unxz "$1"         ;;
            *.tar.zst)   tar --zstd -xvf "$1" ;;
            *.zst)       unzstd "$1"       ;;
            *)           echo "'$1' cannot be extracted via extract" ;;
        esac
    else
        echo "'$1' is not a valid file!"
    fi
}

function maketar() { tar cvzf "${1%%/}.tar.gz"  "${1%%/}/"; }
function makezip() { zip -r "${1%%/}.zip" "$1" ; }

function cheat() { curl "cht.sh/$1"; }

# Brew helpers with fzf
bip() {
  local inst=$(brew search | fzf -m)
  [[ $inst ]] && for prog in $(echo $inst); do brew install $prog; done
}

bup() {
  local upd=$(brew leaves | fzf -m)
  [[ $upd ]] && for prog in $(echo $upd); do brew upgrade $prog; done
}

bcp() {
  local uninst=$(brew leaves | fzf -m)
  [[ $uninst ]] && for prog in $(echo $uninst); do brew uninstall $prog; done
}

# Misc
mcd() { mkdir -p "$1" && cd "$1"; }
cdl() { cd "$1" && ls; }
backup() { cp "$1"{,.bak}; }
lfind() { find . -iname "$@" 2>/dev/null; }
topcmd() { history | awk '{a[$2]++}END{for(i in a){print a[i] " " i}}' | sort -rn | head; }

# Fuzzy tmux session switcher
ts() {
    local session
    session=$(tmux list-sessions -F "#{session_name}" 2>/dev/null | fzf --height 40% --reverse)
    if [[ -n "$session" ]]; then
        tmux switch-client -t "$session" 2>/dev/null || tmux attach -t "$session"
    fi
}

# Git worktree helpers
gwa() {
  if [[ -z $1 ]]; then
    echo "Usage: gwa [branch name]"
    return 1
  fi

  local branch="$1"
  local base="$(basename "$PWD")"
  local wt_path="../${base}--${branch}"

  local existing_wt=$(git worktree list --porcelain | grep -B2 "branch refs/heads/$branch$" | grep "^worktree " | cut -d' ' -f2-)
  if [[ -n "$existing_wt" ]]; then
    if [[ -d "$existing_wt" ]]; then
      echo "Branch already in worktree: $existing_wt"
      cd "$existing_wt"
      return 0
    else
      echo "Pruning stale worktree: $existing_wt"
      git worktree prune
    fi
  fi

  if git show-ref --verify --quiet "refs/heads/$branch"; then
    git worktree add "$wt_path" "$branch"
  else
    git worktree add -b "$branch" "$wt_path"
  fi
  command -v mise &>/dev/null && mise trust "$wt_path"
  cd "$wt_path"
}

gwd() {
  command -v gum &>/dev/null || { echo "gum not installed"; return 1; }
  if gum confirm "Remove worktree and branch?"; then
    local cwd base branch root
    cwd="$(pwd)"
    worktree="$(basename "$cwd")"
    root="${worktree%%--*}"
    branch="${worktree#*--}"
    if [[ $root != $worktree ]]; then
      cd "../$root"
      git worktree remove "$worktree" --force
      git branch -D "$branch"
    fi
  fi
}

# AI-powered git commit
gcm() {
    command -v llm &>/dev/null || { echo "llm not installed"; return 1; }

    generate_commit_message() {
        git diff --cached | llm -m "4o" "
Below is a diff of all staged changes, coming from the command:
\`\`\`
git diff --cached
\`\`\`
Please generate a concise, one-line commit message for these changes using the semantic commit convention prefixes: chore:, fix:, feat:, refactor:, docs:, BREAKING:, etc."
    }

    read_input() {
        if [ -n "$ZSH_VERSION" ]; then
            echo -n "$1"
            read -r REPLY
        else
            read -p "$1" -r REPLY
        fi
    }

    echo "Generating AI-powered commit message..."
    commit_message=$(generate_commit_message)

    while true; do
        echo -e "\nProposed commit message:"
        echo "$commit_message"
        read_input "Do you want to (a)ccept, (e)dit, (r)egenerate, or (c)ancel? "
        choice=$REPLY
        case "$choice" in
            a|A ) git commit -m "$commit_message" && echo "Committed!" && return 0 || return 1 ;;
            e|E ) read_input "Enter your commit message: "; commit_message=$REPLY
                  [ -n "$commit_message" ] && git commit -m "$commit_message" && return 0 || return 1 ;;
            r|R ) echo "Regenerating..."; commit_message=$(generate_commit_message) ;;
            c|C ) echo "Cancelled."; return 1 ;;
            * ) echo "Invalid choice." ;;
        esac
    done
}