feat: add like ranking

Author: sxzz

src/command/index.ts

@@ -4,6 +4,7 @@ import { version } from './version'
 import { user } from './user'
 import { post } from './post'
 import { msg } from './msg'
+import { likeRank } from './like-rank'
 
 export function run() {
   program
@@ -14,6 +15,7 @@ export function run() {
     .addCommand(user)
     .addCommand(post)
     .addCommand(msg)
+    .addCommand(likeRank)
     .addCommand(version)
     .version(versionNumber)
     .parse()

src/command/like-rank.ts

@@ -0,0 +1,93 @@
+import { logger } from '@poppinss/cliui'
+import { createCommand } from 'commander'
+import { limit } from 'jike-sdk'
+import { createClient, displayUser, filterUsers } from '../utils/user'
+import { displayImage, renderDivider } from '../utils/terminal'
+import type { Entity } from 'jike-sdk'
+
+interface LikeRankOptions {
+  top: number
+  count: number
+}
+
+export const likeRank = createCommand('like-rank')
+  .description('Like ranking')
+  .alias('lr')
+  .option('-t, --top <top count>', 'top count', '20')
+  .option('-c, --count <post count>', 'post count, 0 for all', '0')
+  .action(() => {
+    const opts = likeRank.opts<LikeRankOptions>()
+    likeRanking({
+      top: +opts.top,
+      count: +opts.count,
+    })
+  })
+
+export const likeRanking = async ({ top, count }: LikeRankOptions) => {
+  const [user] = filterUsers()
+
+  const spinner = logger.await('Fetching posts')
+
+  const client = createClient(user)
+  const posts = await client.getSelf().queryPersonalUpdate({
+    limit: count > 0 ? limit.limitMaxCount(count) : limit.limitNone(),
+  })
+
+  const userMap: Record<string, { user: Entity.User; count: number }> = {}
+  for (const [i, post] of posts.entries()) {
+    spinner.update(`Fetching post (${i + 1} / ${posts.length})`)
+
+    const users = await post.listLikedUsers()
+    for (const user of users) {
+      const id = user.id
+      if (!userMap[id]) userMap[id] = { user, count: 1 }
+      else userMap[id].count++
+    }
+  }
+
+  const divider = renderDivider()
+
+  const countRanking = [
+    ...new Set(Object.values(userMap).map(({ count }) => count)),
+  ].sort((a, b) => b - a)
+  const getRanking = (count: number) => countRanking.indexOf(count) + 1
+
+  const ranking = await Promise.all(
+    Object.values(userMap)
+      .sort((a, b) => b.count - a.count)
+      .slice(0, top)
+      .map(async ({ user, count }) => {
+        const ranking = getRanking(count)
+        let text = `${renderRanking(ranking)} ${displayUser(
+          user
+        )} 点赞 ${logger.colors.cyan(`${count}`)} 次，${(
+          (count / posts.length) *
+          100
+        ).toFixed(2)}%`
+
+        if (ranking <= 3) {
+          text = `${
+            (await displayImage(user.avatarImage.thumbnailUrl, 3)).result
+          }\n${text}\n${divider}`
+        }
+        return text
+      })
+  )
+
+  spinner.stop()
+
+  process.stdout.write(ranking.join(`\n`))
+
+  function renderRanking(rank: number) {
+    switch (rank) {
+      case 1:
+        return '🥇'
+      case 2:
+        return '🥈'
+      case 3:
+        return '🥉'
+      default:
+        return `${rank.toString().padStart(`${countRanking.length}`.length)}.`
+    }
+  }
+}