`gjZS UdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z mZddlZej eZdZdZdZdZd d d d d d d d dZded<iddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:Zd;ed<<dNd=ZdOdPd>ZdQd?ZdRd@ZdSdAZdTdBZdUdCZ dUdDZ!dUdEZ"dUdFZ#ef dVdGZ$ dW dXdHZ% dY dZdIZ& d[dJZ' d\ d]dKZ( d^ d_dLZ)d`dMZ*y)au= yuanbao_media.py — 元宝平台媒体处理模块 提供 COS 上传、文件下载、TIM 媒体消息构建等功能。 移植自 TypeScript 版 media.ts（yuanbao-openclaw-plugin）， 使用 httpx 替代 cos-nodejs-sdk-v5，避免引入额外 SDK 依赖。 COS 上传流程： 1. 调用 genUploadInfo 获取临时凭证（tmpSecretId/tmpSecretKey/sessionToken） 2. 用临时凭证通过 HMAC-SHA1 签名构建 Authorization 头 3. HTTP PUT 上传到 COS TIM 消息体构建： - buildImageMsgBody() → TIMImageElem - buildFileMsgBody() → TIMFileElem ) annotationsN)OptionalAnyz/api/resource/genUploadInfozyuanbao.tencent.com2T) image/jpegz image/jpg image/gif image/png image/bmp image/webp image/heic image/tiffzdict[str, int]_MIME_TO_IMAGE_FORMAT.jpgr .jpeg.pngr.gifr .webpr.bmpr.heicr.tiffr.icoz image/x-iconz.pdfzapplication/pdfz.doczapplication/mswordz.docxzGapplication/vnd.openxmlformats-officedocument.wordprocessingml.documentz.xlszapplication/vnd.ms-excelz.xlsxzAapplication/vnd.openxmlformats-officedocument.spreadsheetml.sheetz.pptzapplication/vnd.ms-powerpointz.pptxzIapplication/vnd.openxmlformats-officedocument.presentationml.presentationz.txtz text/plainzapplication/zipzapplication/x-tarzapplication/gzipz audio/mpegz video/mp4z audio/wavz audio/oggz video/webm)z.zipz.tarz.gzz.mp3z.mp4z.wavz.oggz.webmdict[str, str] _EXT_TO_MIMEctjj|dj}tj |dS)u*根据文件扩展名猜测 MIME 类型。application/octet-stream)ospathsplitextlowerrget)filenameexts R/root/.hermes/venv/lib/python3.12/site-packages/gateway/platforms/yuanbao_media.pyguess_mime_typer*Ys8 ''  8 $R ( . . 0C   C!; <<c|jdrytjj|dj }|dvS)u判断是否为图片类型。image/Tr > rrrrrrrrr) startswithr"r#r$r%)r' mime_typer(s r)is_imager0_sAH% ''  8 $R ( . . 0C ^ ^^r+cJtj|jdS)u 获取 TIM 图片格式编号。r )rr&r%)r/s r)get_image_formatr2gs $ $Y__%6 <Ir;rwidthheightlenstructunpack)bufwhs r)rArAsb 3x"} 2Aw* dC2J'*A dC2J'*A! $$r+c t|dks|ddk7s|ddk7ryd}|t|dz kr||dk7r|dz }||dz}|dvrItjd ||d z|d zd}tjd ||d z|dzd}||d S|d zt|kr)|dtjd ||dz|dzdzz }n y|t|dz kry)Nr rr rr >z>HrIr rL)rPimarkerrRrQs r)rBrBs 3x!|s1v~Q4 A c#hl  q6T> FA QU \ ! dCAq1u$56q9A dCAq1u$56q9A!, , q53s8  V]]4QUAE):;A>> >A   c#hl  r+ct|dkry|ddjdd}|dvrytjd|ddd }tjd|ddd }||d S) N asciireplaceerrors>GIF87aGIF89ac"go#b'T/c"gQUo dC2J/2V;A dC2J/2V;A!, ,  & s8r>c"go==s2bz215D!#A"*&!+A!, ,  & s8r>RCGqL)SW];q@ARCGqL)SW];q@A!, , r+c Kddlm  |std|d fd }|dzdz}tjddd|gi 4d {} |j |d {}t |jjd dxsd}|dkDr||kDrtd |dz dz d d|d|jd|4d {}|j|jjddjddj}g} d} |jd23d {} | t| z } | |kDrtd|d| j!| =7$7 #tj$rYwxYw77Y6d d d d {7n#1d {7swYnxYwdj# } | fcd d d d {7S#1d {7swYy xYww)uR 下载 URL 内容，返回 (bytes, content_type)。 Args: url: HTTP(S) URL max_size_mb: 最大允许大小（MB），超过则抛出异常 Returns: (data_bytes, content_type_string) Raises: ValueError: 内容超过大小限制 httpx.HTTPError: 网络/HTTP 错误 r) is_safe_urlz&Blocked unsafe URL (SSRF protection): responsecK|jrC|jr6t|jj}|st d|yyyw)Nz.Blocked redirect to private/internal address: ) is_redirect next_requeststrurl ValueError)r~ redirect_urlr}s r)_redirect_guardz%download_url.._redirect_guardsZ   H$9$9x44889L|, D\NS-%: sAAig>@T)timeoutfollow_redirects event_hooksNzcontent-lengthu文件过大: z.1fz MB > z MBGET content-type;iu文件过大: 已超过 u MB 限制r+)r~zhttpx.ResponsereturnNone)tools.url_safetyr}rhttpx AsyncClientheadintheadersr&HTTPStatusErrorstreamraise_for_statussplitstrip aiter_bytesrMappendjoin)r max_size_mbr max_bytesclientrcontent_lengthresp content_typechunks downloadedrzr8r}s @r) download_urlrs*- s A#GHHd"T)I  /!23!"!"  S))D !1!12BA!F!K!LN!ny&@ $^d%:T%A#$Ff[MY\]==, % %  ! ! #<<++NB?EEcJ1MSSUL"$FJ#//6 % %ec%j(  )$2;-zJ e$=!"* $$    % %6 % % % % %xx\!C!"!"!"!"!"sA G8 E1G8G#E7'E4(AE78G#FG#A"F'4F8F 9F<5F'1G84E77F  G# F  G#FF' G# F#!G#'F9 -F0.F9 5G# G8GG8#G5)G, *G51G8c ttj}|xs|d|xs||z} tj|j d| j dt j j} td|jD} td|jD} djd| D} djd| D}djd| D}djd | D}d j|j|||d g}t j |j dj}d jd | |d g}tj| j d|j dt j j}d |d| d| d|d| d| S)u 构建 COS 请求签名（q-sign-algorithm=sha1 方案）。 参考：https://cloud.tencent.com/document/product/436/7778 Args: method: HTTP 方法（小写，如 "put"） path: URL 路径（URL encode 后的小写） params: URL 查询参数 dict（用于签名） headers: 参与签名的请求头 dict（key 需小写） secret_id: 临时 SecretId（tmpSecretId） secret_key: 临时 SecretKey（tmpSecretKey） start_time: 签名起始 Unix 时间戳（默认 now） expire_seconds: 签名有效期（秒，默认 3600） Returns: Authorization header 值（完整字符串） rzutf-8c3K|]A\}}|jtjjt |dfCywrsafeNr%urllibparsequoter.0kvs r) z_cos_sign..:s:hQUQRTUAGGIv||'9'9#a&r'9'JLhAA c3K|]A\}}|jtjjt |dfCywrrrs r)rz_cos_sign..;s:jRVRSUVQWWY (:(:3q6(:(KMjrc3&K|] \}}| ywNr>rr_s r)rz_cos_sign..=s:DAqa:&c30K|]\}}|d|yw=Nr>rs r)rz_cos_sign..>s?AQCq*?c3&K|] \}}| ywrr>rs r)rz_cos_sign..?s8A18rc30K|]\}}|d|ywrr>rs r)rz_cos_sign..@s@AQCq*@r rsha1zq-sign-algorithm=sha1&q-ak=z &q-sign-time=z &q-key-time=z&q-header-list=z&q-url-param-list=z &q-signature=) rtimehmacnewencoder4rr6sorteditemsrr%)methodr#paramsr secret_id secret_key start_timeexpire_secondsnow q_sign_timesign_key sorted_paramssorted_headersurl_param_list url_params header_list header_str http_string sha1_of_httpstring_to_sign signatures r) _cos_signrs6 diik C&3'q**;~)M(NOKxx'"7# ik  hY_YeYeYghhMjZaZgZgZijjNXX:M::N???J((888K@@@J))   K<< 2 27 ;<FFHLYY  N g& ik    } {m + ^,  {  $r+c K| t}|jdt}d||xs|dd}|r||d<||ddd } tjd 4d{} | j || | d{} | j | j} dddd{ jd } | dk7r#| !td| d| jdd| jdxs| }ddg}|Dcgc]}|j|r|}}|rtd||S777#1d{7swYxYwcc}ww)u 调用 genUploadInfo 接口获取 COS 临时密钥及上传配置。 Args: app_key: 应用 Key（用于 X-ID 头） api_domain: API 域名（如 https://bot.yuanbao.tencent.com） token: 当前有效的签票 token（X-Token 头） filename: 待上传的文件名（含扩展名） file_id: 客户端生成的唯一文件 ID（不传则自动生成） bot_id: Bot 账号 ID（用于 X-ID 头） Returns: COS 上传配置 dict，包含以下字段： bucketName (str) — COS Bucket 名称 region (str) — COS 地域 location (str) — 上传 Key（对象路径） encryptTmpSecretId (str) — 临时 SecretId encryptTmpSecretKey(str) — 临时 SecretKey encryptToken (str) — SessionToken startTime (int) — 凭证起始时间戳（Unix） expiredTime (int) — 凭证过期时间戳（Unix） resourceUrl (str) — 上传后的公网访问 URL resourceID (str) — 资源 ID（可选） Raises: RuntimeError: 接口返回非 0 code 或字段缺失 N/zapplication/jsonweb) Content-TypezX-TokenzX-IDzX-Sourcez X-Route-EnvlocalDocr)fileNamefileIddocFrom docOpenIdg.@r)jsonrcoderugenUploadInfo 失败: code=z, msg=msgr8 bucketNamelocationu2genUploadInfo 返回字段不完整: 缺少字段 ) r?rstripUPLOAD_INFO_PATHrrpostrrr& RuntimeError)app_key api_domaintokenr'file_idbot_id route_env upload_urlrbodyrrresultrr8required_fieldsfmissings r)get_cos_credentialsrgsH"$%%c*+,<+=>J+!' G !*   D  .--&[[$[HH !%-- ::f D qyT%)$vfjj6K5L M   ::f  'D#Z0O)=Q!q=G=@ J   K'-H---->smAED1ED77D38$D7 E'D5(AEE E E3D75E7E =E >E  Ec K|jdd}|jdd}|jdd}|jdd} |jdd} |jd} |jd} |r|r| s/td t|d t|d t| tr|d } n|d |d} tj j | d}d| d|jd}|r|dk(rt|r t|}nd}t|}t|}| ||d}ttj}| r| n|}| r | |kDr| |z nd}tdd|jdi|||||}|||d}tj!d||| ||t#j$d4d{}|j'|||d{}|j)dddd{| xs|||d}|j+drt-|}|r|d|d<|d |d <tj!d!|d"||S77z7\#1d{7swYlxYww)#u 通过 httpx PUT 请求将文件上传到 COS。 使用临时凭证（tmpSecretId/tmpSecretKey/sessionToken）构建 HMAC-SHA1 签名。 Args: file_bytes: 文件二进制内容 filename: 文件名（用于辅助计算 MIME、UUID） content_type: MIME 类型（如 "image/jpeg"） credentials: get_cos_credentials() 返回的 dict，包含： encryptTmpSecretId → tmpSecretId encryptTmpSecretKey → tmpSecretKey encryptToken → sessionToken location → COS key（对象路径） resourceUrl → 上传后公网 URL startTime → 凭证起始时间（Unix） expiredTime → 凭证过期时间（Unix） bucket: COS Bucket 名称（如 chatbot-1234567890） region: COS 地域（如 ap-guangzhou） Returns: 上传结果 dict，包含： url (str) — COS 公网访问 URL uuid (str) — 文件内容 MD5 size (int) — 文件大小（字节） width (int, optional) — 图片宽度（仅图片） height (int, optional) — 图片高度（仅图片） Raises: httpx.HTTPStatusError: COS 返回非 2xx 状态 RuntimeError: credentials 字段缺失 encryptTmpSecretIdrencryptTmpSecretKey encryptTokenr resourceUrl startTime expiredTimeu$COS credentials 不完整: secretId=z , secretKey=z , location=z.cos.accelerate.myqcloud.comz.cos.z .myqcloud.comrrzhttps://r!)hostrx-cos-security-tokenput)rr#rrrrrr) Authorizationrrz3COS PUT: bucket=%s region=%s key=%s size=%d mime=%sg^@rN)contentr)ruuidsizer-rJrKu COS 上传成功: url=%s size=%dr)r&rboolCOS_USE_ACCELERATErrrlstripr0r*r9rMrrrloggerinforrr rr.rE) file_bytesr'r credentialsbucketregionrr session_tokencos_key resource_urlr expired_timecos_host encoded_keycos_url file_uuid file_size sign_headersr sign_start sign_expire authorization put_headersrrr size_infos r) upload_to_cosr&sN!__%92>I!oo&;R@J$)rr'r r r1s r)build_file_msg_bodyr6os3&  HE&%!    r+c tjj|}tjj |jS#t $rYywxYw)u从 URL 提取文件名。r)rrurlparser"r#basename Exception)rparseds r)r0r0sF&&s+ww ,, sAA AA)r'rrr)r)r'rr/rrr)r/rrr)r8bytesrr)rr)r8r<rOptional[dict[str, int]])rPr<rr=)rrrrrztuple[bytes, str])Nr)rrr#rrrrrrrrrrz Optional[int]rrrr)fileNrr)rrrrrrr'rr Optional[str]rrrrrdict)rr<r'rrrrr@rrrrrr@)NNrrrr)rrr r?r'r?r rrJrrKrr/rr list[dict])Nr) rrr'rr r?r rrrA)rrrr)+__doc__ __future__rr4rloggingr"r<rNr urllib.parsertypingrrr getLogger__name__rrDEFAULT_API_DOMAINDEFAULT_MAX_SIZE_MBrr__annotations__rr*r0r2r9r?rErArBrCrDrrrr&r2r6r0r>r+r)rLs"#      8 $1*  )~   L  \  K  K    \    K   \  \  N      V  &  P  +   X! " L# $        3  n>= _= ) ! %(%:+D" D"D"D"`!%O O O O O  O  OOO Op!K KK K K  K  KK K\EEEE E  E  E EX"- - --  -  -  ---f       Hr+