koishi-plugin-steam-info-check 1.0.5 → 1.0.6

package/dist/index.js

@@ -127,8 +127,8 @@ function apply(ctx, config) {
                 const profile = await ctx.steam.getUserData(steamId);
                 const image = await ctx.drawer.drawPlayerStatus(profile, steamId);
                 if (typeof image === 'string')
-                    return session.send(image);
-                return session.send(koishi_1.h.image(image, 'image/png'));
+                    return image;
+                return koishi_1.h.image(image, 'image/png');
             }
             catch (err) {
                 exports.logger.error(err);
@@ -148,15 +148,15 @@ function apply(ctx, config) {
                 const summaries = await ctx.steam.getPlayerSummaries(steamIds);
                 if (summaries.length === 0)
                     return session.text('.api_error');
-                const channelInfo = await ctx.database.get('steam_channel', { id: session.channelId });
-                const parentAvatar = channelInfo[0]?.avatar
-                    ? Buffer.from(channelInfo[0].avatar, 'base64')
+                const channelInfo = await ensureChannelMeta(ctx, session);
+                const parentAvatar = channelInfo.avatar
+                    ? Buffer.from(channelInfo.avatar, 'base64')
                     : await ctx.drawer.getDefaultAvatar();
-                const parentName = channelInfo[0]?.name || session.channelId || 'Unknown';
+                const parentName = channelInfo.name || session.channelId || 'Unknown';
                 const image = await ctx.drawer.drawFriendsStatus(parentAvatar, parentName, summaries, binds);
                 if (typeof image === 'string')
-                    return session.send(image);
-                return session.send(koishi_1.h.image(image, 'image/png'));
+                    return image;
+                return koishi_1.h.image(image, 'image/png');
             }
             catch (err) {
                 exports.logger.error(err);
@@ -236,6 +236,34 @@ function apply(ctx, config) {
 }
 // Broadcast Logic
 const statusCache = new Map();
+async function ensureChannelMeta(ctx, session) {
+    const channelId = session.channelId;
+    const existing = await ctx.database.get('steam_channel', { id: channelId });
+    const current = existing[0] || { id: channelId, enable: true, platform: session.platform, assignee: session.selfId };
+    let name = current.name;
+    if (!name && session.event?.channel?.name) {
+        name = session.event.channel.name;
+    }
+    let avatar = current.avatar;
+    if (!avatar) {
+        try {
+            // For QQ group avatars
+            const url = `http://p.qlogo.cn/gh/${channelId}/${channelId}/0`;
+            const buffer = await ctx.http.get(url, { responseType: 'arraybuffer' });
+            avatar = Buffer.from(buffer).toString('base64');
+        }
+        catch {
+            // fallback later to default avatar
+        }
+    }
+    const update = { id: channelId, platform: session.platform, assignee: session.selfId };
+    if (name)
+        update.name = name;
+    if (avatar)
+        update.avatar = avatar;
+    await ctx.database.upsert('steam_channel', [update]);
+    return { ...current, ...update };
+}
 async function seedStatusCache(ctx) {
     const binds = await ctx.database.get('steam_bind', {});
     if (!binds.length)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-steam-info-check",
-  "version": "1.0.5",
+  "version": "1.0.6",
   "description": "Steam friends status broadcast plugin for Koishi",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/index.ts

@@ -157,8 +157,8 @@ export function apply(ctx: Context, config: Config) {
 
           const profile = await ctx.steam.getUserData(steamId)
           const image = await ctx.drawer.drawPlayerStatus(profile, steamId)
-          if (typeof image === 'string') return session.send(image)
-          return session.send(h.image(image, 'image/png'))
+          if (typeof image === 'string') return image
+          return h.image(image, 'image/png')
         } catch (err) {
           logger.error(err)
           return session.text('.error')
@@ -178,15 +178,15 @@ export function apply(ctx: Context, config: Config) {
           
           if (summaries.length === 0) return session.text('.api_error')
 
-          const channelInfo = await ctx.database.get('steam_channel', { id: session.channelId })
-          const parentAvatar = channelInfo[0]?.avatar 
-            ? Buffer.from(channelInfo[0].avatar, 'base64') 
+          const channelInfo = await ensureChannelMeta(ctx, session)
+          const parentAvatar = channelInfo.avatar 
+            ? Buffer.from(channelInfo.avatar, 'base64') 
             : await ctx.drawer.getDefaultAvatar()
-          const parentName = channelInfo[0]?.name || session.channelId || 'Unknown'
+          const parentName = channelInfo.name || session.channelId || 'Unknown'
 
           const image = await ctx.drawer.drawFriendsStatus(parentAvatar, parentName, summaries, binds)
-          if (typeof image === 'string') return session.send(image)
-          return session.send(h.image(image, 'image/png'))
+          if (typeof image === 'string') return image
+          return h.image(image, 'image/png')
         } catch (err) {
           logger.error(err)
           return session.text('.error')
@@ -269,6 +269,36 @@ export function apply(ctx: Context, config: Config) {
 // Broadcast Logic
 const statusCache = new Map<string, any>()
 
+async function ensureChannelMeta(ctx: Context, session: Session) {
+  const channelId = session.channelId
+  const existing = await ctx.database.get('steam_channel', { id: channelId })
+  const current = existing[0] || { id: channelId, enable: true, platform: session.platform, assignee: session.selfId }
+
+  let name = current.name
+  if (!name && session.event?.channel?.name) {
+    name = session.event.channel.name
+  }
+
+  let avatar = current.avatar
+  if (!avatar) {
+    try {
+      // For QQ group avatars
+      const url = `http://p.qlogo.cn/gh/${channelId}/${channelId}/0`
+      const buffer = await ctx.http.get(url, { responseType: 'arraybuffer' })
+      avatar = Buffer.from(buffer).toString('base64')
+    } catch {
+      // fallback later to default avatar
+    }
+  }
+
+  const update: any = { id: channelId, platform: session.platform, assignee: session.selfId }
+  if (name) update.name = name
+  if (avatar) update.avatar = avatar
+  await ctx.database.upsert('steam_channel', [update])
+
+  return { ...current, ...update }
+}
+
 async function seedStatusCache(ctx: Context) {
   const binds = await ctx.database.get('steam_bind', {})
   if (!binds.length) return