alemonjs 1.0.0

package/lib/alemon/puppeteer.js

@@ -0,0 +1,226 @@
+import puppeteer from 'puppeteer';
+import queryString from 'querystring';
+/**
+ * 截图次数
+ */
+let pic = 0;
+/**
+ *  超过200次截图重启工具
+ */
+const RestartControl = 200;
+/**
+ * 实例
+ */
+let browser;
+/**
+ * 实例控制
+ */
+let isBrowser = false;
+/**
+ * 对每个页面进行缓存
+ */
+const pageCache = {};
+/**
+ * 清除缓存
+ */
+function delCache() {
+    for (const item in pageCache) {
+        delete pageCache[item];
+    }
+}
+/**
+ * 实例配置
+ */
+let LaunchCfg;
+/**
+ * 配置浏览器参数
+ * @param val
+ */
+export function setLanchConfig(val) {
+    LaunchCfg = val;
+}
+/**
+ * 得到pup配置
+ * @returns
+ */
+export function getLanchConfig() {
+    return LaunchCfg;
+}
+/**
+ * 启动检查
+ * @returns
+ */
+export async function pupStartCheck() {
+    /**
+     * 检测是否开启
+     */
+    if (isBrowser == false) {
+        if (!(await startChrom()))
+            return false;
+    }
+    if (pic <= RestartControl) {
+        /**
+         * 记录次数
+         */
+        pic++;
+    }
+    else {
+        /**
+         * 重置次数
+         */
+        pic = 0;
+        console.info('[puppeteer] close');
+        delCache();
+        isBrowser = false;
+        browser.close().catch(err => console.error(err));
+        console.info('[puppeteer] reopen');
+        if (!(await startChrom()))
+            return false;
+        pic++;
+    }
+    return true;
+}
+/**
+ * 截图并返回buffer
+ * @param htmlPath 绝对路径
+ * @param tab 截图元素位
+ * @param type 图片类型
+ * @param quality 清晰度
+ * @param timeout 响应检查
+ * @returns
+ */
+export async function screenshotByFile(htmlPath, Options) {
+    if (!pupStartCheck())
+        return false;
+    const { SOptions, tab = 'body', timeout = 120000 } = Options;
+    try {
+        /**
+         * 开始
+         */
+        if (!isBrowser) {
+            if (!(await startChrom()))
+                return false;
+        }
+        console.info('[puppeteer] start');
+        /**
+         * 实例化
+         */
+        const page = await browser.newPage();
+        /**
+         * 挂载网页
+         */
+        await page.goto(`file://${htmlPath}`, {
+            timeout
+        });
+        /**
+         * 获取元素
+         */
+        const body = await page.$(tab);
+        /**
+         * 得到图片
+         */
+        console.info('[puppeteer] success');
+        const buff = await body
+            .screenshot(SOptions)
+            .catch(err => {
+            console.error(err);
+            return false;
+        });
+        /**
+         * 关闭
+         */
+        page.close().catch((err) => console.error(err));
+        /**
+         * 空的
+         */
+        if (!buff)
+            console.error('[puppeteer]', htmlPath);
+        return buff;
+    }
+    catch (err) {
+        console.error(err);
+        return false;
+    }
+}
+/**
+ * 截图
+ * @param val
+ * @returns
+ */
+export async function screenshotByUrl(val) {
+    if (!pupStartCheck())
+        return false;
+    if (!isBrowser) {
+        if (!(await startChrom()))
+            return false;
+    }
+    console.info('[puppeteer] start');
+    const { url, time, rand, params, tab, cache } = val;
+    if (!pageCache[url]) {
+        pageCache[url] = await browser.newPage();
+    }
+    const isurl = params == undefined ? url : `${url}?${queryString.stringify(params ?? {})}`;
+    /**
+     * 启用页面缓存
+     */
+    await pageCache[url].setCacheEnabled(cache == undefined ? true : cache);
+    /**
+     * 启动网页
+     */
+    await pageCache[url].goto(isurl);
+    console.info(`open ${isurl}`);
+    /**
+     * 找到元素
+     */
+    const body = await pageCache[url].$(tab ?? 'body');
+    if (!body) {
+        delete pageCache[url];
+        console.error(`tab err`);
+        return false;
+    }
+    /**
+     * 延迟
+     */
+    await new Promise(resolve => setTimeout(resolve, time ?? 1000));
+    /**
+     * 截图
+     */
+    const buff = await body
+        .screenshot(rand ?? {
+        type: 'jpeg',
+        quality: 90,
+        path: ''
+    })
+        .catch(err => {
+        console.error(err);
+        return false;
+    });
+    /**
+     * 打印错误
+     */
+    if (!buff) {
+        delete pageCache[url];
+        console.error(`buff err:${url}`);
+    }
+    return buff;
+}
+/**
+ * 启动浏览器
+ * @returns
+ */
+export async function startChrom() {
+    try {
+        browser = await puppeteer.launch(LaunchCfg);
+        isBrowser = true;
+        delCache();
+        console.info('[puppeteer] open success');
+        return true;
+    }
+    catch (err) {
+        console.error(err);
+        isBrowser = false;
+        delCache();
+        console.error('[puppeteer] open fail');
+        return false;
+    }
+}

