@mp-assistant/cli 0.0.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Taozi
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/bin/index.d.ts

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/bin/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/bin/index.ts"],"names":[],"mappings":""}

package/dist/bin/index.js

@@ -0,0 +1,22 @@
+#!/usr/bin/env node
+import { program } from 'commander';
+import fs from 'fs';
+import path from 'path';
+import { getRootDir } from '../pathManage.js';
+import { ConfigStore } from '../store/ConfigStore.js';
+import { start } from '../start.js';
+const rootPackageJson = JSON.parse(fs.readFileSync(path.join(getRootDir(), 'package.json')).toString());
+program
+    .version(rootPackageJson.version)
+    .description('小程序助手')
+    // 
+    .command('start')
+    .action(() => {
+    ConfigStore.instance.setConfig({
+        executablePath: 'C:/Program Files/Google/Chrome/Application/chrome.exe',
+        headless: false,
+        port: 3001,
+    });
+    start();
+});
+program.parse(process.argv);

package/dist/event/WSMessageEvent.d.ts

@@ -0,0 +1,7 @@
+import { WSMessage } from "@mp-assistant/common/dist/ws/index.js";
+export declare class WSMessageEvent extends WSMessage.Event {
+    private static __instance;
+    static get instance(): WSMessageEvent;
+    constructor();
+}
+//# sourceMappingURL=WSMessageEvent.d.ts.map

package/dist/event/WSMessageEvent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WSMessageEvent.d.ts","sourceRoot":"","sources":["../../src/event/WSMessageEvent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,uCAAuC,CAAC;AAGlE,qBAAa,cAAe,SAAQ,SAAS,CAAC,KAAK;IAC/C,OAAO,CAAC,MAAM,CAAC,UAAU,CAA+B;IACxD,WAAkB,QAAQ,mBAEzB;;CAeJ"}

package/dist/event/WSMessageEvent.js

@@ -0,0 +1,20 @@
+import { WSMessage } from "@mp-assistant/common/dist/ws/index.js";
+import { WSStore } from "../store/WSStore.js";
+export class WSMessageEvent extends WSMessage.Event {
+    static __instance = null;
+    static get instance() {
+        return this.__instance ?? (this.__instance = new WSMessageEvent());
+    }
+    constructor() {
+        super();
+        /**
+         * 转发消息
+         */
+        this.on('*', (type, event) => {
+            WSStore.instance.broadcast({
+                type,
+                data: event
+            });
+        });
+    }
+}

package/dist/hooks/useLocalStore.d.ts

@@ -0,0 +1,7 @@
+export declare function useLocalStore<T>(name: string, defaultValue: T, options: {
+    storeDir?: string;
+}): {
+    get: () => T;
+    set: (value: T) => void;
+};
+//# sourceMappingURL=useLocalStore.d.ts.map

package/dist/hooks/useLocalStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useLocalStore.d.ts","sourceRoot":"","sources":["../../src/hooks/useLocalStore.ts"],"names":[],"mappings":"AAGA,wBAAgB,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE;IACrE,QAAQ,CAAC,EAAE,MAAM,CAAC;CACrB;;iBAcuB,CAAC;EAQxB"}

package/dist/hooks/useLocalStore.js

@@ -0,0 +1,23 @@
+import path from "path";
+import fs from "fs";
+export function useLocalStore(name, defaultValue, options) {
+    const { storeDir = './' } = options;
+    const localStorePath = path.join(storeDir, `${name}.json`);
+    const get = () => {
+        try {
+            const localStore = JSON.parse(fs.readFileSync(localStorePath, 'utf-8'));
+            return localStore;
+        }
+        catch (error) {
+            set(defaultValue);
+            return defaultValue;
+        }
+    };
+    const set = (value) => {
+        fs.writeFileSync(localStorePath, JSON.stringify(value, null, 2));
+    };
+    return {
+        get,
+        set,
+    };
+}

package/dist/pathManage.d.ts

@@ -0,0 +1,2 @@
+export declare function getRootDir(): string;
+//# sourceMappingURL=pathManage.d.ts.map

