koishi-plugin-rocom 1.0.1

package/lib/commands/merchant.js

@@ -0,0 +1,242 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.register = register;
+const koishi_1 = require("koishi");
+const send_image_1 = require("../send-image");
+const subscription_send_1 = require("../subscription-send");
+const logger = new koishi_1.Logger('rocom-merchant');
+const TEXT = {
+    merchant: '\u8fdc\u884c\u5546\u4eba',
+    subscribe: '\u8ba2\u9605\u8fdc\u884c\u5546\u4eba',
+    unsubscribe: '\u53d6\u6d88\u8ba2\u9605\u8fdc\u884c\u5546\u4eba',
+    viewSubscribe: '\u67e5\u770b\u8fdc\u884c\u5546\u4eba\u8ba2\u9605',
+    unknown: '\u672a\u77e5',
+    notOpen: '\u672a\u5f00\u653e',
+    defaultSource: '\u9ed8\u8ba4',
+    customSource: '\u81ea\u5b9a\u4e49',
+};
+function normalizeTimestamp(value) {
+    if (value === null || value === undefined || value === '')
+        return null;
+    const timestamp = Number(value);
+    if (!Number.isFinite(timestamp))
+        return null;
+    return timestamp < 100000000000 ? timestamp * 1000 : timestamp;
+}
+function formatProductWindow(product) {
+    const start = normalizeTimestamp(product?.start_time);
+    const end = normalizeTimestamp(product?.end_time);
+    if (!start && !end)
+        return '';
+    const formatTime = (timestamp) => {
+        const date = new Date(timestamp);
+        const hour = String(date.getHours()).padStart(2, '0');
+        const minute = String(date.getMinutes()).padStart(2, '0');
+        return `${hour}:${minute}`;
+    };
+    if (start && end)
+        return `${formatTime(start)}-${formatTime(end)}`;
+    if (start)
+        return `${formatTime(start)}+`;
+    return `${formatTime(end)}-`;
+}
+function getMerchantActivity(res) {
+    const activities = res?.merchantActivities || res?.merchant_activities || [];
+    return activities[0] || {};
+}
+function getActiveProducts(res) {
+    const activity = getMerchantActivity(res);
+    const products = activity?.products || activity?.product_list || activity?.get_props || [];
+    return products.filter((p) => {
+        const now = Date.now();
+        const start = normalizeTimestamp(p.start_time) ?? 0;
+        const end = normalizeTimestamp(p.end_time) ?? Infinity;
+        return now >= start && now < end;
+    });
+}
+function getCurrentMerchantRound() {
+    const now = new Date();
+    const hour = now.getHours();
+    const rounds = [8, 12, 16, 20];
+    let currentRound = null;
+    let nextRound = null;
+    for (let i = 0; i < rounds.length; i++) {
+        if (hour >= rounds[i] && (i === rounds.length - 1 || hour < rounds[i + 1])) {
+            currentRound = i + 1;
+            nextRound = i < rounds.length - 1 ? rounds[i + 1] : rounds[0];
+            break;
+        }
+    }
+    if (currentRound === null)
+        nextRound = rounds[0];
+    const nextTime = new Date(now);
+    if (nextRound <= hour)
+        nextTime.setDate(nextTime.getDate() + 1);
+    nextTime.setHours(nextRound, 0, 0, 0);
+    const diff = nextTime.getTime() - now.getTime();
+    const hours = Math.floor(diff / 3600000);
+    const mins = Math.floor((diff % 3600000) / 60000);
+    const datePart = [
+        now.getFullYear(),
+        String(now.getMonth() + 1).padStart(2, '0'),
+        String(now.getDate()).padStart(2, '0'),
+    ].join('-');
+    return {
+        current: currentRound,
+        countdown: `${hours}\u5c0f\u65f6${mins}\u5206\u949f`,
+        is_open: currentRound !== null,
+        round_id: `${datePart}-${currentRound || 'closed'}`,
+    };
+}
+function parseMerchantSubscriptionArgs(args, defaultItems) {
+    const parts = (args || '').split(/\s+/).filter(Boolean);
+    let mentionAll = false;
+    if (parts[0] === '1' || parts[0] === '0') {
+        mentionAll = parts.shift() === '1';
+    }
+    const items = parts.length ? parts : defaultItems;
+    return {
+        mention_all: mentionAll,
+        items,
+        source: parts.length ? TEXT.customSource : TEXT.defaultSource,
+    };
+}
+function getSubscriptionTarget(session) {
+    const platform = session.platform || session.bot?.platform || '';
+    const privateChat = !session.guildId;
+    const channelId = session.channelId || session.guildId || session.userId || '';
+    const key = privateChat ? `private_${session.userId}` : session.guildId;
+    return { key, platform, channelId, privateChat };
+}
+function isBotAdmin(session, adminUserIds) {
+    return adminUserIds.includes(session?.userId || '');
+}
+function sameStringArray(left, right) {
+    return JSON.stringify(left) === JSON.stringify(right);
+}
+async function checkMerchantSubscriptions(deps) {
+    const { ctx, client, merchantSubMgr } = deps;
+    const res = await client.getMerchantInfo(ctx, true);
+    if (!res)
+        return { subscriptions: 0, matched: 0, pushed: 0 };
+    const products = getActiveProducts(res);
+    const productNames = products.map((p) => p.name || '').filter(Boolean);
+    const roundInfo = getCurrentMerchantRound();
+    const subs = merchantSubMgr.getAll();
+    let matchedCount = 0;
+    let pushedCount = 0;
+    for (const [key, sub] of Object.entries(subs)) {
+        const matched = sub.items.filter((item) => productNames.some(n => n.includes(item)));
+        if (!matched.length)
+            continue;
+        matchedCount++;
+        if (sub.last_push_round === roundInfo.round_id && sameStringArray(matched, sub.last_matched_items || []))
+            continue;
+        const msg = `\ud83d\udd14 \u8fdc\u884c\u5546\u4eba\u5237\u65b0\u63d0\u9192\n\u5f53\u524d\u5546\u54c1\uff1a${productNames.join('\u3001')}\n\u5339\u914d\u8ba2\u9605\uff1a${matched.join('\u3001')}`;
+        const platform = sub.platform || ctx.bots[0]?.platform;
+        const channelId = sub.channel_id || sub.group_id || sub.user_id || key;
+        if (!platform || !channelId) {
+            logger.warn(`\u63a8\u9001\u5931\u8d25 ${key}: \u65e0\u6cd5\u786e\u5b9a\u5e73\u53f0\u6216\u9891\u9053`);
+            continue;
+        }
+        const sent = await (0, subscription_send_1.sendScheduledMessage)(ctx, {
+            platform,
+            channelId,
+            guildId: sub.group_id || '',
+            userId: sub.user_id || '',
+        }, sub.mention_all ? `@\u5168\u4f53\n${msg}` : msg);
+        if (!sent)
+            continue;
+        pushedCount++;
+        merchantSubMgr.upsert(key, {
+            ...sub,
+            last_push_round: roundInfo.round_id,
+            last_matched_items: matched,
+        });
+    }
+    return { subscriptions: Object.keys(subs).length, matched: matchedCount, pushed: pushedCount };
+}
+function register(deps) {
+    const { ctx, config, client, merchantSubMgr } = deps;
+    ctx.command(TEXT.merchant, '\u67e5\u770b\u8fdc\u884c\u5546\u4eba\u5546\u54c1')
+        .action(async ({ session }) => {
+        const res = await client.getMerchantInfo(ctx, true);
+        if (!res)
+            return '\u83b7\u53d6\u8fdc\u884c\u5546\u4eba\u6570\u636e\u5931\u8d25\u3002';
+        const products = getActiveProducts(res);
+        const roundInfo = getCurrentMerchantRound();
+        const activity = getMerchantActivity(res);
+        const data = {
+            background: '',
+            title: activity.name || TEXT.merchant,
+            subtitle: activity.start_date || '\u6bcf\u65e5 08:00 / 12:00 / 16:00 / 20:00 \u5237\u65b0',
+            titleIcon: '',
+            product_count: products.length,
+            round_info: roundInfo,
+            products: products.map((p) => ({
+                name: p.name || TEXT.unknown,
+                image: p.icon_url || '',
+                time_label: formatProductWindow(p),
+            })),
+        };
+        const fallback = products.length
+            ? `\u8fdc\u884c\u5546\u4eba\u5f53\u524d\u5546\u54c1\uff1a${products.map((p) => p.name || TEXT.unknown).join('\u3001')}\n\u8f6e\u6b21\uff1a${roundInfo.current || TEXT.notOpen}\n\u5269\u4f59\uff1a${roundInfo.countdown}`
+            : '\u5f53\u524d\u8fdc\u884c\u5546\u4eba\u6682\u65e0\u5546\u54c1\u3002';
+        const png = await deps.renderer.renderHtml(ctx, 'yuanxing-shangren', data);
+        await (0, send_image_1.sendImageWithFallback)(session, png, fallback, 'merchant:yuanxing-shangren', deps.config);
+    });
+    ctx.command(`${TEXT.subscribe} [args:text]`, '\u8ba2\u9605\u8fdc\u884c\u5546\u4eba\u5546\u54c1\u63d0\u9192')
+        .action(async ({ session }, args) => {
+        const target = getSubscriptionTarget(session);
+        if (!target.privateChat && !isBotAdmin(session, config.adminUserIds))
+            return '此指令仅限管理员使用。';
+        if (target.privateChat && !config.merchantPrivateSubscriptionEnabled)
+            return '个人私聊订阅功能已被禁用，请联系机器人管理员。';
+        const parsed = parseMerchantSubscriptionArgs(args, config.merchantSubscriptionItems);
+        const existing = merchantSubMgr.get(target.key);
+        merchantSubMgr.upsert(target.key, {
+            group_id: session.guildId || '',
+            user_id: target.privateChat ? session.userId : '',
+            type: target.privateChat ? '个人订阅' : '群订阅',
+            channel_id: target.channelId,
+            platform: target.platform,
+            items: parsed.items,
+            mention_all: target.privateChat ? false : parsed.mention_all,
+            last_push_round: existing?.last_push_round ?? null,
+            last_matched_items: existing?.last_matched_items ?? [],
+            updated_by: session.userId,
+        });
+        return `\u2705 \u5df2\u8ba2\u9605\u8fdc\u884c\u5546\u4eba\u5546\u54c1\uff1a${parsed.items.join('\u3001')}\uff08${parsed.source}\uff09\uff1b${target.privateChat ? '个人订阅' : (parsed.mention_all ? '\u547d\u4e2d\u540e\u4f1a @\u5168\u4f53' : '\u547d\u4e2d\u540e\u4e0d @\u5168\u4f53')}`;
+    });
+    ctx.command(TEXT.viewSubscribe, '\u67e5\u770b\u5f53\u524d\u4f1a\u8bdd\u7684\u8fdc\u884c\u5546\u4eba\u8ba2\u9605')
+        .action(async ({ session }) => {
+        const target = getSubscriptionTarget(session);
+        if (!target.privateChat && !isBotAdmin(session, config.adminUserIds))
+            return '此指令仅限管理员使用。';
+        const sub = merchantSubMgr.get(target.key);
+        const scopeName = target.privateChat ? '你' : '当前群组';
+        if (!sub)
+            return `${scopeName}未订阅远行商人。\n用法：${TEXT.subscribe} [1/0] [商品名1] [商品名2] ...`;
+        return `${scopeName}订阅商品：${sub.items.join('、')}\n提醒方式：${target.privateChat ? '私聊提醒' : (sub.mention_all ? '@全体' : '普通提醒')}`;
+    });
+    ctx.command(TEXT.unsubscribe, '\u53d6\u6d88\u8fdc\u884c\u5546\u4eba\u8ba2\u9605')
+        .action(async ({ session }) => {
+        const target = getSubscriptionTarget(session);
+        if (!target.privateChat && !isBotAdmin(session, config.adminUserIds))
+            return '此指令仅限管理员使用。';
+        merchantSubMgr.delete(target.key);
+        return '\u2705 \u5df2\u53d6\u6d88\u8fdc\u884c\u5546\u4eba\u8ba2\u9605\u3002';
+    });
+    ctx.command('洛克').subcommand('.调试远行商人订阅', '立即执行一次远行商人订阅检查')
+        .action(async ({ session }) => {
+        if (!isBotAdmin(session, config.adminUserIds))
+            return '此指令仅限管理员使用。';
+        const result = await checkMerchantSubscriptions(deps);
+        return `远行商人订阅检查完成：订阅 ${result.subscriptions} 条，命中 ${result.matched} 条，推送 ${result.pushed} 条。`;
+    });
+    if (config.merchantSubscriptionEnabled) {
+        ctx.setInterval(async () => {
+            await checkMerchantSubscriptions(deps);
+        }, config.merchantCheckInterval);
+    }
+}

package/lib/commands/query.d.ts

@@ -0,0 +1,2 @@
+import { PluginDeps } from '../types';
+export declare function register(deps: PluginDeps): void;