package/lib/alemon/qrcode.js

@@ -0,0 +1,66 @@
+import { toDataURL } from 'qrcode';
+import { writeFile, readFile } from 'fs';
+/**
+ * 链接转化为二维码
+ * @param text 链接
+ * @param localpath 可选,要保存的路径
+ * @returns
+ */
+export async function createQrcode(text, localpath) {
+    try {
+        const qrDataURL = await new Promise((resolve, reject) => {
+            toDataURL(text, {
+                margin: 2,
+                width: 500
+            }, (err, qrDataURL) => {
+                if (err) {
+                    console.error(err);
+                    reject(err);
+                }
+                else {
+                    resolve(qrDataURL);
+                }
+            });
+        });
+        const bufferData = Buffer.from(qrDataURL.split(',')[1], 'base64');
+        if (localpath != undefined) {
+            setBuffer(localpath, bufferData);
+        }
+        return bufferData;
+    }
+    catch (err) {
+        console.error(err);
+        return false;
+    }
+}
+/**
+ * 读取buffer文件
+ * @param localpath 读取地址
+ * @returns
+ */
+export const getBuffer = (localpath) => {
+    return new Promise((resolve, reject) => {
+        readFile(localpath, (err, bufferData) => {
+            if (err) {
+                reject(err);
+            }
+            else {
+                console.info('[BUFFER][READ]', localpath);
+                resolve(bufferData);
+            }
+        });
+    });
+};
+/**
+ * 写入buffer
+ * @param localpath 写入地址
+ * @param bufferData 数据
+ */
+export const setBuffer = (localpath, bufferData) => {
+    writeFile(localpath, bufferData, (err) => {
+        if (err)
+            throw err;
+        console.info('[BUFFER][SET]', localpath);
+    });
+    return;
+};

package/lib/alemon/screenshot.js