package/dist/pathManage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pathManage.d.ts","sourceRoot":"","sources":["../src/pathManage.ts"],"names":[],"mappings":"AAMA,wBAAgB,UAAU,WAEzB"}

package/dist/pathManage.js

@@ -0,0 +1,7 @@
+import { fileURLToPath } from "url";
+import path from "path";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+export function getRootDir() {
+    return path.join(__dirname, "..");
+}

package/dist/server/api/index.d.ts

@@ -0,0 +1,3 @@
+import { FastifyInstance } from "fastify";
+export declare const registerApi: (fastify: FastifyInstance) => void;
+//# sourceMappingURL=index.d.ts.map

package/dist/server/api/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/server/api/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAK1C,eAAO,MAAM,WAAW,GAAI,SAAS,eAAe,SAInD,CAAA"}

package/dist/server/api/index.js

@@ -0,0 +1,8 @@
+import { registerConfigApi } from "./modules/config.js";
+import { registerWorkerApi } from "./modules/worker.js";
+import { registerCommonApi } from "./modules/common.js";
+export const registerApi = (fastify) => {
+    registerCommonApi(fastify);
+    registerConfigApi(fastify);
+    registerWorkerApi(fastify);
+};

package/dist/server/api/modules/common.d.ts

@@ -0,0 +1,3 @@
+import { FastifyInstance } from "fastify";
+export declare const registerCommonApi: (fastify: FastifyInstance) => void;
+//# sourceMappingURL=common.d.ts.map

package/dist/server/api/modules/common.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"common.d.ts","sourceRoot":"","sources":["../../../../src/server/api/modules/common.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAW1C,eAAO,MAAM,iBAAiB,GAAI,SAAS,eAAe,SA4BzD,CAAC"}

package/dist/server/api/modules/common.js

@@ -0,0 +1,30 @@
+import { Api } from "@mp-assistant/common/dist/api/index.js";
+import { getErrorApiResponse, getSuccessApiResponse } from "@mp-assistant/common/dist/api/utils.js";
+import { pipeline } from "node:stream/promises";
+import fs from "node:fs";
+import path from "node:path";
+import dayjs from "dayjs";
+import { getUUID } from "@mp-assistant/common/dist/utils/index.js";
+import { getFilesDir } from "@mp-assistant/common/dist/pathManage.js";
+import { pathNormalize } from "@mp-assistant/common/dist/utils/node.js";
+export const registerCommonApi = (fastify) => {
+    fastify.post(Api.Common.UploadFile.url, async function (req, reply) {
+        const data = await req.file();
+        if (!data) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        const fileDir = path.join(getFilesDir(), dayjs().format('YYYY/MM/DD'));
+        const fileName = `${getUUID()}.${data.filename.split('.')?.[1] || ''}`;
+        if (!fs.existsSync(fileDir)) {
+            fs.mkdirSync(fileDir, { recursive: true });
+        }
+        await pipeline(data.file, fs.createWriteStream(path.join(fileDir, fileName)));
+        return getSuccessApiResponse(pathNormalize(path.join(fileDir, fileName)));
+    });
+    fastify.get(Api.Common.GetFile.url, async function (req, reply) {
+        const { filePath } = req.query;
+        const fileDir = path.dirname(decodeURIComponent(filePath));
+        const fileName = path.basename(decodeURIComponent(filePath));
+        return reply.sendFile(fileName, fileDir);
+    });
+};

package/dist/server/api/modules/config.d.ts

@@ -0,0 +1,3 @@
+import { FastifyInstance } from "fastify";
+export declare const registerConfigApi: (fastify: FastifyInstance) => void;
+//# sourceMappingURL=config.d.ts.map

package/dist/server/api/modules/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../../src/server/api/modules/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAM1C,eAAO,MAAM,iBAAiB,GAAI,SAAS,eAAe,SAWzD,CAAA"}

package/dist/server/api/modules/config.js

