koishi-plugin-stock 1.0.4 → 1.0.8

package/lib/index.d.ts

@@ -1,5 +1,10 @@
 import { Context, Schema } from 'koishi';
 export interface Config {
+    activeMarketCapBlacklist?: string[];
+    stockAlertBlacklist?: string[];
+    limitUpBoardBlacklist?: string[];
+    stockSelectionBlacklist?: string[];
+    allCommandsBlacklist?: string[];
 }
 export declare const Config: Schema<Config>;
-export declare function apply(ctx: Context): void;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -3,17 +3,58 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.Config = void 0;
 exports.apply = apply;
 const koishi_1 = require("koishi");
-exports.Config = koishi_1.Schema.object({});
-function apply(ctx) {
+exports.Config = koishi_1.Schema.object({
+    allCommandsBlacklist: koishi_1.Schema.array(String).description('全部指令黑名单用户ID'),
+    activeMarketCapBlacklist: koishi_1.Schema.array(String).description('活跃市值指令黑名单用户ID'),
+    stockAlertBlacklist: koishi_1.Schema.array(String).description('异动指令黑名单用户ID'),
+    limitUpBoardBlacklist: koishi_1.Schema.array(String).description('涨停看板指令黑名单用户ID'),
+    stockSelectionBlacklist: koishi_1.Schema.array(String).description('选股指令黑名单用户ID'),
+});
+function apply(ctx, config) {
+    // 检查用户是否在特定指令的黑名单中
+    function isUserInSpecificBlacklist(session, commandName) {
+        const userId = session.userId;
+        // 检查特定指令黑名单
+        switch (commandName) {
+            case '活跃市值':
+                if (config.activeMarketCapBlacklist?.includes(userId)) {
+                    return true;
+                }
+                break;
+            case '异动':
+                if (config.stockAlertBlacklist?.includes(userId)) {
+                    return true;
+                }
+                break;
+            case '涨停看板':
+                if (config.limitUpBoardBlacklist?.includes(userId)) {
+                    return true;
+                }
+                break;
+            case '选股':
+                if (config.stockSelectionBlacklist?.includes(userId)) {
+                    return true;
+                }
+                break;
+        }
+        // 检查全局黑名单
+        if (config.allCommandsBlacklist?.includes(userId)) {
+            return true;
+        }
+        return false;
+    }
     // 监听活跃市值命令
     ctx.command('活跃市值', '获取活跃市值数据')
         .action(async ({ session }) => {
+        if (isUserInSpecificBlacklist(session, '活跃市值')) {
+            return '您已被加入黑名单，无法使用此功能。';
+        }
         try {
             // 使用Koishi的HTTP服务发起请求获取数据
             // 根据测试，API返回的是文本格式而非JSON
             const responseText = await ctx.http.get('http://stock.svip886.com/api/indexes', { responseType: 'text' });
             // 直接返回API返回的数据
-            return `📊 活跃市值数据：\n\n${responseText}`;
+            return `📊 指数看板：\n\n${responseText}`;
         }
         catch (error) {
             console.error('获取活跃市值数据失败:', error);
@@ -23,6 +64,9 @@ function apply(ctx) {
     // 监听异动命令，接受股票代码参数
     ctx.command('异动 <stockCode:text>', '获取指定股票的异动分析数据')
         .action(async ({ session }, stockCode) => {
+        if (isUserInSpecificBlacklist(session, '异动')) {
+            return '您已被加入黑名单，无法使用此功能。';
+        }
         if (!stockCode) {
             return '请输入股票代码，格式：异动 [股票代码]';
         }
@@ -40,6 +84,9 @@ function apply(ctx) {
     // 监听涨停看板命令
     ctx.command('涨停看板', '获取涨停看板图片')
         .action(async ({ session }) => {
+        if (isUserInSpecificBlacklist(session, '涨停看板')) {
+            return '您已被加入黑名单，无法使用此功能。';
+        }
         try {
             // 使用Koishi的HTTP服务下载图片
             const imageUrl = 'http://stock.svip886.com/api/limit_up.png';
@@ -58,6 +105,9 @@ function apply(ctx) {
     // 监听选股命令
     ctx.command('选股 <strategy:text>', '根据指定策略选股（支持策略：N型、填坑、少妇、突破、补票、少妇pro）')
         .action(async ({ session }, strategy) => {
+        if (isUserInSpecificBlacklist(session, '选股')) {
+            return '您已被加入黑名单，无法使用此功能。';
+        }
         if (!strategy) {
             return '请输入选股策略，格式：选股 [策略名称或编号]\n支持的策略：N型(1)、填坑(2)、少妇(3)、突破(4)、补票(5)、少妇pro(6)';
         }
@@ -102,6 +152,9 @@ function apply(ctx) {
     ctx.middleware(async (session, next) => {
         const content = session.content?.trim();
         if (content === '活跃市值') {
+            if (isUserInSpecificBlacklist(session, '活跃市值')) {
+                return '您已被加入黑名单，无法使用此功能。';
+            }
             try {
                 // 使用Koishi的HTTP服务发起请求获取数据
                 const responseText = await ctx.http.get('http://stock.svip886.com/api/indexes', { responseType: 'text' });
@@ -114,6 +167,9 @@ function apply(ctx) {
             }
         }
         else if (content?.startsWith('异动 ')) {
+            if (isUserInSpecificBlacklist(session, '异动')) {
+                return '您已被加入黑名单，无法使用此功能。';
+            }
             // 解析股票代码
             const match = content.match(/^异动\s+(.+)$/);
             if (match) {
@@ -131,6 +187,9 @@ function apply(ctx) {
             }
         }
         else if (content === '涨停看板') {
+            if (isUserInSpecificBlacklist(session, '涨停看板')) {
+                return '您已被加入黑名单，无法使用此功能。';
+            }
             try {
                 // 使用Koishi的HTTP服务下载图片
                 const imageUrl = 'http://stock.svip886.com/api/limit_up.png';
@@ -147,6 +206,9 @@ function apply(ctx) {
             }
         }
         else if (content?.startsWith('选股 ')) {
+            if (isUserInSpecificBlacklist(session, '选股')) {
+                return '您已被加入黑名单，无法使用此功能。';
+            }
             // 解析选股策略
             const match = content.match(/^选股\s+(.+)$/);
             if (match) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-stock",
-  "version": "1.0.4",
+  "version": "1.0.8",
   "description": "A Koishi plugin that fetches stock data when user sends '活跃市值' or '异动 [stock code]'",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",