kKjv dZddlmZddlmZmZGddeZdZdZeefZ de d zd e d e d e d e f dZ de d efdZ de d e fdZ de d zd e fdZy )uDocket and Redis key encoding for background tasks. The compound Docket task key embeds the auth boundary so that the parser can reject cross-scope access without consulting Redis. Authenticated and anonymous tasks live in disjoint keyspaces: auth:{enc_scope}:{client_task_id}:{task_type}:{enc_identifier} anon:{client_task_id}:{task_type}:{enc_identifier} The same `auth/anon` partition is used for the per-task Redis prefix (``fastmcp:task:auth:{enc_scope}`` vs ``fastmcp:task:anon``) — see ``task_redis_prefix``. ``task_scope`` is the raw scope identifier (typically derived from ``client_id`` or ``client_id|sub``); encoding happens once, at the boundary, in this module. ) TypedDict)quoteunquotec@eZdZUdZedzed<eed<eed<eed<y) TaskKeyPartszDecoded segments of a Docket task key. ``task_scope`` is ``None`` for anonymous tasks, the raw scope string otherwise. N task_scopeclient_task_id task_typecomponent_identifier)__name__ __module__ __qualname____doc__str__annotations__j/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/server/tasks/keys.pyrrs$ d NrrauthanonrNr r r returnc t|d}|td|d|d|St|d}td|d|d|d| S)azBuild Docket task key with embedded metadata. When ``task_scope`` is ``None`` the task is anonymous and lives in the ``anon`` keyspace. Otherwise it lives under ``auth:{enc_scope}``. Args: task_scope: Raw authorization scope, or ``None`` for anonymous tasks client_task_id: Client-provided task ID task_type: Type of task ("tool", "prompt", "resource") component_identifier: Tool name, prompt name, or resource URI Returns: Encoded task key for Docket Examples: >>> build_task_key("client-a", "task456", "tool", "my_tool") 'auth:client-a:task456:tool:my_tool' >>> build_task_key(None, "task456", "tool", "my_tool") 'anon:task456:tool:my_tool' >>> build_task_key("client-a", "task456", "resource", "file://data.txt") 'auth:client-a:task456:resource:file%3A%2F%2Fdata.txt' safe:)r _ANON_TAG _AUTH_TAG)rr r r encoded_identifier encoded_scopes rbuild_task_keyr!)sl<3"=An-Qyk;M:NOO*2.M +Q}oQ~&6a {!DVCWXrtask_keyc|jd\}}}|tvs|std|dtd|tk(rE|j dd}t |dk7rtd|d|\}}}d ||t |d S|j dd}t |d k7rtd |d |\}}}}t |||t |d S)aParse Docket task key to extract metadata. Args: task_key: Encoded task key from Docket Returns: Dict with keys: ``task_scope`` (``str | None``), ``client_task_id``, ``task_type``, ``component_identifier``. Raises: ValueError: If the key has an unrecognized tag or wrong segment count. Examples: >>> parse_task_key("auth:client-a:task456:tool:my_tool") `{'task_scope': 'client-a', 'client_task_id': 'task456', 'task_type': 'tool', 'component_identifier': 'my_tool'}` >>> parse_task_key("anon:task456:tool:my_tool") `{'task_scope': None, 'client_task_id': 'task456', 'task_type': 'tool', 'component_identifier': 'my_tool'}` rzInvalid task key format: z. Expected leading tag in .zInvalid anonymous task key: zD. Expected: anon:{client_task_id}:{task_type}:{component_identifier}N)rr r r z Invalid authenticated task key: zP. Expected: auth:{enc_scope}:{client_task_id}:{task_type}:{component_identifier}) partition _VALID_TAGS ValueErrorrsplitlenr) r"tag_restpartsr r rr s rparse_task_keyr1Ps*(%%c*LCD +T'z2''2m1 6   i 3" u:?.xj9[\ 9>5 #5,"$+,>$?    JJsA E 5zQ.xj9e f  DI@M>9.@m,( '(: ;  rct|dS)a]Extract just the client task ID from a task key. Args: task_key: Full encoded task key Returns: Client-provided task ID Examples: >>> get_client_task_id_from_key("auth:client-a:task456:tool:my_tool") 'task456' >>> get_client_task_id_from_key("anon:task456:tool:my_tool") 'task456' r )r1)r"s rget_client_task_id_from_keyr3s ( #$4 55rcF| dtSdtdt|dS)zReturn the Redis key prefix that owns a given scope. Authenticated tasks live under ``fastmcp:task:auth:{enc_scope}``; anonymous tasks live under ``fastmcp:task:anon``. Callers append ``f":{task_id}:..."`` to compose the final key. z fastmcp:task:rrr)rrr)rs rtask_redis_prefixr5s2yk** 9+QuZb'A&B CCr)rtypingr urllib.parserrrrrr)rr!r1r3r5rrrr8s$' 9    )$ $d $$$ $  $N6S6\6r6#6#6& D#* D Dr