@@ -0,0 +1,13 @@
+import { ConfigStore } from "../../../store/ConfigStore.js";
+import { Api } from "@mp-assistant/common/dist/api/index.js";
+import { getSuccessApiResponse } from "@mp-assistant/common/dist/api/utils.js";
+export const registerConfigApi = (fastify) => {
+    fastify.get(Api.Config.GetConfig.url, async (request, reply) => {
+        return getSuccessApiResponse(ConfigStore.instance.config);
+    });
+    fastify.post(Api.Config.SetConfig.url, async (request, reply) => {
+        const config = request.body;
+        ConfigStore.instance.setConfig(config);
+        return getSuccessApiResponse(ConfigStore.instance.config);
+    });
+};

package/dist/server/api/modules/worker.d.ts

@@ -0,0 +1,3 @@
+import { FastifyInstance } from "fastify";
+export declare const registerWorkerApi: (fastify: FastifyInstance) => void;
+//# sourceMappingURL=worker.d.ts.map

package/dist/server/api/modules/worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../../../../src/server/api/modules/worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAW1C,eAAO,MAAM,iBAAiB,GAAI,SAAS,eAAe,SA2JzD,CAAA"}

package/dist/server/api/modules/worker.js

@@ -0,0 +1,138 @@
+import { WorkerStore } from "../../../store/WorkerStore.js";
+import { Api } from "@mp-assistant/common/dist/api/index.js";
+import { createWorker, isWXWorker } from "@mp-assistant/core/dist/worker/index.js";
+import { ConfigStore } from "../../../store/ConfigStore.js";
+import { createTask } from "@mp-assistant/core/dist/worker/wx/task/index.js";
+import { getSuccessApiResponse, getErrorApiResponse } from "@mp-assistant/common/dist/api/utils.js";
+import { WSStore } from "../../../store/WSStore.js";
+import { WSMessage } from "@mp-assistant/common/dist/ws/message.js";
+import { WSMessageEvent } from "../../../event/WSMessageEvent.js";
+export const registerWorkerApi = (fastify) => {
+    fastify.get(Api.Worker.GetWorkerList.url, async (request, reply) => {
+        const workerInfos = WorkerStore.instance.workerList.map(item => {
+            return item.info();
+        });
+        return getSuccessApiResponse(workerInfos);
+    });
+    fastify.get(Api.Worker.GetWorkerDetail.url, async (request, reply) => {
+        const { key } = request.query;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        return getSuccessApiResponse(worker.info());
+    });
+    fastify.post(Api.Worker.AddWorker.url, async (request, reply) => {
+        const { type, name } = request.body;
+        const worker = createWorker(type, {
+            name,
+            wsMessageEventHandler: WSMessageEvent.instance,
+        });
+        await worker.init({
+            executablePath: ConfigStore.instance.config.executablePath,
+            headless: ConfigStore.instance.config.headless,
+        });
+        WorkerStore.instance.addWorker(worker);
+        WSStore.instance.broadcast(WSMessage.Worker.ListChange.createMessage());
+        return getSuccessApiResponse(worker.info());
+    });
+    fastify.delete(Api.Worker.RemoveWorker.url, async (request, reply) => {
+        const { key } = request.query;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        await worker.destroy();
+        WorkerStore.instance.removeWorker(worker);
+        WSStore.instance.broadcast(WSMessage.Worker.ListChange.createMessage());
+        return getSuccessApiResponse(undefined);
+    });
+    fastify.put(Api.Worker.UpdateWorker.url, async (request, reply) => {
+        const { key } = request.query;
+        const { name } = request.body;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        worker.name = name || worker.name;
+        WSStore.instance.broadcast(WSMessage.Worker.DetailChange.createMessage({ key }));
+        return getSuccessApiResponse(worker.info());
+    });
+    fastify.post(Api.Worker.WorkerLogin.url, async (request, reply) => {
+        const { key } = request.query;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        if (isWXWorker(worker)) {
+            worker.login();
+            return getSuccessApiResponse(undefined);
+        }
+        else {
+            return getErrorApiResponse('Worker type not supported', 400);
+        }
+    });
+    fastify.post(Api.Worker.WorkerLogout.url, async (request, reply) => {
+        const { key } = request.query;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        if (isWXWorker(worker)) {
+            worker.logout();
+            return getSuccessApiResponse(undefined);
+        }
+        else {
+            return getErrorApiResponse('Worker type not supported', 400);
+        }
+    });
+    fastify.get(Api.Worker.WorkerUpdateWxaList.url, async (request, reply) => {
+        const { key } = request.query;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        if (isWXWorker(worker)) {
+            worker.updateWxaList();
+            return getSuccessApiResponse(undefined);
+        }
+        else {
+            return getErrorApiResponse('Worker type not supported', 400);
+        }
+    });
+    fastify.post(Api.Worker.AddTask.url, async (request, reply) => {
+        const { key } = request.query;
+        const { type, options } = request.body;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        const task = createTask(type, options);
+        if (!task) {
+            return getErrorApiResponse('Task not found', 400);
+        }
+        worker.addTask(task);
+        return getSuccessApiResponse(worker.info());
+    });
+    fastify.delete(Api.Worker.RemoveTask.url, async (request, reply) => {
+        const { key, taskKey } = request.query;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        await worker.removeTask(taskKey);
+        return getSuccessApiResponse(worker.info());
+    });
+    fastify.get(Api.Worker.TaskDetail.url, async (request, reply) => {
+        const { key, taskKey } = request.query;
+        const worker = WorkerStore.instance.workerList.find(item => item.key === key);
+        if (!worker) {
+            return getErrorApiResponse('Worker not found', 404);
+        }
+        const task = worker.taskList.find(item => item.key === taskKey);
+        if (!task) {
+            return getErrorApiResponse('Task not found', 404);
+        }
+        return getSuccessApiResponse(task.info());
+    });
+};

