.i_dZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZmZmZdd lmZe eZd Zd Zd Zd ZdZdZdZGddZdS)uCSlack Interactive Gateway — Block Kit control panel for zhihu-hunter. Manages two human-in-the-loop approval gates: Gate 1 [✍️ 生成草稿] / [🚫 忽略] Triggered when ZhihuHunter surfaces candidate questions. Gate 2 [✅ 确认发布] / [🔄 重新生成] / [❌ 取消] Triggered after an answer draft has been generated. Usage: gateway = SlackInteractiveGateway(slack_client, hunter, engine, channel) gateway.register(app) # call once at bot startup gateway.post_question_cards(questions) # call from dispatcher N)Path)Optional)App) WebClient) setup_logger) AnswerDraft ZhihuHunter ZhihuQuestion)ZhihuPlaywrightEnginezhihu_generate_draft zhihu_ignore zhihu_publishzhihu_regenerate zhihu_cancelzhihu_regen_modali c eZdZdZ d'dededededee ddf d Z d e ddfd Z d'd e ed eeddfdZded ededdfdZd(dZd(dZd(dZd(dZd(dZd(dZdede efdZdedede efdZdede efdZdede efdZdede efdZdedee fd Zd!ed"eddfd#Zdedefd$Z dedeefd%Z!deddfd&Z"dS))SlackInteractiveGatewayaBlock Kit gateway providing two approval gates for the zhihu-hunter workflow. Args: slack_client: Authenticated WebClient (same token as the Bolt App). hunter: ZhihuHunter instance for draft generation. engine: ZhihuPlaywrightEngine instance for publishing. notify_channel: Default Slack channel/DM for outbound cards. state_dir: Directory for storing draft JSON files between interactions. Defaults to /data/zhihu_drafts. N slack_clienthunterenginenotify_channel state_dirreturnc||_||_||_||_|p6t t jddz dz |_|j dddS)Ndata zhihu_draftsTparentsexist_ok) clientrrrr__file__resolverrmkdir)selfrrrrrs B/root/projects/butler/slack_bot/zhihu/slack_interactive_gateway.py__init__z SlackInteractiveGateway.__init__8sz* # # ,& NN " " $ $ ,Q /& 8> I  TD99999appc|t|j|t|j|t |j|t|j|t|j | t|j tddS)zAttach all action handlers to a Slack Bolt App instance. Call this once during bot startup, after the App object is created. Args: app: The Slack Bolt App that receives Socket Mode events. z-ZhihuHunter: Slack action handlers registeredN)action _ACT_GENERATE_handle_generate_draft _ACT_IGNORE_handle_ignore _ACT_PUBLISH_handle_publish _ACT_REGEN_handle_regenerate _ACT_CANCEL_handle_cancelview _VIEW_REGEN_handle_regenerate_submitloggerinfo)r%r)s r&registerz SlackInteractiveGateway.registerKs " =!!$"=>>> ; 3444  <  !5666 :t6777 ; 3444d<=== CDDDDDr( questionschannelc|p|j}|s|j|ddS|j|dt|d|D]w} |j|||d|j<#t $r/}td|jd |Yd}~pd}~wwxYwdS) u:Post one Block Kit card per discovered question (Gate 1). Each card includes [✍️ 生成草稿] and [🚫 忽略] buttons. Args: questions: Candidate questions returned by ZhihuHunter.scan_and_hunt(). channel: Override channel. Defaults to self.notify_channel. uA🔍 本次未发现符合条件的知乎问题，下次再试。)r=textNu 🔍 发现 *u,* 个知乎相关问题，请逐一确认：u🔍 知乎问题: )r=blocksr?z"Failed to post question card for 'z': ) rr!chat_postMessagelen_question_blockstitle Exceptionr9error)r%r<r=chqes r&post_question_cardsz+SlackInteractiveGateway.post_question_cards]s> ++  K ( (X )    F $$]Y]]] %    S SA S ,,00338qw88-  S S S Q!'QQaQQRRRRRRRR S S Ss8B C%C  Cdraft thread_tsc||} |j|||||d|jjdS#t $r(}td|Yd}~dSd}~wwxYw)uWPost an answer draft card as a thread reply (Gate 2). Includes [✅ 确认发布], [🔄 重新生成], and [❌ 取消] buttons. Args: draft: Generated AnswerDraft. channel: Channel containing the original question card. thread_ts: ts of the question card message (reply target). u+📝 草稿已生成，等待确认发布: r=rLr@r?zFailed to post draft card: N) _save_draftr!rA _draft_blocksquestionrDrEr9rF)r%rKr=rLdraft_idrIs r&post_draft_cardz'SlackInteractiveGateway.post_draft_cards##E** < K ( (#))%::Y5>CWYY )       < < < LL:q:: ; ; ; ; ; ; ; ; ; *u草稿已生成zDraft generation failed: Texc_infou草稿生成失败：u草稿生成失败N) chat_update_loading_blockr model_validate_jsonr draft_answer_save_to_vault model_copystrrOrArPrD _done_blockurlrEr9rF _error_block) r%ackbodyr!r=msg_tsraw_valrQrKr[rRrIs r&r-z.SlackInteractiveGateway._handle_generate_drafts:  ?4( ?4( ?1%g. &&'NOO%    $ $8AAH{//99E,,U33J Q((s:0O(PP''..H  # # ))%::>hn>> $      ''zYaYezzhphvzzz'          LL8Q884L H H H   (()D)D)DEE)           s(C1E F>%AF99F>c\||dd|ddS)u4Gate 1 → [🚫 忽略]: dismiss the question card.Tu🚫 已忽略u🚫 已忽略此问题replace_originalr?r@Nrir%rlresponds r&r/z&SlackInteractiveGateway._handle_ignoresH !!##$=>>      r(c ||dd}|dd}|ddd}||||dd  ||}|td |d |jstd|j|jj|j }|||| d|jjd|jj d|dd| |j r| |j |||d S#t$rZ} t d| d||||d| dd Yd } ~ d Sd } ~ wwxYw)uEGate 2 → [✅ 确认发布]: login to Zhihu and publish the answer.r=rUrVrWrXrrYu.⏳ 正在登录知乎并发布，请稍候...u 发布中...rZNu草稿文件不存在（id=u），可能已过期u'知乎登录失败，请扫码后重试) question_url answer_textu ✅ *发布成功！* 问题: ** 回答: u✅ 已发布: zPublish failed: Tr`u发布失败：u7 你可以重新点击发布，或联系排查日志。u 发布失败)rbrc _load_draft RuntimeErrorrensure_logged_inpublish_answerrQrjcontentrirDr[_update_vault_published _delete_draftrEr9rFrk) r%rlrmr!r=rnrRrK answer_urlrIs r&r1z'SlackInteractiveGateway._handle_publishsj  ?4( ?4( ?1%g.&&'WXX    ' $$X..E}"#`#`#`#`aaa;//11 N"#LMMM33"^/!M4J   ''D!&!3DD6;n6JDD *DDD 4z33     K,,U-=zJJJ   x ( ( ( ( (    LL/A//$L ? ? ?   ((aaaaa$           s(DE++ G5AG  Gc||dd}|dd}|ddd} tj|}|j}n#t$rd}YnwxYwt j|||d } ||d d td d dd ddd dd| ddd|ddddddd dddddd ddddgdd S)!uQGate 2 → [🔄 重新生成]: open a modal for user to enter revision feedback.r=rUrVrWrXrrYu（问题信息丢失）)r=rn regen_value trigger_idmodal plain_textu重新生成草稿typer?u生成u取消sectionmrkdwn*u=* 请输入修改意见，Bot 会据此重新生成草稿。inputfeedback_blockTu 修改意见plain_text_inputfeedback_inputu`例：内容太技术了，偏向大众一些；字数控制在500字以内；结尾加一句诗)r action_id multiline placeholder)rblock_idoptionallabelelement)r callback_idrDsubmitcloseprivate_metadatar@)rr6N)r rdrDrEjsondumps views_openr7) r%rlrmr!r=rnrorQquestion_titlers r&r3z*SlackInteractiveGateway._handle_regenerates  ?4( ?4( ?1%g. ,$8AAH&^NN 8 8 87NNN 8  :! "' '    L)*".8LMM#/BB#/BB$4!*$,$w$w$w$w!!!($4$(*6!O!O$6)9)-(4)K,, $$  " " " " " sA A'&A'c |tj|dd}|d}|d}|d}|ddddid id pd }||||d d  t j|} |j| |} | | } | r%| dt| i} | | } |||| | | d| jdS#t$rY} t d| d||||d| dYd} ~ dSd} ~ wwxYw)u5Modal submit → regenerate draft with user feedback.r6rr=rnrstatevaluesrrrYu#🔄 重新生成中，请稍候...u重新生成中...rZ)feedbackr[r\u📝 草稿已重新生成: zRegenerate submit failed: Tr`u重新生成失败：u重新生成失败N)rloadsgetrbrcr rdrrerfrgrhrOrPrDrEr9rFrk)r%rlrmr!metadatar=rnrorrQrKr[rRrIs r&r8z1SlackInteractiveGateway._handle_regenerate_submitKs5 JtF|,>?@@Y'X&]+ L !( + S!2 & & S!2 & & S\\   &&'LMM%     $8AAH{//8/LLE,,U33J Q((s:0O(PP''..H   ))%::DHNDD          LL9a99DL I I I   (()D)D)DEE)           sB6E99 GAGGc\||dd|ddS)u0Gate 2 → [❌ 取消]: dismiss the draft card.Tu ❌ 已取消u❌ 已取消本次发布rqNrsrts r&r5z&SlackInteractiveGateway._handle_cancel{sH ! ##$?@@      r(rQc|d|jpdddi}|}d|jd|jdg}|jr%|d |jdd |jr1|d d |jd ddd|dddddddddt|ddddddtddgdgS)zBuild Gate 1 Block Kit blocks for a single question. Args: question: ZhihuQuestion to display. Returns: List of Slack Block Kit block dicts. snippetrNdr\z* u _关键词： · _rr rrXbuttonru✍️ 生成草稿Trr?emojiprimaryrr?stylerrYu 🚫 忽略ignorerr?rrYrelements) rgrmodel_dump_jsonrjrDappendkeywordsjoinr,r.)r%rQ q_for_button btn_value body_liness r&rCz(SlackInteractiveGateway._question_blockssf ** 0 6B=>+  !0022 <8<<<(.<<<=   =   ;8#3DSD#9;; < < <   Q   Ofkk(:K.L.LOOO P P P"!)499Z3H3HII   "!))5?T_c d d!*%2!* !))5}W[ \ \%0!)      r(rRc |jdt}t|jtkr|dt|jdz }|jd|jjpdddi}|}dd d |jjd |jjd d d ddidd |d d g}|j r?| dd dd |j ddd gd|ddidddddddt|ddddddt|ddddddd td!dgdgz }|S)"zBuild Gate 2 Block Kit blocks for a draft answer. Args: draft: AnswerDraft to display. draft_id: UUID key used as the publish button value. Returns: List of Slack Block Kit block dicts. Nu _…（共 u 字，发布时使用全文）_rrrr\rru📝 *草稿已生成* | rrdividercontextu📚 RAG 来源：rrrXrru✅ 确认发布Trrru🔄 重新生成ru ❌ 取消dangercancel)r}_PREVIEW_CHARSrBrQrgrrrjrD rag_sourcesrrr0r2r4)r%rKrRpreview q_for_btnrr@s r&rPz%SlackInteractiveGateway._draft_blockss-0 u}   . . [s5='9'9[[[ [GN-- 6 <"dsdCD.   //11 "$I!N.II161EIII  Y !!)7;;   $    MM!$UU=NrPQr=R1S1SUU     Y !!))5?Q\` a a!*%1!) !))5?R]a b b%/!, !))5|VZ [ [!)%0!)     8 r(rVcdd|ddgS)zSimple single-section block for loading state. Args: message: Text to display (supports mrkdwn). Returns: List with one section block. rrrr%rVs r&rcz&SlackInteractiveGateway._loading_block#Xw,O,OPPQQr(cdd|ddgS)zSingle-section block for a completed/dismissed state. Args: message: Text to display (supports mrkdwn). Returns: List with one section block (no action buttons). rrrrrs r&riz#SlackInteractiveGateway._done_block rr(cddd|ddgS)uSingle-section block for error display. Args: message: Error description (supports mrkdwn). Returns: List with one section block prefixed with ⚠️. rru⚠️ rrrs r&rkz$SlackInteractiveGateway._error_blocks/%/B/B/BCC    r(c |ddl}ddlm} |jjdz }|dd|d}|dd |j j dd  d }||d |d z }| d |j j d |j j d|d|j j d|jd dtd||S#t"$r(}td|Yd}~dSd}~wwxYw)zPersist a draft as a Markdown file in /zhihu/. Args: draft: The generated AnswerDraft to save. Returns: Absolute path of the written file, or None on failure. rN)datezhihuTrz%Y-%m-%dz[^\w\u4e00-\u9fff]+-(z.mdz --- title: "z" url: z date: z$ status: draft tags: [zhihu] --- # z rutf-8encodingzDraft saved to vault: zFailed to save draft to vault: )redatetimerr vault_pathr$todaystrftimesubrQrDstrip write_textrjr}r9r:rEwarning) r%rKr_date zhihu_dirdate_strslug file_pathrIs r&rfz&SlackInteractiveGateway._save_to_vault's  ******  .8I OOD4O 8 8 8{{}}--j99H660#u~7KLLSbSQWWX[\\D!x$;$;$$;$;$;;I  %!N0%%*%%"%%^)%%=%%%! !    KK<<< = = =     NN@Q@@ A A A44444 sC*>>G MM'GM 4 4 4 KK<<< = = = = = @ @ @ NN>1>> ? ? ? ? ? ? ? ? ? @s#B A$B B?B::B?ctjjdd}|j|dz }||dt d||S)zPersist a draft to disk and return its UUID key. Args: draft: AnswerDraft to save. Returns: UUID string key (used as publish button value). N .jsonrrz Draft saved: )uuiduuid4hexrrrr9debug)r%rKrRrs r&rOz#SlackInteractiveGateway._save_draftcsuJLL$SbS)N%7%7%77 U2244wGGG 0Y00111r(c |j|dz }|sdS tj|dS#t $r+}t d|d|Yd}~dSd}~wwxYw)zLoad a draft by its UUID key. Args: draft_id: Key returned by _save_draft(). Returns: AnswerDraft if found, None if the file does not exist. rNrrzFailed to load draft : )rrrrdrrEr9rFr%rRrrIs r&ryz#SlackInteractiveGateway._load_draftrsN%7%7%77 !! 4 293F3FPW3F3X3XYY Y    LL@@@Q@@ A A A44444 s'A B A==Bc|j|dz } |ddS#t$r+}td|d|Yd}~dSd}~wwxYw)zwRemove a draft file after successful publish. Args: draft_id: Key of the draft to delete. rT) missing_okzCould not delete draft file rN)runlinkrEr9rrs r&rz%SlackInteractiveGateway._delete_drafts N%7%7%77  K     - - - - - K K K NNI(IIaII J J J J J J J J J Ks' A AA)N)rN)#__name__ __module__ __qualname____doc__rr r rhrrr'rr;listr rJrrSr-r/r1r3r8r5dictrCrPrcrirkrfr~rOryrrr(r&rr,sH  "%) ::::& :  : D> : ::::&ECEDEEEE*"&!S!S &!S#!S  !S!S!S!SJ<<< <  <<<<83333j    5555n6 6 6 6 p....`    , , 4:, , , , \K;K#K$t*KKKKZ Rc Rd4j R R R R R3 R4: R R R R  C  DJ     $K$HTN$$$$L@#@3@4@@@@,      CH[,A$ Kc Kd K K K K K Kr(r)rrrpathlibrtypingr slack_boltr slack_sdkrhealth.utils.logging_configrslack_bot.zhihu.zhihu_hunterrr r 'slack_bot.zhihu.zhihu_playwright_enginer rr9r,r.r0r2r4r7rrrr(r&rs)  444444PPPPPPPPPPIIIIII h  '   "  # b Kb Kb Kb Kb Kb Kb Kb Kb Kb Kr(