# bash completion for ai_research_os
# Usage: source <(python ai_research_os.py --bash-completion)

_bash_complete() {
    local cur prev words cword
    _init_completion || return

    # All arguments with their types
    case "$prev" in
        --root|--category|--concept-dir|--comparison-dir|--pdf)
            _filedir
            return
            ;;
        --ocr-lang|--api-key|--base-url|--model)
            return
            ;;
        --tags)
            # No completion for custom tags
            return
            ;;
        --max-pages|--ai-max-papers|--ai-max-chars|--ocr-zoom)
            return
            ;;
        --ai|--ocr|--no-pdfminer|--structured|--ai-cnote)
            # These are flags, complete next word as positional arg
            ;;
    esac

    # Complete positional arg (input: arXiv URL/ID or DOI)
    if [[ "$cword" -eq 1 ]]; then
        # Suggest common arXiv prefixes
        COMPREPLY=( $(compgen -W "https://arxiv.org/abs/ https://arxiv.org/pdf/ 10.48550/arXiv." -- "$cur") )
        return
    fi

    # Complete short/long options
    if [[ "$cur" == -* ]]; then
        COMPREPLY=( $(compgen -W \
            "--root --category --tags --concept-dir --comparison-dir \
             --max-pages --pdf --ocr --ocr-lang --ocr-zoom \
             --no-pdfminer --structured --ai --ai-cnote \
             --ai-max-papers --api-key --model --base-url --ai-max-chars" \
            -- "$cur") )
    fi
} && complete -F _bash_complete ai_research_os.py

# Alias for common usage patterns
alias airos='python /path/to/ai_research_os.py'