package/dist/server/ws/index.d.ts

@@ -0,0 +1,3 @@
+import { FastifyInstance } from "fastify";
+export declare const registerWebSocket: (fastify: FastifyInstance) => Promise<void>;
+//# sourceMappingURL=index.d.ts.map

package/dist/server/ws/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/server/ws/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAI1C,eAAO,MAAM,iBAAiB,GAAU,SAAS,eAAe,kBAK/D,CAAA"}

package/dist/server/ws/index.js

@@ -0,0 +1,7 @@
+import { WSStore } from "../../store/WSStore.js";
+import { WSUrl } from "@mp-assistant/common/dist/ws/index.js";
+export const registerWebSocket = async (fastify) => {
+    fastify.get(WSUrl, { websocket: true }, (socket, request) => {
+        WSStore.instance.connection(socket);
+    });
+};

package/dist/start.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Run the server!
+ */
+export declare const start: () => Promise<void>;
+//# sourceMappingURL=start.d.ts.map

package/dist/start.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"start.d.ts","sourceRoot":"","sources":["../src/start.ts"],"names":[],"mappings":"AAiBA;;GAEG;AACH,eAAO,MAAM,KAAK,qBAYjB,CAAA"}

package/dist/start.js

@@ -0,0 +1,72 @@
+import path from 'path';
+import { createRequire } from 'module';
+import Fastify from 'fastify';
+import fastifyStatic from '@fastify/static';
+import fastifyWebsocket from '@fastify/websocket';
+import { ConfigStore } from './store/ConfigStore.js';
+import { registerWebSocket } from './server/ws/index.js';
+import { registerApi } from './server/api/index.js';
+import { ApiPrefix } from '@mp-assistant/common/dist/api/index.js';
+import { getErrorApiResponse } from '@mp-assistant/common/dist/api/utils.js';
+import { WorkerStore } from './store/WorkerStore.js';
+import fastifyCors from '@fastify/cors';
+import multipart from '@fastify/multipart';
+const require = createRequire(import.meta.url);
+/**
+ * Run the server!
+ */
+export const start = async () => {
+    startServer();
+    /**
+     * 初始化worker
+     */
+    for (const worker of WorkerStore.instance.workerList) {
+        await worker.init({
+            executablePath: ConfigStore.instance.config.executablePath,
+            headless: ConfigStore.instance.config.headless,
+        });
+    }
+};
+const startServer = async () => {
+    const fastify = Fastify({
+    // logger: true
+    });
+    // 注册 WebSocket 插件
+    await fastify.register(fastifyWebsocket);
+    fastify.register(multipart, {
+    //
+    });
+    // 设置跨域
+    fastify.register(fastifyCors, {
+        origin: '*',
+        methods: ['GET', 'POST', 'PUT', 'DELETE', 'OPTIONS'],
+        allowedHeaders: ['Content-Type', 'Authorization'],
+    });
+    // 注册静态目录，指向 @mp-assistant/dashboard 包的 dist 目录
+    const dashboardDir = path.join(path.dirname(require.resolve('@mp-assistant/dashboard/package.json')), 'dist');
+    await fastify.register(fastifyStatic, {
+        root: dashboardDir,
+        prefix: '/',
+    });
+    // WebSocket 路由
+    await fastify.register(registerWebSocket);
+    // api 路由
+    await fastify.register(registerApi, { prefix: ApiPrefix });
+    // 统一错误响应格式
+    fastify.setErrorHandler((error, request, reply) => {
+        const statusCode = error.statusCode ?? 500;
+        reply.status(200).send(getErrorApiResponse(error.message || 'Internal Server Error', ([400, 401, 403, 404, 500].includes(statusCode) ? statusCode : 500)));
+    });
+    // 统一 404 响应格式
+    fastify.setNotFoundHandler((request, reply) => {
+        reply.status(200).send(getErrorApiResponse(`Route ${request.method} ${request.url} not found`, 404));
+    });
+    try {
+        await fastify.listen({ port: ConfigStore.instance.config.port });
+        console.log(`Server is running on http://localhost:${ConfigStore.instance.config.port}`);
+    }
+    catch (err) {
+        fastify.log.error(err);
+        process.exit(1);
+    }
+};

