YU 2 d Z ddlZddlZddlmZmZmZ dedefdZdej deeedf ee f fd Z
dej dee fd
Z dej d eedf d eeedf ee f fd
Z deeedf ee f deedf dee fdZ
ddej dedefdZ ddej dedefdZ ddej dedefdZeeedZ ee Z ddej dededefdZdS )z
Shell auto-completion script generators for the CLI.
Introspects an argparse parser to enumerate all commands, subcommands, and
flags, then emits completion scripts for bash, zsh, and fish.
N)DictListTupleprogreturnc . t j dd| S )zDConvert a prog name to a safe shell identifier (letters, digits, _).z [^a-zA-Z0-9]_)resub)r s V/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/completions.py_safe_identifierr
s
6/3 - -- parser.c , i }t | d| |S )z@Walk the parser tree and return {command_path: [flags]} mapping. ) _walk_parser)r results r _collect_commandsr s 02FV$$$ Mr c g }| j D ]W}t |t j r t |t j r8|j D ]}| | Xt | S )z:Extract all optional flags from a parser (excluding help).)_actions
isinstanceargparse _HelpAction_SubParsersActionoption_stringsappendsorted)r flagsactionopts r
_get_flagsr! s E/ fh 2 3 3 fh 8 9 9 ( C LL %==r pathr c t | }|||< | j D ]P}t |t j r4|j D ]\ }}t |||fz | QdS )z>Recursively walk subparsers and collect command paths + flags.N)r! r r r r choicesitemsr )r r" r r r name subparsers r r r ( s
v EF4L/ @ @ fh 8 9 9 @#)>#7#7#9#9
@
@i Yw????@ @r commandsprefixc t }| D ][}t | t | dz k r6|dt | |k r| |d \t | S )z,Get immediate subcommands of a given prefix. N)setlenaddr )r( r) subsr" s r _get_subcommandsr1 7 ss
55D t99F a ' 'D 3v;; ,?6,I,I HHT"X $<<r imunify360-agentc , t | }g }| d| | d| d | d | dt | d | d | d | d | d
| d | d | d
| d | d | d | d | d | d | d | d | d | d | d | d | d | d | d t | d }|D ]}|st || }|| }d ||z }d | } | d | d | d | d! | d" t |d# }
| d#g } d |
| z } | d$ | d | d! | d" | d% | d& | d | d't | d(| | d d) | S )*z"Generate a bash completion script.z# bash completion for # Auto-generated by z completions bash r z_completions() {z local cur prev words cwordz. if type _init_completion &>/dev/null; thenz" _init_completion || returnz elsez COMPREPLY=()z' cur="${COMP_WORDS[COMP_CWORD]}"z* prev="${COMP_WORDS[COMP_CWORD-1]}"z" words=("${COMP_WORDS[@]}")z cword=$COMP_CWORDz fiz' # Build the command path from wordsz local cmd_path=""z local iz% for (( i=1; i < cword; i++ )); doz case "${words[i]}" inz -*) continue ;;zB *) cmd_path="${cmd_path:+${cmd_path} }${words[i]}" ;;z esac donez case "$cmd_path" inc & t | | fS Nr. ps r z generate_bash..g Awl r key "")z% COMPREPLY=($(compgen -W "z
" -- "$cur"))z return ;;r "") esac}z
complete -F _z
_completions
)r r r
r keysr1 joinget)
r r r( lines all_pathsr" r0 r completionspattern root_subs
root_flagsroot_completionss
r
generate_bashrQ C s !((H E LL0$00111 LL????@@@ LL LL>%d++>>>??? LL1222 LLABBB LL5666 LL LL'((( LL:;;; LL=>>> LL5666 LL,--- LL LL LL:;;; LL())) LL LL8999 LL0111 LL./// LLL
LL LL LL LL*+++ x}},B,BCCCI . . $//hhte|,, ((4..
,,,,---
NK N N N
,---- !2..I b"%%Jxx J 677 LL LLO0@OOO
LL())) LL LL LL LLL!1$!7!7LLdLLMMM LL 99U r c 2 t | }dt | }g }| d| | d| | d| d | d | | d | d | d | d | d
| d | d | d
| d | d | d | d | d t | d }|D ]}|st || }|| }d | } | d| d |r8d d |D }
| d|
d |r8d d |D } | d| d | |rd nd d | d | d t |d } | d g }
d d! | D }| d" | d| d |
r8d d# |
D } | d| d | d | d | d$ | d% | d | | | d d& | S )'z!Generate a zsh completion script.r z #compdef z# zsh completion for r4 z completions zshr5 z() {z local -a commands flagsz local cmd_pathz# # Build command path from wordsz cmd_path=()z" for word in ${words[2,-1]}; doz% [[ $word == -* ]] && continuez5 [[ $word == "$words[$CURRENT]" ]] && continuez cmd_path+=($word)r6 z case "${cmd_path[*]}" inc & t | | fS r8 r9 r: s r r< z generate_zsh.. r= r r> r@ rA rB c 3 " K | ]
}d | d V dS "Nr .0ss r z generate_zsh.. s* 8 8aQ 8 8 8 8 8 8r z commands=()c 3 " K | ]
}d | d V dS rU r rX fs r rZ z generate_zsh.. s* 9 9aQ 9 9 9 9 9 9r z flags=(z; _describe 'command' commands -- flags && returnz compadd -- z
&& returnz ;;r c 3 " K | ]
}d | d V dS rU r rW s r rZ z generate_zsh.. s* 55aQ555555r rC c 3 " K | ]
}d | d V dS rU r r] s r rZ z generate_zsh.. s* : :!XXXX : : : : : :r rD rE rF )r r
r r rG r1 rH rI )r r r( func_namerJ rK r" r0 r rM desc_list flag_listrN rO root_descs r generate_zshre s; !((H,$T**,,I E LL#T##$$$ LL///000 LL>>>>??? LL LLI$$$%%% LL./// LL%&&& LL LL6777 LL"### LL5666 LL8999 LLHIII LL,--- LL LL LL/000x}},B,BCCCI ' ' $//((4..
,,,,--- @ 8 84 8 8 888I LL>)>>> ? ? ? = 9 95 9 9 999I LL;y;;; < < <
G I IF388E??FFF
%&&&& !2..I b"%%J55955555I LL LL6)666777 9HH : :z : : :::
79777888 LLNOOO LL!""" LL LL LL LLI LL 99U r c t | }g }| d| | d| d | d t | d D ]}t || }|| }|sdd t |d }|D ]!}| d
| d | d | d
"|D ]{} | d r(| d
| d | d| dd d
?| d r'| d
| d | d| dd d
|g }
|D ]} |
d| d |
}|} | r|dd | z
}|D ]!}| d
| d | d | d
"|D ]{} | d r(| d
| d | d| dd d
?| d r'| d
| d | d| dd d
|| d d | S )z"Generate a fish completion script.z# fish completion for r4 z completions fishr5 c $ t | | fS r8 r9 r: s r r< z generate_fish.. s s1vvqk r r> z not __fish_seen_subcommand_from r@ r z complete -c z -n 'z ' -f -a ''z--z' -l ' N-z' -s 'r+ z __fish_seen_subcommand_from z && z$ && not __fish_seen_subcommand_from rF )r r r rG r1 rH
startswith)
r r r( rJ r" r0 r conditionr flag
seen_partsr;
child_subss
r
generate_fishrp s !((H E LL0$00111 LL????@@@ LL x}},A,ABBB 1 1 $// - ?HH -h ; ;<<? ?
H4HHiHH#HHH
??4(( LLNtNN)NN48NNN __S)) LLNtNN)NN48NNN
J
F
F!!"D"D"DEEEE J//I J
/,,/ /
H4HHiHH#HHH
??4(( LLNtNN)NN48NNN __S)) LLNtNN)NN48NNN
LL 99U r )bashzshfishshellc t | }|-t d| dd t || | S )zfGenerate completion script for the given shell.
Raises ValueError if shell is not supported.
NzUnsupported shell: z. Supported shells: z, )
GENERATORSrI
ValueErrorrH SUPPORTED_SHELLS)r rt r generators r generate_completionsrz sl u%%I
?%
?
?!%+;!r s $ $ $ $ $ $ $ $ $ $.3 .3 . . . .
# %S/49
$%
x.
49
@ # @
S/ @
sCx $s)+ , @ @ @ @ 5c?DI-. #s(O
#Y 2D@ @ #@ +.@ @ @ @ @ H 2D> > #> +.> > > > > D 2D@ @ #@ +.@ @ @ @ @ H
6*//++,, ## # ## # # # # # # # #r