From 9d5bc39fdd0a25888f764d0c5220ac70c6a22773 Mon Sep 17 00:00:00 2001
From: xtaodada <xtao@xtaolink.cn>
Date: Wed, 23 Aug 2023 20:55:52 +0800
Subject: [PATCH] :bug: Fix post cover

---
 src/render/article.py        | 8 ++++++--
 src/templates/article.jinja2 | 4 ++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/render/article.py b/src/render/article.py
index bdb51f1..18b2403 100644
--- a/src/render/article.py
+++ b/src/render/article.py
@@ -42,8 +42,8 @@ def get_description(soup: BeautifulSoup) -> str:
 
 def format_image_url(url: str) -> str:
     if url.endswith(".png") or url.endswith(".jpg"):
-        return f'<img src="{url}{Hyperion.get_images_params()}"/>'
-    return url
+        url += Hyperion.get_images_params()
+    return f'<img src="{url}"/>'
 
 
 def parse_tag(tag: Union[Tag, PageElement]) -> str:
@@ -100,6 +100,9 @@ def parse_stat(stat: PostStat):
 
 
 def get_public_data(game_id: str, post_id: int, post_info: PostInfo) -> Dict:
+    cover = post_info.cover
+    if (not post_info.cover) and post_info.image_urls:
+        cover = post_info.image_urls[0]
     return {
         "url": f"https://www.miyoushe.com/{game_id}/article/{post_id}",
         "published_time": datetime.fromtimestamp(post_info.created_at).strftime(
@@ -108,6 +111,7 @@ def get_public_data(game_id: str, post_id: int, post_info: PostInfo) -> Dict:
         "channel": CHANNEL_MAP.get(game_id, "HSRCN"),
         "stat": parse_stat(post_info.stat),
         "game_id": game_id,
+        "cover": cover,
         "post": post_info,
         "author": post_info["post"]["user"],
     }
diff --git a/src/templates/article.jinja2 b/src/templates/article.jinja2
index 5c81db9..2998461 100644
--- a/src/templates/article.jinja2
+++ b/src/templates/article.jinja2
@@ -18,11 +18,11 @@ Embed MiYouShe posts, videos, polls, and more on Telegram
     <meta property="twitter:site" content="{{ author.nickname }}"/>
     <meta property="twitter:creator" content="{{ author.nickname }}"/>
     <meta property="twitter:title" content="{{ post.subject }} ({{ author.nickname }})"/>
-    <meta property="twitter:image" content="{{ post.cover }}"/>
+    <meta property="twitter:image" content="{{ cover }}"/>
     <meta property="twitter:card" content="summary_large_image"/>
 
     <meta property="og:url" content="{{ url }}"/>
-    <meta property="og:image" content="{{ post.cover }}"/>
+    <meta property="og:image" content="{{ cover }}"/>
     <meta property="og:title" content="{{ post.subject }} ({{ author.nickname }})"/>
     <meta property="og:description" content="{{ description }}"/>
     <meta property="og:site_name" content="{{ post.subject }} - {{ author.nickname }} - 米游社"/>