@@ -0,0 +1,126 @@
+import lodash from 'lodash';
+import { join, basename } from 'path';
+import { readFileSync, watch, mkdirSync } from 'fs';
+import { getAppDir } from './dealmsg.js';
+/**
+ * 源码缓存
+ */
+const html = {};
+/**
+ * 监听器
+ */
+const watcher = {};
+/**
+ * 数据缓存
+ */
+const CacheData = {};
+/**
+ * 模板缓存
+ */
+const absolutePathTemplateCache = {};
+/**
+ * 缓存监听
+ * @param tplFile 模板地址
+ * @returns
+ */
+function watchCT(tplFile) {
+    /**
+     * 监听存在,直接返回
+     */
+    if (watcher[tplFile])
+        return;
+    /**
+     * 监听不存在,增加监听
+     */
+    watcher[tplFile] = watch(tplFile)
+        .on('change', () => {
+        /**
+         * 模板改变,删除模板
+         */
+        delete html[tplFile];
+        console.info('[HTML][UPDATA]', tplFile);
+    })
+        .on('close', () => {
+        /**
+         * 监听器被移除,删除监听器
+         */
+        delete watcher[tplFile];
+    });
+}
+/**
+ * 如果control为真则需重新用art渲染
+ * @param Options
+ * @returns
+ */
+export function createStr(Options) {
+    const { AppName, tplFile, data } = Options;
+    const appDir = getAppDir();
+    /**
+     * 插件路径
+     */
+    const basePath = join(process.cwd(), appDir, AppName);
+    /**
+     * 写入地址
+     */
+    const AdressHtml = join(process.cwd(), 'data', AppName, basename(tplFile));
+    /**
+     * 确保写入目录存在
+     */
+    mkdirSync(join(process.cwd(), 'data', AppName), { recursive: true });
+    /**
+     * 判断初始模板是否改变
+     */
+    let control = false;
+    /**
+     * 缓存不存在
+     */
+    if (!html[tplFile]) {
+        /**
+         * 读取文件
+         */
+        html[tplFile] = readFileSync(tplFile, 'utf8');
+        /**
+         * 读取后监听文件
+         */
+        watchCT(tplFile);
+        control = true;
+    }
+    /**
+     * 模板对象不同需要更新数据
+     */
+    if (!lodash.isEqual(CacheData[tplFile], data ?? {})) {
+        CacheData[tplFile] = data ?? {};
+        control = true;
+    }
+    /**
+     * 模板更改和数据更改都会生成生成html
+     */
+    if (control) {
+        const reg = /url\(['"](@[^'"]+)['"]\)|href=['"](@[^'"]+)['"]|src=['"](@[^'"]+)['"]/g;
+        absolutePathTemplateCache[tplFile] = html[tplFile].replace(reg, (match, urlPath, hrefPath, srcPath) => {
+            const relativePath = urlPath ?? hrefPath ?? srcPath;
+            /**
+             * 去掉路径开头的 @ 符号
+             * 转义\/
+             */
+            const absolutePath = join(basePath, relativePath.substr(1)).replace(/\\/g, '/');
+            if (urlPath)
+                return `url('${absolutePath}')`;
+            if (hrefPath)
+                return `href='${absolutePath}'`;
+            if (srcPath)
+                return `src='${absolutePath}'`;
+        });
+        /**
+         * 打印反馈生成后的地址
+         */
+        console.info('[HTML][CREATE]', AdressHtml);
+    }
+    return {
+        control,
+        // 模板地址
+        AdressHtml,
+        // 模板字符
+        template: absolutePathTemplateCache[tplFile]
+    };
+}

package/lib/alemon/typings.js

