eiW3 dZddlZddlZddlZddlmZmZmZeje Z ej Z dZ dZdZd>dZdZd d d d d dd dddddddddddd ddddgddZdedefdZd Zd!d"d d d dd d#dd ddddd$ddd ddd%dd&gddZdedefd'Zd Zd(d)d d d dd d*dd d+dd dddd,gd-ddZdedefd.Zd/Zd0d1d d d dd d2dd dddd3dd4gddZdedefd5Zejd d6eeegdd7d89 ejd!d6eeegdd:d89 ejd(d6eeegdd;d<9 ejd0d6eeegdd=d<9 dS)?aFeishu Drive Tools -- document comment operations via Feishu/Lark API. Provides tools for listing, replying to, and adding document comments. Uses the same lazy-import + BaseRequest pattern as feishu_comment.py. The lark client is injected per-thread by the comment event handler. N)registry tool_error tool_resultc|t_dS)zFStore a lark client for the current thread (called by feishu_comment).N)_localclient)rs 9/home/div/.hermes/hermes-agent/tools/feishu_drive_tool.py set_clientr s FMMMc.ttddS)z7Return the lark client for the current thread, or None.rN)getattrrr r get_clientrs 68T * **r c2 ddl}dS#t$rYdSwxYw)NrTF) lark_oapi ImportError)rs r _check_feishurs:t uus  cddlm}ddlm}ddlm}|dkr|jn|j} | |  | |j h} |r| |} |r| |} || |} | } || } t%| dd} t%| dd }i}t%| d d}|r]t'|d rM t)j|j}|d i}n#t(jt2f$rYnwxYw|sJt%| d d}t5|t6r|}n!|rt'|d rt9|}| ||fS)z?Build and execute a BaseRequest, return (code, msg, data_dict).r)AccessTokenType) HttpMethod) BaseRequestGETNcodemsgrawcontentdata__dict__)rrlark_oapi.core.enumr!lark_oapi.core.model.base_requestrrPOSTbuilder http_methoduri token_typesTENANTpathsqueriesbodybuildrequestr hasattrjsonloadsrgetJSONDecodeErrorAttributeError isinstancedictvars)rmethodr%r(r)r*rrrr$r#r,responserrrr body_json resp_datas r _do_requestr:&s))))))......======$*eOO*..K  [ ! ! S o,- . .   '--&&+//'** ,,t$$mmooG~~g&&H 8VT * *D (E2 & &C D (E4 ( (C wsI&&  3;//I==,,DD$n5    D  #Hfd33 i & & #DD  #79j99 # ??D d?s0/E E98E9z./open-apis/drive/v1/files/:file_token/commentsfeishu_drive_list_commentsz[List comments on a Feishu document. Use is_whole=true to list whole-document comments only.objectstringzThe document file token.)type descriptionzFile type (default: docx).docx)r>r?defaultbooleanz-If true, only return whole-document comments.Fintegerz&Number of comments per page (max 100).dzPagination token for next page.) file_token file_typeis_whole page_size page_tokenrE)r> propertiesrequired)namer? parametersargsreturnc t}|tdS|dd}|stdS|ddpd}|dd}|d d }|d d}d|fd d t |fg}|r|d |r|d |ft |dtd|i|\} } } | dkrtd| d| St| S)NFeishu client not availablerErzfile_token is requiredrFr@rGFrHrDrI user_id_typeopen_id)rGtruerr(r)rzList comments failed: code= msg=) rrr0stripstrappendr:_LIST_COMMENTS_URIr) rNkwargsrrErFrGrHrIr)rrrs r _handle_list_commentsr]si \\F ~7888,++1133J 42333f--7Ixx E**Hc**I,++J i # c)nn%G -+,,,3 j1222!)Z(OD#t  qyyHHH3HHIII t  r zB/open-apis/drive/v1/files/:file_token/comments/:comment_id/replies!feishu_drive_list_comment_repliesz:List all replies in a comment thread on a Feishu document.z#The comment ID to list replies for.z%Number of replies per page (max 100).)rE comment_idrFrHrIr_c t}|tdS|dd}|dd}|r|stdS|ddpd}|dd }|d d}d|fd dt |fg}|r|d |ft |d t||d |\} } } | dkrtd| d| St| S)NrQrErr_z&file_token and comment_id are requiredrFr@rHrDrIrRrrEr_rVrzList replies failed: code=rW) rrr0rXrYrZr:_LIST_REPLIES_URIr) rNr\rrEr_rFrHrIr)rrrs r _handle_list_repliesrcsf \\F ~7888,++1133J,++1133J DZDBCCCf--7Ic**I,++J i # c)nn%G 3 j1222!('zBBOD#t  qyyGtGG#GGHHH t  r feishu_drive_reply_commentzReply to a local comment thread on a Feishu document. Use this for local (quoted-text) comments. For whole-document comments, use feishu_drive_add_comment instead.zThe comment ID to reply to.z6The reply text content (plain text only, no markdown).)rEr_rrF)rEr_rc Bt}|tdS|dd}|dd}|dd}|r|r|stdS|ddpd}dd d d |id gii}t |d t ||dd|fg|\}} } |dkrtd|d| St d| S)NrQrErr_rz0file_token, comment_id, and content are requiredrFr@elementstext_runtext)r>rgr"ra)r(r)r*rzReply comment failed: code=rWTsuccessr)rrr0rXr:_REPLY_COMMENT_URIr) rNr\rrEr_rrFr*rrrs r _handle_reply_commentrlsd \\F ~7888,++1133J,++1133Jhhy"%%++--G NZNwNLMMMf--7I  &!' 1  D"*'zBBy)*  OD#t  qyyHHH3HHIII t$ / / //r z2/open-apis/drive/v1/files/:file_token/new_commentsfeishu_drive_add_commentzAdd a new whole-document comment on a Feishu document. Use this for whole-document comments or as a fallback when reply_comment fails with code 1069302.z8The comment text content (plain text only, no markdown).)rErrFrc t}|tdS|dd}|dd}|r|stdS|ddpd}|d|d gd }t |d t d|i| \}}} |d krtd|d|St d| S)NrQrErrz#file_token and content are requiredrFr@rh)r>rh)rFreply_elementsr")r(r*rzAdd comment failed: code=rWTri)rrr0rXr:_ADD_COMMENT_URIr) rNr\rrErrFr*rrrs r _handle_add_commentrq]s" \\F ~7888,++1133Jhhy"%%++--G AWA?@@@f--7IW - -   D"(Z( OD#t  qyyFdFFFFGGG t$ / / //r feishu_drivezList document commentsu💬) rLtoolsetschemahandlercheck_fn requires_envis_asyncr?emojizList comment replieszReply to a document commentu✉️zAdd a whole-document comment)NNN) __doc__r.logging threadingtools.registryrrr getLogger__name__loggerlocalrr rrr:r[!FEISHU_DRIVE_LIST_COMMENTS_SCHEMAr4rYr]rb FEISHU_DRIVE_LIST_REPLIES_SCHEMArcrkFEISHU_DRIVE_REPLY_SCHEMArlrpFEISHU_DRIVE_ADD_COMMENT_SCHEMArqregisterrr r rs~ <<<<<<<<<<  8 $ $    +++ ....jF ) B!9 !;! "N  "G !@)  2"N7 #%#%!L  3    NY 0O!9 !D !;! "F !@'  0"<05$$ Dt#JZ ) M !9 !< !W !;!  &<;;+B!0!03!0!0!0!0PH ' 1 !9 !Y !;!  "9-###:0d00000D %  , !  (      ,  +  &      %  $ !  -      #  *   .       r