package/dist/store/ConfigStore.d.ts

@@ -0,0 +1,10 @@
+import { Config } from "@mp-assistant/common/dist/types/config.js";
+export declare class ConfigStore {
+    private static __instance;
+    static get instance(): ConfigStore;
+    private __config;
+    get config(): Config;
+    constructor();
+    setConfig(config: Partial<Config>): void;
+}
+//# sourceMappingURL=ConfigStore.d.ts.map

package/dist/store/ConfigStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConfigStore.d.ts","sourceRoot":"","sources":["../../src/store/ConfigStore.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAanE,qBAAa,WAAW;IACpB,OAAO,CAAC,MAAM,CAAC,UAAU,CAA4B;IACrD,WAAkB,QAAQ,gBAEzB;IAED,OAAO,CAAC,QAAQ,CAEf;IAED,IAAI,MAAM,WAET;;IAMD,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC;CAOpC"}

package/dist/store/ConfigStore.js

@@ -0,0 +1,32 @@
+import { useLocalStore } from "../hooks/useLocalStore.js";
+import { getStoreDir } from "@mp-assistant/common/dist/pathManage.js";
+const DEFAULT_CONFIG = {
+    executablePath: '',
+    headless: true,
+    port: 3001,
+};
+const { get: getConfigLocalStore, set: setConfigLocalStore } = useLocalStore('config', DEFAULT_CONFIG, {
+    storeDir: getStoreDir(),
+});
+export class ConfigStore {
+    static __instance = null;
+    static get instance() {
+        return this.__instance ?? (this.__instance = new ConfigStore());
+    }
+    __config = {
+        ...DEFAULT_CONFIG,
+    };
+    get config() {
+        return this.__config;
+    }
+    constructor() {
+        this.__config = getConfigLocalStore();
+    }
+    setConfig(config) {
+        this.__config = {
+            ...this.__config,
+            ...config,
+        };
+        setConfigLocalStore(this.__config);
+    }
+}