@@ -0,0 +1,96 @@
+/**
+ * 平台枚举
+ */
+export const PlatformEnum = [
+    'qq',
+    'kook',
+    'discord',
+    'villa',
+    'ntqq',
+    'wechat',
+    'telegram',
+    'dodo'
+];
+/**
+ * 消息枚举
+ */
+export const EventEnum = [
+    /**
+     * 频道消息
+     */
+    'GUILD',
+    /**
+     * 子频道消息
+     */
+    'CHANNEL',
+    /**
+     * 成员频道进出变动消息
+     */
+    'GUILD_MEMBERS',
+    /**
+     * 审核消息
+     */
+    'MESSAGE_AUDIT',
+    /**
+     * 私聊会话消息
+     */
+    'DIRECT_MESSAGE',
+    /**
+     * 论坛主题
+     */
+    'FORUMS_THREAD',
+    /**
+     * 论坛POST
+     */
+    'FORUMS_POST',
+    /**
+     * 论坛评论
+     */
+    'FORUMS_REPLY',
+    /**
+     * 会话消息:公私合并
+     */
+    'MESSAGES',
+    /**
+     * 小写兼容层
+     */
+    'message',
+    /**
+     * 频道表情点击会话消息
+     */
+    'GUILD_MESSAGE_REACTIONS',
+    /**
+     * 互动事件监听
+     */
+    'INTERACTION',
+    /**
+     * 音频事件
+     */
+    'AUDIO_FREQUENCY',
+    /**
+     * 麦克风事件
+     */
+    'AUDIO_MICROPHONE',
+    /**
+     * 兼容不响应
+     */
+    'notice.*.poke'
+];
+/**
+ * 消息判断
+ */
+export const EventType = ['CREATE', 'UPDATE', 'DELETE'];
+export const CacrdEnum = [
+    /**
+     * qq频道
+     */
+    'qq_embed',
+    /**
+     * QQ频道
+     */
+    'qq_ark',
+    /**
+     * kook
+     */
+    'kook_card'
+];

package/lib/config.js

@@ -0,0 +1,25 @@
+import { readFileSync, writeFileSync, existsSync } from 'fs';
+import { join } from 'path';
+import { parse } from '@iarna/toml';
+import { stringify } from '@iarna/toml';
+let urlVal = '/alemon.toml';
+/**
+ * 读取配置文件
+ * @param url
+ * @returns
+ */
+export function getToml(url = urlVal) {
+    // 检验文件是否存在
+    urlVal = url;
+    const u = join(process.cwd(), urlVal);
+    if (!existsSync(u))
+        return {};
+    return parse(readFileSync(u, 'utf8'));
+}
+/**
+ * 写入配置文件
+ * @param data
+ */
+export function writeToml(data) {
+    writeFileSync(join(process.cwd(), urlVal), stringify(data));
+}

package/lib/discord/alemon/api.js

@@ -0,0 +1,44 @@
+import { existsSync, createReadStream } from 'fs';
+import { Readable, isReadable } from 'stream';
+import { basename } from 'path';
+import { fileTypeFromBuffer, fileTypeFromStream } from 'file-type';
+import { AttachmentBuilder } from 'discord.js';
+/**
+ * 发送资源
+ * @param file
+ * @param name
+ * @returns
+ */
+export async function postImage(file, name = 'image.jpg') {
+    let picData;
+    if (typeof file === 'string') {
+        if (!existsSync(file)) {
+            return false;
+        }
+        if (!name) {
+            name = basename(file);
+        }
+        picData = createReadStream(file);
+    }
+    else if (Buffer.isBuffer(file)) {
+        if (!name) {
+            name = 'file.' + (await fileTypeFromBuffer(file)).ext;
+        }
+        picData = new Readable();
+        picData.push(file);
+        picData.push(null);
+    }
+    else if (isReadable(file)) {
+        if (!name) {
+            name = 'file.' + (await fileTypeFromStream(file)).ext;
+        }
+        picData = file;
+    }
+    else {
+        return false;
+    }
+    const attach = new AttachmentBuilder(picData, {
+        name
+    });
+    return attach;
+}

package/lib/discord/alemon/bot.js

@@ -0,0 +1,11 @@
+let bot = {
+    id: '',
+    name: '',
+    avatar: 'string'
+};
+export function setBotMsgByDiscord(val) {
+    bot = val;
+}
+export function getBotMsgByDiscord() {
+    return bot;
+}

package/lib/discord/alemon/conversation.js

@@ -0,0 +1,7 @@
+import { PUBLIC_GUILD_MESSAGES_DISCORD } from './message/PUBLIC_GUILD_MESSAGES.js';
+/**
+ * 消息接收入口
+ * @param req
+ * @param res
+ */
+export const callBackByDisdord = PUBLIC_GUILD_MESSAGES_DISCORD;