Ki] TdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z ddl mZmZmZddlmZddlmZdd lmZmZdd lmZmZe rdd lmZGd d eZeGddZdddee e!dfeffdZ"de defdZ#de!de!fdZ$dde!de!de!fdZ%d de!de!de&de!fdZ'y)!zqShared shell completion infrastructure. Provides data extraction, type analysis, and text processing utilities. N)Enum)Path) TYPE_CHECKINGAnyget_args get_origin)ITERABLE_TYPES is_annotatedis_union)ArgumentCollection) CycloptsError)RegisteredCommandgroups_from_app)frozenis_class_and_subclass)AppceZdZdZdZdZdZy)CompletionActionz'Shell-agnostic completion action types.nonefiles directoriesN)__name__ __module__ __qualname____doc__NONEFILES DIRECTORIESe/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/cyclopts/completion/_base.pyrrs1 D EKr rc6eZdZUdZded<eeed<eed<y)CompletionDataz#Completion data for a command path.r argumentscommands help_formatN)rrrr__annotations__listrstrrr r!r#r#s-##$%%r r#apprreturn.cNiddttdfffd S)a:Recursively extract completion data for app and all subcommands. Parameters ---------- app : App The Cyclopts application to extract completion data from. Returns ------- dict[tuple[str, ...], CompletionData] Mapping from command path tuples to their completion data. command_path.c jt|\}}}|d}t}j|}j|5|D]2}|js|j!d } |j#| 4 d d d g} t%|d D]F\} } | j&s| D]/} | j(j&s| | vs| j+| 1H|jjdd }t|| | |<| D]1} | j,D] }|j/dr||fz"3y #tttf$r}}t j jdrtjd|d|djjdd }ttg| |<Yd }~y d }~wwxYw#1swY[xYw)z@Recursively extract completion data for command and subcommands.CYCLOPTS_COMPLETION_DEBUGz3Failed to extract completion data for command path z: ) stacklevelr&markdown)fallback)r$r%r&NT)parse_docstring) resolve_lazy-)parse_commandsr(r ValueError TypeErrorosenvirongetwarningswarn app_stackresolver#r _get_resolution_contextdefault_commandassemble_argument_collectionextendrshowr*appendnames startswith)r-_execution_path command_apper&r$apps_for_paramssubapp app_argumentsr%groupregistered_commandsregistered_commandcmd_name_extractr*completion_datas r!rUz)extract_completion_data.._extract7s ##5d<6H#I A~q(,K'( 55nE ]]> * 4) 4))$*$G$GX\$G$]M$$]3 4 4 *9+TX*Y < &E&zz*=<&)--227IQY7Y (:;< < "++33MJ3W (6U]kv(w %"* 9 .44 9**3/\XK78 9 9;z95 zz~~9: MMOP\O__abcaders t--// /SK,:,.-OL )   4 4s*#EG$%%G$G!$A3GG!$G.)r)tupler))r*rUrVs`@@r!extract_completion_datarX(s->@O%9uS#X%9N J r type_hintct|rtt|dSt|rPt|D]2}|t dust|}|t j k7s0|cSt j St|}t|ttr)t|}|rt|dk\rt|dS|xs|}|tust|trt jSt j S)zGet completion action from type hint. Parameters ---------- type_hint : Any Type annotation. Returns ------- CompletionAction Completion action for type. rN)r get_completion_actionrr typerrrrrWr lenrr)rYargactionoriginargs target_types r!r\r\bsI$Xi%8%;<< I& "C$t*$.s3-222!M  "  $$$  "FVU>%:; " CIN(a1 1%IKd3KF%%%   r textc~tjdd|}tjdd|j}|S)zClean choice text without shell-specific escaping. Parameters ---------- text : str Raw choice text. Returns ------- str Cleaned text (not shell-escaped). [\x00-\x1f\x7f]\s+ )resubstrip)rds r!clean_choice_textrms7 66$b$ /D 66&#t $ * * ,D Kr namecharscb|jdd}|D]}|j|d|}|S)aEscape glob/pattern characters for shell case patterns. Both bash and zsh case patterns treat glob characters as special even inside quotes. This function escapes them with backslashes for literal matching. Parameters ---------- name : str String to escape. chars : str Characters to escape. Default covers basic glob chars. For zsh, also pass "()|" for extended patterns. Returns ------- str Escaped string safe for shell case patterns. \z\\)replace)rnroresultchars r!escape_for_shell_patternrus>(\\$ 'F34&k23 Mr format max_lengthcddlm}ddlm}|j ||}||}t j dd|}t j dd|j}t||kDr |d |d z d z}|S) a#Strip markup and render to plain text for shell completions. Converts formatted text (markdown/RST/rich) to plain text suitable for shell completion descriptions. Removes control characters, normalizes whitespace, and truncates if needed. Parameters ---------- text : str Text with markup. format : str Markup format: "markdown", "rst", "rich", or "plaintext". max_length : int Maximum length before truncation. Returns ------- str Plain text (not shell-escaped). r) extract_text) InlineText)rvrfrgrhriNr[u…) cyclopts._markuprycyclopts.help.inline_textrz from_formatrjrkrlr^)rdrvrwryrzinlines r! strip_markuprs~*.4  # #D # 8F  D 66$b$ /D 66&#t $ * * ,D 4y:$j1n%- Kr )z*?[])r3P)(rr;rjr>enumrpathlibrtypingrrrrcyclopts.annotationsr r r cyclopts.argumentr cyclopts.exceptionsr cyclopts.group_extractorsrrcyclopts.utilsrrcycloptsrrr#dictrWr)rXr\rmruintrrr r!rs  ;;GG0-H8 t 774c3h0O+P7t&!S&!-=&!RCC$3s4!s!C!#!s!r