i'P UdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z mZddlZejeZdZdZdZdZd 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<<dyd@Zdzd{dDZd|dFZd}dIZd~dJZddLZddNZ ddOZ!ddPZ"ddQZ#efddUZ$ ddd`Z% dddjZ&ddpZ' dddvZ( dddwZ)ddxZ*dS)u= 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_MIMEfilenamestrreturnctj|d}t|dS)u*根据文件扩展名猜测 MIME 类型。application/octet-stream)ospathsplitextlowerrget)rexts D/home/ubuntu/.hermes/hermes-agent/gateway/platforms/yuanbao_media.pyguess_mime_typer,Ys? '  8 $ $R ( . . 0 0C   C!; < << mime_typeboolc|drdStj|d}|dvS)u判断是否为图片类型。image/Tr#> rrrrrrrrr) startswithr%r&r'r()rr/r*s r+is_imager4_sOH%%t '  8 $ $R ( . . 0 0C ^ ^^r-intc\t|dS)u 获取 TIM 图片格式编号。r )rr)r()r/s r+get_image_formatr7gs" $ $Y__%6%6 < <s r+parse_image_sizerLysF  " D ! ! " 4  " D ! ! r-bufct|dkrdS|dddkrdStjd|ddd}tjd|ddd}||dS) Nr sPNGz>IrArwidthheightlenstructunpack)rMwhs r+rHrHsx 3xx"}}t 2A2w*t dC2J''*A dC2J''*A! $ $$r-c :t|dks|ddks |ddkrdSd}|t|dz kr||dkr|dz }(||dz}|dvrWtjd ||d z|d zd}tjd ||d z|dzd}||d S|d zt|kr0|dtjd ||dz|dzdzz }nn|t|dz kdS)Nr rr rr )z>HrQr rT)rMimarkerrYrXs r+rIrIs> 3xx!||s1v~~Q4t A c#hhl   q6T>> FA QU \ ! ! dCAq1u $566q9A dCAq1u $566q9A!,, , q53s88   V]4QUAE\):;;A>> >AA  c#hhl   4r-c t|dkrdS|dddd}|dvrdStjd|ddd }tjd|ddd }||d S) N asciireplaceerrors)GIF87aGIF89a>c"goo#b'T//c"gQUoo dC2J//2V;A dC2J//2V;A!,, , & s88r>>c"goo=s2b5z2215D!#A"*&!+A!,, , & s88r>>RCGqL)SW];q@ARCGqL)SW];q@A!,, , 4r-url max_size_mbtuple[bytes, str]cK|dzdz}tjdd4d{V} ||d{V}t|jddpd}|dkr#||krt d|dz dz d d |d n#tj$rYnwxYw|d |4d{V}| |jd d dd }g}d} | d23d{V} | t| z } | |krt d|d|| H6 dddd{Vn#1d{VswxYwYd|} | |fcdddd{VS#1d{VswxYwYdS)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 错误 ig>@T)timeoutfollow_redirectsNzcontent-lengthru文件过大: z.1fz MB > z MBGET content-typer.;iu文件过大: 已超过 u MB 限制r-)httpx AsyncClientheadr5headersr) ValueErrorHTTPStatusErrorstreamraise_for_statussplitstrip aiter_bytesrUappendjoin) rr max_bytesclientrcontent_lengthresp content_typechunks downloadedrr8s r+ download_urlrsd$d"T)I EEE""""""" S))))))))D !1!12BA!F!F!K!LLN!!ny&@&@ ]^d%:T%A]]][]]]$    D ==,, % % % % % % %  ! ! # # #<++NB??EEcJJ1MSSUUL"$FJ#//66 % % % % % % %ec%jj(  ))$J;JJJ e$$$$ 76 % % % % % % % % % % % % % % % % % % % % % % % % % % %xx\!;""""""""""""""""""""""""""""""sgGA.BGB*'G)B**G A4F=FAFG F# #G&F# 'G G"Gmethodr&paramsr secret_id secret_key start_time Optional[int]expire_secondsc ttj}|p|d|p||z} tj|d| dt j} td| D} td| D} d d| D} d d| D}d d| D}d d | D}d | |||d g}t j|d}d d | |d g}tj| d|dt 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|]I\}}|tjt |dfVJdSr.safeNr(urllibparsequoter .0kvs r+ z_cos_sign..&sQhhQUQRTUAGGIIv|'9'9#a&&r'9'J'JLhhhhhhr-c3K|]I\}}|tjt |dfVJdSrrrs r+rz_cos_sign..'sQjjRVRSUVQWWYY (:(:3q66(:(K(KMjjjjjjr-c3 K|] \}}|V dSNrDrr_s r+rz_cos_sign..)s&::DAqa::::::r-&c3*K|]\}}|d|VdS=NrDrs r+rz_cos_sign..*s0??AQ****??????r-c3 K|] \}}|V dSrrDrs r+rz_cos_sign..+s&88A1888888r-c3*K|]\}}|d|VdSrrDrs r+rz_cos_sign..,s0@@AQ****@@@@@@r- r.sha1zq-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=) r5timehmacnewencoder;rr=sorteditemsrr()rr&rrrrrrnow 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_signrsy6 dikk  C&3OO**;~)MOOKx'""7## ikk  hhY_YeYeYgYghhhhhMjjZaZgZgZiZijjjjjNXX::M:::::N???????J((8888888K@@@@@@@J))   K< 2 27 ; ;<<FFHHLYY  N  g&& ikk  $ $ $# $ $# $ $&  $ $ ,  $ $ "  $ $r-fileapp_key api_domaintokenfile_id Optional[str]bot_id route_envdictc K|t}|dt}d||p|dd}|r||d<||ddd } tjd 4d{V} | || | d{V} | | } dddd{Vn#1d{VswxYwY| d } | dkr+| )td| d| dd| dp| ddg}fd|D}|rtd|S)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 bucketNamelocationc>g|]}||SrD)r))rfr8s r+ z'get_cos_credentials..s(===Q!=q===r-u2genUploadInfo 返回字段不完整: 缺少字段 ) rErstripUPLOAD_INFO_PATHrrpostrrr) RuntimeError)rrrrrrr upload_urlrbodyrrresultrrequired_fieldsmissingr8s @r+get_cos_credentialsrSsIH"$$%%c**>,<>>J+!' G +!*    D ...-------&[[$[HHHHHHHH !%--------------------------- ::f  D qyyT% M$ M Mfjj6K6K M M    ::f   'D#Z0O====/===G  J J J    Ks$AB== C C file_bytesr credentialsbucketregionc K|dd}|dd}|dd}|dd} |dd} |d} |d} |r|r| s?td t|d t|d t| tr|d } n|d |d} tj| d}d| d|d}|r|dkr!t|rt|}nd}t|}t|}| ||d}ttj }| r| n|}| r | |kr| |z nd}tdd|di|||||}|||d}td||| ||t#jd4d{V}||||d{V}|dddd{Vn#1d{VswxYwY| p|||d}|dr't-|}|r|d|d<|d |d <td!|d"||S)#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 字段缺失 encryptTmpSecretIdr.encryptTmpSecretKey encryptTokenr resourceUrl startTime expiredTimeu$COS credentials 不完整: secretId=z , secretKey=z , location=z.cos.accelerate.myqcloud.comz.cos.z .myqcloud.comrrzhttps://r$)hostrx-cos-security-tokenrput)rr&rrrrrr) Authorizationrr z3COS PUT: bucket=%s region=%s key=%s size=%d mime=%sg^@rN)contentr)ruuidsizer2rRrSu COS 上传成功: url=%s size=%dr)r)rr0COS_USE_ACCELERATErrrlstripr4r,r?rUr5rrloggerinforrr rr3rL)rrrrrrrr 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"s8N!__%92>>I!oo&;R@@J$<rCs"#"""""     8 $ $1*  ) )      L  \  K  K    \    K   \  \  N      V  &  P  +   X! " L# $        3    >==== _____==== )))) !!!!    %%%%(%%%%:+0"0"0"0"0"x!%OOOOOp!KKKKK\EEEEX"-----f Hr-