n@j]BdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ej e ZdZGdd eZed Gd d ZdCdZdDdZdEdZdFdZdGdZdHdZdIdJdZd ed!dKd&ZdLd(ZdLd)ZdMd,Zd-d-d.dNd1ZdOd3Zd4Z d5Z!d6Z"dCd7Z#dPd9Z$d-d:d;dQd>Z%dRd?Z&dSd@Z'dTdAZ(dUdBZ)dS)VupOn-disk pet store — install / list / resolve pets. Pets live under ``get_hermes_home()/pets//`` so every profile gets its own set (we deliberately do **not** reuse petdex's ``~/.codex/pets`` default — that's owned by the petdex npm CLI and isn't profile-aware). Each installed pet directory holds: pets// pet.json # {id, displayName, description, spritesheetPath} spritesheet.webp # (or .png) The active pet is resolved from the caller-supplied ``display.pet.slug`` config value (falling back to the first installed pet), so this module stays free of the config loader. ) annotationsN) dataclass)Path)get_hermes_homegN@ceZdZdZdS) PetStoreErrorzRaised on install/IO failures.N)__name__ __module__ __qualname____doc__5/home/thesage/.hermes/hermes-agent/agent/pet/store.pyrr s((((rrT)frozenceZdZUdZded<ded<ded<ded<ded<d Zded <edd ZeddZdS) InstalledPetzA pet present on disk.strslug display_name descriptionr directory spritesheet created_byreturnboolc4|jSN)ris_fileselfs rexistszInstalledPet.exists/s'')))rc|jdkS)N generator)rr s r generatedzInstalledPet.generated3s+--rN)rr) r r r r __annotations__rpropertyr"r%r rrrr$s   IIIOOOJ ***X*...X...rrrrcVtdz }|dd|S)z=Return the profile-scoped pets directory (created on demand).petsTparentsexist_ok)rmkdirpaths rpets_dirr08s-   v %DJJtdJ+++ Krrdictc|dz }|siS tj|dS#tt f$r(}t d||icYd}~Sd}~wwxYw)Npet.jsonutf-8encodingzunreadable pet.json in %s: %s)rjsonloads read_textOSError ValueErrorloggerdebug)rpet_jsonexcs r_read_pet_jsonr@?s:%H      z(,,g,>>??? Z  4iEEE s'AA>A93A>9A>metact|ddpd}|r||z }|r|SdD]}||z }|r|cS |dz S)zFind the spritesheet for a pet dir. Honors ``spritesheetPath`` from pet.json, else probes the conventional filenames (``spritesheet.{webp,png}`` and petdex R2's ``sprite.webp``). spritesheetPathr)spritesheet.webpzspritesheet.pngz sprite.webpz sprite.pngrD)rgetstripr)rrAdeclared candidatenames r_resolve_spritesheetrJJs 488-r228b99??AAH(       T$           ) ))rrrcxtt|j}|dvrdS|S)aSNormalize a slug to a single bare path segment. Pet slugs index into ``pets_dir()//`` for load/remove, so a value carrying path separators (``../``, absolute paths) could escape the pets directory. Strip every separator and reject ``.``/``..`` so callers can only ever name a direct child of the pets directory. )r.z..r)rrrFrI)rsegments r _safe_slugrN]s93t99??$$%%*G/!!r NrInstalledPet | Nonec t|}|sdSt|z }|sdSt|}t |t |ddp|t |ddpd|t||t |ddpdS)zCReturn the :class:`InstalledPet` for *slug*, or ``None`` if absent.N displayNamerr createdBy)rrrrrr)rNr0is_dirr@rrrErJ)rrrAs rload_petrTks d  D t T!I     t ) $ $D  -44<==339r::(D99txx R006B77    rlist[InstalledPet]cg}ttD]I}|st |j}|r|jr||J|S)zBReturn every installed pet (dirs containing a usable spritesheet).)sortedr0iterdirrSrTrIr"append)outchildpets rinstalled_petsr]~sv C **,,--||~~  uz""  3:  JJsOOO Jrconfigured_slug str | Nonec|r,t|}|r |jr|St}|r|dndS)zResolve which pet to display. Precedence: the configured slug (``display.pet.slug``) if it's installed, otherwise the first installed pet alphabetically, otherwise ``None``. rN)rTrFr"r])r^r\r)s rresolve_active_petrasY ,,..//  3: J   D $477$rF)forcetimeoutrbrrcfloatcxddlm}t|}|stdt |}|r |jr|s|S|||}|td|dt |jstd|d t|z }| d d |j d d d rd nd}|d|z }t|j||i} |jr_t |jrK t|j|} n3#t $r&} t"d|| Yd} ~ nd} ~ wwxYwt'| t(r| s ||jdd} |j| d<| d|| d|j|dz t3j| ddt |} | | jstd|d| S)zDownload *slug* from the manifest into the pets directory. Idempotent: a fully-installed pet is returned as-is unless *force*. Raises :class:`PetStoreError` / :class:`~agent.pet.manifest.ManifestError` on failure. r) find_entryzinvalid pet slug)rcNpet 'z' is not in the petdex manifestz*refusing non-petdex spritesheet host for ''Tr*?.pngz.webprz pet.json fetch failed for %s: %sr)idrQrrCrkrQr3indentr4r5z install of '' did not produce a spritesheet)agent.pet.manifestrfrNrrTr"_is_petdex_hostspritesheet_urlr0r-lowersplitendswith _download pet_json_url_download_json Exceptionr<r= isinstancer1rrI setdefault write_textr7dumps) rrbrcrfexistingentryr sprite_ext sprite_pathrAr?r\s r install_petrs.----- d  D 0.///~~HHOE JtW - - -E }IDIIIJJJ 50 1 1RPPPPQQQ T!I OOD4O000 06688>>sCCAFOOPVWWd]dJ8J888K e#['BBBBD Hoe.@AAH H!%"4gFFFDD H H H LL;T3 G G G G G G G G H dD ! !RR5+=bQQ).D OOD$OOM5#5666'' 4(B(B(BW'UUU 4..C {#*{P4PPPQQQ Js:E FE<<FrIctjdd|pdd}|pdS)zFLowercase, hyphenate, and strip a display name into a filesystem slug.z [^a-z0-9]+-rr\)resubrFrs)rIrs rslugifyrsI 6-tzr&8&8&:&:&@&@&B&B C C I I# N ND =5rct|}|}d}t|z r/|d|}|dz }t|z /|S)zGA :func:`slugify` result that doesn't collide with an existing pet dir.rlr)rr0r")rIbasercounters r unique_slugrsv 4==D DG ::  $ $ & &""""1  ::  $ $ & & KrdestNonect|ttfr$|t|dSddlm}t|t tfrC||5}| d}dddn #1swxYwYn| d}| |ddddd dS) zHWrite *source* (PIL image, bytes, or path) as a lossless WebP at *dest*.NrImageRGBAWEBPTd)formatlosslessqualitymethodexact) rzbytes bytearray write_bytesPILrrropenconvertsave)sourcerropenedimages r_write_spritesheetrs&5),-- v'''&3+&&' ZZ   +6NN6**E + + + + + + + + + + + + + + +v&& JJtFT3qPTJUUUUUs8BB!Br)rrrrct|}t|z }|dd|dz } t||n(#t$r}t d|d||d}~wwxYw||p||pd|jdd }|d z tj |d d t|}||j st d|d|S)awWrite a locally-generated pet into the store and return it. *spritesheet* may be a PIL image, raw WebP/PNG bytes, or a path. The pet appears in :func:`installed_pets` immediately, and because :func:`install_pet` returns an already-on-disk pet before consulting the manifest, it can be adopted (``pet.select`` / ``/pet ``) without a manifest entry. Tr*rDz!could not write spritesheet for 'z': Nrr$)rkrQrrCrRr3rlrmr4r5zregister of generated pet 'ro) rr0r-rryrrIr|r7r}rTr") rrrrrrr?rAr\s rregister_local_petrs* 4==D T!I OOD4O00000KY; 4444 YYYNNNNNOOUXXY#+t"(b&+   D'' 4(B(B(BW'UUU 4..C {#*{_$___``` JsA A4A//A4tuple[str, bytes]cddl}ddl}t}||z }|j|ks|std|d|j}| }| |d|j 5}t| D]P}|r:|jds |||d|jQ dddn #1swxYwY|d|fS) uZip an installed pet's folder (pet.json + spritesheet) → (filename, bytes). Dotfiles (cached thumbs, backups) are skipped so the archive is a clean, re-importable pet package. Raises :class:`PetStoreError` if not installed. rNrgz' is not installedwrL/z.zip)iozipfiler0rFresolveparentrSrrIBytesIOZipFile ZIP_DEFLATEDrWrXr startswithwritegetvalue) rrrrootrrIbufarchiver/s r export_petrs IIINNN ::Dtzz||#I!T\\^^339;K;K;M;M3D **,,C c7#7 8 8;G9,,..// ; ;D||~~ ;di&:&:3&?&? ; dt$9$9di$9$9::: ;;;;;;;;;;;;;;;;===#,,.. ((s;A3D;;D?D?`cVtdz }|dd|S)Nz.thumbsTr*)r0r-r.s r _thumbs_dirr3s+ :: !DJJtdJ+++ Krurlcddlm} ||jpd}n#t$rYdSwxYw|dkp|dS)uHTrue only for petdex.dev hosts — bounds server-side fetch (anti-SSRF).r)urlparserFz petdex.devz .petdex.dev) urllib.parserhostnamersr;ru)rrhosts rrqrq9sz%%%%%% &,"3355 uu <  ?4==#?#??s $- ;;g>@) source_urlrcr bytes | Nonec z|}|sdSt|dz }|r% |S#t$rYnwxYwd}t |}|r4|jr- |j}n#t$rd}YnwxYw||r~t|ro ddl }| ||dddi}| |j }n3#t$r&}td||Yd}~nd}~wwxYw|sdS ddl} dd lm} | | |5} | d ddt/t0| jt/t4| jf} t9t:t4zt0z } | t:| f| j} | }| |d |!}dddn #1swxYwYn4#t$r'}td ||Yd}~dSd}~wwxYw |"|n#t$rYnwxYw|S)uReturn a small idle-frame PNG for *slug*, cached on disk. Crops the top-left (idle, frame 0) cell of the spritesheet and downsamples it to a thumbnail. Source preference: an installed spritesheet on disk, else *source_url* — but only when it points at petdex (so the gateway never fetches an arbitrary client-supplied URL). Returns ``None`` when there's no usable source or Pillow/network fails; callers render a placeholder. Doing this server-side sidesteps the renderer's CSP / R2 hotlink limits that break a direct ```` and lets the result ride the authenticated gateway as a same-origin data URL. NrjrT User-Agenthermes-agent-petdexrcfollow_redirectsheaderszthumb fetch failed for %s: %srrPNG)rzthumb crop failed for %s: %s)#rFrr read_bytesr:rTr"rrqhttpxrEraise_for_statuscontentryr<r=rrrrrrcropmin_THUMB_FRAME_Wwidth_THUMB_FRAME_Hheightround_THUMB_WresizeNEARESTrrr)rrrccache sheet_bytesr\rrespr?rrimframerrdatas r thumbnail_pngrDsa ::<2844c.")6T6TUE8n4~EFFFLL(F!3U]CCE**,,C JJs5J ) ) )<<>>D " " " " " " " " " " " " " " "  3T3???ttttt  $      KsA A#"A#B B*)B*:C<< D,D''D,42I &C"I I II II J*J  JJ++ J87J8cJddl}t|}|sdS t|dz dn#t$rYnwxYwt |z }|sdS||d| S)zIDelete an installed pet directory. Returns True if anything was removed.rNFrjT) missing_ok) ignore_errors) shutilrNrunlinkr:r0rSrmtreer")rrrs r remove_petrsMMM d  D u  D &..$.????       T!I     u MM)4M000!! !!s(A AAct|}|pd}|r|sdSt|z }|dz }|sdS t j|d}n#ttf$ri}YnwxYwt|tsi}||d<|}t|}|r||krt|z s | t|z  t|dz t|dz n#t$rYnwxYwt|z }|dz }|}||d<n#t$r|}YnwxYw |t j|d dn#t$rYdSwxYw|S) uRename a pet's ``displayName`` AND realign its slug/dir to match. Generated pets are hatched under a provisional, prompt-derived slug; when the user names the pet on the reveal screen we make that name the real identity so lists/subtitles show what they typed, not the prompt. The dir is renamed to ``slugify(name)`` (and the cached thumbnail moved alongside it) whenever that yields a free, different slug — otherwise the slug is left as is. Returns the resulting slug on success, or ``None`` on failure. rNr3r4r5rQrjrkrlrm)rNrFr0rr7r8r9r:r;rzr1rr"renamerr|r})rrrr>rAnew_slugdesireds r rename_petrs: d  D &B--//L |t T!I:%H     tz(,,g,>>?? Z  dD ! !&DHl##G 7d??HJJ,@+H+H+J+J?    XZZ'1 2 2 2 D.66{}}'GWGWGW7WXXXX     W,I :-HH!DJJ   HHH DJtA666IIII tt OsZ(BBB4$F9EF E FE  F FF+G G Gc ddl} |d||dddi5}|||jdz}|d5}|D]}|| dddn #1swxYwY||ddddS#1swxYwYdS#t$r}td |d ||d}~wwxYw) NrGETTrrrz.partwbzdownload failed for z: ) rstreamr with_suffixsuffixr iter_bytesrreplaceryr) rrrcrrtmpfhchunkr?s rrvrvsLLLJ \\  !!#89      ! ! # # #""4;#899C$ $2!__..$$EHHUOOOO$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ KK                      JJJ?3??#??@@cIJs_CAC)-B# C#B' 'C*B' +C CCCCC D'C==Dcddl}|||dddi}||}t |t r|niS)NrTrrr)rrErr7rzr1)rrcrrrs rrxrxsoLLL 99 45   D   99;;DdD)) 144r1r)rr)rrrr1)rrrAr1rr)rrrr)rrrrO)rrUr)r^r_rrO)rrrbrrcrdrr)rIrrr)rrrr)rrrrrrrr)rrrr)rrrr)rrrrrcrdrr)rrrr)rrrrrr_)rrrrrcrdrr)rrrcrdrr1)*r __future__rr7loggingr dataclassesrpathlibrhermes_constantsr getLoggerr r<_DOWNLOAD_TIMEOUT RuntimeErrorrrr0r@rJrNrTr]rarrrrrrrrrrrqrrrrvrxr rrrs #"""""  !!!!!!,,,,,,  8 $ $)))))L))) $........&****&    &     % % % % %-2DU444444n  V V V V( ######L))))0  @@@@35tHHHHHHV"""".....bJJJJ* 2 2 2 2 2 2r