From d1c371dfd82e7939bd0313e84114e725db3a8007 Mon Sep 17 00:00:00 2001
From: Rongrong <15956627+Rongronggg9@users.noreply.github.com>
Date: Sun, 23 Jan 2022 00:03:49 +0800
Subject: [PATCH] =?UTF-8?q?feat(route)(picuki/profile):=20=E6=94=AF?=
 =?UTF-8?q?=E6=8C=81=E5=A4=9A=E5=9B=BE=E7=89=87/=E8=A7=86=E9=A2=91?=
 =?UTF-8?q?=E8=A7=A3=E6=9E=90=20(#8303)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lib/routes/picuki/profile.js | 67 ++++++++++++++++++++++++++++--------
 1 file changed, 53 insertions(+), 14 deletions(-)

diff --git a/lib/routes/picuki/profile.js b/lib/routes/picuki/profile.js
index 58ec1921bd..0e91e10d8b 100644
--- a/lib/routes/picuki/profile.js
+++ b/lib/routes/picuki/profile.js
@@ -4,7 +4,7 @@ const chrono = require('chrono-node');
 
 module.exports = async (ctx) => {
     const id = ctx.params.id;
-    const displayVideo = ctx.params.displayVideo ? true : false;
+    const displayVideo = !!ctx.params.displayVideo;
 
     const url = `https://www.picuki.com/profile/${id}`;
 
@@ -17,27 +17,66 @@ module.exports = async (ctx) => {
 
     const list = $('ul.box-photos [data-s="media"]').get();
 
-    async function getVideo(url) {
+    async function getMedia(url) {
         const response = await got.get(url);
         const $ = cheerio.load(response.data);
-        return $('.single-photo').html();
+        // Instagram 允许最多 10 条图片/视频任意混合于一条 post，picuki 在所有情况下都会将它（们）置于 .single-photo 内
+        const media_items = $('.single-photo img,video').get();
+        let html = '';
+        media_items.forEach((item) => {
+            item = $(item);
+            html += item.toString(); // 可能是视频，也可能是图片，格式都比较友好，这里直接添加
+        });
+        return html;
+    }
+
+    function deVideo(media) {
+        const $ = cheerio.load(media);
+        const media_list = $('img,video').get();
+        let media_deVideo = '';
+        media_list.forEach((medium) => {
+            medium = $(medium);
+            const poster = medium.attr('poster');
+            // 如果有 poster 属性，表明它是视频，将它替换为它的 poster；如果不是，就原样返回
+            media_deVideo += poster ? `<img src='${poster}' alt='video poster'>` : medium.toString();
+        });
+        return media_deVideo;
     }
 
     const items = await Promise.all(
         list.map(async (post) => {
             post = $(post);
-            const isVideo = post.find('.video-icon').length !== 0 && displayVideo;
-            const post_image = post.find('.post-image').attr('src');
-            const post_description = post.find('.photo-description').text().trim();
+
             const post_link = post.find('.photo > a').attr('href');
-            const post_time = post.find('.time');
-            return {
-                title: post_description || 'Untitled',
-                author: `@${id}`,
-                description: (isVideo ? await getVideo(post_link) : `<img src="${post_image}">`) + (post_description.length > 100 ? `<p>${post_description}</p>` : ''),
-                link: post_link,
-                pubDate: post_time ? chrono.parseDate(post_time.text()) : new Date(),
-            };
+            const cache = await ctx.cache.get(post_link);
+
+            let cache_d;
+            if (cache) {
+                cache_d = JSON.parse(cache);
+            } else {
+                const post_description = post.find('.photo-description').text().trim();
+                const post_time = post.find('.time');
+                const media_displayVideo = await getMedia(post_link);
+                const media = deVideo(media_displayVideo);
+
+                cache_d = {
+                    title: post_description || 'Untitled',
+                    author: `@${id}`,
+                    media_prefix: media,
+                    media_displayVideo_prefix: media_displayVideo,
+                    description_body: post_description.length > 100 ? `<p>${post_description}</p>` : '',
+                    link: post_link,
+                    pubDate: post_time ? chrono.parseDate(post_time.text()) : new Date(),
+                };
+
+                ctx.cache.set(post_link, JSON.stringify(cache_d));
+            }
+
+            cache_d.description = (displayVideo ? cache_d.media_displayVideo_prefix : cache_d.media_prefix) + cache_d.description_body;
+            delete cache_d.media_prefix;
+            delete cache_d.media_displayVideo_prefix;
+            delete cache_d.description_body;
+            return cache_d;
         })
     );