inquiry_robot/inquiry-agent/agent/channel/h5/wecom_close.py

"""
企微内嵌 H5 成功后自动关页。

本文件职责：给附件上传成功、选线路成功页一段共用脚本。
手机端只调一次 WeixinJSBridge.closeWindow，关掉当前 H5 后留在询价对话里。
再调 ww/wx/window.close 会把询价小助手整个退出，回到企业微信首页。
电脑企微即使注入了 WeixinJSBridge，这一下也关不掉当前页，必须按电脑走 JS-SDK。
失败、过期页不要挂这段。
副作用：只产出 HTML 片段；关窗签名走浏览器短请求，不占智能体请求线程做慢活。
"""

from __future__ import annotations

from html import escape

from agent.config import get_settings

# 签名完成后再稍停，让销售看见「已成功」。
_CLOSE_DELAY_MS = 400


def wecom_auto_close_script() -> str:
    """
    成功页脚：加载企微 JS-SDK，register 后关掉当前 H5 窗口。

    线程：浏览器端执行。签名接口与复制页共用，ticket 已缓存在 Redis。
    为何这样写：电脑企微也会带上 WeixinJSBridge，但关不掉当前页。
    若一发现桥就只走手机关闭，成功页会停在「正在返回企微」。手机仍只关一次。
    """
    settings = get_settings()
    corp = escape((settings.wecom_corp_id or "").strip(), quote=True)
    agent = escape(str(settings.wecom_agent_id or "1000010").strip(), quote=True)
    delay = _CLOSE_DELAY_MS
    return f"""<script src="https://wwcdn.weixin.qq.com/node/wework/open/js/wecom-jssdk-2.3.2.js"></script>
<script>
(function () {{
  var corpId = "{corp}";
  var agentId = Number("{agent}") || 0;

  function sign(kind) {{
    return function (url) {{
      return fetch("/inquiry-form/jsapi-signature?kind=" + encodeURIComponent(kind)
        + "&url=" + encodeURIComponent(url))
        .then(function (r) {{
          if (!r.ok) throw new Error("sign");
          return r.json();
        }});
    }};
  }}

  var closed = false;

  function isPhone() {{
    return /Android|iPhone|iPad|iPod|Mobile/i.test(navigator.userAgent || "");
  }}

  function closeOnce(fn) {{
    if (closed) return;
    closed = true;
    try {{ fn(); }} catch (e) {{}}
  }}

  // 手机企微有 WeixinJSBridge：只关当前 H5。再调第二套关闭会把询价小助手整个退出，回到企业微信首页。
  function closeMobile() {{
    closeOnce(function () {{ WeixinJSBridge.call("closeWindow"); }});
  }}

  // 电脑企微即使有 WeixinJSBridge，也关不掉当前页。注册 JS-SDK 后再关一次。
  // 不要调 window.close：手机会把询价小助手整个退出。
  function closeDesktop() {{
    closeOnce(function () {{
      if (window.ww && typeof ww.closeWindow === "function") ww.closeWindow();
      else if (window.wx && typeof wx.closeWindow === "function") wx.closeWindow();
    }});
  }}

  function boot() {{
    if (isPhone()) {{
      if (typeof WeixinJSBridge !== "undefined") {{
        setTimeout(closeMobile, {delay});
        return;
      }}
      document.addEventListener("WeixinJSBridgeReady", function () {{
        setTimeout(closeMobile, {delay});
      }}, false);
      return;
    }}
    var ready = Promise.resolve();
    if (window.ww && ww.register && corpId) {{
      ready = ww.register({{
        corpId: corpId,
        agentId: agentId,
        jsApiList: ["closeWindow"],
        getConfigSignature: sign("corp"),
        getAgentConfigSignature: sign("agent")
      }}).catch(function () {{ return null; }});
    }}
    var timed = Promise.race([
      ready,
      new Promise(function (resolve) {{ setTimeout(resolve, 2500); }})
    ]);
    timed.then(function () {{
      setTimeout(closeDesktop, {delay});
    }}).catch(function () {{
      setTimeout(closeDesktop, {delay});
    }});
  }}

  if (document.readyState === "loading") {{
    document.addEventListener("DOMContentLoaded", boot);
  }} else {{
    boot();
  }}
}})();
</script>"""