kokkoro-plugin-setu/lib/index.js

hso，我都不看这些的

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const path_1 = require("path");
const amesu_1 = require("amesu");
const core_1 = require("@kokkoro/core");
const service_1 = require("./service");
const option = {
    apply: true,
    lock: false,
    max_lsp: 5,
    r18: false,
    unsend: 0,
    size: 'regular',
    // anti_harmony: true,
};
const pkg = require('../package.json');
const plugin = new core_1.Plugin('setu', option);
const service = new service_1.Service(plugin);
plugin
    .version(pkg.version)
    .schedule('0 0 5 * *', () => service.clearLspMap());
plugin
    .command('random', 'group')
    .description('随机发送本地涩图')
    .sugar(/^来[点张份][涩瑟色]图$/)
    .prevent(async (ctx) => {
    await ctx.reply('不可以色色！', true);
})
    .action(async (ctx, bot) => {
    const { option, sender } = ctx;
    const { r18, unsend } = option;
    const is_ban = service.smallBlackRoom(ctx);
    if (is_ban) {
        return;
    }
    const setuInfo = service.getRandomSetu(r18);
    const { setu_url, image_info, setu_file } = setuInfo;
    await ctx
        .reply(image_info)
        .then(() => amesu_1.segment.image(setu_url))
        .then(image => ctx.reply([image]))
        .then((message_ret) => {
        const sendInfo = {
            unsend,
            user_id: sender.user_id,
            message_id: message_ret.message_id,
        };
        service.unsendSetu(bot, sendInfo);
        service.emit('setu.send.success', bot, setu_url, setu_file);
    });
});
plugin
    .command('search <...tags>', 'group')
    .description('检索在线涩图')
    .sugar(/^来[点张份](?<tags>.+)[涩瑟色]图$/)
    .prevent(async (ctx) => {
    await ctx.reply('不可以色色！', true);
})
    .action(async (ctx, bot) => {
    const { option, query, sender } = ctx;
    const { tags } = query;
    const { unsend, r18 } = option;
    const is_ban = service.smallBlackRoom(ctx);
    if (is_ban) {
        return;
    }
    ctx
        .reply('图片检索中，请耐心等待喵~', true)
        .then(() => service.searchLoliconImage(tags, option))
        .then(async (setu_info) => {
        const { setu_url, image_info } = setu_info;
        await ctx.reply(image_info);
        return amesu_1.segment.image(setu_url);
    })
        .then(image => ctx.reply([image]))
        .then((message_ret) => {
        const sendInfo = {
            unsend,
            user_id: sender.user_id,
            message_id: message_ret.message_id,
        };
        service.unsendSetu(bot, sendInfo);
    })
        .catch(async (error) => {
        const { setu_url, image_info, setu_file } = service.getRandomSetu(r18);
        ctx.reply(`Error: ${error.message}\n将为你随机发送本地色图`)
            .then(() => ctx.reply(image_info))
            .then(() => amesu_1.segment.image(setu_url))
            .then(image => ctx.reply([image]))
            .then((message_ret) => {
            const sendInfo = {
                unsend,
                user_id: sender.user_id,
                message_id: message_ret.message_id,
            };
            service.unsendSetu(bot, sendInfo);
            service.emit('setu.send.success', bot, setu_url, setu_file);
        })
            .catch(error => ctx.reply(error.message));
    });
});
plugin
    .command('multi <number>', 'group')
    .description('获取多张色图')
    .sugar(/^来(?<number>[1-9]\d*)[点张份][涩瑟色]图$/)
    .prevent(async (ctx) => {
    await ctx.reply('不可以色色！', true);
})
    .action(async (ctx, bot) => {
    const { option, query } = ctx;
    const { number } = query;
    const { r18 } = option;
    const is_ban = service.smallBlackRoom(ctx);
    if (is_ban) {
        return;
    }
    else if (number > 10) {
        return ctx.reply(`数量不能大于 10，人别冲死了`);
    }
    // 多张转发不删除本地图片
    const setus = service.getSetus(r18);
    const images = [];
    for (let i = 0; i < number; i++) {
        const ran = Math.floor(Math.random() * setus.length);
        images.push(setus.splice(ran, 1)[0]);
    }
    ;
    const forwardMessage = [];
    for (let i = 0; i < number; i++) {
        const setu_file = images[i];
        const setu_path = r18 ? service_1.r18_path : service_1.r17_path;
        const setu_url = (0, path_1.join)(setu_path, setu_file);
        const [uid, author, pid, title] = setu_file.split('@');
        const image_info = `作者:\n  ${author} (${uid})\n标题:\n  ${title} (${pid})`;
        const message = {
            message: [image_info, '\n', amesu_1.segment.image(setu_url)],
            user_id: ctx.self_id,
            nickname: bot.nickname,
        };
        forwardMessage.push(message);
    }
    ctx.group
        .makeForwardMsg(forwardMessage)
        .then(xml => ctx.reply(xml))
        .catch(error => ctx.reply(error.message));
});
// function downloadImage(url: string) {
//   return new Promise((resolve, reject) => {
//     axios.get(url, { responseType: 'arraybuffer', timeout: 5000, })
//       .then((response) => {
//         resolve(segment.image(response.data));
//       })
//       .catch((error: Error) => {
//         reject(new Error(`${error.message}\n图片下载失败，地址:\n${url}`));
//       })
//   })
// }