zz.sh

#!/usr/bin/env bash
set -euo pipefail

# zz - flat workspace + zellij with fuzzy finder

#=============================================================================
# Configuration
#=============================================================================

# Workspace directory (repos are direct children)
ZZ_WORKSPACE="$HOME/workspace"

#=============================================================================
# Utilities
#=============================================================================

die() {
    echo "Error: $1" >&2
    exit 1
}

# List all repo paths (absolute), excluding git worktrees
list_repos() {
    local dir
    for dir in "$ZZ_WORKSPACE"/*/; do
        [[ -d "$dir" ]] || continue
        dir="${dir%/}"
        # worktrees have a .git file, not a .git directory
        [[ -d "$dir/.git" || ! -e "$dir/.git" ]] && echo "$dir"
    done
}

# List repos that have active zz sessions
list_session_repos() {
    local sessions
    sessions=$(list_zz_sessions)
    [[ -z "$sessions" ]] && return 1
    while read -r s; do
        lookup_path_from_session "$s" 2>/dev/null
    done <<< "$sessions"
}

# Full path -> session name
path_to_session() {
    local relpath="${1#$HOME/}"
    echo "zz:${relpath//\//.}"
}

# Session name -> full path (O(n) lookup via list_repos)
lookup_path_from_session() {
    local session=${1#zz:}
    while IFS= read -r repo_path; do
        if [[ "$(path_to_session "$repo_path")" == "zz:$session" ]]; then
            echo "$repo_path"
            return 0
        fi
    done < <(list_repos)
    return 1
}

# Check if string contains all words
contains_all() {
    local str="$1"; shift
    for word in "$@"; do
        [[ "$str" != *"$word"* ]] && return 1
    done
}

# Query matching (see SPEC.md for details)
# 1. path query (starts with . or /)
# 2. exact basename match
# 3. partial basename match (all words)
# 4. full path match (all words)
query_match() {
    local candidates="$1"; shift
    local -a words=("$@")
    local query="$*"

    # 1. path query
    if [[ "$query" == .* || "$query" == /* ]]; then
        local resolved
        resolved=$(realpath -m "$query")
        if echo "$candidates" | grep -qxF "$resolved"; then
            echo "$resolved"
            return
        fi
        die "No match: $query"
    fi

    # 2. exact basename match
    local match
    match=$(echo "$candidates" | while IFS= read -r p; do
        [[ "$(basename "$p")" == "$query" ]] && echo "$p"
    done)
    [[ -n "$match" ]] && { head -1 <<< "$match"; return; }

    # 3. partial basename match
    match=$(echo "$candidates" | while IFS= read -r p; do
        contains_all "$(basename "$p")" "${words[@]}" && echo "$p"
    done)
    [[ -n "$match" ]] && { head -1 <<< "$match"; return; }

    # 4. full path match
    match=$(echo "$candidates" | while IFS= read -r p; do
        contains_all "$p" "${words[@]}" && echo "$p"
    done)
    [[ -n "$match" ]] && { head -1 <<< "$match"; return; }

    die "No match: $query"
}

select_repo() {
    local candidates
    if [[ "$SESSION_ONLY" == true ]]; then
        candidates=$(list_session_repos) || die "No zz sessions found."
    else
        candidates=$(list_repos)
    fi
    [[ -z "$candidates" ]] && die "No repos found."

    if [[ $# -gt 0 ]]; then
        query_match "$candidates" "$@"
    else
        local prompt="Repo: "
        [[ "$SESSION_ONLY" == true ]] && prompt="Session: "
        fzf --prompt="$prompt" -1 <<< "$candidates" || die "No selection"
    fi
}

#=============================================================================
# Zellij helpers
#=============================================================================

list_zz_sessions() {
    zellij list-sessions -s 2>/dev/null | grep '^zz:' || true
}

open_zellij_session() {
    local session_name="$1" repo_path="$2"
    if zellij list-sessions -s 2>/dev/null | grep -qx "$session_name"; then
        zellij attach "$session_name"
    else
        zellij -s "$session_name" options --default-cwd "$repo_path"
    fi
}

#=============================================================================
# Commands
#=============================================================================

cmd_help() {
    cat <<'EOF'
Usage: zz [command] [args]

Workspace + zellij session manager

Commands:
  [query]           Select repo (basename match) → zellij session
  a, attach [q]     Alias for default (explicit attach)
  query [q]         Print the full path of the selected repo
  get <url>         Clone repo (git clone into $ZZ_CLONE_DIR)
  list, ls          List existing zz sessions
  list -a, --all    List all repos with session status
  delete, d [q]     Delete zellij session
  delete -a, --all  Delete all zz sessions

Flags:
  -s, --session     Select from existing zz sessions only
  -h, --help        Show this help

Examples:
  zz             # interactive select repo (fzf) → zellij session
  zz myrepo      # attach to repo matching "myrepo" by basename
  zz query       # print selected repo path
  zz query foo   # print path of repo matching "foo"
  zz ls          # list existing sessions
  zz ls -a       # list all repos with session status
  zz -s          # select from existing sessions
  zz d myrepo    # delete session matching "myrepo"
  zz d -a        # delete all zz sessions
EOF
}

cmd_attach() {
    [[ -n "${ZELLIJ:-}" ]] && die "cannot switch sessions from inside zellij. Detach first with Ctrl+o d"
    local repo_path
    repo_path=$(select_repo "$@")
    open_zellij_session "$(path_to_session "$repo_path")" "$repo_path"
}

cmd_ls() {
    local sessions_full
    sessions_full=$(zellij list-sessions -n 2>/dev/null | grep '^zz:' || true)

    print_repo() {
        local repo_path="$1" session_name
        session_name=$(path_to_session "$repo_path")
        if [[ "$sessions_full" == *"$session_name "*EXITED* ]]; then
            echo -e "${RED}○${RESET} $session_name"
        elif [[ "$sessions_full" == *"$session_name "* ]]; then
            echo -e "${GREEN}●${RESET} $session_name"
        else
            echo "  $session_name"
        fi
    }

    local repos
    if [[ "$LIST_ALL" == true ]]; then
        repos=$(list_repos) || die "No repositories found."
    else
        repos=$(list_session_repos) || die "No zz sessions found."
    fi

    while read -r repo_path; do
        print_repo "$repo_path"
    done <<< "$repos"
}

cmd_delete() {
    local sessions
    sessions=$(list_zz_sessions)
    [[ -z "$sessions" ]] && die "No zz sessions found."

    if [[ "$DELETE_ALL" == true ]]; then
        while read -r session; do
            zellij delete-session -f "$session"
            echo "Deleted session: $session"
        done <<< "$sessions"
    else
        local session
        session=$(echo "$sessions" | fzf --prompt="Delete session: " -1 -q "$*") || die "No session selected"
        zellij delete-session -f "$session"
        echo "Deleted session: $session"
    fi
}

cmd_get() {
    git clone "$1" "$ZZ_WORKSPACE/$(basename "${1%.git}")"
}

#=============================================================================
# Main
#=============================================================================

# Global constants
GREEN='\033[32m' RED='\033[31m' RESET='\033[0m'

# Global flags
SHOW_HELP=false
SESSION_ONLY=false
LIST_ALL=false
DELETE_ALL=false

args=()
parse_flags=true
for arg in "$@"; do
    if [[ "$parse_flags" == true ]]; then
        case "$arg" in
            --) parse_flags=false ;;
            -h|--help) SHOW_HELP=true ;;
            -s|--session) SESSION_ONLY=true ;;
            -a|--all) LIST_ALL=true; DELETE_ALL=true ;;
            get) args+=("$arg"); parse_flags=false ;;
            -*) die "Unknown flag: $arg" ;;
            *) args+=("$arg") ;;
        esac
    else
        args+=("$arg")
    fi
done
set -- "${args[@]+"${args[@]}"}"

if [[ "$SHOW_HELP" == true ]]; then
    cmd_help
    exit 0
fi

case "${1:-}" in
    get)           shift; cmd_get "$@" ;;
    query)         shift; select_repo "$@" ;;
    list|ls)       cmd_ls ;;
    delete|d)      shift; cmd_delete "$@" ;;
    a|attach)      shift; cmd_attach "$@" ;;
    *)             cmd_attach "$@" ;;
esac