o i @s4dZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZdd lmZdd lmZmZdd lmZmZe rVdd lmZGd ddeZeGdddZdddee e!dfeffddZ"de defddZ#de!de!fddZ$d#de!de!d e%de!fd!d"Z&dS)$zqShared shell completion infrastructure. Provides data extraction, type analysis, and text processing utilities. N)Enum)Path) TYPE_CHECKINGAnyget_args get_origin)ITERABLE_TYPES)is_union)ArgumentCollection) CycloptsError)RegisteredCommandgroups_from_app)frozenis_class_and_subclass)Appc@seZdZdZdZdZdZdS)CompletionActionz'Shell-agnostic completion action types.nonefiles directoriesN)__name__ __module__ __qualname____doc__NONEFILES DIRECTORIESrr_/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cyclopts/completion/_base.pyrs rc@s.eZdZUdZded<eeed<eed<dS)CompletionDataz#Completion data for a command path.r argumentscommands help_formatN)rrrr__annotations__listr strrrrrr s   rapprreturn.cs.iddttdfffdd 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. r command_path.c szt|\}}}|d}Wn<tttfyL}z-tjdr"tj d|d|ddj j ddd }t t g|d |<WYd }~d Sd }~wwt }|} ||D]}|jrm|jd d } || q]Wd n1sxwYg} t|D]\} } | jr| D]} | jjr| | vr| | qq|j j ddd }t || |d |<| D]} | jD]}|dsÈ||fqqd S)z@Recursively extract completion data for command and subcommands.CYCLOPTS_COMPLETION_DEBUGz3Failed to extract completion data for command path z: ) stacklevelr!markdown)fallback)rr r!NT)parse_docstring-)parse_commandsr#r ValueError TypeErrorosenvirongetwarningswarn app_stackresolverr _get_resolution_contextdefault_commandassemble_argument_collectionextendr showr%appendnames startswith)r'_execution_path command_apper!rapps_for_paramssubapp app_argumentsr groupregistered_commandsregistered_commandcmd_name_extractr%completion_datarrrN8sP          z)extract_completion_data.._extractN)r)tupler$)r%rrMrextract_completion_data)s 'rQ type_hintcCst|r!t|D]}|tdurt|}|tjkr|SqtjSt|}t|tt r>t|}|r>t |dkr>t|dS|pA|}|t usKt|t rNtj StjS)zGet completion action from type hint. Parameters ---------- type_hint : Any Type annotation. Returns ------- CompletionAction Completion action for type. Nr) r rtypeget_completion_actionrrrrrPrlenrr)rRargactionoriginargs target_typerrrrUcs"    rUtextcCs$tdd|}tdd|}|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)r\rrrclean_choice_texts rdr,Pformat max_lengthcCsrddlm}ddlm}|j||d}||}tdd|}tdd|}t||kr7|d |d d }|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)rfr]r^r_r`NrSu…) cyclopts._markuprhcyclopts.help.inline_textri from_formatrarbrcrV)r\rfrgrhriinlinerrr strip_markups   rn)r,re)'rr3rar6enumrpathlibrtypingrrrrcyclopts._convertrcyclopts.annotationsr cyclopts.argumentr cyclopts.exceptionsr cyclopts.group_extractorsr r cyclopts.utilsrrcycloptsrrrdictrPr$rQrUrdintrnrrrrs,       ":&