mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-09-07 10:07:15 +00:00
leonoxo
855ae2bdba
The LINE adapter passed text through as a single Plain component, ignoring the mention payload (mentions[].index/length/isSelf) that the Line Messaging API includes in the webhook. As a result: - At(target=bot_account_id) never appeared in the message chain, so the 'at-bot' group respond rule silently dropped every @bot mention. - The bot only replied when the message happened to match the prefix rule (e.g. starting with 'ai'). Now LINEMessageConverter reads message.message.mention and builds the chain per mention position: - Bot mention (isSelf) -> At(target=bot_account_id) so AtBotRule matches the same way as other adapters (dingtalk/lark etc.). - Other mentions -> At(target=<line user id>, display=<mention text>). At.__str__ already prepends '@', so the display text carries no double '@' and the rendered text (prefix/regexp rules, quotes, session context) is byte-identical to before. - Missing/out-of-bounds mentions are skipped defensively. target2yiri becomes an instance method (like wechatpad/aiocqhttp) so the converters can hold bot_account_id; LINEAdapter passes it in from its own config.
335 lines
13 KiB
Python
335 lines
13 KiB
Python
import typing
|
||
import quart
|
||
|
||
|
||
import traceback
|
||
import asyncio
|
||
import base64
|
||
import datetime
|
||
|
||
|
||
import langbot_plugin.api.definition.abstract.platform.adapter as abstract_platform_adapter
|
||
import langbot_plugin.api.entities.builtin.platform.message as platform_message
|
||
import langbot_plugin.api.entities.builtin.platform.events as platform_events
|
||
import langbot_plugin.api.entities.builtin.platform.entities as platform_entities
|
||
from ..logger import EventLogger
|
||
from ...utils import bounded_executor
|
||
|
||
|
||
from linebot.v3 import WebhookHandler
|
||
from linebot.v3.exceptions import InvalidSignatureError
|
||
from linebot.v3.messaging import Configuration, ApiClient, MessagingApi, ReplyMessageRequest, TextMessage, ImageMessage
|
||
from linebot.v3.webhooks import (
|
||
MessageEvent,
|
||
TextMessageContent,
|
||
ImageMessageContent,
|
||
VideoMessageContent,
|
||
AudioMessageContent,
|
||
UserMentionee,
|
||
)
|
||
|
||
# from linebot import WebhookParser
|
||
from linebot.v3.webhook import WebhookParser
|
||
from linebot.v3.messaging import MessagingApiBlob
|
||
|
||
MAX_LINE_MEDIA_BYTES = 10 * 1024 * 1024
|
||
|
||
|
||
def _validate_line_media_content(content: bytes) -> bytes:
|
||
if len(content) > MAX_LINE_MEDIA_BYTES:
|
||
raise ValueError(f'LINE media exceeds the {MAX_LINE_MEDIA_BYTES}-byte limit')
|
||
return content
|
||
|
||
|
||
class LINEMessageConverter(abstract_platform_adapter.AbstractMessageConverter):
|
||
@staticmethod
|
||
async def yiri2target(message_chain: platform_message.MessageChain, api_client: ApiClient) -> typing.Tuple[list]:
|
||
content_list = []
|
||
for component in message_chain:
|
||
if isinstance(component, platform_message.At):
|
||
content_list.append({'type': 'at', 'target': component.target})
|
||
elif isinstance(component, platform_message.Plain):
|
||
content_list.append({'type': 'text', 'content': component.text})
|
||
elif isinstance(component, platform_message.Image):
|
||
# Only add image if it has a valid URL
|
||
if component.url:
|
||
content_list.append({'type': 'image', 'image': component.url})
|
||
elif isinstance(component, platform_message.Voice):
|
||
content_list.append({'type': 'voice', 'url': component.url, 'length': component.length})
|
||
|
||
return content_list
|
||
|
||
def __init__(self, bot_account_id: str = ''):
|
||
self.bot_account_id = bot_account_id
|
||
|
||
async def target2yiri(self, message, bot_client) -> platform_message.MessageChain:
|
||
lb_msg_list = []
|
||
msg_create_time = datetime.datetime.fromtimestamp(int(message.timestamp) / 1000)
|
||
|
||
lb_msg_list.append(platform_message.Source(id=message.webhook_event_id, time=msg_create_time))
|
||
|
||
if isinstance(message.message, TextMessageContent):
|
||
lb_msg_list.extend(
|
||
self._build_text_components(message.message.text, getattr(message.message, 'mention', None))
|
||
)
|
||
elif isinstance(message.message, AudioMessageContent):
|
||
pass
|
||
elif isinstance(message.message, VideoMessageContent):
|
||
pass
|
||
elif isinstance(message.message, ImageMessageContent):
|
||
message_content = await asyncio.to_thread(
|
||
MessagingApiBlob(bot_client).get_message_content,
|
||
message.message.id,
|
||
)
|
||
_validate_line_media_content(message_content)
|
||
|
||
base64_string = await asyncio.to_thread(lambda: base64.b64encode(message_content).decode('utf-8'))
|
||
|
||
# 如果需要Data URI格式（用于直接嵌入HTML等）
|
||
# 首先需要知道图片类型，LINE图片通常是JPEG
|
||
data_uri = f'data:image/jpeg;base64,{base64_string}'
|
||
lb_msg_list.append(platform_message.Image(base64=data_uri))
|
||
return platform_message.MessageChain(lb_msg_list)
|
||
|
||
def _build_text_components(self, text: str, mention) -> list:
|
||
"""Build message components from text, inserting At components for mentions.
|
||
|
||
LINE provides mention positions (index/length) and is_self per mentionee in the
|
||
webhook payload. Mapping the bot mention to At(target=bot_account_id) makes the
|
||
'at-bot' group respond rule work for LINE, consistent with other adapters.
|
||
"""
|
||
components: list = []
|
||
if not mention or not mention.mentionees:
|
||
if text:
|
||
components.append(platform_message.Plain(text=text))
|
||
return components
|
||
segments: list[tuple[int, int, object]] = sorted((m.index, m.index + m.length, m) for m in mention.mentionees)
|
||
cursor = 0
|
||
for start, end, mentionee in segments:
|
||
if start < cursor:
|
||
start, end = cursor, min(end, len(text))
|
||
if start < cursor or end <= start or end > len(text):
|
||
continue
|
||
if start > cursor:
|
||
components.append(platform_message.Plain(text=text[cursor:start]))
|
||
if isinstance(mentionee, UserMentionee):
|
||
target = self.bot_account_id if mentionee.is_self else mentionee.user_id
|
||
if not target:
|
||
target = text[start:end]
|
||
else:
|
||
target = text[start:end]
|
||
# At.__str__ already prepends '@', so strip one from the LINE text token.
|
||
display = text[start:end].lstrip('@')
|
||
components.append(platform_message.At(target=str(target), display=display))
|
||
cursor = end
|
||
if cursor < len(text):
|
||
components.append(platform_message.Plain(text=text[cursor:]))
|
||
return components
|
||
|
||
|
||
class LINEEventConverter(abstract_platform_adapter.AbstractEventConverter):
|
||
def __init__(self, bot_account_id: str = ''):
|
||
self.bot_account_id = bot_account_id
|
||
self.message_converter = LINEMessageConverter(bot_account_id)
|
||
|
||
@staticmethod
|
||
async def yiri2target(
|
||
event: platform_events.MessageEvent,
|
||
) -> MessageEvent:
|
||
pass
|
||
|
||
async def target2yiri(self, event, bot_client) -> platform_events.Event:
|
||
message_chain = await self.message_converter.target2yiri(event, bot_client)
|
||
|
||
if event.source.type == 'user':
|
||
return platform_events.FriendMessage(
|
||
sender=platform_entities.Friend(
|
||
id=event.source.user_id,
|
||
nickname=event.source.user_id,
|
||
remark='',
|
||
),
|
||
message_chain=message_chain,
|
||
time=event.timestamp,
|
||
source_platform_object=event,
|
||
)
|
||
else:
|
||
# 'group' and 'room' sources carry the stable chat id under different
|
||
# field names; user_id may be absent for some members, so fall back
|
||
# to the group/room id rather than the per-message id.
|
||
group_id = event.source.group_id if event.source.type == 'group' else event.source.room_id
|
||
member_id = event.source.user_id or group_id
|
||
|
||
return platform_events.GroupMessage(
|
||
sender=platform_entities.GroupMember(
|
||
id=member_id,
|
||
member_name=member_id,
|
||
permission=platform_entities.Permission.Member,
|
||
group=platform_entities.Group(
|
||
id=group_id,
|
||
name='',
|
||
permission=platform_entities.Permission.Member,
|
||
),
|
||
special_title='',
|
||
),
|
||
message_chain=message_chain,
|
||
time=event.timestamp,
|
||
source_platform_object=event,
|
||
)
|
||
|
||
|
||
class LINEAdapter(abstract_platform_adapter.AbstractMessagePlatformAdapter):
|
||
bot: MessagingApi
|
||
api_client: ApiClient
|
||
parser: WebhookParser
|
||
|
||
bot_account_id: str # 用于在流水线中识别at是否是本bot，直接以bot_name作为标识
|
||
message_converter: LINEMessageConverter
|
||
event_converter: LINEEventConverter
|
||
|
||
listeners: typing.Dict[
|
||
typing.Type[platform_events.Event],
|
||
typing.Callable[[platform_events.Event, abstract_platform_adapter.AbstractMessagePlatformAdapter], None],
|
||
]
|
||
|
||
config: dict
|
||
bot_uuid: str = None
|
||
|
||
card_id_dict: dict[str, str] # 消息id到卡片id的映射，便于创建卡片后的发送消息到指定卡片
|
||
|
||
seq: int # 用于在发送卡片消息中识别消息顺序，直接以seq作为标识
|
||
|
||
def __init__(self, config: dict, logger: EventLogger):
|
||
configuration = Configuration(access_token=config['channel_access_token'])
|
||
line_webhook = WebhookHandler(config['channel_secret'])
|
||
parser = WebhookParser(config['channel_secret'])
|
||
api_client = ApiClient(configuration)
|
||
|
||
bot_account_id = config.get('bot_account_id', 'langbot')
|
||
|
||
super().__init__(
|
||
config=config,
|
||
logger=logger,
|
||
listeners={},
|
||
card_id_dict={},
|
||
seq=1,
|
||
event_converter=LINEEventConverter(bot_account_id),
|
||
message_converter=LINEMessageConverter(bot_account_id),
|
||
line_webhook=line_webhook,
|
||
parser=parser,
|
||
configuration=configuration,
|
||
api_client=api_client,
|
||
bot=MessagingApi(api_client),
|
||
bot_account_id=bot_account_id,
|
||
)
|
||
|
||
async def send_message(self, target_type: str, target_id: str, message: platform_message.MessageChain):
|
||
pass
|
||
|
||
async def reply_message(
|
||
self,
|
||
message_source: platform_events.MessageEvent,
|
||
message: platform_message.MessageChain,
|
||
quote_origin: bool = False,
|
||
):
|
||
content_list = await self.message_converter.yiri2target(message, self.api_client)
|
||
|
||
for content in content_list:
|
||
if content['type'] == 'text':
|
||
await asyncio.to_thread(
|
||
self.bot.reply_message_with_http_info,
|
||
ReplyMessageRequest(
|
||
reply_token=message_source.source_platform_object.reply_token,
|
||
messages=[TextMessage(text=content['content'])],
|
||
),
|
||
)
|
||
elif content['type'] == 'image':
|
||
# LINE ImageMessage requires original_content_url and preview_image_url
|
||
image_url = content['image']
|
||
await asyncio.to_thread(
|
||
self.bot.reply_message_with_http_info,
|
||
ReplyMessageRequest(
|
||
reply_token=message_source.source_platform_object.reply_token,
|
||
messages=[ImageMessage(original_content_url=image_url, preview_image_url=image_url)],
|
||
),
|
||
)
|
||
|
||
async def is_muted(self, group_id: int) -> bool:
|
||
return False
|
||
|
||
def register_listener(
|
||
self,
|
||
event_type: typing.Type[platform_events.Event],
|
||
callback: typing.Callable[
|
||
[platform_events.Event, abstract_platform_adapter.AbstractMessagePlatformAdapter], None
|
||
],
|
||
):
|
||
self.listeners[event_type] = callback
|
||
|
||
def unregister_listener(
|
||
self,
|
||
event_type: typing.Type[platform_events.Event],
|
||
callback: typing.Callable[
|
||
[platform_events.Event, abstract_platform_adapter.AbstractMessagePlatformAdapter], None
|
||
],
|
||
):
|
||
self.listeners.pop(event_type)
|
||
|
||
def set_bot_uuid(self, bot_uuid: str):
|
||
"""设置 bot UUID（用于生成 webhook URL）"""
|
||
self.bot_uuid = bot_uuid
|
||
|
||
async def handle_unified_webhook(self, bot_uuid: str, path: str, request):
|
||
"""处理统一 webhook 请求。
|
||
|
||
Args:
|
||
bot_uuid: Bot 的 UUID
|
||
path: 子路径（如果有的话）
|
||
request: Quart Request 对象
|
||
|
||
Returns:
|
||
响应数据
|
||
"""
|
||
try:
|
||
signature = request.headers.get('X-Line-Signature')
|
||
body = await request.get_data(as_text=True)
|
||
|
||
# Check if signature header exists
|
||
if not signature:
|
||
await self.logger.warning('Missing X-Line-Signature header')
|
||
return quart.Response('Missing X-Line-Signature header', status=400)
|
||
|
||
try:
|
||
events = self.parser.parse(body, signature) # 解密解析消息
|
||
except InvalidSignatureError:
|
||
await self.logger.info(
|
||
f'Invalid signature. Please check your channel access token/channel secret.{traceback.format_exc()}'
|
||
)
|
||
return quart.Response('Invalid signature', status=400)
|
||
|
||
# 处理事件
|
||
if events and len(events) > 0:
|
||
lb_event = await self.event_converter.target2yiri(events[0], self.api_client)
|
||
if lb_event.__class__ in self.listeners:
|
||
await self.listeners[lb_event.__class__](lb_event, self)
|
||
|
||
return {'code': 200, 'message': 'ok'}
|
||
except Exception:
|
||
await self.logger.error(f'Error in LINE callback: {traceback.format_exc()}')
|
||
print(traceback.format_exc())
|
||
return {'code': 500, 'message': 'error'}
|
||
|
||
async def run_async(self):
|
||
# 统一 webhook 模式下，不启动独立的 Quart 应用
|
||
# 保持运行但不启动独立端口
|
||
|
||
# 打印 webhook 回调地址
|
||
async def keep_alive():
|
||
while True:
|
||
await asyncio.sleep(1)
|
||
|
||
await keep_alive()
|
||
|
||
async def kill(self) -> bool:
|
||
await bounded_executor.run_blocking_cleanup(self.api_client.close)
|
||
return True
|