# claude-account.sh
# Usage: source this once per shell session, then toggle with:
#   claude-pro     -> unsets the ollama-routing vars for this shell (uses your Pro subscription)
#   claude-ollama  -> restores the local ollama vars
#   claude-status  -> shows which mode is currently active

_claude_ollama_vars=(ANTHROPIC_AUTH_TOKEN ANTHROPIC_BASE_URL ANTHROPIC_MODEL ANTHROPIC_SMALL_FAST_MODEL CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC)

claude-pro() {
  for v in "${_claude_ollama_vars[@]}"; do
    if [ -n "${!v+x}" ]; then
      export "_SAVED_$v=${!v}"
      unset "$v"
    fi
  done
  echo "claude -> Pro subscription (ollama routing unset for this shell)"
}

claude-ollama() {
  for v in "${_claude_ollama_vars[@]}"; do
    saved="_SAVED_$v"
    if [ -n "${!saved+x}" ]; then
      export "$v=${!saved}"
      unset "$saved"
    fi
  done
  echo "claude -> local ollama routing restored"
}

claude-status() {
  if [ -n "${ANTHROPIC_BASE_URL+x}" ]; then
    echo "Currently routed to: $ANTHROPIC_BASE_URL (ollama)"
  else
    echo "Currently routed to: Anthropic API (Pro/Max subscription)"
  fi
}