Function main

nanoresearch/feishu_bot.py:235–317

()

# ═══════════════════════════════════════════════════════════════

def main() -> None:
    import ssl
    import certifi

    import websockets
    if os.environ.get("NANORESEARCH_ALLOW_INSECURE_SSL") == "1":
        logger.warning(
            "NANORESEARCH_ALLOW_INSECURE_SSL=1: disabling Feishu WebSocket SSL verification"
        )
        ssl_context = ssl.create_default_context()
        ssl_context.check_hostname = False
        ssl_context.verify_mode = ssl.CERT_NONE

        _original_connect = websockets.connect

        def _connect_with_insecure_ssl(*args, **kwargs):
            if 'ssl' not in kwargs:
                kwargs['ssl'] = ssl_context
            return _original_connect(*args, **kwargs)

        websockets.connect = _connect_with_insecure_ssl

    logging.basicConfig(
        level=logging.INFO,
        format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
    )

    app_id, app_secret = _load_feishu_credentials()
    bot = FeishuBot(app_id, app_secret)

    logger.info("NanoResearch 飞书助手启动中...")

    def on_message(data: lark.im.v1.P2ImMessageReceiveV1) -> None:
        try:
            event = data.event
            message = event.message
            sender = event.sender

            if message.message_type != "text":
                return

            chat_id = message.chat_id
            message_id = message.message_id
            sender_id = sender.sender_id.open_id if sender.sender_id else ""

            try:
                content = json.loads(message.content)
                text = content.get("text", "")
            except (json.JSONDecodeError, TypeError):
                text = ""

            logger.info("on_message: raw_content=%r text=%r", message.content, text[:200] if text else "")

            if not text.strip():
                return

            bot.handle_message(chat_id, message_id, text, sender_id)