@kotori-bot/kotori-plugin-webui 1.0.0 → 1.1.1

package/README.md

@@ -1,7 +1,13 @@
-# kotori-bot-admin-server
-
-Web front end repository: [kotorijs/webui](https://github.com/kotorijs/webui)
-
-## Reference
-
-- [Kotori Docs](https://kotori.js.org/)
+# kotori-bot-admin-server
+
+Web front end repository: [kotorijs/webui](https://github.com/kotorijs/webui)
+
+## Reference
+
+- [Kotori Docs](https://kotori.js.org/)
+
+## TODO
+
+- [] Add authorization for websocket server
+- [] Login at the same time
+- [] Console outputting support for cmd adapter

package/lib/index.js

@@ -1,128 +1,41 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.main = exports.config = exports.inject = void 0;
-const kotori_bot_1 = require("kotori-bot");
-const path_1 = __importStar(require("path"));
-const fs_1 = require("fs");
-const common_1 = require("./common");
-const const_1 = require("./const");
-exports.inject = ['server', 'file'];
-exports.config = kotori_bot_1.Tsu.Object({
-    username: kotori_bot_1.Tsu.String().default(const_1.DEFAULT_USERNAME),
-    password: kotori_bot_1.Tsu.String().default(const_1.DEFAULT_PASSWORD)
-});
-function main(ctx, con) {
-    const loadToken = () => ctx.file.load('token', 'txt');
-    const updateToken = () => ctx.file.save('token', (0, common_1.getToken)(), 'txt');
-    const checkToken = (token) => token && token === loadToken();
+const path_1 = __importDefault(require("path"));
+const webui_1 = require("./utils/webui");
+const routers_1 = __importDefault(require("./routers"));
+const ws_1 = __importDefault(require("./ws"));
+const router_1 = __importDefault(require("./routers/router"));
+exports.inject = ['server', 'file', 'cache'];
+var webui_2 = require("./utils/webui");
+Object.defineProperty(exports, "config", { enumerable: true, get: function () { return webui_2.config; } });
+function main(ctx, cfg) {
+    /* Starts webui service */
+    ctx.service('webui', new webui_1.Webui(ctx, cfg));
+    ctx.inject('webui');
+    ctx.on('ready', () => {
+        (0, ws_1.default)(ctx, ctx.server.wss('/webui'));
+    });
+    /* Sets up routes */
     const app = ctx.server;
     app.use(app.static(path_1.default.resolve(__dirname, '../dist')));
     app.use(app.json());
     /* eslint-disable-next-line @typescript-eslint/no-explicit-any */
     app.all('*', (req, res, next) => {
-        if (req.path !== '/api/info')
-            ctx.logger.label(req.method).trace(req.path);
+        ctx.logger.label(req.method).trace(req.path);
         res.header('Access-Control-Allow-Origin', '*');
         res.header('Access-Control-Allow-Headers', 'Content-Type');
         res.header('Access-Control-Allow-Methods', '*');
         res.header('Content-Type', 'application/json;charset=utf-8');
-        if (req.path !== '/api/login' && !checkToken(String(req.query.token))) {
-            res.status(200).json({});
-            return;
-        }
-        next();
-    });
-    const router = app.router();
-    router.get('/bots', (_, res) => {
-        const bots = [];
-        ctx[kotori_bot_1.Symbols.bot].forEach((bot) => bot.forEach((api) => bots.push({
-            ...api.adapter.status,
-            platform: api.adapter.platform,
-            identity: api.adapter.identity,
-            id: api.adapter.selfId,
-            lang: api.adapter.config.lang
-        })));
-        res.status(200).json(bots);
-    });
-    router.get('/modules', (_, res) => {
-        const modules = [];
-        ctx[kotori_bot_1.Symbols.modules].forEach((module) => modules.push([module[0].pkg, module[1]]));
-        res.status(200).json(modules);
-    });
-    router.get('/data', (_, res) => {
-        res.status(200).json({
-            midwares: ctx[kotori_bot_1.Symbols.midware].size,
-            commands: ctx[kotori_bot_1.Symbols.command].size,
-            regexps: ctx[kotori_bot_1.Symbols.regexp].size,
-            bots: ctx[kotori_bot_1.Symbols.bot].size,
-            adapters: ctx[kotori_bot_1.Symbols.adapter].size,
-            modules: ctx[kotori_bot_1.Symbols.modules].size,
-            dir: ctx.baseDir.root,
-            pkg: ctx.pkg,
-            node: process.version
-        });
-    });
-    router.get('/info', (_, res) => {
-        res.status(200).json({
-            cpu: (0, common_1.getCpuRate)(),
-            ram: (0, common_1.getRamRate)(),
-            version: JSON.parse((0, fs_1.readFileSync)((0, path_1.resolve)(__dirname, '../package.json')).toString()).version
-        });
-    });
-    router.get('/login', (req, res) => {
-        const { user, pwd } = req.query;
-        if (user === con.username && pwd === con.password) {
-            updateToken();
-            res
-                .status(200)
-                .json({ token: loadToken(), default: con.username === const_1.DEFAULT_USERNAME && con.password === const_1.DEFAULT_PASSWORD });
-        }
-        else {
-            res.status(200).json({});
-        }
-    });
-    router.get('/config', (req, res) => {
-        const { type, id } = req.query;
-        if (type === 'module') {
-            if (typeof id === 'string') {
-                const split = id.split('/');
-                const handle = (0, kotori_bot_1.stringRightSplit)(split.length > 1 ? split[1] : split[0], kotori_bot_1.PLUGIN_PREFIX);
-                res.status(200).json(handle in ctx.config.plugin ? ctx.config.plugin[handle] : {});
-            }
-            else {
-                res.status(200).json({});
-            }
-        }
-        else if (type === 'adapter') {
-            res.status(200).json(typeof id === 'string' && id in ctx.config.adapter ? ctx.config.adapter[id] : {});
-        }
-        else {
-            res.status(200).json(ctx.config);
-        }
+        if (!router_1.default.find((item) => item.path === req.path || req.path.startsWith(item.path)))
+            return res.status(404).send();
+        if (req.path === '/api/accounts/login' || ctx.webui.checkToken(req.headers.authorization))
+            return next();
+        return res.status(401);
     });
-    app.use('/api', router);
+    app.use('/', (0, routers_1.default)(ctx, app));
 }
 exports.main = main;

package/lib/routers/api/accounts.js

@@ -0,0 +1,32 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const constant_1 = require("../../constant");
+/* eslint-disable-next-line func-names */
+function default_1(ctx, app) {
+    const router = app.router();
+    router.post('/login', (req, res) => {
+        const { username: user, password: pwd } = req.body;
+        const { username, password } = ctx.webui.config;
+        const loginStats = ctx.webui.getLoginStats();
+        if (user === username && pwd === password) {
+            ctx.logger.label('server').trace('Login successful');
+            loginStats.success += 1;
+            ctx.webui.setLoginStats(loginStats);
+            ctx.webui.updateToken();
+            return res.json({
+                token: ctx.webui.getToken(),
+                isDefault: username + password === constant_1.DEFAULT_USERNAME + constant_1.DEFAULT_PASSWORD
+            });
+        }
+        ctx.logger.label('server').trace('Login failed');
+        loginStats.failed += 1;
+        ctx.webui.setLoginStats(loginStats);
+        return res.status(401).json({ message: 'Invalid username or password' });
+    });
+    router.post('/logout', (_, res) => {
+        ctx.webui.updateToken();
+        res.status(204).send();
+    });
+    return router;
+}
+exports.default = default_1;

package/lib/routers/api/config.js

@@ -0,0 +1,72 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const kotori_bot_1 = require("kotori-bot");
+/* eslint-disable-next-line func-names */
+function default_1(ctx, app) {
+    const getPluginConfig = () => Object.entries(ctx.config.plugin).map(([name, origin]) => ({ name, origin, schema: {} }));
+    const getBotConfig = () => Object.entries(ctx.config.adapter).map(([id, origin]) => ({ id, origin, schema: {} }));
+    const router = app.router();
+    router.get('/plugins/:scope?/:name?', (req, res) => {
+        const { scope, name } = req.params;
+        if (!scope)
+            return res.json(getPluginConfig());
+        const pluginName = (0, kotori_bot_1.stringRightSplit)(name ?? scope, kotori_bot_1.PLUGIN_PREFIX);
+        const pluginConfig = getPluginConfig().find((plugin) => plugin.name === pluginName);
+        return pluginConfig ? res.json(pluginConfig) : res.status(404).json({ message: 'Plugin not found' });
+    });
+    router.put('/plugins/:scope?/:name?', (req, res) => {
+        const { scope, name } = req.params;
+        const { body } = req;
+        if (!scope || !name)
+            return res.status(400).json({ message: 'Invalid plugin scope' });
+        if (typeof body !== 'object')
+            return res.status(400).json({ message: 'Invalid body' });
+        const pluginName = (0, kotori_bot_1.stringRightSplit)(name ?? scope, kotori_bot_1.PLUGIN_PREFIX);
+        const pluginConfig = getPluginConfig().find((plugin) => plugin.name === pluginName);
+        if (!pluginConfig)
+            return res.status(404).json({ message: 'Plugin not found' });
+        Object.keys(body).forEach((key) => {
+            if (key in pluginConfig)
+                pluginConfig[key] = body[key];
+        });
+        return res.status(204).send();
+    });
+    router.get('/bots/:name?', (req, res) => {
+        const { name } = req.params;
+        if (!name)
+            return res.json(getBotConfig());
+        const botConfig = getBotConfig().find((bot) => bot.id === name);
+        return botConfig ? res.json(botConfig) : res.status(404).json({ message: 'Bot not found' });
+    });
+    router.put('/bots/:name?', (req, res) => {
+        const { name } = req.params;
+        const { body } = req;
+        if (!name)
+            return res.status(400).json({ message: 'Invalid bot name' });
+        if (typeof body !== 'object')
+            return res.status(400).json({ message: 'Invalid body' });
+        const botConfig = ctx.config.adapter[name];
+        if (!botConfig)
+            return res.status(404).json({ message: 'Bot not found' });
+        Object.keys(body).forEach((key) => {
+            if (key in botConfig)
+                botConfig[key] = body[key];
+        });
+        return res.status(204).send();
+    });
+    router.get('/global', (_, res) => {
+        res.json(ctx.config.global);
+    });
+    router.put('/global', (req, res) => {
+        const { body } = req;
+        if (typeof body !== 'object')
+            return res.status(400).json({ message: 'Invalid body' });
+        Object.keys(body).forEach((key) => {
+            if (key in ctx.config.global)
+                ctx.config.global[key] = body[key];
+        });
+        return res.status(204).send();
+    });
+    return router;
+}
+exports.default = default_1;

package/lib/routers/api/data.js

@@ -0,0 +1,88 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const os_1 = __importDefault(require("os"));
+const kotori_bot_1 = require("kotori-bot");
+const common_1 = require("../../utils/common");
+/* eslint-disable-next-line func-names */
+function default_1(ctx, app) {
+    const getModuleData = () => {
+        const list = [];
+        ctx[kotori_bot_1.Symbols.modules].forEach((module) => list.push(module[0].pkg));
+        return list;
+    };
+    const getBotData = () => {
+        const list = [];
+        ctx[kotori_bot_1.Symbols.bot].forEach((bot) => bot.forEach((api) => list.push({
+            status: api.adapter.status,
+            platform: api.adapter.platform,
+            identity: api.adapter.identity,
+            id: String(api.adapter.selfId),
+            lang: api.adapter.config.lang
+        })));
+        return list;
+    };
+    const router = app.router();
+    router.get('/modules/:scope?/:name?', (req, res) => {
+        const { scope, name } = req.params;
+        if (!scope)
+            return res.json(getModuleData());
+        const moduleName = name ? `${scope}/${name}` : scope;
+        const moduleData = getModuleData().find((module) => module.name === moduleName);
+        return moduleData ? res.json(moduleData) : res.status(404).json({ message: 'Modules not found' });
+    });
+    router.get('/bots/:name', (req, res) => {
+        const { name } = req.params;
+        if (!name)
+            return res.json(getBotData());
+        const botData = getBotData().find((bot) => bot.identity === name);
+        return botData ? res.json(botData) : res.status(404).json({ message: 'Bot not found' });
+    });
+    router.get('/stats', async (_, res) => {
+        const botsStatus = getBotData().map((bot) => bot.status.value === 'online');
+        const msgTotal = (0, common_1.calcGrandRecord)(ctx.webui.getMsgTotal().origin);
+        const { success: loginSuccess, failed: loginFailed } = ctx.webui.getLoginStats();
+        const chats = { received: [], sent: [] };
+        [0, 1, 2, 3, 4, 5, 6, 7].forEach((day) => {
+            const { received, sent } = (0, common_1.calcGrandRecord)(ctx.webui.getMsgDay(day).origin);
+            chats.received.push(received || 0);
+            chats.sent.push(sent || 0);
+        });
+        res.json({
+            chats,
+            count: {
+                midwares: ctx[kotori_bot_1.Symbols.midware].size,
+                commands: ctx[kotori_bot_1.Symbols.command].size,
+                regexps: ctx[kotori_bot_1.Symbols.regexp].size,
+                bots: ctx[kotori_bot_1.Symbols.bot].size,
+                adapters: ctx[kotori_bot_1.Symbols.adapter].size,
+                modules: ctx[kotori_bot_1.Symbols.modules].size
+            },
+            system: {
+                type: os_1.default.type(),
+                arch: os_1.default.arch(),
+                uptime: os_1.default.uptime(),
+                hostname: os_1.default.hostname(),
+                homedir: os_1.default.homedir(),
+                node: process.version
+            },
+            info: [
+                { name: '累计消息收:发量', value: `${msgTotal.received}:${msgTotal.sent}` },
+                { name: '实例在线数/总数', value: `${botsStatus.filter((status) => status).length}/${botsStatus.length}` },
+                { name: '登录成功:失败次数', value: `${loginSuccess}:${loginFailed}` },
+                { name: '内存使用量', value: `${Math.round(process.memoryUsage().heapUsed / 1024 / 1024)} MB` }
+            ]
+        });
+    });
+    router.get('/status', (_, res) => {
+        res.json({
+            ...ctx.webui.getStats(),
+            mode: ctx.options.mode,
+            core: ctx.pkg.version
+        });
+    });
+    return router;
+}
+exports.default = default_1;

package/lib/routers/api/demo.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+/* eslint-disable-next-line func-names */
+function default_1(ctx, app) {
+    const router = app.router();
+    return router;
+}
+exports.default = default_1;

package/lib/routers/index.js

@@ -0,0 +1,15 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const router_1 = __importDefault(require("./router"));
+/* eslint-disable-next-line func-names */
+function default_1(ctx, app) {
+    const router = app.router();
+    router_1.default.forEach((page) => {
+        router.use(page.path, page.handler(ctx, app));
+    });
+    return router;
+}
+exports.default = default_1;

package/lib/routers/router.js

@@ -0,0 +1,25 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const accounts_1 = __importDefault(require("./api/accounts"));
+const config_1 = __importDefault(require("./api/config"));
+const data_1 = __importDefault(require("./api/data"));
+function defineRouter(config) {
+    return config;
+}
+exports.default = defineRouter([
+    {
+        path: '/api/accounts',
+        handler: accounts_1.default
+    },
+    {
+        path: '/api/config',
+        handler: config_1.default
+    },
+    {
+        path: '/api/data',
+        handler: data_1.default
+    }
+]);

package/lib/types/index.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Context = void 0;
+var kotori_bot_1 = require("kotori-bot");
+Object.defineProperty(exports, "Context", { enumerable: true, get: function () { return kotori_bot_1.Context; } });

package/lib/utils/common.js

@@ -0,0 +1,47 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.calcGrandRecord = exports.getDate = exports.generateMessage = exports.getCpuData = exports.getRamData = exports.generateToken = void 0;
+const kotori_bot_1 = require("kotori-bot");
+const os_1 = __importDefault(require("os"));
+function generateToken() {
+    return (0, kotori_bot_1.getUuid)().replaceAll('-', '');
+}
+exports.generateToken = generateToken;
+function getRamData() {
+    const total = os_1.default.totalmem() / 1024 / 1024 / 1024;
+    const unused = os_1.default.freemem() / 1024 / 1024 / 1024;
+    const used = total - unused;
+    const rate = (used / total) * 100;
+    return { total, unused, used, rate };
+}
+exports.getRamData = getRamData;
+function getCpuData() {
+    const cpuData = os_1.default.cpus();
+    let rate = 0;
+    let speed = 0;
+    cpuData.forEach((value) => {
+        const { times, speed: spd } = value;
+        rate += (1 - times.idle / (times.idle + times.user + times.nice + times.sys + times.irq)) * 100;
+        speed += spd / cpuData.length;
+    });
+    return { rate, speed };
+}
+exports.getCpuData = getCpuData;
+function generateMessage(type, data) {
+    return JSON.stringify(type === 'error' ? { type, message: data } : { type, data });
+}
+exports.generateMessage = generateMessage;
+function getDate() {
+    return `${new Date().getFullYear()}-${new Date().getMonth() + 1}-${new Date().getDay()}`;
+}
+exports.getDate = getDate;
+function calcGrandRecord(data) {
+    return Object.values(data).reduce((acc, cur) => ({
+        received: acc.received + cur.received,
+        sent: acc.sent + cur.sent
+    }), { received: 0, sent: 0 });
+}
+exports.calcGrandRecord = calcGrandRecord;

package/lib/utils/transport.js

@@ -0,0 +1,34 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+require("../types");
+const kotori_bot_1 = __importStar(require("kotori-bot"));
+class WebuiTransport extends kotori_bot_1.Transport {
+    handle(data) {
+        (0, kotori_bot_1.none)(this);
+        kotori_bot_1.default.emit('console_output', data);
+    }
+}
+exports.default = WebuiTransport;

package/lib/utils/webui.js

@@ -0,0 +1,111 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Webui = exports.config = void 0;
+const date_fns_1 = require("date-fns");
+const kotori_bot_1 = require("kotori-bot");
+const common_1 = require("./common");
+const constant_1 = require("../constant");
+const transport_1 = __importDefault(require("./transport"));
+exports.config = kotori_bot_1.Tsu.Object({
+    username: kotori_bot_1.Tsu.String().default(constant_1.DEFAULT_USERNAME),
+    password: kotori_bot_1.Tsu.String().default(constant_1.DEFAULT_PASSWORD),
+    interval: kotori_bot_1.Tsu.Number().default(5)
+});
+class Webui extends kotori_bot_1.Service {
+    timer;
+    constructor(ctx, cfg) {
+        super(ctx, cfg, 'webui');
+    }
+    start() {
+        /* Add webui transport to logger */
+        const logger = this.ctx.get('logger');
+        logger.options.transports.push(new transport_1.default({}));
+        /* Load records from file to cache */
+        const msgTotal = this.ctx.file.load(`${"msg-total" /* KEY.MSG_TOTAL */}.json`, 'json', {});
+        const msgDay = this.ctx.file.load(`${"msg-day" /* KEY.MSG_DAY */}${(0, common_1.getDate)()}.json`, 'json', {});
+        this.setMsgTotal({ origin: msgTotal });
+        this.setMsgDay({ day: (0, common_1.getDate)(), origin: msgDay });
+        /* Auto save records from cache to file */
+        this.timer = setInterval(() => {
+            this.ctx.file.save(`${"msg-total" /* KEY.MSG_TOTAL */}.json`, this.getMsgTotal().origin, 'json');
+            const msgDay = this.getMsgDay();
+            if (msgDay.day === (0, common_1.getDate)()) {
+                this.ctx.file.save(`${"msg-day" /* KEY.MSG_DAY */}${(0, common_1.getDate)()}.json`, msgDay.origin, 'json');
+            }
+            else {
+                this.setMsgDay({ day: (0, common_1.getDate)(), origin: {} });
+            }
+        }, this.config.interval * 1000);
+        /* Update records */
+        this.ctx.on('send', (data) => {
+            const { identity } = data.api.adapter;
+            const { origin: dataTotal } = this.getMsgTotal();
+            const { origin: dataDay } = this.getMsgDay();
+            if (!(identity in dataTotal))
+                dataTotal[identity] = { sent: 0, received: 0 };
+            if (!(identity in dataDay))
+                dataDay[identity] = { sent: 0, received: 0 };
+            dataTotal[identity].sent = (dataTotal[identity].sent || 0) + 1;
+            dataDay[identity].sent = (dataDay[identity].sent || 0) + 1;
+        });
+        this.ctx.midware((next, session) => {
+            next();
+            const { identity } = session.api.adapter;
+            const { origin: dataTotal } = this.getMsgTotal();
+            const { origin: dataDay } = this.getMsgDay();
+            if (!(identity in dataTotal))
+                dataTotal[identity] = { sent: 0, received: 0 };
+            if (!(identity in dataDay))
+                dataDay[identity] = { sent: 0, received: 0 };
+            dataTotal[identity].received = (dataTotal[identity].received || 0) + 1;
+            dataDay[identity].received = (dataDay[identity].received || 0) + 1;
+        }, 10);
+    }
+    getToken() {
+        return this.ctx.file.load('token', 'txt');
+    }
+    updateToken() {
+        return this.ctx.file.save('token', (0, common_1.generateToken)(), 'txt');
+    }
+    checkToken(token) {
+        return token && token === `Bearer ${this.getToken()}`;
+    }
+    getStats() {
+        (0, kotori_bot_1.none)(this);
+        return {
+            ram: (0, common_1.getRamData)(),
+            cpu: (0, common_1.getCpuData)()
+        };
+    }
+    getMsgTotal() {
+        return this.ctx.cache.get("msg-total" /* KEY.MSG_TOTAL */);
+    }
+    setMsgTotal(data) {
+        this.ctx.cache.set("msg-total" /* KEY.MSG_TOTAL */, data);
+    }
+    getMsgDay(days = 0) {
+        if (!days)
+            return this.ctx.cache.get("msg-day" /* KEY.MSG_DAY */);
+        const dateString = (0, date_fns_1.format)((0, date_fns_1.addDays)(new Date(), -days), 'yyyy-M-d');
+        return {
+            day: dateString,
+            origin: this.ctx.file.load(`${"msg-day" /* KEY.MSG_DAY */}${dateString}.json`, 'json', {
+                sent: 0,
+                received: 0
+            })
+        };
+    }
+    setMsgDay(data) {
+        this.ctx.cache.set("msg-day" /* KEY.MSG_DAY */, data);
+    }
+    getLoginStats() {
+        return this.ctx.file.load(`${"login-stats" /* KEY.LOGIN_STATS */}.json`, 'json', { success: 0, failed: 0 });
+    }
+    setLoginStats(stats) {
+        this.ctx.file.save(`${"login-stats" /* KEY.LOGIN_STATS */}.json`, stats, 'json');
+    }
+}
+exports.Webui = Webui;

package/lib/ws/index.js

@@ -0,0 +1,43 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+require("../types");
+const common_1 = require("../utils/common");
+function main(ctx, server) {
+    server.on('connection', (ws) => {
+        // if (!ctx.webui.checkToken(req.headers.authorization)) ws.close(1008, 'Invalid token');
+        /* Setup interval to send stats to client */
+        const interval = ctx.webui.config.interval * 1000;
+        const timer = setInterval(() => {
+            if (ws.readyState === ws.CLOSED || ws.readyState === ws.CLOSING) {
+                clearInterval(timer);
+                return;
+            }
+            ws.send((0, common_1.generateMessage)('stats', ctx.webui.getStats()));
+        }, interval);
+        ctx.on('dispose', () => clearInterval(timer));
+        /* Listen for messages from client */
+        ws.on('message', (message) => {
+            let data;
+            try {
+                data = JSON.parse(message.toString());
+            }
+            catch {
+                return;
+            }
+            switch (data.action) {
+                case 'command':
+                    if (!data.command) {
+                        ws.send((0, common_1.generateMessage)('error', 'No command provided.'));
+                        break;
+                    }
+                    process.stdin.emit('data', data.command);
+                    ctx.logger.label('server').trace(`Received command from client: ${data.command}`);
+                    break;
+                default:
+            }
+        });
+        /* Listen for console output from server */
+        ctx.on('console_output', (data) => ws.send((0, common_1.generateMessage)('console_output', data)));
+    });
+}
+exports.default = main;