koishi-plugin-rocom 1.0.12 → 1.0.17

package/lib/client.js

@@ -3,6 +3,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.RocomClient = void 0;
 const koishi_1 = require("koishi");
 const logger = new koishi_1.Logger('rocom-client');
+// ingame 异步任务状态分类（来自服务端 /ingame/tasks/{task_id} 返回的 status 字段）
+const INGAME_PENDING_STATUSES = ['queued', 'pending', 'running', 'processing', 'accepted'];
+const INGAME_FAILED_STATUSES = ['failed', 'error', 'timeout', 'cancelled', 'canceled'];
+const INGAME_COMPLETED_STATUSES = ['done', 'success', 'succeeded', 'completed', 'finished'];
 class RocomClient {
     baseUrl;
     apiKey;
@@ -349,6 +353,117 @@ class RocomClient {
     async getIngameTask(ctx, taskId, includeApiKey = true) {
         return this.requestWithStatus(ctx, 'GET', `/api/v1/games/rocom/ingame/tasks/${taskId}`, this.wegameHeaders('', '', '', '', includeApiKey), { acceptedStatuses: [200, 202] });
     }
+    static normalizeTaskStatus(value) {
+        return String(value ?? '').trim().toLowerCase();
+    }
+    static extractTaskId(payload) {
+        return String(payload?.task_id || payload?.taskId || payload?.taskID || '').trim();
+    }
+    // 判断一份 payload 是否为“已完成的业务结果”（而非排队占位）。
+    // 家园接口完成时返回 rows/home_info；玩家、商店完成时返回 source/title/rows；
+    // 家园详情（pet/data）完成时返回 npc_pets/npc_pet。
+    static isCompletedGatewayPayload(payload) {
+        if (!payload || typeof payload !== 'object')
+            return false;
+        if (Array.isArray(payload.rows))
+            return true;
+        if (payload.home_info !== undefined)
+            return true;
+        if (payload.source !== undefined)
+            return true;
+        if (Array.isArray(payload.npc_pets))
+            return true;
+        if (payload.npc_pet && typeof payload.npc_pet === 'object')
+            return true;
+        if (String(payload.title || '').trim())
+            return true;
+        return false;
+    }
+    static extractCompletedIngamePayload(payload) {
+        if (!payload || typeof payload !== 'object')
+            return null;
+        if (RocomClient.isCompletedGatewayPayload(payload))
+            return payload;
+        if (RocomClient.isCompletedGatewayPayload(payload.result))
+            return payload.result;
+        if (RocomClient.isCompletedGatewayPayload(payload.data))
+            return payload.data;
+        return null;
+    }
+    static taskErrorMessage(payload, fallbackStatus = '') {
+        const candidates = [
+            payload?.message, payload?.error, payload?.error_message, payload?.reason,
+            payload?.result?.message, payload?.result?.error, payload?.result?.error_message,
+            payload?.data?.message, payload?.data?.error,
+        ];
+        for (const candidate of candidates) {
+            const text = String(candidate ?? '').trim();
+            if (text && text.toLowerCase() !== 'failed')
+                return text;
+        }
+        return String(payload?.status || fallbackStatus || '').trim() || '任务执行失败';
+    }
+    // 排队等候轮询：把 ingame 接口提交后返回的 202（task_id/status=queued）轮询到出结果或超时。
+    // 入参 first 是首请求拿到的 { status, data, usedApiKey }。
+    async pollIngameTask(ctx, first, options, labels) {
+        const { status, data, usedApiKey } = first;
+        // 服务端在 wait_ms 内已直接返回业务结果，或任务壳里已经带了完成结果。
+        const initialCompletedPayload = RocomClient.extractCompletedIngamePayload(data);
+        if (status !== null && initialCompletedPayload)
+            return initialCompletedPayload;
+        const taskId = RocomClient.extractTaskId(data);
+        if (!taskId) {
+            if (status === 202)
+                this.setLastError(labels.queuedNoTaskId);
+            // 既不是任务也不是已知完成结构：把原始 data 透传给上层兼容旧逻辑。
+            return status === null ? null : data;
+        }
+        // 已进入队列，回调通知“排队中”。
+        if (typeof options.onQueued === 'function') {
+            try {
+                await options.onQueued(taskId);
+            }
+            catch (e) {
+                logger.warn(`ingame 排队提示回调失败: ${e}`);
+            }
+        }
+        const intervalMs = Math.max(300, Number(options.intervalMs) || 3000);
+        const timeoutMs = Math.max(intervalMs, Number(options.timeoutMs) || 180000);
+        const startedAt = Date.now();
+        let lastStatus = RocomClient.normalizeTaskStatus(data?.status);
+        while (Date.now() - startedAt < timeoutMs) {
+            if (lastStatus && INGAME_FAILED_STATUSES.includes(lastStatus)) {
+                this.setLastError(RocomClient.taskErrorMessage(data, lastStatus));
+                return null;
+            }
+            await new Promise(resolve => setTimeout(resolve, intervalMs));
+            const task = await this.getIngameTask(ctx, taskId, usedApiKey);
+            if (task.status === null)
+                return null;
+            const completedPayload = RocomClient.extractCompletedIngamePayload(task.data);
+            if (completedPayload)
+                return completedPayload;
+            lastStatus = RocomClient.normalizeTaskStatus(task.data?.status);
+            if (INGAME_FAILED_STATUSES.includes(lastStatus)) {
+                this.setLastError(RocomClient.taskErrorMessage(task.data, lastStatus));
+                return null;
+            }
+            // 状态标记完成，但结果可能嵌套在 result/data 中。
+            if (INGAME_COMPLETED_STATUSES.includes(lastStatus)) {
+                this.setLastError(`Ingame 任务已完成但未返回可解析结果：${taskId}`);
+                return null;
+            }
+            if (lastStatus && !INGAME_PENDING_STATUSES.includes(lastStatus)) {
+                this.setLastError(`Ingame 任务状态异常：${lastStatus}`);
+                return null;
+            }
+            // 既无 task_id 也无状态，说明拿到的就是裸结果。
+            if (!RocomClient.extractTaskId(task.data) && !lastStatus)
+                return task.data;
+        }
+        this.setLastError(labels.stillQueued(taskId));
+        return null;
+    }
     async qqQrLogin(ctx, userIdentifier) {
         const params = { client_type: 'bot', client_id: 'koishi', provider: 'rocom' };
         if (userIdentifier)
@@ -496,9 +611,24 @@ class RocomClient {
         return this.get(ctx, '/api/v1/games/rocom/exchange/posters', this.wegameHeaders(fwToken, userIdentifier), params);
     }
     async getMerchantInfo(ctx, refresh = false) {
-        return this.get(ctx, '/api/v1/games/rocom/merchant/info', this.wegameHeaders(), { refresh: refresh ? 'true' : 'false' });
+        return this.get(ctx, '/api/v1/games/rocom/merchant/info', this.wegameHeaders(), {
+            refresh: refresh ? 'true' : 'false',
+            random_goods: 'all',
+        });
     }
     async queryPetSize(ctx, diameter, weight, sameRideEgg = false, userIdentifier = '') {
+        // 新版蛋尺寸反查：/wiki/pet-size/query（items[].pet/egg_size/match 结构），失败时回退旧接口。
+        const newParams = this.scopedParams({
+            diameter,
+            weight,
+            pool: sameRideEgg ? 'ride' : 'magic',
+            include_display_only: 'false',
+            page_no: 1,
+            page_size: 30,
+        }, userIdentifier);
+        const newRes = await this.get(ctx, '/api/v1/games/rocom/wiki/pet-size/query', this.wegameHeaders('', userIdentifier, 'bot', 'koishi'), newParams, { silentFailureDetails: true });
+        if (newRes && Array.isArray(newRes.items))
+            return newRes;
         const params = this.scopedParams({ diameter, weight }, userIdentifier);
         if (sameRideEgg)
             params.sameRideEgg = 1;
@@ -562,39 +692,47 @@ class RocomClient {
             params.after_event_id = afterEventId;
         return this.get(ctx, '/api/v1/games/rocom/community/egg-exchange-events', this.wegameHeaders('', userIdentifier, 'bot', 'koishi'), this.scopedParams(params, userIdentifier));
     }
-    async ingameHomeInfo(ctx, uid, waitMs = 5000) {
+    async ingameHomeInfo(ctx, uid, options = {}) {
         const sanitizedUid = this.sanitizeUid(uid);
         if (!sanitizedUid) {
             this.setLastError('UID 不能为空');
             return null;
         }
+        const waitMs = Number(options.waitMs) || 5000;
         const path = '/api/v1/games/rocom/ingame/home/info';
         const payload = { uid: sanitizedUid, wait_ms: waitMs };
-        const { status, data, usedApiKey } = await this.requestIngameWithFallback(ctx, path, payload);
-        if (status === 200 && data && !(data.task_id || data.taskId || data.taskID))
-            return data;
-        const taskId = data?.task_id || data?.taskId || data?.taskID;
-        if (!taskId) {
-            if (status === 202)
-                this.setLastError('家园查询任务已入队，但未返回 task_id');
-            return null;
-        }
-        for (let i = 0; i < 10; i++) {
-            await new Promise(resolve => setTimeout(resolve, 1000));
-            const task = await this.getIngameTask(ctx, taskId, usedApiKey);
-            if (task.status === 200)
-                return task.data;
-            if (task.status === null)
-                return null;
-        }
-        this.setLastError(`Home query task is still queued, please retry later (task_id: ${taskId})`);
-        return null;
+        const first = await this.requestIngameWithFallback(ctx, path, payload);
+        return this.pollIngameTask(ctx, first, options, {
+            queuedNoTaskId: '家园查询任务已入队，但未返回 task_id',
+            stillQueued: (taskId) => `家园查询任务仍在排队，请稍后重试（task_id: ${taskId}）`,
+        });
     }
     async ingameMerchantInfo(ctx, shopId) {
         const params = { shop_id: shopId };
         const { status, data } = await this.requestIngameWithFallback(ctx, '/api/v1/games/rocom/ingame/merchant/info', params);
         return status === null ? null : data;
     }
+    // 家园详情：批量或单只查询家园摆放精灵的完整 ingame 数据（上游 v3.6.1）。
+    // UID 参数名为 target_uin；单只查询需同时提供 pet_gid 和 npc_id。
+    async ingamePetData(ctx, uid, extras = {}, options = {}) {
+        const sanitizedUid = this.sanitizeUid(uid);
+        if (!sanitizedUid) {
+            this.setLastError('UID 不能为空');
+            return null;
+        }
+        const waitMs = Number(options.waitMs) || 20000;
+        const payload = { target_uin: sanitizedUid, wait_ms: waitMs };
+        if (extras.petGid !== undefined && extras.petGid !== null && String(extras.petGid) !== '')
+            payload.pet_gid = extras.petGid;
+        if (extras.npcId !== undefined && extras.npcId !== null && String(extras.npcId) !== '')
+            payload.npc_id = extras.npcId;
+        const path = '/api/v1/games/rocom/ingame/pet/data';
+        const first = await this.requestIngameWithFallback(ctx, path, payload);
+        return this.pollIngameTask(ctx, first, options, {
+            queuedNoTaskId: '家园详情任务已入队，但未返回 task_id',
+            stillQueued: (taskId) => `家园详情任务仍在排队，请稍后重试（task_id: ${taskId}）`,
+        });
+    }
     async getFriendship(ctx, fwToken, userIds, userIdentifier = '') {
         return this.get(ctx, '/api/v1/games/rocom/social/friendship', this.rocomHeaders(fwToken, userIdentifier), { user_ids: userIds });
     }
@@ -626,5 +764,67 @@ class RocomClient {
             params.name = options.name;
         return this.get(ctx, '/api/v1/games/rocom/pet/detail', this.wegameHeaders(), params);
     }
+    // ===== 新版 RoCom Wiki API（上游 v3.6.0）=====
+    wikiPagedParams(q = '', pageNo = 1, pageSize = 10, filters = {}) {
+        const params = {
+            page_no: Math.max(Number(pageNo) || 1, 1),
+            page_size: Math.min(Math.max(Number(pageSize) || 10, 1), 100),
+        };
+        if (q)
+            params.q = q;
+        for (const [key, value] of Object.entries(filters)) {
+            if (value !== undefined && value !== null && value !== '')
+                params[key] = value;
+        }
+        return params;
+    }
+    async listWikiPets(ctx, q = '', pageNo = 1, pageSize = 10, filters = {}) {
+        return this.get(ctx, '/api/v1/games/rocom/wiki/pets', this.wegameHeaders(), this.wikiPagedParams(q, pageNo, pageSize, filters));
+    }
+    async getWikiPet(ctx, petId) {
+        return this.get(ctx, `/api/v1/games/rocom/wiki/pets/${petId}`, this.wegameHeaders());
+    }
+    async getWikiPetProfile(ctx, petId) {
+        return this.get(ctx, `/api/v1/games/rocom/wiki/pets/${petId}/profile`, this.wegameHeaders(), undefined, { silentFailureDetails: true });
+    }
+    async getWikiPetSkills(ctx, petId) {
+        return this.get(ctx, `/api/v1/games/rocom/wiki/pets/${petId}/skills`, this.wegameHeaders(), undefined, { silentFailureDetails: true });
+    }
+    async getWikiPetFamily(ctx, petId) {
+        return this.get(ctx, `/api/v1/games/rocom/wiki/pets/${petId}/family`, this.wegameHeaders(), undefined, { silentFailureDetails: true });
+    }
+    async getWikiPetHandbook(ctx, petId) {
+        return this.get(ctx, `/api/v1/games/rocom/wiki/pets/${petId}/handbook`, this.wegameHeaders(), undefined, { silentFailureDetails: true });
+    }
+    async listWikiSkills(ctx, q = '', pageNo = 1, pageSize = 10, filters = {}) {
+        return this.get(ctx, '/api/v1/games/rocom/wiki/skills', this.wegameHeaders(), this.wikiPagedParams(q, pageNo, pageSize, filters));
+    }
+    async getWikiSkill(ctx, skillId) {
+        return this.get(ctx, `/api/v1/games/rocom/wiki/skills/${skillId}`, this.wegameHeaders(), undefined, { silentFailureDetails: true });
+    }
+    async getWikiSkillPets(ctx, skillId) {
+        return this.get(ctx, `/api/v1/games/rocom/wiki/skills/${skillId}/pets`, this.wegameHeaders(), undefined, { silentFailureDetails: true });
+    }
+    async getWikiCatalogs(ctx) {
+        return this.get(ctx, '/api/v1/games/rocom/wiki/catalogs', this.wegameHeaders());
+    }
+    async getWikiOptions(ctx) {
+        return this.get(ctx, '/api/v1/games/rocom/wiki/options', this.wegameHeaders(), undefined, { silentFailureDetails: true });
+    }
+    async getWikiPath(ctx, path, params = {}) {
+        const trimmed = String(path || '').trim();
+        if (!trimmed.startsWith('/api/v1/games/rocom/wiki/')) {
+            this.setLastError('非法 Wiki 路径');
+            return null;
+        }
+        return this.get(ctx, trimmed, this.wegameHeaders(), params);
+    }
+    async listWikiCatalogItems(ctx, path, q = '', pageNo = 1, pageSize = 10, search = true) {
+        const params = this.wikiPagedParams(search ? q : '', pageNo, pageSize);
+        return this.getWikiPath(ctx, path, params);
+    }
+    get wikiAssetBaseUrl() {
+        return this.baseUrl;
+    }
 }
 exports.RocomClient = RocomClient;

package/lib/commands/account.js

@@ -97,6 +97,7 @@ function register(deps) {
     const { ctx, client, userMgr } = deps;
     ctx.command('洛克').subcommand('.QQ登录', 'QQ 扫码登录')
         .alias('洛克QQ登录')
+        .alias('洛克qq登录')
         .action(async ({ session }) => {
         const userId = session.userId;
         const qrData = await client.qqQrLogin(ctx, userId);