#!/bin/zsh
# Zsh completions for Claude Code automation scripts
# Source this file: source ~/.claude/automation/completions.zsh

# Memory add completion
_memory_add() {
    local -a categories
    categories=('preference:Save as preference' 'decision:Save as decision' 'project:Save as project context' 'fact:Save as fact' 'auto:Auto-categorize')

    _arguments \
        '1:category:->category' \
        '*:content:' && return 0

    case $state in
        category)
            _describe 'category' categories
            ;;
    esac
}

# Memory list completion
_memory_list() {
    local -a categories
    categories=('preferences' 'decisions' 'projects' 'facts')

    _arguments \
        '1:category:->category' \
        '--all[Include deprecated items]' && return 0

    case $state in
        category)
            _describe 'category' categories
            ;;
    esac
}

# Restore completion
_claude_restore() {
    local backup_dir="${HOME}/.claude/backups"

    if [[ -d "${backup_dir}" ]]; then
        _files -g "${backup_dir}/*.tar.gz"
    fi
}

# Register completions
compdef _memory_add memory-add.py
compdef _memory_list memory-list.py
compdef _claude_restore restore.sh

# Aliases
alias claude-validate='~/.claude/automation/validate-setup.sh'
alias claude-status='~/.claude/automation/quick-status.sh'
alias claude-backup='~/.claude/automation/backup.sh'
alias claude-restore='~/.claude/automation/restore.sh'
alias claude-clean='~/.claude/automation/clean.sh'
alias claude-memory-add='python3 ~/.claude/automation/memory-add.py'
alias claude-memory-list='python3 ~/.claude/automation/memory-list.py'

echo "Claude Code completions loaded (zsh)"