`gj]BdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z eje ZdZGdd eZed Gd d Zd+dZd,dZd-dZd.dZd/dZd0dZd1d2dZdedd3dZd4dZd4dZd5dZddd d6dZd7dZdZ d Z!d!Z"d+d"Z#d8d#Z$dd$d%d9d&Z%d:d'Z&d;d(Z'dupOn-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@ceZdZdZy) PetStoreErrorzRaised on install/IO failures.N)__name__ __module__ __qualname____doc__B/root/.hermes/venv/lib/python3.12/site-packages/agent/pet/store.pyrr s(rrT)frozencveZdZUdZded<ded<ded<ded<ded<d Zded <edd Zedd Zy ) InstalledPetzA pet present on disk.strslug display_name descriptionr directory spritesheet created_byc6|jjSN)ris_fileselfs rexistszInstalledPet.exists/s''))rc |jdk(S)N generator)rrs r generatedzInstalledPet.generated3s+--rN)returnbool) r r r r __annotations__rpropertyr r#r rrrr$sQ IOJ **..rrcFtdz }|jdd|S)z=Return the profile-scoped pets directory (created on demand).petsTparentsexist_ok)rmkdirpaths rpets_dirr08s$  v %DJJtdJ+ Krc|dz }|jsiS tj|jdS#tt f$r#}t jd||icYd}~Sd}~wwxYw)Npet.jsonutf-8encodingzunreadable pet.json in %s: %s)rjsonloads read_textOSError ValueErrorloggerdebug)rpet_jsonexcs r_read_pet_jsonr??sj:%H     zz(,,g,>?? Z  4iE s$>A0 A+%A0+A0ct|jddxsdj}|r||z }|jr|SdD]}||z }|js|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.pngrB)rgetstripr)rmetadeclared candidatenames r_resolve_spritesheetrIJs 488-r28b9??AH(     T$      ) ))rchtt|jj}|dvry|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)rrrDrH)rsegments r _safe_slugrM]s/3t9??$%**G/! Nrc Tt|}|syt|z }|jsyt|}t |t |j ddxs|t |j ddxsd|t||t |j ddxsdS)zCReturn the :class:`InstalledPet` for *slug*, or ``None`` if absent.N displayNamerr createdBy)rrrrrr)rMr0is_dirr?rrrCrI)rrrEs rload_petrRks d D  T!I     ) $D  -4<=39r:(D9txx R06B7  rcg}ttjD]I}|jst |j }|s,|j s9|j|K|S)zBReturn every installed pet (dirs containing a usable spritesheet).)sortedr0iterdirrQrRrHr append)outchildpets rinstalled_petsrZ~s[ C **,-||~ uzz" 3:: JJsO  Jrc~|r)t|j}|r|jr|St}|r|dSdS)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)rRrDr rZ)configured_slugrYr)s rresolve_active_petr]sA ,,./ 3::J  D47$$rF)forcetimeoutc,ddlm}t|}|s tdt |}|r|j r|s|S|||}|td|dt |jstd|d t|z }|jd d |jjjd djd rd nd}|d|z }t|j||i} |jr-t |jr t|j|} t'| t(r| s||j*dd} |j,| d<| j/d|| j/d|j*|dz j1t3j4| ddt |} | | j std|d| S#t $r!} t"j%d|| Yd} ~ d} ~ wwxYw)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)r_Npet 'z' is not in the petdex manifestz*refusing non-petdex spritesheet host for ''Tr*?.pngz.webprz pet.json fetch failed for %s: %sr)idrOrrArfrOr2indentr3r4z install of '' did not produce a spritesheet)agent.pet.manifestrarMrrRr _is_petdex_hostspritesheet_urlr0r-lowersplitendswith _download pet_json_url_download_json Exceptionr;r< isinstancedictrrH setdefault write_textr6dumps) rr^r_raexistingentryr sprite_ext sprite_pathrEr>rYs r install_petr~s. d D .//~HHOOE tW -E }eD6)HIJJ 500 1HaPQQ T!I OOD4O0 00668>>sCAFOOPVW]dJ J<88K e##['BD oe.@.@A H!%"4"4gFD dD !5+=+=bQ)..D OOD$OOM5#5#56'' 4(BW'U 4.C {#**l4&0OPQQ J H LL;T3 G G HsG)) H2HHctjdd|xsdjjjd}|xsdS)zFLowercase, hyphenate, and strip a display name into a filesystem slug.z [^a-z0-9]+-rrY)resubrDrn)rHrs rslugifyrs? 66-tzr&8&8&:&@&@&B C I I# ND =5rct|}|}d}t|z jr(|d|}|dz }t|z jr(|S)zGA :func:`slugify` result that doesn't collide with an existing pet dir.rgr)rr0r )rHbasercounters r unique_slugrs] 4=D DG :  $ $ &q "1  :  $ $ & Krc^t|ttfr|jt|yddlm}t|t tfr,|j|5}|jd}dddn|jd}j|ddddd y#1swY!xYw) zHWrite *source* (PIL image, bytes, or path) as a lossless WebP at *dest*.NrImageRGBAWEBPTd)formatlosslessqualitymethodexact) rubytes bytearray write_bytesPILrrropenconvertsave)sourcedestropenedimages r_write_spritesheetrs&5),- v'&3+& ZZ  +6NN6*E + +v& JJtFT3qPTJU  + +s B##B,r)rrct|}t|z }|jdd|dz } t||||xs||xsd|j dd }|d z jtj|d d t|}| |jst d|d|S#t$r}t d|d||d}~wwxYw)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*rBz!could not write spritesheet for 'z': Nrr")rfrOrrArPr2rgrhr3r4zregister of generated pet 'rj) rr0r-rrtrrHrxr6ryrRr ) rrrrrr}r>rErYs rregister_local_petrs 4=D T!I OOD4O000KY; 4 #+t"(b&++  D'' 4(BW'U 4.C {#**9$?^_`` J Y?vSNOUXXYs B-- C 6CC cddl}ddl}t}||jz }|j j |j k7s|j std|d|j}|j}|j|d|j5}t|jD]P}|js|jjdr0|j!||d|jR ddd|d|j#fS#1swYxYw) 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. rNrbz' is not installedwrK/z.zip)iozipfiler0rDresolveparentrQrrHBytesIOZipFile ZIP_DEFLATEDrTrUr startswithwritegetvalue) rrrrootrrHbufarchiver/s r export_petrs   :Dtzz|#I!!T\\^39;K;K;MeD6);<== >>D **,C c7#7#7 8;G9,,./ ;D||~dii&:&:3&? dtfAdii[$9: ;;V4=#,,. (( ;;s)-D4D43#D44D=`cFtdz }|jdd|S)Nz.thumbsTr*)r0r-r.s r _thumbs_dirr3s# : !DJJtdJ+ Krcddlm} ||jxsdj}|dk(xs|j dS#t$rYywxYw)uHTrue only for petdex.dev hosts — bounds server-side fetch (anti-SSRF).r)urlparserFz petdex.devz .petdex.dev) urllib.parserhostnamernr:rp)urlrhosts rrlrl9sV% &&,"335 <  ?4==#?? s$A AAg>@) source_urlr_c |j}|syt|dz }|jr |jSd}t |}|r'|j r |jj}|E|rCt|r8 ddl }|j||dddi}|j|j}|sy ddl} dd lm} | j'| j)|5} | j+d j-ddt/t0| j2t/t4| j6f} t9t:t4zt0z } | j=t:| f| j>} | j)}| jA|d |jC}ddd |jE|S#t$rYwxYw#t$rd}YvwxYw#t$r"}tjd||Yd}~Yd}~wwxYw#1swYmxYw#t$r!}tjd ||Yd}~yd}~wwxYw#t$rYSwxYw)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. NrerT User-Agenthermes-agent-petdexr_follow_redirectsheaderszthumb fetch failed for %s: %srrPNG)rzthumb crop failed for %s: %s)#rDrr read_bytesr9rRr rrlhttpxrCraise_for_statuscontentrtr;r<rrrrrrcropmin_THUMB_FRAME_Wwidth_THUMB_FRAME_Hheightround_THUMB_WresizeNEARESTrrr)rrr_cache sheet_bytesrYrrespr>rrimframerrdatas r thumbnail_pngrDsA ::2884c."))6TUE8n4~EFFLL(F!3U]]CE**,C JJs5J )<<>D "  $ Kg    K  E LL8$ D D E " "  3T3?   K s~G!G# 7G5*H/0CH#7H/I G G # G21G25 H >HH #H,(H// I8II I)(I)c ddl}t|}|sy t|dz jdt |z }|j sy|j|d|j S#t$rYMwxYw)zIDelete an installed pet directory. Returns True if anything was removed.rNFreT) missing_ok) ignore_errors) shutilrMrunlinkr9r0rQrmtreer )rrrs r remove_petrs d D   D6 &..$.? T!I     MM)4M0! !!    s A66 BBct|}|xsdj}|r|syt|z }|dz }|jsy t j |j d}t|tsi}||d<|}t|}|r||k7rt|z jsd |jt|z  t|dz jt|dz t|z }|dz }|}||d< |jt j |d d|S#ttf$ri}YwxYw#t$rYfwxYw#t$r|}Y^wxYw#t$rYywxYw) 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. rNr2r3r4rOrerfrgrh)rMrDr0rr6r7r8r9r:rurvrr renamerrxry)rrrr=rEnew_slugdesireds r rename_petrs d D &B--/L | T!I:%H    zz(,,g,>? dD !&DHl#G7d?HJ,@+H+H+J    XZ'1 2 D6.66{}'RVGW7WX! W,I :-HH!DJDJJtA6I O5 Z     H  sZ %E5E)-E?E)(E:EE E&#E)%E&&E)) E76E7: FFcddl} |jd||dddi5}|j|j|jdz}|j d5}|j D]}|j| ddd|j|dddy#1swY#xYw#1swYyxYw#t$r}td |d ||d}~wwxYw) NrGETTrrrz.partwbzdownload failed for z: ) rstreamr with_suffixsuffixr iter_bytesrreplacertr) rrr_rrtmpfhchunkr>s rrqrqsJ \\  !!#89     ! ! #""4;;#89C$ $2!__.$EHHUO$ $ KK    $ $   J23%r#?@cIJsMCAB4'B(B4C(B1 -B44B=9C=C C  CC cddl}|j||dddi}|j|j}t |t r|SiS)NrTrrr)rrCrr6rurv)rr_rrrs rrsrssW 99 45  D   99;DdD)41r1r)r$r)rrr$rv)rrrErvr$r)rrr$r)rrr$InstalledPet | None)r$zlist[InstalledPet]r)r\ str | Noner$r)rrr^r%r_floatr$r)rHrr$r)rrr$None)rrrrrrr$r)rrr$ztuple[str, bytes])rrr$r%)rrrrr_rr$z bytes | None)rrr$r%)rrrrr$r)rrrrr_rr$r)rrr_rr$rv)*r __future__rr6loggingr dataclassesrpathlibrhermes_constantsr getLoggerr r;_DOWNLOAD_TIMEOUT RuntimeErrorrrr0r?rIrMrRrZr]r~rrrrrrrrrrlrrrrqrsr rrrs #  !,   8 $)L) $...&*& &  %-2DU4n  V( # # #  #  #L)0  @35tHV"..bJ* 2r