package/dist/store/WSStore.d.ts

@@ -0,0 +1,20 @@
+import { WebSocket } from "ws";
+import { EventEmitter } from "@mp-assistant/common/dist/event/EventEmitter.js";
+import { WSMessageFormat } from "@mp-assistant/common/dist/ws/index.js";
+export interface EventMap {
+    /** 收到ws消息 */
+    message: WSMessageFormat<any>;
+}
+export declare class WSStore extends EventEmitter<EventMap> {
+    private static __instance;
+    static get instance(): WSStore;
+    private __connectionWsList;
+    constructor();
+    connection(ws: WebSocket): void;
+    /**
+     * 广播消息
+     * @param message
+     */
+    broadcast(message: WSMessageFormat<any>): Promise<void>;
+}
+//# sourceMappingURL=WSStore.d.ts.map

package/dist/store/WSStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WSStore.d.ts","sourceRoot":"","sources":["../../src/store/WSStore.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,YAAY,EAAE,MAAM,iDAAiD,CAAC;AAC/E,OAAO,EAAa,eAAe,EAAE,MAAM,uCAAuC,CAAC;AAEnF,MAAM,WAAW,QAAQ;IACrB,aAAa;IACb,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC;CACjC;AAED,qBAAa,OAAQ,SAAQ,YAAY,CAAC,QAAQ,CAAC;IAC/C,OAAO,CAAC,MAAM,CAAC,UAAU,CAAuB;IAChD,WAAkB,QAAQ,YAEzB;IAED,OAAO,CAAC,kBAAkB,CAKjB;;IAmBT,UAAU,CAAC,EAAE,EAAE,SAAS;IAgCxB;;;OAGG;IACG,SAAS,CAAC,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC;CAehD"}

package/dist/store/WSStore.js

@@ -0,0 +1,76 @@
+import { getUUID } from "@mp-assistant/common/dist/utils/index.js";
+import { WebSocket } from "ws";
+import { EventEmitter } from "@mp-assistant/common/dist/event/EventEmitter.js";
+import { WSMessage } from "@mp-assistant/common/dist/ws/index.js";
+export class WSStore extends EventEmitter {
+    static __instance = null;
+    static get instance() {
+        return this.__instance ?? (this.__instance = new WSStore());
+    }
+    __connectionWsList = [];
+    constructor() {
+        super();
+        // 定时发送心跳消息
+        setInterval(() => {
+            this.broadcast(WSMessage.Heartbeat.createMessage());
+        }, WSMessage.Heartbeat.loopInterval);
+        // 定时检查连接状态，关闭超时的连接
+        setInterval(() => {
+            this.__connectionWsList.forEach(item => {
+                if (Date.now() - item.activeTime > WSMessage.Heartbeat.timeout) {
+                    item.ws.close();
+                }
+            });
+        }, 500);
+    }
+    connection(ws) {
+        const key = getUUID();
+        this.__connectionWsList.push({
+            key,
+            ws,
+            activeTime: Date.now(),
+        });
+        // 监听消息 
+        ws.on('message', (data) => {
+            try {
+                const message = JSON.parse(data.toString());
+                // 只处理非心跳消息。
+                if (message.type !== WSMessage.Heartbeat.type) {
+                    this.emit('message', message);
+                }
+                // 更新活跃时间
+                const currentWs = this.__connectionWsList.find(item => item.key === key);
+                if (currentWs) {
+                    currentWs.activeTime = Date.now();
+                }
+            }
+            catch (e) {
+                // 不是json格式的消息，忽略
+                return;
+            }
+        });
+        // 监听关闭
+        ws.on('close', () => {
+            this.__connectionWsList = this.__connectionWsList.filter(item => item.ws !== ws);
+        });
+    }
+    /**
+     * 广播消息
+     * @param message
+     */
+    async broadcast(message) {
+        await Promise.allSettled(this.__connectionWsList.map(item => {
+            return new Promise((res, rej) => {
+                if (item.ws.readyState === WebSocket.OPEN) {
+                    item.ws.send(JSON.stringify(message), (error) => {
+                        if (error) {
+                            rej();
+                            return;
+                        }
+                        res();
+                    });
+                }
+            });
+        }));
+    }
+}

