feat: bsky support video

This commit is contained in:
xtaodada 2024-10-22 17:10:05 +08:00
parent 70369ad436
commit c2f6c1255c
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
2 changed files with 23 additions and 5 deletions

View File

@ -100,6 +100,16 @@ class Timeline:
parse_mode=ParseMode.HTML,
reply_markup=Timeline.get_button(post),
)
elif post.video:
return await bot.send_video(
reply.cid,
post.video,
caption=text,
thumb=post.video_thumbnail,
reply_to_message_id=reply.mid,
parse_mode=ParseMode.HTML,
reply_markup=Timeline.get_button(post),
)
elif not post.images:
return await bot.send_message(
reply.cid,

View File

@ -27,6 +27,7 @@ if TYPE_CHECKING:
)
TZ = pytz.timezone("Asia/Shanghai")
XRPC_DOMAIN = "bsky.social"
class HumanAuthor(BaseModel):
@ -99,6 +100,7 @@ class HumanPost(BaseModel, frozen=False):
images: Optional[list[str]] = None
gif: Optional[str] = None
video: Optional[str] = None
video_thumbnail: Optional[str] = None
external: Optional[str] = None
created_at: datetime
@ -138,12 +140,18 @@ class HumanPost(BaseModel, frozen=False):
@staticmethod
def parse_view(post: Union["PostView", "BskyViewRecordRecord"]) -> "HumanPost":
record = post.value if isinstance(post, BskyViewRecordRecord) else post.record
# author
author = HumanAuthor.parse(post.author)
embed = (
(post.embeds[0] if post.embeds else None)
if isinstance(post, BskyViewRecordRecord)
else post.embed
)
content = bsky_html_parser.unparse(record.text, record.facets) if record.facets else record.text
content = (
bsky_html_parser.unparse(record.text, record.facets)
if record.facets
else record.text
)
created_at = record.created_at
# images
images = []
@ -151,9 +159,10 @@ class HumanPost(BaseModel, frozen=False):
for image in embed.images:
images.append(image.fullsize)
# video
video = None
video, video_thumbnail = None, None
if isinstance(embed, BskyViewVideo):
video = embed.playlist # m3u8
video = f"https://{XRPC_DOMAIN}/xrpc/com.atproto.sync.getBlob?did={author.did}&cid={embed.cid}"
video_thumbnail = embed.thumbnail
# gif
gif, extra = None, None
if isinstance(embed, BskyViewExternal):
@ -162,14 +171,13 @@ class HumanPost(BaseModel, frozen=False):
gif = uri
else:
extra = uri
# author
author = HumanAuthor.parse(post.author)
return HumanPost(
cid=post.cid,
content=content,
images=images,
gif=gif,
video=video,
video_thumbnail=video_thumbnail,
external=extra,
created_at=created_at,
like_count=post.like_count,