koishi-plugin-gl-bot 0.0.9 → 0.0.11

package/lib/gl/index.d.ts

@@ -1,7 +1,9 @@
 import { Context, Schema } from 'koishi';
+import { NapCat } from '../napCat';
 export declare class GLBot {
     private ctx;
     private config;
+    static inject: string[];
     static Config: Schema<Schemastery.ObjectS<{
         wsServer: Schema<"客户端" | "服务端", "客户端" | "服务端">;
         wsHost: Schema<string, string>;
@@ -35,6 +37,9 @@ export declare class GLBot {
         mcManagerHost: Schema<string, string>;
         mcManagerWs: Schema<string, string>;
         mcManagerMaxConnectWs: Schema<number, number>;
+    }> | Schemastery.ObjectS<{
+        napCatBaseUrl: Schema<string, string>;
+        napCatToken: Schema<string, string>;
     }>, {
         wsServer: "客户端" | "服务端";
         wsHost: string;
@@ -68,9 +73,13 @@ export declare class GLBot {
         mcManagerHost: string;
         mcManagerWs: string;
         mcManagerMaxConnectWs: number;
+    } & {
+        napCatBaseUrl: string;
+        napCatToken: string;
     }>;
     private mcSyncMsg;
     private mcsManager;
+    napCat: NapCat;
     constructor(ctx: Context, config: GLBotConfigType);
     private initialize;
     private globalCommand;

package/lib/gl/index.js

@@ -7,6 +7,7 @@ exports.GLBotBase = exports.GLBot = void 0;
 const koishi_1 = require("koishi");
 const constants_1 = require("../constants");
 const mcsManager_1 = require("../mcsManager");
+const napCat_1 = require("../napCat");
 const queQiao_1 = __importDefault(require("../queQiao"));
 const logger = new koishi_1.Logger('gl-bot');
 class GLBot {
@@ -14,7 +15,8 @@ class GLBot {
         this.ctx = ctx;
         this.config = config;
         this.mcSyncMsg = new queQiao_1.default(ctx, config);
-        this.mcsManager = new mcsManager_1.MCManager(ctx, config);
+        this.mcsManager = new mcsManager_1.MCManager(this, ctx, config);
+        this.napCat = new napCat_1.NapCat(ctx, config);
         this.initialize();
     }
     initialize() {
@@ -41,7 +43,12 @@ class GLBot {
     }
 }
 exports.GLBot = GLBot;
-GLBot.Config = koishi_1.Schema.intersect([queQiao_1.default.Config, mcsManager_1.MCManager.Config]);
+GLBot.inject = ['database'];
+GLBot.Config = koishi_1.Schema.intersect([
+    queQiao_1.default.Config,
+    mcsManager_1.MCManager.Config,
+    napCat_1.NapCat.Config,
+]);
 class GLBotBase {
     constructor(ctx, config) {
         this.ctx = ctx;

package/lib/index.d.ts

@@ -1,3 +1,2 @@
 import { GLBot } from './gl';
-export declare const name = "gl-bot";
 export default GLBot;

package/lib/index.js

@@ -1,6 +1,4 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.name = void 0;
 const gl_1 = require("./gl");
-exports.name = 'gl-bot';
 exports.default = gl_1.GLBot;

package/lib/mcsManager/api.d.ts

@@ -1,15 +1,16 @@
 import { HTTP } from 'koishi';
-import { ServiceInstanceConnectAuth, ServiceRemoteInstanceItem, ServiceRemoteItem, UserInfo } from './type';
+import { CreateInstanceConfig, CreateInstanceData, ServiceInstanceConnectAuth, ServiceRemoteInstanceItem, ServiceRemoteItem, UserInfo } from './type';
 export declare class MCSManagerAPI {
     private http;
     private baseUrl;
+    private wsUrl;
     private authCookie?;
     userInfo: UserInfo;
     get requestHeaders(): {
         cookie: string;
         'x-requested-with': string;
     };
-    constructor(http: HTTP, baseUrl: string, authCookie?: string);
+    constructor(http: HTTP, baseUrl: string, wsUrl: string, authCookie?: string);
     login(username: string, password: string): Promise<boolean>;
     getUserInfo(): Promise<UserInfo>;
     getServiceRemoteList(): Promise<ServiceRemoteItem[]>;
@@ -21,5 +22,19 @@ export declare class MCSManagerAPI {
         page_size?: number;
     }): Promise<ServiceRemoteInstanceItem[]>;
     getServiceInstanceConnectAuth(remoteUUID: string, instanceId: string): Promise<ServiceInstanceConnectAuth | null>;
-    restartRemoteInstance(daemonId: string, instanceId: string): Promise<HTTP.Response<any>>;
+    restartRemoteInstance(daemonId: string, instanceId: string): Promise<boolean>;
+    stopRemoteInstance(daemonId: string, instanceId: string): Promise<boolean>;
+    startRemoteInstance(daemonId: string, instanceId: string): Promise<boolean>;
+    createInstance(daemonId: string, data: CreateInstanceData): Promise<boolean>;
+    instanceUploadByZip(daemonId: string, config: Partial<CreateInstanceConfig>, file: {
+        filename: string;
+        size: number;
+        path: string;
+    }, onProcessHandle?: (value: string) => void): Promise<{
+        addr: string;
+        instanceUuid: string;
+        password: string;
+        remoteMappings: string[];
+    }>;
+    uploadFileToRemoteInstance(daemonId: string, filePath: string): Promise<boolean>;
 }

package/lib/mcsManager/api.js

@@ -1,6 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MCSManagerAPI = void 0;
+const fs_1 = require("fs");
+const constants_1 = require("../constants");
+const createInstanceUpload = require('./json/createInstanceUpload.json');
 class MCSManagerAPI {
     get requestHeaders() {
         return {
@@ -8,9 +11,10 @@ class MCSManagerAPI {
             'x-requested-with': 'XMLHttpRequest',
         };
     }
-    constructor(http, baseUrl, authCookie) {
+    constructor(http, baseUrl, wsUrl, authCookie) {
         this.http = http;
         this.baseUrl = baseUrl;
+        this.wsUrl = wsUrl;
         this.authCookie = authCookie;
     }
     async login(username, password) {
@@ -72,19 +76,165 @@ class MCSManagerAPI {
         })).data.data ?? null);
     }
     async restartRemoteInstance(daemonId, instanceId) {
-        console.log({
-            daemonId,
-            uuid: instanceId,
-            token: this.userInfo.token,
-        });
-        return this.http(`${this.baseUrl}/protected_instance/restart`, {
+        return ((await this.http(`${this.baseUrl}/protected_instance/restart`, {
+            headers: this.requestHeaders,
+            params: {
+                daemonId,
+                uuid: instanceId,
+                token: this.userInfo.token,
+            },
+        })).data.data?.instanceUuid === instanceId);
+    }
+    async stopRemoteInstance(daemonId, instanceId) {
+        return ((await this.http(`${this.baseUrl}/protected_instance/stop`, {
             headers: this.requestHeaders,
             params: {
                 daemonId,
                 uuid: instanceId,
                 token: this.userInfo.token,
             },
+        })).data.data?.instanceUuid === instanceId);
+    }
+    async startRemoteInstance(daemonId, instanceId) {
+        return ((await this.http(`${this.baseUrl}/protected_instance/open`, {
+            headers: this.requestHeaders,
+            params: {
+                daemonId,
+                uuid: instanceId,
+                token: this.userInfo.token,
+            },
+        })).data.data?.instanceUuid === instanceId);
+    }
+    // 创建实例
+    async createInstance(daemonId, data) {
+        const mergeConfig = Object.assign({}, data, {
+            nickname: '',
+            startCommand: '',
+            stopCommand: 'stop',
+            cwd: '.',
+            ie: 'utf-8',
+            oe: 'utf-8',
+            processType: 'general',
+            lastDatetime: '',
+            type: 'minecraft/java',
+            tag: [],
+            maxSpace: null,
+            endTime: '',
+            docker: {
+                containerName: '',
+                image: '',
+                ports: [],
+                extraVolumes: [],
+                networkMode: 'bridge',
+                networkAliases: [],
+                cpusetCpus: '',
+                workingDir: '/data',
+                changeWorkdir: false,
+                env: [],
+            },
+        });
+        const result = await this.http(`${this.baseUrl}/instance`, {
+            method: 'POST',
+            headers: this.requestHeaders,
+            params: {
+                daemonId,
+                token: this.userInfo.token,
+            },
+            data: mergeConfig,
+        });
+        if (result.status === 200) {
+            return true;
+        }
+        return;
+    }
+    async instanceUploadByZip(daemonId, config, file, onProcessHandle) {
+        const mergeConfig = Object.assign({}, createInstanceUpload, config);
+        const uploadDir = '.';
+        const { status, data } = await this.http(`${this.baseUrl}/instance/upload`, {
+            method: 'POST',
+            headers: this.requestHeaders,
+            params: {
+                upload_dir: uploadDir,
+                daemonId,
+                token: this.userInfo.token,
+            },
+            data: mergeConfig,
+        });
+        if (status !== 200 && !data.data.instanceUuid) {
+            throw new Error('创建实例上传任务失败');
+        }
+        const newFile = await this.http(`${this.wsUrl}/upload-new/${data.data.password}`, {
+            headers: this.requestHeaders,
+            method: 'POST',
+            params: {
+                filename: file.filename,
+                size: file.size,
+                sum: '',
+                overwrite: false,
+                unzip: true,
+                code: 'gbk',
+                token: this.userInfo.token,
+            },
+        });
+        if (newFile.status !== 200 || !newFile.data.data.id) {
+            throw new Error('上传文件失败');
+        }
+        // 分片上传文件
+        const fileBuffer = (0, fs_1.readFileSync)(file.path);
+        const chunkSize = 2 * 1024 * 1024; // 2MB 每个分片
+        const totalSize = fileBuffer.length;
+        let offset = 0;
+        // console.log(
+        //   `开始分片上传，文件大小: ${totalSize} bytes，分片大小: ${chunkSize} bytes`,
+        // );
+        onProcessHandle?.(`正在解压文件...`);
+        while (offset < totalSize) {
+            const end = Math.min(offset + chunkSize, totalSize);
+            const chunk = fileBuffer.slice(offset, end);
+            const chunkBlob = new Blob([chunk], {
+                type: 'application/octet-stream',
+            });
+            const formData = new FormData();
+            formData.append('file', chunkBlob, file.filename);
+            // const progress = ((end / totalSize) * 100).toFixed(0);
+            if (constants_1.IS_DEV) {
+                console.log(`上传分片: offset=${offset}, size=${chunk.length}, progress=${((end / totalSize) * 100).toFixed(1)}%`);
+            }
+            const response = await this.http(`${this.wsUrl}/upload-piece/${newFile.data.data.id}`, {
+                method: 'POST',
+                params: {
+                    offset,
+                    token: this.userInfo.token,
+                },
+                data: formData,
+            });
+            if (response.status !== 200) {
+                throw new Error(`分片上传失败，offset: ${offset}`);
+            }
+            offset = end;
+        }
+        return data.data;
+    }
+    // 上传文件至实例
+    async uploadFileToRemoteInstance(daemonId, filePath) {
+        const fileBuffer = (0, fs_1.readFileSync)(filePath);
+        const fileBlob = new Blob([fileBuffer], {
+            type: 'application/octet-stream',
+        });
+        const formData = new FormData();
+        formData.append('file', fileBlob);
+        const result = await this.http(`${this.baseUrl}/protected_instance/upload_file`, {
+            method: 'POST',
+            headers: {
+                ...this.requestHeaders,
+            },
+            params: {
+                daemonId,
+                token: this.userInfo.token,
+            },
+            data: formData,
         });
+        return result.status === 200;
     }
 }
 exports.MCSManagerAPI = MCSManagerAPI;

package/lib/mcsManager/bot.d.ts

@@ -1,14 +1,16 @@
 import { Context } from 'koishi';
+import { MCManager } from '.';
 import { GLBotConfigType } from '../gl';
-import { MCBotResetCommand } from './commands/reset';
+import { MCBotListCommand } from './commands';
 import { MCSManagerPanel } from './panel';
 export declare class MCSManagerBot {
+    readonly manager: MCManager;
     readonly ctx: Context;
     readonly config: GLBotConfigType;
     readonly panel: MCSManagerPanel;
-    constructor(ctx: Context, config: GLBotConfigType, panel: MCSManagerPanel);
+    constructor(manager: MCManager, ctx: Context, config: GLBotConfigType, panel: MCSManagerPanel);
     help(): string[];
-    commands(): (typeof MCBotResetCommand)[];
+    commands(): (typeof MCBotListCommand)[];
     initialize(): Promise<void>;
     private registerCommands;
 }

package/lib/mcsManager/bot.js

@@ -1,9 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MCSManagerBot = void 0;
-const reset_1 = require("./commands/reset");
+const commands_1 = require("./commands");
+const create_1 = require("./commands/mc/create");
 class MCSManagerBot {
-    constructor(ctx, config, panel) {
+    constructor(manager, ctx, config, panel) {
+        this.manager = manager;
         this.ctx = ctx;
         this.config = config;
         this.panel = panel;
@@ -20,7 +22,19 @@ class MCSManagerBot {
         ];
     }
     commands() {
-        return [reset_1.MCBotResetCommand];
+        return [
+            commands_1.MCBotRestartCommand,
+            commands_1.MCBotListCommand,
+            commands_1.MCBotStartCommand,
+            commands_1.MCBotStopCommand,
+            create_1.MCBotCreateCommand,
+            commands_1.MCBotOnlineTimeCommand,
+            commands_1.MCBotGameOnline,
+            commands_1.ARKBotListCommand,
+            commands_1.ARKBotRestartCommand,
+            commands_1.ARKBotStartCommand,
+            commands_1.ARKBotStopCommand,
+        ];
     }
     async initialize() {
         this.ctx.on('ready', async () => {
@@ -29,14 +43,8 @@ class MCSManagerBot {
     }
     registerCommands() {
         for (const Command of this.commands()) {
-            new Command(this);
+            const _ = new Command(this);
         }
-        // this.ctx
-        //   .command('/服务器 <action> [...name]')
-        //   .option('debug', '-d')
-        //   .action(({ options }, action, ...name) => {
-        //     return `收到指令：${action} ${name.join(' ')}，调试模式：${options.debug}`;
-        //   });
     }
 }
 exports.MCSManagerBot = MCSManagerBot;

package/lib/mcsManager/commands/ark/create.d.ts

@@ -0,0 +1,15 @@
+import { Argv } from 'koishi';
+import { MCSManagerBot } from '../../bot';
+import { BotCommandBase, BotCommandRole } from '../base';
+/**
+ * 服务器创建指令
+ *
+ * @example 服务器 创建
+ */
+export declare class ARKBotCreateCommand extends BotCommandBase {
+    readonly bot: MCSManagerBot;
+    command: string[];
+    roles: BotCommandRole[];
+    constructor(bot: MCSManagerBot);
+    handle({ args, session, ...opt }: Argv, status?: string[]): Promise<string>;
+}

package/lib/mcsManager/commands/ark/create.js

@@ -0,0 +1,66 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ARKBotCreateCommand = void 0;
+const fs_1 = require("fs");
+const koishi_1 = require("koishi");
+const base_1 = require("../base");
+/**
+ * 服务器创建指令
+ *
+ * @example 服务器 创建
+ */
+class ARKBotCreateCommand extends base_1.BotCommandBase {
+    constructor(bot) {
+        super(bot);
+        this.bot = bot;
+        this.command = ['方舟.创建 <name...>', 'ARK.创建 <name...>'];
+        this.roles = [base_1.BotCommandRole.All];
+        this.initialize();
+    }
+    async handle({ args, session, ...opt }, status) {
+        const { elements } = session?.event?.message?.quote ?? {};
+        if (elements.length > 0) {
+            const fileElements = koishi_1.h.select(elements, 'file');
+            if (fileElements.length > 0) {
+                const results = [`检测到 ${fileElements.length} 个文件:`];
+                for (const [index, element] of fileElements.entries()) {
+                    const fileInfo = {
+                        src: element.attrs?.src,
+                        fileName: element.attrs?.file || element.attrs?.src,
+                        fileId: element.attrs?.['fileId'],
+                        fileSize: element.attrs?.['fileSize'],
+                    };
+                    results.push(`\n文件 ${index + 1}:`);
+                    results.push(`  名称: ${fileInfo.fileName}`);
+                    results.push(`  文件ID: ${fileInfo.fileId}`);
+                    if (fileInfo.fileSize) {
+                        const sizeInMB = (parseInt(fileInfo.fileSize, 10) /
+                            1024 /
+                            1024).toFixed(2);
+                        results.push(`  大小: ${sizeInMB} MB`);
+                    }
+                    results.push(`-> 开始尝试下载群文件...`);
+                    session.send(results.join('\n'));
+                    results.length = 0;
+                    // 尝试下载文件
+                    try {
+                        const result = await this.bot.manager.gl.napCat.api.getFile(fileInfo.fileId);
+                        const fileExists = result ? (0, fs_1.existsSync)(result?.file) : false;
+                        if (result && fileExists) {
+                            session.send('下载完毕，正在解压并创建服务器实例...');
+                        }
+                        else {
+                            results.push(`状态: 处理失败`);
+                        }
+                    }
+                    catch (error) {
+                        results.push(`  状态: 处理失败 ${error}`);
+                    }
+                }
+                return results.join('\n');
+            }
+        }
+        return `需要引用包含服务端压缩包的文件喵~`;
+    }
+}
+exports.ARKBotCreateCommand = ARKBotCreateCommand;

package/lib/mcsManager/commands/ark/list.d.ts

@@ -0,0 +1,14 @@
+import { MCSManagerBot } from '../../bot';
+import { BotCommandBase, BotCommandRole } from '../base';
+/**
+ * 服务器列表指令
+ *
+ * @example 服务器 列表
+ */
+export declare class ARKBotListCommand extends BotCommandBase {
+    readonly bot: MCSManagerBot;
+    command: string[];
+    roles: BotCommandRole[];
+    constructor(bot: MCSManagerBot);
+    handle(_: any, status?: string[]): Promise<string>;
+}

package/lib/mcsManager/commands/ark/list.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ARKBotListCommand = void 0;
+const utils_1 = require("../../../utils");
+const constants_1 = require("../../constants");
+const base_1 = require("../base");
+/**
+ * 服务器列表指令
+ *
+ * @example 服务器 列表
+ */
+class ARKBotListCommand extends base_1.BotCommandBase {
+    constructor(bot) {
+        super(bot);
+        this.bot = bot;
+        this.command = ['方舟.列表 <status>', 'ARK.列表 <status>'];
+        this.roles = [base_1.BotCommandRole.All];
+        this.initialize();
+    }
+    async handle(_, status) {
+        await this.bot.panel.handleRemoteServices();
+        const filteredStatus = status?.at(0);
+        const nameInstances = (await this.bot.panel.searchInstanceByName('')).filter(item => !filteredStatus ||
+            constants_1.RemoteInstanceStatusName[item.instance.cfg.status] === filteredStatus);
+        return `${'='.repeat(10)}服务器列表${'='.repeat(10)}\n${nameInstances
+            .map(item => {
+            const { cfg } = item.instance;
+            const duration = (0, utils_1.formatDuration)(cfg.config.lastDatetime -
+                new Date(cfg.config.createDatetime).getTime());
+            return `- [${constants_1.RemoteInstanceStatusName[cfg.status]}] ${cfg.config.nickname} 「${duration}」`;
+        })
+            .join('\n')}\n ${'='.repeat(28)} `;
+    }
+}
+exports.ARKBotListCommand = ARKBotListCommand;

package/lib/mcsManager/commands/ark/restart.d.ts

@@ -0,0 +1,15 @@
+import { Argv } from 'koishi';
+import { MCSManagerBot } from '../../bot';
+import { BotCommandBase, BotCommandRole } from '../base';
+/**
+ * 服务器重启指令
+ *
+ * @example 服务器 重启 神话
+ */
+export declare class ARKBotRestartCommand extends BotCommandBase {
+    readonly bot: MCSManagerBot;
+    command: string[];
+    roles: BotCommandRole[];
+    constructor(bot: MCSManagerBot);
+    handle({ session }: Argv, args: string[]): Promise<string>;
+}

package/lib/mcsManager/commands/ark/restart.js

@@ -0,0 +1,62 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ARKBotRestartCommand = void 0;
+const lodash_1 = require("lodash");
+const constants_1 = require("../../constants");
+const base_1 = require("../base");
+const tempSelections = new Map();
+/**
+ * 服务器重启指令
+ *
+ * @example 服务器 重启 神话
+ */
+class ARKBotRestartCommand extends base_1.BotCommandBase {
+    constructor(bot) {
+        super(bot);
+        this.bot = bot;
+        this.command = ['方舟.重启 <name...>', 'ARK.重启 <name...>'];
+        this.roles = [base_1.BotCommandRole.All];
+        this.initialize();
+    }
+    async handle({ session }, args) {
+        let name = (0, lodash_1.isString)(args) ? args : args.join(' ');
+        let selectIndex = -1;
+        const userId = Number(session.event.user.id);
+        // 溯源前搜索项
+        if (tempSelections.has(userId) && !isNaN(Number(name))) {
+            selectIndex = Number(name) - 1;
+            name = tempSelections.get(userId);
+        }
+        tempSelections.delete(userId);
+        const nameInstances = await this.bot.panel.searchInstanceByName(name);
+        if (nameInstances.length === 0) {
+            return `未找到名称包含 "${name}" 的服务器`;
+        }
+        if (nameInstances.length > 1 && selectIndex === -1) {
+            tempSelections.set(userId, name);
+            return `请输入序号以选择:\n\n${nameInstances
+                .map((item, index) => `${index + 1}. [${constants_1.RemoteInstanceStatusName[item.instance.cfg.status]}] ${item.instance.cfg.config.nickname}`)
+                .join('\n')}\n ==== 例如发送： (服务器 重启 1) ====`;
+        }
+        const targetInstance = selectIndex !== -1 ? nameInstances[selectIndex] : nameInstances[0];
+        if (!targetInstance || !targetInstance.instance) {
+            return `未找到名称包含 "${name}" 的服务器`;
+        }
+        const { cfg } = targetInstance.instance;
+        switch (cfg.status) {
+            // 关闭状态：启动
+            case constants_1.RemoteInstanceStatusEnum.RUNNING:
+                await targetInstance.instance.restartInstance();
+                this.bot.panel.handleRemoteServices();
+                return `已向服务器实例 "${cfg.config.nickname}" 发送重启操作`;
+            // 启动状态：重启
+            case constants_1.RemoteInstanceStatusEnum.STOPPED:
+                await targetInstance.instance.startInstance();
+                this.bot.panel.handleRemoteServices();
+                return `已向服务器实例 "${cfg.config.nickname}" 发送启动操作`;
+            default:
+                return `服务器实例 "${cfg.config.nickname}" 当前状态为 ${cfg.status}，无法执行重启操作`;
+        }
+    }
+}
+exports.ARKBotRestartCommand = ARKBotRestartCommand;

package/lib/mcsManager/commands/ark/start.d.ts

@@ -0,0 +1,15 @@
+import { Argv } from 'koishi';
+import { MCSManagerBot } from '../../bot';
+import { BotCommandBase, BotCommandRole } from '../base';
+/**
+ * 服务器启动指令
+ *
+ * @example 服务器 启动 神话
+ */
+export declare class ARKBotStartCommand extends BotCommandBase {
+    readonly bot: MCSManagerBot;
+    command: string[];
+    roles: BotCommandRole[];
+    constructor(bot: MCSManagerBot);
+    handle({ session }: Argv, args: string[]): Promise<string>;
+}