jKjlUdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z ddl mZmZmZdd lmZdd lmZdd lmZmZGd d Ze GddZdZej6Zded<ddZddZddZ ddZ!ddZ" d d!dZ# d d"dZ$d#dZ%y)$aPer-server tracking state, the bounded identity LRU, and identity resolution. Per-server state lives in a module-level ``weakref.WeakKeyDictionary`` keyed by the server object, so state is isolated per server and garbage-collected with it (the Python equivalent of the TS ``WeakMap``). ) annotationsN) OrderedDict) dataclassfield)datetimetimezone)AnyDictOptional)log) McpEventSink)MCPAnalyticsOptions UserIdentityc:eZdZdZdd dZd dZd dZd dZd dZy) IdentityCachezmBounded LRU of session identities, isolated per server so identities never bleed across server instances.c0t|_||_yN)r_cache _max_size)selfmax_sizes f/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/mcp/_internal.py__init__zIdentityCache.__init__s8C  !cx|jj|}|y|jj||Sr)rget move_to_endr session_ididentitys rrzIdentityCache.get#s4;;??:.    +rc||jvr|j|=n>t|j|jk\r|jjd||j|<yNF)last)rlenrpopitemrs rsetzIdentityCache.set*sR  $ J'   / KK  U  +"* Jrc||jvSr)rrr s rhaszIdentityCache.has1sT[[((rc,t|jSr)r%r)rs rsizezIdentityCache.size4s4;;rN))rintreturnNone)r strr/Optional[UserIdentity])r r1r!rr/r0)r r1r/bool)r/r.) __name__ __module__ __qualname____doc__rrr'r*r,rrrrs &"+) rrc6eZdZUdZded<dZded<dZded <d Zded <dZd ed <e dZ ded<e e Z ded<e e Zded<e e Zded<e eZded<dZd ed<dZd ed<e ej*Zded<ddZy)MCPAnalyticsDatazAll per-server tracking state.roptionsNzOptional[McpEventSink]sinkr1r generatedsession_sourcez Optional[str]last_mcp_session_idcHtjtjSr)rnowrutcr8rrzMCPAnalyticsData.AsHLLc',,.11 ::gcldC @DJJ CLD#E    rc||St|ji|jxsi|jxsi|j|jS|jS)N)r_rcrb)rr_rcrb)previousnxts rmerge_identitiesrnush OOLx**0bLcnn6JL ZZ3szz :B rczKtj|stj|r |d{S|S7wr)rS iscoroutineisfuture)values r _maybe_awaitrss35!W%5%5e%<{ Ls 0;9;cK|jjsy |jj}t|tr|}nt |||d{}|st d|y|j j|}t||}|xr t|| }|j j|||rSddl m } t d||t|| j||dtj t"j$dS y7#t&$r} t d|d | Yd} ~ yd} ~ wwxYww) zResolve the optional ``identify`` callback, dedupe against the identity cache, and return an ``$identify`` event to emit only when the identity has materially changed (otherwise ``None``).Nz>Warning: Supplied identify function returned null for session r )MCPAnalyticsEventTypezIdentified session )requestextra)r resource_name event_type parameters timestampz9Error: identify function threw while identifying session z - )r;identify isinstancerrsr rGrrnrjr' _event_typesru_get_request_resource_nameIDENTIFYrrBrrC Exception) rZr rvrwr|identity_resultrlmerged has_changedruerrors rhandle_identifyrsJ <<   <<(( h -6>O$0'51I$JJO PQ[P\] ++// ;!(O<#N(DDD.E/B(DEED E%D;6E;EEcK|jjsy t|jj||d{xsdS7#t$r}t d|Yd}~yd}~wwxYww)Nz!event_properties callback error: )r;event_propertiesrsrr )rZrvrwes rresolve_event_propertiesrsg << ( (!$,,"?"?"OPPXTXXP  /s 34s@A2(AA A A2 A A/A*%A2*A//A2ct|tsy|jd}t|tsy|jd}t|tr|SdS)NUnknownparamsname)r}rRrr1)rvrrs rrrsL gt $ [[ "F fd # ::f DdC(47i7r)rWr r/zOptional[MCPAnalyticsData])rWr rZr:r/r0)rerrfrr/r3)rlr2rmrr/r)rrr r/r r) rZr:r r1rvDict[str, Any]rwOptional[Dict[str, Any]]r/r)rZr:rvrrwrr/r)rvr r/r1)&r7 __future__rrSr`weakref collectionsr dataclassesrrrrtypingr r r loggerr _sinkrtypesrrrr:rOWeakKeyDictionaryrUrQrXr[rjrnrsrrrr8rrrs # #('&&4  : :: :2!GD ($& $ +7  '+ - --- $ -  -f'+      $  8r