3 |
2 |
3 |
' +
27 | pushCommand +
28 | '',
29 | {
30 | parse_mode: 'HTML'
31 | }
32 | );
33 | await ctx.answerCbQuery();
34 | } catch (err) {
35 | logger.error(err, "error occured when processing 'publish' action");
36 | if (err instanceof Error) return await ctx.answerCbQuery(err.message);
37 | }
38 | });
39 |
--------------------------------------------------------------------------------
/src/bot/middlewares/actions/delete.ts:
--------------------------------------------------------------------------------
1 | import { Telegraf } from 'telegraf';
2 | import { deleteContribution } from '~/database/operations/contribution';
3 | import { CallbackQuery } from 'telegraf/typings/core/types/typegram';
4 |
5 | export default Telegraf.action(/delete-/, async ctx => {
6 | const query = ctx.callbackQuery as CallbackQuery & { data: string };
7 | const message_id = parseInt(query.data.split('-')[1]);
8 |
9 | try {
10 | const delete_count = await deleteContribution(message_id);
11 | if (delete_count > 0) {
12 | await ctx.answerCbQuery('投稿已删除~');
13 | await ctx.deleteMessage(message_id);
14 | await ctx.deleteMessage();
15 | }
16 | await ctx.answerCbQuery('投稿删除失败');
17 | } catch (err) {
18 | if (err instanceof Error) {
19 | ctx.answerCbQuery('操作失败:' + err.message);
20 | }
21 | ctx.answerCbQuery('操作失败: 未知原因');
22 | }
23 | });
24 |
--------------------------------------------------------------------------------
/src/bot/middlewares/commands/debug.ts:
--------------------------------------------------------------------------------
1 | import { Telegraf } from 'telegraf';
2 | import {
3 | Message,
4 | MessageOriginChannel
5 | } from 'telegraf/typings/core/types/typegram';
6 | import { getArtwork } from '~/database/operations/artwork';
7 | import { getMessage } from '~/database/operations/message';
8 | import { Artwork } from '~/types/Artwork';
9 | import logger from '~/utils/logger';
10 | import { parseParams } from '~/utils/param-parser';
11 |
12 | export default Telegraf.command('debug', async ctx => {
13 | const command = parseParams(ctx.message.text);
14 | if (!command.target) {
15 | return await ctx.reply('No debug type specfiied!', {
16 | reply_parameters: {
17 | message_id: ctx.message.message_id,
18 | allow_sending_without_reply: true
19 | }
20 | });
21 | }
22 | switch (command.target) {
23 | case 'chat_id':
24 | return await ctx.reply(ctx.chat.id.toString(), {
25 | reply_parameters: {
26 | message_id: ctx.message.message_id,
27 | allow_sending_without_reply: true
28 | }
29 | });
30 | case 'file_id':
31 | const msg = ctx.message.reply_to_message as Partial<
32 | Message.PhotoMessage & Message.DocumentMessage
33 | >;
34 |
35 | let file_id = '';
36 |
37 | if (msg?.photo) file_id = msg.photo[msg.photo.length - 1].file_id;
38 | if (msg?.document) file_id = msg.document.file_id;
39 |
40 | return await ctx.reply(
41 | file_id
42 | ? `file_id: ${file_id}`
43 | : 'No file found in the message',
44 | {
45 | reply_parameters: {
46 | message_id: ctx.message.message_id,
47 | allow_sending_without_reply: true
48 | },
49 | parse_mode: 'HTML'
50 | }
51 | );
52 | case 'dump':
53 | return await ctx.reply(
54 | '' + 55 | JSON.stringify( 56 | ctx.message.reply_to_message 57 | ? ctx.message.reply_to_message 58 | : ctx.message, 59 | undefined, 60 | ' ' 61 | ) + 62 | '', 63 | { 64 | reply_parameters: { 65 | message_id: ctx.message.message_id, 66 | allow_sending_without_reply: true 67 | }, 68 | parse_mode: 'HTML' 69 | } 70 | ); 71 | case 'parser': 72 | if (!ctx.message.reply_to_message) 73 | return await ctx.reply('Reply a message to parse', { 74 | reply_parameters: { 75 | message_id: ctx.message.message_id, 76 | allow_sending_without_reply: true 77 | } 78 | }); 79 | 80 | const target_msg = ctx.message 81 | .reply_to_message as Message.TextMessage; 82 | 83 | return await ctx.reply( 84 | '
' + 85 | JSON.stringify( 86 | parseParams(target_msg.text, target_msg.entities), 87 | undefined, 88 | ' ' 89 | ) + 90 | '', 91 | { 92 | reply_parameters: { 93 | message_id: ctx.message.message_id, 94 | allow_sending_without_reply: true 95 | }, 96 | parse_mode: 'HTML' 97 | } 98 | ); 99 | case 'artwork': 100 | try { 101 | const reply_to_message = ctx.message 102 | .reply_to_message as Message.CommonMessage; 103 | if (!reply_to_message.is_automatic_forward) return; 104 | const message = await getMessage( 105 | (reply_to_message.forward_origin as MessageOriginChannel) 106 | .message_id 107 | ); 108 | const artwork = await getArtwork(message.artwork_index); 109 | 110 | // In case that the title or desc field has special characters 111 | // Just remove them 112 | 113 | artwork.title = '...'; 114 | artwork.desc = '...'; 115 | 116 | return await ctx.reply( 117 | '
' + 118 | JSON.stringify(artwork, undefined, ' ') + 119 | '', 120 | { 121 | reply_parameters: { 122 | message_id: ctx.message.message_id, 123 | allow_sending_without_reply: true 124 | }, 125 | parse_mode: 'HTML' 126 | } 127 | ); 128 | } catch (err) { 129 | logger.error(err); 130 | return await ctx.reply('Cannot get artwork info', { 131 | reply_parameters: { 132 | message_id: ctx.message.message_id, 133 | allow_sending_without_reply: true 134 | } 135 | }); 136 | } 137 | default: 138 | return await ctx.reply(`Debug type '${command.target}' not found`, { 139 | reply_parameters: { 140 | message_id: ctx.message.message_id, 141 | allow_sending_without_reply: true 142 | } 143 | }); 144 | } 145 | }); 146 | -------------------------------------------------------------------------------- /src/bot/middlewares/commands/delete.ts: -------------------------------------------------------------------------------- 1 | import { delArtwork } from '~/services/artwork-service'; 2 | import { getMessage } from '~/database/operations/message'; 3 | import { wrapCommand } from '~/bot/wrappers/command-wrapper'; 4 | import { Artwork } from '~/types/Artwork'; 5 | import { getArtwork } from '~/database/operations/artwork'; 6 | import { MessageOriginChannel } from 'telegraf/typings/core/types/typegram'; 7 | 8 | export default wrapCommand('delete', async ctx => { 9 | if (!ctx.command.params['index'] && !ctx.is_reply) 10 | return await ctx.directlyReply( 11 | '命令语法不正确，请回复一条消息或指定要删除的作品序号!' 12 | ); 13 | await ctx.wait('正在删除作品...', true); 14 | let artwork_index = -1; 15 | if (ctx.is_reply) { 16 | if (!ctx.reply_to_message?.is_automatic_forward) 17 | return await ctx.resolveWait('回复的消息不是有效的频道消息！'); 18 | const message = await getMessage( 19 | (ctx.reply_to_message.forward_origin as MessageOriginChannel) 20 | .message_id 21 | ); 22 | artwork_index = message.artwork_index; 23 | } 24 | let artwork: Artwork; 25 | if (ctx.reply_to_message) { 26 | const message_id: number = ( 27 | ctx.reply_to_message.forward_origin as MessageOriginChannel 28 | ).message_id; 29 | const message = await getMessage(message_id); 30 | artwork = await getArtwork(message.artwork_index); 31 | artwork_index = artwork.index; 32 | } else if (ctx.command.params['index']) 33 | artwork_index = parseInt(ctx.command.params['index']); 34 | const result = await delArtwork(artwork_index); 35 | if (result.succeed) return; 36 | else await ctx.resolveWait('作品删除失败: ' + result.message); 37 | }); 38 | -------------------------------------------------------------------------------- /src/bot/middlewares/commands/file.ts: -------------------------------------------------------------------------------- 1 | import { Message } from 'telegraf/typings/core/types/typegram'; 2 | import { wrapCommand } from '~/bot/wrappers/command-wrapper'; 3 | import { insertFile } from '~/database/operations/file'; 4 | import { File } from '~/types/File'; 5 | 6 | export default wrapCommand('file', async ctx => { 7 | const documet_message = ctx.reply_to_message as Message.DocumentMessage; 8 | if (!documet_message?.document) 9 | return ctx.directlyReply('请回复一个文件消息！'); 10 | if (!ctx.command.target) return ctx.directlyReply('请指定一个文件名称！'); 11 | const file_name = ctx.command.target; 12 | const regex = /^[a-zA-Z0-9_]+$/; 13 | if (!regex.test(file_name)) 14 | return ctx.directlyReply('文件名称只能包含字母、数字和下划线！'); 15 | const file: File = { 16 | name: file_name, 17 | file_id: documet_message.document.file_id, 18 | description: ctx.command.params['desc'], 19 | create_time: new Date() 20 | }; 21 | 22 | await insertFile(file); 23 | return ctx.resolveWait( 24 | `文件成功添加到数据库~\n链接：https://t.me/${ctx.me}?start=file-${file.name}` 25 | ); 26 | }); 27 | -------------------------------------------------------------------------------- /src/bot/middlewares/commands/grant.ts: -------------------------------------------------------------------------------- 1 | import { wrapCommand } from '~/bot/wrappers/command-wrapper'; 2 | import { grantPermissons } from '~/database/operations/admin'; 3 | import { AdminPermission } from '~/types/Admin'; 4 | 5 | export default wrapCommand('grant', async ctx => { 6 | if (!ctx.reply_to_message?.from) 7 | return await ctx.directlyReply( 8 | '命令语法不正确，请回复一条其他用户的消息!' 9 | ); 10 | if (!ctx.command.target) 11 | return await ctx.directlyReply('命令语法不正确，请指定要授予的权限!'); 12 | const user_id = ctx.reply_to_message.from.id; 13 | const str_array = 14 | ctx.command.target.search(',') == -1 15 | ? [ctx.command.target] 16 | : ctx.command.target.split(','); 17 | const permissons: Array
${artwork_info.desc}\n`
70 | // caption += `尺寸: ${artwork_info.size.width}x${artwork_info.size.height}`
71 | // await ctx.replyWithDocument({
72 | // source: path.resolve(config.TEMP_DIR, file_name),
73 | // filename: file_name
74 | // }, {
75 | // caption: caption,
76 | // parse_mode: 'HTML'
77 | // })
78 | // return await ctx.deleteMessage(waiting_reply.message_id)
79 | // }
80 | // catch (err) {
81 | // console.log(err)
82 | // if (err instanceof Error) {
83 | // return await ctx.reply('操作失败: ' + err.message, {
84 | // reply_to_message_id: ctx.message.message_id
85 | // })
86 | // }
87 | // return await ctx.reply('操作失败: 未知错误', {
88 | // reply_to_message_id: ctx.message.message_id
89 | // })
90 | // }
91 | // })
92 |
--------------------------------------------------------------------------------
/src/bot/middlewares/commands/push.ts:
--------------------------------------------------------------------------------
1 | // import { adminPredicate, NonAdminHandler } from "~/bot/middlewares/guard"
2 | import { Message } from 'telegraf/typings/core/types/typegram';
3 | import getArtworkInfoByUrl from '~/platforms';
4 | import { publishArtwork } from '~/services/artwork-service';
5 | import { Contribution } from '~/types/Contribution';
6 | import { getContributionById } from '~/database/operations/contribution';
7 | import { wrapCommand } from '~/bot/wrappers/command-wrapper';
8 | import logger from '~/utils/logger';
9 | import { semiIntArray } from '~/utils/param-parser';
10 |
11 | export default wrapCommand('push', async ctx => {
12 | if (!ctx.command.urls || ctx.command.urls.length == 0)
13 | return await ctx.directlyReply(
14 | '命令语法不正确\n命令语法: /push [args(key=value)] ${err.message}`,
41 | 'HTML'
42 | );
43 | }
44 | return await _ctx.resolveWait('操作失败: 未知原因');
45 | }
46 | });
47 | }
48 |
49 | export class WarpperContext extends Context${escapeHtmlTags( 90 | artwork.desc 91 | )}\n`; 92 | 93 | caption += '\n'; 94 | 95 | if (artwork.quality) caption += '#精选 '; 96 | 97 | for (const tag of artwork.tags) { 98 | caption += `#${tag.name} `; 99 | } 100 | 101 | // When the caption length is longer than 1024, cut off the description 102 | if (caption.length > MAX_CAPTION_LENGTH) { 103 | artwork.desc = cutDescription( 104 | escapeHtmlTags(artwork.desc), 105 | caption.length 106 | ); 107 | caption = artworkCaption(artwork, artist, event_info); 108 | } 109 | 110 | return caption; 111 | } 112 | 113 | export function infoCmdCaption(artwork_info: ArtworkInfo) { 114 | let caption = '图片下载成功!\n\n'; 115 | if (artwork_info.title) 116 | caption += `${replaceHtmlBrackets(artwork_info.title)}\n`; 117 | if (artwork_info.artist) { 118 | caption += `画师: `; 119 | caption += `${ 120 | artwork_info.artist.name 121 | }\n`; 122 | } 123 | caption += `尺寸: `; 124 | 125 | caption += artwork_info.photos 126 | .map(photo => `${photo.size.width}x${photo.size.height}`) 127 | .join('/'); 128 | if (artwork_info.desc) 129 | caption += `
${escapeHtmlTags( 130 | artwork_info.desc 131 | )}\n`; 132 | if (artwork_info.raw_tags && artwork_info.raw_tags.length > 0) { 133 | caption += '\n'; 134 | caption += '
'; 135 | caption += artwork_info.raw_tags.map(str => `#${str}`).join(' '); 136 | caption += ''; 137 | } 138 | 139 | // When the caption length is longer than 1024, cut off the description 140 | if (caption.length > MAX_CAPTION_LENGTH) { 141 | artwork_info.desc = cutDescription( 142 | escapeHtmlTags(artwork_info.desc), 143 | caption.length 144 | ); 145 | caption = infoCmdCaption(artwork_info); 146 | } 147 | 148 | return caption; 149 | } 150 | 151 | export function contributeCaption(artwork_info: ArtworkInfo) { 152 | let caption = '感谢投稿 ! 正在召唤 @Revincx_Rua \n\n'; 153 | caption += `链接: ${artwork_info.post_url}\n`; 154 | caption += `尺寸: ${artwork_info.photos[0].size.width}x${artwork_info.photos[0].size.height}\n\n`; 155 | if (artwork_info.raw_tags && artwork_info.raw_tags.length > 0) { 156 | caption += '
'; 157 | caption += artwork_info.raw_tags.map(str => `#${str}`).join(' '); 158 | caption += ''; 159 | caption += '\n'; 160 | } 161 | 162 | return caption; 163 | } 164 | 165 | export function randomCaption( 166 | artwork: Artwork | ArtworkWithMessages, 167 | tags?: string[] 168 | ) { 169 | let caption = ''; 170 | if (artwork.title && artwork.source.post_url) 171 | caption += `${replaceHtmlBrackets( 172 | artwork.title 173 | )}\n\n`; 174 | 175 | caption += `这是你要的`; 176 | 177 | if (tags && tags.length > 0) { 178 | caption += ' '; 179 | caption += tags.map(str => `#${str}`).join(' '); 180 | caption += ' '; 181 | } 182 | 183 | caption += '壁纸~'; 184 | 185 | return caption; 186 | } 187 | -------------------------------------------------------------------------------- /src/utils/decorators.ts: -------------------------------------------------------------------------------- 1 | // import { ExecResult } from "~/types/Event" 2 | 3 | // export function ErrCatch(target: Function): ExecResult 4 | // { 5 | // try{ 6 | // target() 7 | // } 8 | // catch (err) { 9 | // if(err instanceof Error) 10 | // { 11 | // console.log(err.message) 12 | // return { 13 | // succeed: false, 14 | // message: err.message 15 | // } 16 | // } 17 | // return { 18 | // succeed: false, 19 | // message: "未知原因" 20 | // } 21 | // } 22 | // } 23 | -------------------------------------------------------------------------------- /src/utils/download.ts: -------------------------------------------------------------------------------- 1 | import path from 'path'; 2 | import fs from 'fs'; 3 | import axios from 'axios'; 4 | import config from '~/config'; 5 | import logger from './logger'; 6 | 7 | export default async function downloadFile( 8 | url: string, 9 | file_name?: string, 10 | sub_dir = '' 11 | ): Promise