+jdZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZmZddlmZdd lmZejeZd!d Zd"dZd#dZddd$dZd%dZd&dZd'd ZdS)(u LightClaw — remote file storage client. Mirrors: src/file-storage.ts All binary transfer with ai-server happens here. HTTP contract (reverse-engineered from the TS client): Upload : POST /drive/save Content-Type: multipart/form-data Fields : file (binary) + filePath (string) Auth : Bearer , x-product: channel Success : { code: 0, data: { uploaded: true, ... } } Download : GET /drive/preview?filePath= Auth : Bearer , x-product: channel Success : HTTP 200, body = raw file bytes The server-side ``filePath`` is constructed client-side as ``${Date.now()}/${fileName}`` so that concurrent uploads of the same name never collide. The returned public URL is constructed locally as ``${SERVER_UPLOAD_BASE_URL}${API_PATH_DOWNLOAD}?filePath=`` rather than trusting the server response — matches TS behaviour and means the public URL is deterministic given a known ``filePath``. ) annotationsN)OptionalTuple)quote)API_PATH_DOWNLOADAPI_PATH_UPLOADDOWNLOAD_TIMEOUTSERVER_UPLOAD_BASE_URLUPLOAD_TIMEOUT)guess_mime_by_ext)build_auth_headers file_pathstrreturnc||sdS|dr|Sttdt|dS)zBuild a public ``/drive/preview`` URL for *file_path*. Does not perform any network IO. Mirrors TS ``getFileDownloadUrl``. zhttp://zhttps://z ?filePath=)safe) startswithr rr)rs @/home/agentuser/.hermes/plugins/lightclawbot/src/file_storage.pyget_file_download_urlr4sV r344$ ^&7 ^ ^5Y[C\C\C\ ^ ^^ file_namecRttjdzd|S)zDGenerate the remote ``filePath`` (millisecond timestamp dir + name).i/)inttime)rs r_make_server_file_pathrAs)$)++$%% 3 3 3 33rdatabytesmimeapi_keyTuple[str, str]cKddl}t|}|}|d||||d|tt }t |} |||| |t4d{V } | j dkr<| d{Vdd} td | j d | | d d{V} dddd{Vn#1d{VswxYwYn0#tj$r} td td | d} ~ wwxYwt!| t"std| | dpi}| ddkr(|ddur|t'|fStd| )zInternal: POST /drive/save and return ``(file_path, public_url)``. Raises ``RuntimeError`` on any failure (HTTP error, timeout, server rejection). The caller decides whether to fall back gracefully. rNfile)filename content_typefilePathtotal)r headerstimeoutz Upload HTTP : )r(zUpload timeout after sz#Upload rejected: non-JSON response r codeuploadedTzUpload rejected: )aiohttprFormData add_fieldr r rpost ClientTimeoutr statustext RuntimeErrorjsonasyncio TimeoutError isinstancedictgetr)r rr"r#sessionr3rformurlr,respbodyresultexc data_fields r _do_uploadrIFsNNN&y11I     DNN64)$NGGGNN:y))) # 6_ 6 6C ))G O<< )))??    8 8 8 8 8 8 8 8 {c!!"iikk))))))4C40"#G$+#G#G#G#GHHH99$977777777F 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8  OOOD>DDDEE3NO fd # #MKKKLLLF##)rJ zz&Q:>>*#=#=#E#E/ :::: 3633 4 44s=98D:2A$D(D:( D22D:5D26D::E' E""E')custom_file_name local_pathrJ Optional[str]c K|rtj|std|tj|st d||ptj|}t|}t|d5}| }dddn #1swxYwYt|||||d{V\}} t d|| || fS)aUpload a local file to ai-server. Returns ``(server_file_path, public_url)``. :param local_path: absolute path of the file to upload. :param api_key: tenant API key (Bearer auth). :param session: an ``aiohttp.ClientSession``. :param custom_file_name: override the remote file name. :raises FileNotFoundError: when *local_path* does not exist or is not a regular file. :raises RuntimeError: for any network / server rejection. zFile not found: zNot a regular file: rbNr rr"r#rAu[lightclaw] uploaded %s → %s) ospathexistsFileNotFoundErrorisfiler:basenamer openreadrIloggerinfo) rKr#rArJrr"fppayloadr public_urls rupload_file_to_serverr]zsk& ARW^^J77A ?: ? ?@@@ 7>>* % %@>*>>??? @BG$4$4Z$@$@I Y ' 'D j$  2''))#-  ###Iz KK0*jIII j  s'CC C buffercK|tdt||pd|pd||d{V\}}tdt ||||fS)zKUpload an in-memory buffer. Same contract as :func:`upload_file_to_server`.Nzbuffer must not be Noner&application/octet-streamrOu-[lightclaw] uploaded buffer (%d bytes) → %s)r:rIrXrYlen)r^rr"r#rArr\s rupload_buffer_to_serverrbs~4555", y2F  //###Iz  KK?VjYYY j  rc@Kt|||d{V\}}|S)z;Convenience wrapper: upload and return only the public URL.)r#rAN)r])rKr#rA_rCs rupload_and_get_public_urlresL)GWFAs Jrfile_path_or_urlTuple[bytes, str, str]c,Kddl}|std|dr|nt|}t |} ||||t4d{V }|jdkr<| d{Vdd}td|jd || d{V}|j d d } dddd{Vn#1d{VswxYwYn0#tj $r} td td | d} ~ wwxYw|ddd} | dddpd} || | fS)uGET ``/drive/preview`` for *file_path_or_url*. Returns ``(buffer, file_name, content_type)``. If a full http(s) URL is passed, it's used directly — otherwise it's treated as a remote ``filePath`` and wrapped with :func:`get_file_download_url`. rNzfile_path_or_url is requiredrr*)r,r-r.zDownload HTTP r/z content-typer`zDownload timeout after r0?rrr&)r3r:rrrr@r7r r8r9rWr,r<r=splitrsplit) rfr#rAr3rCr,rDrEbufr(rGtailrs rdownload_file_from_serverrosNNN ;9:::))*ABB 8  %&677!))G S;; ))0@)AA  X X X X X X X X{c!!"iikk))))))4C40"#IDK#I#I4#I#IJJJ ######C<++Nr{s2#""""" """"""""%$$$$$''''''  8 $ $ _ _ _ _4444 -5-5-5-5r'+ #!#!#!#!#!#!L!!!!(    "'('('('('('(r