package/dist/store/WorkerStore.d.ts

@@ -0,0 +1,11 @@
+import type { BaseWorker } from "@mp-assistant/core/dist/worker/BaseWorker.js";
+export declare class WorkerStore {
+    private static __instance;
+    static get instance(): WorkerStore;
+    private __workerList;
+    get workerList(): BaseWorker[];
+    constructor();
+    addWorker(worker: BaseWorker): void;
+    removeWorker(worker: BaseWorker): void;
+}
+//# sourceMappingURL=WorkerStore.d.ts.map

package/dist/store/WorkerStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WorkerStore.d.ts","sourceRoot":"","sources":["../../src/store/WorkerStore.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,8CAA8C,CAAC;AAc/E,qBAAa,WAAW;IACpB,OAAO,CAAC,MAAM,CAAC,UAAU,CAA4B;IACrD,WAAkB,QAAQ,gBAEzB;IAED,OAAO,CAAC,YAAY,CAAoB;IAExC,IAAI,UAAU,iBAEb;;IAYD,SAAS,CAAC,MAAM,EAAE,UAAU;IAS5B,YAAY,CAAC,MAAM,EAAE,UAAU;CAIlC"}

package/dist/store/WorkerStore.js

@@ -0,0 +1,38 @@
+import { getStoreDir } from "@mp-assistant/common/dist/pathManage.js";
+import { useLocalStore } from "../hooks/useLocalStore.js";
+import { createWorker } from "@mp-assistant/core/dist/worker/index.js";
+import { WSMessageEvent } from "../event/WSMessageEvent.js";
+const { get: getWorkerLocalStoreList, set: setWorkerLocalStoreList } = useLocalStore('workerList', [], {
+    storeDir: getStoreDir(),
+});
+export class WorkerStore {
+    static __instance = null;
+    static get instance() {
+        return this.__instance ?? (this.__instance = new WorkerStore());
+    }
+    __workerList = [];
+    get workerList() {
+        return [...this.__workerList];
+    }
+    constructor() {
+        this.__workerList = getWorkerLocalStoreList().map(item => {
+            return createWorker(item.type, {
+                key: item.key,
+                name: item.name,
+                wsMessageEventHandler: WSMessageEvent.instance,
+            });
+        });
+    }
+    addWorker(worker) {
+        this.__workerList.push(worker);
+        setWorkerLocalStoreList([...getWorkerLocalStoreList(), {
+                key: worker.key,
+                type: worker.type,
+                name: worker.name,
+            }]);
+    }
+    removeWorker(worker) {
+        this.__workerList = this.__workerList.filter(w => w.key !== worker.key);
+        setWorkerLocalStoreList(getWorkerLocalStoreList().filter(item => item.key !== worker.key));
+    }
+}

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@mp-assistant/cli",
+  "version": "0.0.1",
+  "description": "",
+  "type": "module",
+  "main": "index.js",
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "@fastify/cors": "^11.2.0",
+    "@fastify/multipart": "^9.4.0",
+    "@fastify/static": "^9.0.0",
+    "@fastify/websocket": "^11.2.0",
+    "commander": "^14.0.3",
+    "dayjs": "^1.11.13",
+    "fastify": "^5.7.4",
+    "ws": "^8.19.0",
+    "@mp-assistant/common": "0.0.1",
+    "@mp-assistant/dashboard": "0.0.1",
+    "@mp-assistant/core": "0.0.1"
+  },
+  "devDependencies": {
+    "@types/ws": "^8.18.1",
+    "@mp-assistant/scripts": "0.0.1",
+    "@mp-assistant/configuration": "0.0.1"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/taozijiankang/mp-assistant.git"
+  },
+  "scripts": {
+    "clear-build-product": "mp-c clear-build-product",
+    "dev": "tsc --watch",
+    "build": "tsc"
+  }
+}