$jv dZddlmZddlmZmZGddeZdZdZeefZ de d zd e d e d e d e f dZ de d efdZ de d e fdZ de d zd e fdZd S)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)quoteunquotecBeZdZUdZedzed<eed<eed<eed<dS) 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__a/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/server/tasks/keys.pyrrsO d NNNrrauthanonrNr 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' safeN:)r _ANON_TAG _AUTH_TAG)rr r r encoded_identifier encoded_scopes rbuild_task_keyr!)s<3"===OOnOOyOO;MOOO*2...M XX}XX~XX XXDVXXrtask_keyc |d\}}}|tvs|std|dtd|tkrV|dd}t |dkrtd|d|\}}}d ||t |d S|dd}t |d krtd |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 6 6 6'2 6 6 6    i 3"" u::??\x\\\ 9>5 #5,"$+,>$?$?     JJsA  E 5zzQ fx f f f   DI@M>9.@m,,( '(: ; ;   rc,t|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 55rcR| 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. Nz fastmcp:task:rrr)rrr)rs rtask_redis_prefixr5s;*y*** C9 C CuZb'A'A'A C CCr)rtypingr urllib.parserrrrrr)rr!r1r3r5rrrr8s?$''''''''     9      )$ $d $$$ $  $$$$N6S6\6666r6#6#6666& D#* D D D D D D Dr