feat: add support for slack

This commit is contained in:
wangcham
2025-03-30 22:24:53 -04:00
parent c0dbf6fd13
commit be1328cee9
4 changed files with 40 additions and 24 deletions

View File

@@ -32,13 +32,13 @@ class SlackMessageConverter(adapter.MessageConverter):
return content_list
@staticmethod
async def target2yiri(message:str,message_id:str,pic_url:str):
async def target2yiri(message:str,message_id:str,pic_url:str,bot:SlackClient):
yiri_msg_list = []
yiri_msg_list.append(
platform_message.Source(id=message_id,time=datetime.datetime.now())
)
if pic_url is not None:
base64_url = await image.get_slack_image_to_base64(pic_url=pic_url)
base64_url = await image.get_slack_image_to_base64(pic_url=pic_url,bot_token=bot.bot_token)
yiri_msg_list.append(
platform_message.Image(base64=base64_url)
)
@@ -55,9 +55,9 @@ class SlackEventConverter(adapter.EventConverter):
return event.source_platform_object
@staticmethod
async def target2yiri(event:SlackEvent):
async def target2yiri(event:SlackEvent,bot:SlackClient):
yiri_chain = await SlackMessageConverter.target2yiri(
message=event.text,message_id=event.message_id,pic_url=event.pic_url
message=event.text,message_id=event.message_id,pic_url=event.pic_url,bot=bot
)
if event.type == 'channel':
@@ -138,7 +138,6 @@ class SlackAdapter(adapter.MessagePlatformAdapter):
for content in content_list:
if slack_event.type == 'channel':
print("fasong1")
await self.bot.send_message_to_channle(
content['content'],slack_event.channel_id
)
@@ -162,7 +161,7 @@ class SlackAdapter(adapter.MessagePlatformAdapter):
self.bot_account_id = "SlackBot"
try:
return await callback(
await self.event_converter.target2yiri(event),self
await self.event_converter.target2yiri(event,self.bot),self
)
except:
traceback.print_exc()

View File

@@ -215,5 +215,20 @@ async def extract_b64_and_format(image_base64_data: str) -> typing.Tuple[str, st
return base64_str, image_format
async def get_slack_image_to_base64(pic_url:str):
pass
async def get_slack_image_to_base64(pic_url:str,bot_token:str):
"""
将Slack图片转换为base64
"""
async def get_slack_image_to_base64(pic_url:str, bot_token:str):
headers = {"Authorization": f"Bearer {bot_token}"}
try:
async with aiohttp.ClientSession() as session:
async with session.get(pic_url, headers=headers) as resp:
image_data = await resp.read()
return base64.b64encode(image_data).decode('utf-8')
except Exception as e:
raise(e)