fix(completion): replace bash completion script to handle colons in task names

Author: nekomoyi

Cargo.lock

@@ -0,0 +1,64 @@
+_clap_reassemble_words() {
+    if [[ "$COMP_WORDBREAKS" != *:* ]]; then
+        return
+    fi
+    local i j=0 line=$COMP_LINE
+    words=()
+    _CLAP_COMPLETE_INDEX=0
+    for ((i = 0; i < ${#COMP_WORDS[@]}; i++)); do
+        if ((i > 0 && j > 0)) && [[ "${COMP_WORDS[i]}" == :* || "${words[j-1]}" == *: ]] && [[ "$line" != [[:blank:]]* ]]; then
+            words[j-1]="${words[j-1]}${COMP_WORDS[i]}"
+        else
+            words[j]="${COMP_WORDS[i]}"
+            ((j++))
+        fi
+        if ((i == COMP_CWORD)); then
+            _CLAP_COMPLETE_INDEX=$((j - 1))
+        fi
+        line=${line#*"${COMP_WORDS[i]}"}
+    done
+}
+
+_clap_trim_completions() {
+    local cur="${words[_CLAP_COMPLETE_INDEX]}"
+    if [[ "$cur" != *:* || "$COMP_WORDBREAKS" != *:* ]]; then
+        return
+    fi
+    local colon_word=${cur%"${cur##*:}"}
+    local i=${#COMPREPLY[*]}
+    while [[ $((--i)) -ge 0 ]]; do
+        COMPREPLY[$i]=${COMPREPLY[$i]#"$colon_word"}
+    done
+}
+
+_clap_complete_vp() {
+    local IFS=$'\013'
+    local _CLAP_COMPLETE_INDEX=${COMP_CWORD}
+    local _CLAP_COMPLETE_COMP_TYPE=${COMP_TYPE}
+    if compopt +o nospace 2> /dev/null; then
+        local _CLAP_COMPLETE_SPACE=false
+    else
+        local _CLAP_COMPLETE_SPACE=true
+    fi
+    local words=("${COMP_WORDS[@]}")
+    _clap_reassemble_words
+    COMPREPLY=( $( \
+        _CLAP_IFS="$IFS" \
+        _CLAP_COMPLETE_INDEX="$_CLAP_COMPLETE_INDEX" \
+        _CLAP_COMPLETE_COMP_TYPE="$_CLAP_COMPLETE_COMP_TYPE" \
+        _CLAP_COMPLETE_SPACE="$_CLAP_COMPLETE_SPACE" \
+        VP_COMPLETE="bash" \
+        "vp" -- "${words[@]}" \
+    ) )
+    if [[ $? != 0 ]]; then
+        unset COMPREPLY
+    elif [[ $_CLAP_COMPLETE_SPACE == false ]] && [[ "${COMPREPLY-}" =~ [=/:]$ ]]; then
+        compopt -o nospace
+    fi
+    _clap_trim_completions
+}
+if [[ "${BASH_VERSINFO[0]}" -eq 4 && "${BASH_VERSINFO[1]}" -ge 4 || "${BASH_VERSINFO[0]}" -gt 4 ]]; then
+    complete -o nospace -o bashdefault -o nosort -F _clap_complete_vp vp
+else
+    complete -o nospace -o bashdefault -F _clap_complete_vp vp
+fi

crates/vite_global_cli/completion-register.bash

@@ -1487,7 +1487,6 @@ fn should_force_global_delegate(command: &str, args: &[String]) -> bool {
 ///
 /// Delegates to the local vite-plus CLI to run `vp run` without arguments,
 /// which returns a list of available tasks in the format "task_name: description".
-
 fn run_tasks_completions(current: &OsStr) -> Vec<clap_complete::CompletionCandidate> {
     let Some(cwd) = std::env::current_dir()
         .ok()
@@ -1497,6 +1496,7 @@ fn run_tasks_completions(current: &OsStr) -> Vec<clap_complete::CompletionCandid
         return vec![];
     };
 
+    // Unescape hashtag and trim quotes for better matching
     let current = current
         .to_string_lossy()
         .replace("\\#", "#")

crates/vite_global_cli/src/cli.rs

@@ -17,6 +17,7 @@ mod shim;
 mod tips;
 
 use std::{
+    env,
     io::{IsTerminal, Write},
     process::{ExitCode, ExitStatus},
 };
@@ -229,11 +230,18 @@ async fn main() -> ExitCode {
     // Initialize tracing
     vite_shared::init_tracing();
 
+    let mut args: Vec<String> = std::env::args().collect();
+
+    // Replace bash completion script to fix completion for items containing ':'
+    if env::var_os("VP_COMPLETE").is_some_and(|shell| shell == "bash") && args.len() == 1 {
+        print!("{}", include_str!("../completion-register.bash"));
+        return ExitCode::SUCCESS;
+    }
+
     // Handle shell completion
     CompleteEnv::with_factory(command_with_help).var("VP_COMPLETE").complete();
 
     // Check for shim mode (invoked as node, npm, or npx)
-    let mut args: Vec<String> = std::env::args().collect();
     let argv0 = args.first().map(|s| s.as_str()).unwrap_or("vp");
     tracing::debug!("argv0: {argv0}");