# Gemini CLI Context Management

# Base directory for Gemini contexts
export GEMINI_CONTEXT_DIR="${XDG_CONFIG_HOME:-$HOME/.config}/gemini"

# Ensure the base context directory exists
mkdir -p "$GEMINI_CONTEXT_DIR"

# Template settings file
GEMINI_TEMPLATE_SETTINGS="${GEMINI_CONTEXT_DIR}/settings.json"

# Create a new Gemini context
gemini-context-create() {
  if [ -z "$1" ]; then
    echo "Usage: gemini-context-create <name>"
    return 1
  fi

  local context_name="$1"
  local context_path="$GEMINI_CONTEXT_DIR/$context_name"

  if [[ "$context_name" =~ [^a-zA-Z0-9_-] ]]; then
    echo "Error: Context name should only contain alphanumeric characters, underscores, or dashes."
    return 1
  fi

  if [[ "$context_name" == .* ]]; then
    echo "Error: Context name cannot start with a dot."
    return 1
  fi

  if [ -d "$context_path" ]; then
    echo "Error: Context '$context_name' already exists at $context_path"
    return 1
  fi

  echo "Creating context '$context_name' at $context_path"
  mkdir -p "$context_path/.gemini"

  if [ -f "$GEMINI_TEMPLATE_SETTINGS" ]; then
    cp "$GEMINI_TEMPLATE_SETTINGS" "$context_path/.gemini/settings.json"
    echo "Initialized with template settings."
  else
    echo "Warning: Template settings not found at $GEMINI_TEMPLATE_SETTINGS"
    touch "$context_path/.gemini/settings.json"
  fi
  echo "Context '$context_name' created."
}

# List available Gemini contexts (directories only, containing .gemini)
_gemini_context_list_internal() {
  local contexts
  # Use Zsh globbing: find directories containing a .gemini subdir
  contexts=($GEMINI_CONTEXT_DIR/*/.gemini(N/:h:t))
  [[ ${#contexts} -gt 0 ]] && print -l "${contexts[@]}"
}

gemini-context-list() {
  echo "Available Gemini contexts:"
  local contexts=($(_gemini_context_list_internal))
  if [ ${#contexts} -eq 0 ]; then
    echo "  (No contexts found)"
    return
  fi
  for context in "${contexts[@]}"; do
    if [ "$GEMINI_CLI_HOME" = "$GEMINI_CONTEXT_DIR/$context" ]; then
      echo "  * $context (active)"
    else
      echo "    $context"
    fi
  done
}

# Delete a Gemini context
gemini-context-delete() {
  local context_name
  if [ -z "$1" ]; then
    if command -v fzf >/dev/null; then
      context_name=$(_gemini_context_list_internal | fzf --prompt="Select Gemini Context to DELETE: ")
      [[ -z "$context_name" ]] && return 1
    else
      echo "Usage: gemini-context-delete <name>"
      return 1
    fi
  else
    context_name="$1"
  fi

  local context_path="$GEMINI_CONTEXT_DIR/$context_name"
  if [ ! -d "$context_path" ]; then
    echo "Error: Context '$context_name' not found."
    return 1
  fi

  if read -q "confirm?Are you sure you want to delete context '$context_name'? [y/N] "; then
    echo
    rm -rf "$context_path"
    echo "Context '$context_name' deleted."
    if [ "$GEMINI_CLI_HOME" = "$context_path" ]; then
      gemini-context-unset
    fi
  else
    echo "\nDeletion cancelled."
  fi
}

# Rename a Gemini context
gemini-context-rename() {
  if [ -z "$1" ] || [ -z "$2" ]; then
    echo "Usage: gemini-context-rename <old_name> <new_name>"
    return 1
  fi

  local old_path="$GEMINI_CONTEXT_DIR/$1"
  local new_path="$GEMINI_CONTEXT_DIR/$2"

  if [ ! -d "$old_path" ]; then
    echo "Error: Source context '$1' not found."
    return 1
  fi

  if [ -d "$new_path" ]; then
    echo "Error: Destination context '$2' already exists."
    return 1
  fi

  mv "$old_path" "$new_path"
  echo "Context '$1' renamed to '$2'."
  if [ "$GEMINI_CLI_HOME" = "$old_path" ]; then
    gemini-context-use -q "$2"
  fi
}

# Use a specific Gemini context
gemini-context-use() {
  local context_name
  local quiet=0

  if [ "$1" = "-q" ] || [ "$1" = "--quiet" ]; then
    quiet=1
    shift
  fi

  if [ -z "$1" ]; then
    if command -v fzf >/dev/null; then
      context_name=$(_gemini_context_list_internal | fzf --prompt="Select Gemini Context: ")
      if [ -z "$context_name" ]; then
        return 1
      fi
    else
      echo "Usage: gemini-context-use [-q] <name>"
      echo "fzf not found for interactive selection."
      gemini-context-list
      return 1
    fi
  else
    context_name="$1"
  fi

  local context_path="$GEMINI_CONTEXT_DIR/$context_name"

  if [ ! -d "$context_path/.gemini" ]; then
    [[ "$quiet" -eq 0 ]] && echo "Error: '$context_name' is not a valid Gemini context (missing .gemini directory)."
    return 1
  fi

  export GEMINI_CLI_HOME="$context_path"
  [[ "$quiet" -eq 0 ]] && echo "Switched to Gemini context '$context_name'"
}

# Edit a context's settings
gemini-context-edit() {
  local context_name
  if [ -z "$1" ]; then
    if command -v fzf >/dev/null; then
      context_name=$(_gemini_context_list_internal | fzf --prompt="Select Gemini Context to Edit: ")
      if [ -z "$context_name" ]; then
        return 1
      fi
    else
      echo "Usage: gemini-context-edit <name>"
      gemini-context-list
      return 1
    fi
  else
    context_name="$1"
  fi

  local context_path="$GEMINI_CONTEXT_DIR/$context_name"
  local settings_file="$context_path/.gemini/settings.json"

  if [ ! -d "$context_path" ]; then
    echo "Error: Context '$context_name' not found."
    return 1
  fi

  if [ ! -f "$settings_file" ]; then
    echo "Error: settings.json not found for context '$context_name'."
    return 1
  fi

  ${EDITOR:-vim} "$settings_file"
}

# Show the current Gemini context
gemini-context-current() {
  if [ -n "$GEMINI_CLI_HOME" ]; then
    local current_context=${GEMINI_CLI_HOME:t}
    if [ "$GEMINI_CONTEXT_DIR/$current_context" = "$GEMINI_CLI_HOME" ]; then
      echo "Current Gemini context: $current_context"
    else
      echo "Current Gemini context: Custom ($GEMINI_CLI_HOME)"
    fi
  else
    echo "No Gemini context set, using default."
  fi
}

# Unset the current Gemini context
gemini-context-unset() {
  unset GEMINI_CLI_HOME
  echo "Unset Gemini context, reverted to default."
}

# Alias for gemini-context-use
alias gemctx='gemini-context-use'

# Zsh Completion
_gemini_contexts() {
  local -a contexts
  contexts=($(_gemini_context_list_internal))
  _describe 'gemini contexts' contexts
}

compdef _gemini_contexts gemini-context-use gemctx gemini-context-edit gemini-context-delete gemini-context-rename