Uj; dZddlmZddlZddlZddlZddlZddlZddl Zddl Zddl m Z ddl m Z ddlmZe ej#j$j$Ze ej(j+deej$dz Zej(j+d d Ze ej(j+d ee j2d z d z dz dz ZdZd#dZd$dZ d% d&dZd'dZd(dZ d)dZ!e"dZ#e"dZ$d*dZ%d+dZ&d'dZ'd*dZ(d,d-dZ)d*dZ*d,d.dZ+d*d Z,d'd!Z-d"Z.y)/u~Storage HUD — a fast, lazy file browser over Supabase Storage. The bucket 'jarvis-files' (on Ahmed's Supabase) is the WRITABLE shared file store that any Jarvis / Claude on any device can read, save, and delete from. This module is the engine side of the HUD's "pull up the storage" panel. DESIGN RULE (the whole point): the LIST is METADATA-ONLY. `list()` / `show()` fetch names + type + size + updated timestamp and NOTHING ELSE — no file bodies. A file is downloaded ONLY when Ahmed actually opens it (`open_file` / `download`), and even then it's cached to disk and re-fetched ONLY when the remote copy is newer. This is deliberately unlike the memory graph, which was slow because it bulk-loaded everything. Ports the proven REST logic from jarvis-tools-service/storage.py (stdlib urllib, no new deps). Endpoints under {SUPABASE_URL}/storage/v1/...: POST /object/list/{bucket} → metadata list (name, updated_at, metadata.size…) GET /object/{bucket}/{name} → the file body (only on open) POST /object/{bucket}/{name} → upload (x-upsert) DELETE /object/{bucket}/{name} Creds come from the process env, else the Ultron repo's .env (the single source of truth — SUPABASE_URL + SUPABASE_SECRET, the service_role key). MCP_STORAGE=0 disables the whole thing. Engine → HUD event (voice.events.emit): storage {files:[{name,type,size,updated}]} — open / refresh the panel ) annotationsN)datetime)Path)emit ULTRON_ENVz.envSTORAGE_BUCKETz jarvis-filesJARVIS_STORAGE_CACHELibraryzApplication SupportJarvisstoragez.emptyFolderPlaceholderc2tjjdd}tjjdd}|r|stj r tj j D]r}|j}|jdsd|vr)|jd\}}}|j|j}}|dk(r|s|}h|dk(sn|rq|}t |r|sy|jd|fS#t$rY#wxYw)zG(base_url, service_key) from env, falling back to the Ultron repo .env. SUPABASE_URLSUPABASE_SECRET#=N/) osenvironget _ULTRON_ENVexists read_text splitlinesstrip startswith partition Exceptionrstrip)urlkeylinek_vs C/Users/ahmed/devFolder/Ultron/claude-voice/voice/storage_control.py_credsr';s **.. ,C **..*B /C C[//1 #--/::<zz|??3'3d?..-1awwy!'')1&sC++CC= C ::c?C     sBD *D -D DDcbtjjdddk(rytduS)zBOn when the Supabase service creds are available and not disabled. MCP_STORAGE10FN)rrrr'r&enabledr.Rs) zz~~mS)S0 84 r-c t}|s td|\}}tjj |d|||}|j dd||j d||xsij D]\} } |j | |  tjj||5} | jcdddS#1swYyxYw#tjj$rA} td| jd | jdd jd d d} ~ wwxYw) Nz/storage not configured (no SUPABASE_URL/SECRET)z /storage/v1)datamethod AuthorizationzBearer apikey)timeoutzHTTP z: utf-8ignore) r' RuntimeErrorurllibrequestRequest add_headeritemsurlopenreaderror HTTPErrorcodedecode) r1pathbodyheadersr4credsbaser!reqr#r%res r&_dorL]s& HE LMMID# .. D6TF!;$(. ! 0CNN?gcUO4NN8S!B%%'1 q!(J ^^ # #C # 9Q668: 9 9 << ! !JU166("ffhtn33GXFGIJ JJs6!C!;C C!CC!C!!D?> u @@ @s2B ct|jjdj}|r|S|xsij dxst j |dxsd}d|vr|jddSdS)zA short type tag for the row icon: the extension if any, else the mime subtype (e.g. 'pdf', 'png', 'plain'). Lowercased, no dot..mimetyperrr)rsuffixlstriplowerr mimetypes guess_typesplit)rOmetaextmimes r&_kindrl~s{ t*   " "3 ' - - /C  JB  J ' N9+?+?+Ea+H NBD"%+4::c?2 525r-c|sy tj|jddjS#t$rYywxYw)znParse an ISO-8601 timestamp (with 'Z' / fractional seconds) to epoch seconds. 0.0 on anything unparseable.gZz+00:00)r fromisoformatreplace timestampr)isos r&_epochrssF %%ckk#x&@AKKMM s28 AAzCabcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789.-() 0123456789abcdefct|jjd}djd|j dDS)Nrrc3`K|]&}t|tvr t|nd|d(yw)r$02xN)chr_KEEP).0bs r& z_enc_key..s320q!Vu_3q6AaW+=0s,.r6)strrrdjoinrZrOs r&_enc_keyrsD t9??  # #C (D 772 KK02 22r-c`t}dt|}}||kr||dk(rN|dz|krF||dztvr8||dztvr*|jt ||dz|dzd|dz }n%|jt ||dz|dz }||kr|j dd S) Nrr$r6rp) bytearraylen_HEXappendintordrC)r!r{r]ns r& _dec_namers A c#hqA a% q6S=QUaZC!H,<QqSTAQ HHSQqS1r* + FA HHSQ[4' ( FA a% 88GY ''r-c g}tD]}|jdxsi}|jd}||jd}t|d}|j|t ||t |t tfr t |nd|jdxs|jdxsdd |S) zdMetadata-only listing: [{name, type, size, updated}] sorted by name. NEVER fetches file content.metadatasize contentLengthrOr updated_at lastModifiedr)rOtyperupdated)r^rrrrl isinstancerfloat)outr]rirdisplays r&listrsC [uuZ &Bxx <88O,DAfI& '4(!+D3,!?CIQuu\*Ldhh~.FL"    Jr-ct|}tD]U}|jd|k(s|jdxsi}|jdxs|jdxsdcSy)z*BHb H r-c t|jjd}|s tdt|z }||n t |}t |}|jr-|r+|jj|dz k\r t|Stddtdtjjt|d}|j j#dd|j%||r$ t'j(|||ft|St|S#t*$r Yt|SwxYw) zDownload ONE file to the local cache and return its path. Re-downloads only when the remote copy is newer than the cached one (compared via the remote updated_at vs the cached file's mtime). This is the ONLY path that ever pulls bytes.rzdownload: need a file namerGET/object/safeT)parentsexist_ok)r}rrd ValueErrorCACHErrsis_filestatst_mtimerLrXr9parsequoterparentmkdir write_bytesrutimeOSError)rOrdest remote_iso remote_tsr0s r&downloadrs$ t9??  # #C (D 566 4D: <<8 #   fd^ , K WW_ LL  K   j$/ 0 K : )#a&#,899 K:s"*A=(A=#A== B+B&&B+c tjjt|j }|rtjj |st d||r(t|j jdndxstjj|}tj|dxsd}t|d5}|j}dddtdd tdtj j#t%|d |d d |S#1swYNxYw) zvUpload a LOCAL file into the bucket (upsert). `name` defaults to the file's own basename. Returns the stored name.z no file at rrrzapplication/octet-streamrbNrNrrtrue)rVzx-upsert)rrD expanduserr}risfileFileNotFoundErrorrdbasenamerfrgrr?rLrXr9rrr) local_pathrOlprkfrEs r&uploadrs   C O113 4B RWW^^B'+bV 455-1CIOO  $ $S )r Rbgg>N>Nr>RD    %a ( F,FD b$1vvx (6(!FLL$6$6x~C$6$P#QRTX624 K s D99Ec @t|jjd}|s tdt ddt dt jjt|d t|z jd|S#t$rY|SwxYw)z8Remove a file from the bucket. Returns the deleted name.rzdelete: need a file nameDELETErrT) missing_ok) r}rrdrrLrXr9rrrrunlinkrrs r&deleters t9??  # #C (D 344HVHAfll&8&8$c&8&R%STU . K   K s5B BBc t}td||S#t$r3}tdt |ddtdggcYd}~Sd}~wwxYw)uList (metadata only) and push a `storage` event so the HUD opens/refreshes the file panel. Returns the same list for the caller. Never raises — a viewer refresh must not crash the assistant.r )filesz" [storage] panel refresh failed: Nr)rrrrr})rrKs r&showrsV Ye$  23q6$3<.AB Yb! s A(A AAcddlddlm}m}dd|ddidfd }|ddd tidfd }|d d ttd dfd }|ddd tidfd }|dd||||gS)Nr)toolcreate_sdk_mcp_servercdd|dgiS)Ncontenttext)rrr,)msgs r&_textzbuild_server.._text-sVS9:;;r- storage_showuBPull up Ahmed's file storage on the HUD and list what's there. Use for 'pull up the storage', 'show my files', 'open storage', 'what files do I have'. Opens a fast panel (names + types only — nothing is downloaded) and returns the list. He can then open/delete files there, or you can with storage_open / storage_delete.ctK jtd{}|sdS|Dcgc]'}d|d|jdr d|ddndz)}}t|dd j |zS7fcc}w#t $r!}d t |dd cYd}~Sd}~wwxYww) NzStorage is empty.z- rOrz (z bytes)rz# file(s) in storage (panel is up):  zstorage_show failed: ) to_threadrrrr~rr})argsrrlinesrKrasyncios r&rz"build_server..storage_show0s A!++D11E011$%#!F)%45EE&M#ai[0rK# %CJ<'KL99U+,- - 2%  A0Q?@ @ As_B8B B B B8B ,B'B B8B B B5B0*B5+B80B55B8 storage_openuOpen ONE stored file by name — downloads it (cached; only re-fetched if it changed) and opens it in its default app so Ahmed can view or edit it. `name` is the exact file name from storage_show.rOc&Kt|jddj}|sdS jt|d{d|dS7#t $r!}dt|ddcYd}~Sd}~wwxYww)NrOrz&storage_open failed: need a file name.zOpened r`zstorage_open failed: r)r}rrrrrrrOrKrrs r&rz"build_server..storage_openCs 488FB'(..0AB B A##It4 4 474&*+ + 5 A0Q?@ @ AsF4BA$A"A$!B"A$$ B-B BB BBstorage_uploadzUpload a LOCAL file (by absolute path on this Mac) into Ahmed's storage so it syncs everywhere. `local_path` = the path; `name` = optional name to store it as (defaults to the file's own name). Refreshes the HUD panel.)rrOc Kt|jddj}|sdS jt|t|jddjxsdd{}jt d{d|dS7-7#t $r!}dt|ddcYd}~Sd}~wwxYww) Nrrz)storage_upload failed: need a local_path.rOz Uploaded z to storage.zstorage_upload failed: r)r}rrrrrr)rrrOrKrrs r&rz$build_server..storage_uploadRs ,+ , 2 2 4DE E C **C 45;;=EGGD##D) ) )9TF,78 8G ) C23q6$3<.AB B CsY4CAB/=B+>B/B-B/*C+B/-B// C8CCCCCstorage_deletezXDelete a file from Ahmed's storage by name (from storage_show). Refreshes the HUD panel.cdKt|jddj}|sdS jt|d{jt d{d|dS7-7#t $r!}dt|ddcYd}~Sd}~wwxYww)NrOrz(storage_delete failed: need a file name.zDeleted z from storage.zstorage_delete failed: r)r}rrrrrrrs r&rz$build_server..storage_deletecs488FB'(..0CD D C##FD1 1 1##D) ) )8D689 9 2 ) C23q6$3<.AB B CsX4B0BA?B.B/B>B0?BB B- B("B-#B0(B--B0r z1.0.0)rOversiontools)rr}returndict)rrrr)rclaude_agent_sdkrrr})rrrrrrrrs @@r& build_serverr(s<< . ;=? @ A @ A . F 3-  A  A  %69#&F H C H C  %'-sm5 C5 C ! \>>J r-)rztuple[str, str] | None)rbool)NN) r1r}rDr}rEz bytes | NonerFz dict | Noner4rrbytes)rz list[dict])rOr}rirrr})rrr}rr)rOr}rr})r!r}rr})N)rOr}r str | Nonerr})rr}rOrrr})/__doc__ __future__rrfrrr urllib.errorr9 urllib.parseurllib.requestrpathlibr voice.eventsr__file__resolver_PROJECTrrr}rrXhomerr\r'r.rLr^rlrs frozensetryrrrrrrrrrrrr,r-r&rsk6#   > ! ! # * * 1 12::>>,"%hoo&>"?AB (. 9 RZZ^^  i"77(BYNOQ R)  . 6:46JJ.1J;@J( @6 I K#$2 ("&0    $Lr-