+1

.gitignore

reviewed

··· 1 1 # Environment variables 2 2 .env 3 3 + announcements.json 3 4 4 5 # Queue files 5 6 queue/queue.json

+611 -2

bot/telegramBot.js

reviewed

··· 8 8 const queueManager = require('../queue/queueManager'); 9 9 const scraperManager = require('../utils/scraperManager'); 10 10 const mediaCache = require('../utils/mediaCache'); 11 11 + const AnnouncementManager = require('../utils/announcementManager'); 11 12 const discordWebhook = require('./discordWebhook'); 12 13 13 14 class StagehandBot { 14 15 constructor() { 15 16 this.bot = new TelegramBot(config.botToken, { polling: true }); 16 17 this.serviceName = 'telegram'; 18 18 + this.channelId = config.channelId; 19 19 + this.announcements = new AnnouncementManager(this); 17 20 this.init(); 18 21 } 19 22 20 20 - init() { 23 23 + async init() { 24 24 + await this.announcements.init(); 21 25 this.registerCommands(); 22 26 this.registerCallbacks(); 23 27 console.log('Telegram bot started...'); ··· 37 41 Stagehand Bot Commands: 38 42 /queue - Show current queue status with interactive management 39 43 /send - Post the next image in the queue 40 40 - /schedule [cron] - Set posting schedule (cron syntax) 44 44 + /schedule [cron] - Set posting schedule (cron syntax, use https://crontab.guru/ for help) 41 45 /setcount [number] - Set number of images per scheduled post (default: 1) 42 46 /clear - Clear the queue 43 47 /cleancache - Clean expired items from media cache ··· 288 292 } 289 293 }); 290 294 295 295 + // Command to add a text announcement 296 296 + this.bot.onText(/\/announce(?:\s+(.+))?/, async (msg, match) => { 297 297 + const chatId = msg.chat.id; 298 298 + 299 299 + if (!this.isAuthorized(msg.from.id)) { 300 300 + this.bot.sendMessage(chatId, 'You are not authorized to use this command.'); 301 301 + return; 302 302 + } 303 303 + 304 304 + // Check if there's text after the command 305 305 + const text = match && match[1] ? match[1].trim() : ''; 306 306 + 307 307 + if (!text) { 308 308 + // No parameters - show usage help 309 309 + this.bot.sendMessage( 310 310 + chatId, 311 311 + 'Usage: /announce [message text]\n\nThis will initiate the announcement creation process. After sending the message text, you\'ll be prompted to set a name, schedule, and an optional button link for the announcement.' 312 312 + ); 313 313 + return; 314 314 + } 315 315 + 316 316 + // Store the message text in session and ask for a name 317 317 + this.pendingAnnouncements = this.pendingAnnouncements || {}; 318 318 + this.pendingAnnouncements[msg.from.id] = { message: text }; 319 319 + 320 320 + this.bot.sendMessage( 321 321 + chatId, 322 322 + 'Please enter a name for this announcement (or type "skip" for auto-generated name):', 323 323 + { reply_markup: { force_reply: true } } 324 324 + ).then(namePrompt => { 325 325 + // Set up a one-time listener for the name response 326 326 + this.bot.onReplyToMessage(chatId, namePrompt.message_id, async (nameMsg) => { 327 327 + const announcementName = nameMsg.text === 'skip' ? '' : nameMsg.text; 328 328 + this.pendingAnnouncements[msg.from.id].name = announcementName; 329 329 + 330 330 + // Now ask for a schedule 331 331 + this.bot.sendMessage( 332 332 + chatId, 333 333 + 'Please enter a cron schedule for when this announcement should run (use https://crontab.guru/ for help):', 334 334 + { reply_markup: { force_reply: true } } 335 335 + ).then(schedulePrompt => { 336 336 + // Set up a one-time listener for the schedule response 337 337 + this.bot.onReplyToMessage(chatId, schedulePrompt.message_id, async (scheduleMsg) => { 338 338 + const cronSchedule = scheduleMsg.text; 339 339 + 340 340 + // Ask if they want to add a button 341 341 + this.bot.sendMessage( 342 342 + chatId, 343 343 + 'Would you like to add a button with a link to this announcement?', 344 344 + { 345 345 + reply_markup: { 346 346 + inline_keyboard: [ 347 347 + [ 348 348 + { text: 'Yes', callback_data: 'add_button' }, 349 349 + { text: 'No', callback_data: 'skip_button' } 350 350 + ] 351 351 + ] 352 352 + } 353 353 + } 354 354 + ).then(buttonPrompt => { 355 355 + // Callback handler for yes/no button selection 356 356 + this.bot.once('callback_query', async (query) => { 357 357 + await this.bot.answerCallbackQuery(query.id); 358 358 + 359 359 + // Delete the yes/no prompt 360 360 + await this.bot.deleteMessage(chatId, buttonPrompt.message_id); 361 361 + 362 362 + if (query.data === 'add_button') { 363 363 + // User wants to add a button 364 364 + this.bot.sendMessage( 365 365 + chatId, 366 366 + 'Please enter the button text:', 367 367 + { reply_markup: { force_reply: true } } 368 368 + ).then(buttonTextPrompt => { 369 369 + this.bot.onReplyToMessage(chatId, buttonTextPrompt.message_id, async (buttonTextMsg) => { 370 370 + const buttonText = buttonTextMsg.text; 371 371 + 372 372 + // Now ask for the button URL 373 373 + this.bot.sendMessage( 374 374 + chatId, 375 375 + 'Please enter the button URL:', 376 376 + { reply_markup: { force_reply: true } } 377 377 + ).then(buttonUrlPrompt => { 378 378 + this.bot.onReplyToMessage(chatId, buttonUrlPrompt.message_id, async (buttonUrlMsg) => { 379 379 + const buttonUrl = buttonUrlMsg.text; 380 380 + 381 381 + // Create the button object 382 382 + const button = { 383 383 + text: buttonText, 384 384 + url: buttonUrl 385 385 + }; 386 386 + 387 387 + try { 388 388 + const announcement = await this.announcements.addAnnouncement( 389 389 + this.pendingAnnouncements[msg.from.id].message, 390 390 + cronSchedule, 391 391 + this.pendingAnnouncements[msg.from.id].name, 392 392 + button 393 393 + ); 394 394 + 395 395 + delete this.pendingAnnouncements[msg.from.id]; 396 396 + 397 397 + this.bot.sendMessage( 398 398 + chatId, 399 399 + `✅ Announcement "${announcement.name}" created!\n\n`+ 400 400 + `Scheduled for: ${announcement.cronSchedule}\n\n`+ 401 401 + `Button: "${button.text}" → ${button.url}\n\n`+ 402 402 + `You can manage all announcements with /announcements` 403 403 + ); 404 404 + } catch (error) { 405 405 + this.bot.sendMessage( 406 406 + chatId, 407 407 + `Error creating announcement: ${error.message}\n\nPlease try again.` 408 408 + ); 409 409 + } 410 410 + }); 411 411 + }); 412 412 + }); 413 413 + }); 414 414 + } else { 415 415 + // User doesn't want to add a button 416 416 + try { 417 417 + const announcement = await this.announcements.addAnnouncement( 418 418 + this.pendingAnnouncements[msg.from.id].message, 419 419 + cronSchedule, 420 420 + this.pendingAnnouncements[msg.from.id].name 421 421 + ); 422 422 + 423 423 + delete this.pendingAnnouncements[msg.from.id]; 424 424 + 425 425 + this.bot.sendMessage( 426 426 + chatId, 427 427 + `✅ Announcement "${announcement.name}" created!\n\nScheduled for: ${announcement.cronSchedule}\n\nYou can manage all announcements with /announcements` 428 428 + ); 429 429 + } catch (error) { 430 430 + this.bot.sendMessage( 431 431 + chatId, 432 432 + `Error creating announcement: ${error.message}\n\nPlease try again with a valid cron expression.` 433 433 + ); 434 434 + } 435 435 + } 436 436 + }); 437 437 + }); 438 438 + }); 439 439 + }); 440 440 + }); 441 441 + }); 442 442 + }); 443 443 + 444 444 + // Command to list and manage all announcements 445 445 + this.bot.onText(/\/announcements/, async (msg) => { 446 446 + const chatId = msg.chat.id; 447 447 + 448 448 + if (!this.isAuthorized(msg.from.id)) { 449 449 + this.bot.sendMessage(chatId, 'You are not authorized to use this command.'); 450 450 + return; 451 451 + } 452 452 + 453 453 + const announcements = this.announcements.getAnnouncements(); 454 454 + 455 455 + if (announcements.length === 0) { 456 456 + this.bot.sendMessage( 457 457 + chatId, 458 458 + 'No announcements configured. Use /announce to create a new announcement.' 459 459 + ); 460 460 + return; 461 461 + } 462 462 + 463 463 + // Format the list of announcements with inline buttons 464 464 + let message = '📣 *Text Announcements*\n\n'; 465 465 + 466 466 + const inlineKeyboard = []; 467 467 + 468 468 + for (let i = 0; i < announcements.length; i++) { 469 469 + const announcement = announcements[i]; 470 470 + 471 471 + // Add announcement details to message 472 472 + message += `*${i+1}. ${announcement.name}*\n`; 473 473 + message += `Schedule: \`${announcement.cronSchedule}\`\n`; 474 474 + message += `Last run: ${announcement.lastRun ? new Date(announcement.lastRun).toLocaleString() : 'Never'}\n`; 475 475 + 476 476 + // Show button info if present 477 477 + if (announcement.button && announcement.button.text && announcement.button.url) { 478 478 + message += `Button: "${announcement.button.text}" → ${announcement.button.url}\n`; 479 479 + } 480 480 + 481 481 + message += `Message: "${announcement.message.substring(0, 50)}${announcement.message.length > 50 ? '...' : ''}"\n\n`; 482 482 + 483 483 + // Add buttons for this announcement 484 484 + inlineKeyboard.push([ 485 485 + { 486 486 + text: `▶️ Run #${i+1}`, 487 487 + callback_data: `run_announcement_${announcement.id}` 488 488 + }, 489 489 + { 490 490 + text: `✏️ Edit #${i+1}`, 491 491 + callback_data: `edit_announcement_${announcement.id}` 492 492 + }, 493 493 + { 494 494 + text: `❌ Delete #${i+1}`, 495 495 + callback_data: `delete_announcement_${announcement.id}` 496 496 + } 497 497 + ]); 498 498 + } 499 499 + 500 500 + // Add a button to create a new announcement 501 501 + inlineKeyboard.push([ 502 502 + { 503 503 + text: '➕ Add New Announcement', 504 504 + callback_data: 'new_announcement' 505 505 + } 506 506 + ]); 507 507 + 508 508 + await this.bot.sendMessage(chatId, message, { 509 509 + parse_mode: 'Markdown', 510 510 + reply_markup: { 511 511 + inline_keyboard: inlineKeyboard 512 512 + } 513 513 + }); 514 514 + }); 515 515 + 291 516 // Handle URL links 292 517 this.bot.on('message', async (msg) => { 293 518 if (msg.text && msg.text.startsWith('http')) { 294 519 const chatId = msg.chat.id; 295 520 521 521 + // Skip processing if this is part of an announcement setup 522 522 + const isInAnnouncementFlow = this.pendingAnnouncements && this.pendingAnnouncements[msg.from.id]; 523 523 + const isInButtonEditFlow = this.editingAnnouncementButton && this.editingAnnouncementButton[msg.from.id]; 524 524 + 525 525 + if (isInAnnouncementFlow || isInButtonEditFlow) { 526 526 + // This URL is part of an announcement setup, so we should not process it as a link 527 527 + return; 528 528 + } 529 529 + 296 530 if (!this.isAuthorized(msg.from.id)) { 297 531 this.bot.sendMessage(chatId, 'You are not authorized to use this bot.'); 298 532 return; ··· 441 675 } 442 676 } else { 443 677 await this.bot.answerCallbackQuery(query.id, { text: 'Item not found' }); 678 678 + } 679 679 + break; 680 680 + } 681 681 + 682 682 + // New announcement management callback handlers 683 683 + case 'run': { 684 684 + if (data[1] === 'announcement') { 685 685 + const announcementId = data[2]; 686 686 + await this.bot.answerCallbackQuery(query.id, { text: 'Sending announcement...' }); 687 687 + 688 688 + try { 689 689 + const result = await this.announcements.sendAnnouncementNow(announcementId); 690 690 + if (result) { 691 691 + await this.bot.sendMessage(chatId, `✅ Announcement sent successfully!`); 692 692 + } else { 693 693 + await this.bot.sendMessage(chatId, `❌ Failed to send announcement.`); 694 694 + } 695 695 + } catch (error) { 696 696 + await this.bot.sendMessage(chatId, `❌ Error: ${error.message}`); 697 697 + } 698 698 + 699 699 + // Refresh announcements list 700 700 + await this.bot.deleteMessage(chatId, query.message.message_id); 701 701 + await this.bot.onText.handlers.find(h => h.regexp.toString().includes('/announcements'))?._callback({ chat: { id: chatId } }); 702 702 + } 703 703 + break; 704 704 + } 705 705 + 706 706 + case 'delete': { 707 707 + if (data[1] === 'announcement') { 708 708 + const announcementId = data[2]; 709 709 + 710 710 + // Get the announcement to show its name 711 711 + const announcement = this.announcements.getAnnouncementById(announcementId); 712 712 + if (!announcement) { 713 713 + await this.bot.answerCallbackQuery(query.id, { text: 'Announcement not found.' }); 714 714 + return; 715 715 + } 716 716 + 717 717 + // Show confirmation dialog 718 718 + await this.bot.answerCallbackQuery(query.id); 719 719 + 720 720 + const confirmMessage = await this.bot.sendMessage( 721 721 + chatId, 722 722 + `Are you sure you want to delete the announcement "${announcement.name}"?`, 723 723 + { 724 724 + reply_markup: { 725 725 + inline_keyboard: [ 726 726 + [ 727 727 + { text: '✅ Yes, delete it', callback_data: `confirm_delete_announcement_${announcementId}` }, 728 728 + { text: '❌ No, cancel', callback_data: 'cancel_delete_announcement' } 729 729 + ] 730 730 + ] 731 731 + } 732 732 + } 733 733 + ); 734 734 + } 735 735 + break; 736 736 + } 737 737 + 738 738 + case 'confirm': { 739 739 + if (data[1] === 'delete' && data[2] === 'announcement') { 740 740 + const announcementId = data[3]; 741 741 + 742 742 + try { 743 743 + const result = await this.announcements.removeAnnouncement(announcementId); 744 744 + if (result) { 745 745 + await this.bot.answerCallbackQuery(query.id, { text: 'Announcement deleted successfully.' }); 746 746 + } else { 747 747 + await this.bot.answerCallbackQuery(query.id, { text: 'Failed to delete announcement.' }); 748 748 + } 749 749 + 750 750 + // Delete confirmation message 751 751 + await this.bot.deleteMessage(chatId, query.message.message_id); 752 752 + 753 753 + // Refresh announcements list 754 754 + await this.bot.onText.handlers.find(h => h.regexp.toString().includes('/announcements'))?._callback({ chat: { id: chatId } }); 755 755 + } catch (error) { 756 756 + await this.bot.answerCallbackQuery(query.id, { text: `Error: ${error.message}` }); 757 757 + } 758 758 + } 759 759 + break; 760 760 + } 761 761 + 762 762 + case 'cancel': { 763 763 + if (data[1] === 'delete' && data[2] === 'announcement') { 764 764 + await this.bot.answerCallbackQuery(query.id, { text: 'Delete cancelled.' }); 765 765 + await this.bot.deleteMessage(chatId, query.message.message_id); 766 766 + } 767 767 + break; 768 768 + } 769 769 + 770 770 + case 'edit': { 771 771 + if (data[1] === 'announcement') { 772 772 + const announcementId = data[2]; 773 773 + const announcement = this.announcements.getAnnouncementById(announcementId); 774 774 + 775 775 + if (!announcement) { 776 776 + await this.bot.answerCallbackQuery(query.id, { text: 'Announcement not found.' }); 777 777 + return; 778 778 + } 779 779 + 780 780 + await this.bot.answerCallbackQuery(query.id); 781 781 + 782 782 + // Show edit options 783 783 + const editMessage = await this.bot.sendMessage( 784 784 + chatId, 785 785 + `Editing announcement: *${announcement.name}*\n\nWhat would you like to edit?`, 786 786 + { 787 787 + parse_mode: 'Markdown', 788 788 + reply_markup: { 789 789 + inline_keyboard: [ 790 790 + [ 791 791 + { 792 792 + text: '📝 Edit Message', 793 793 + callback_data: `edit_announcement_message_${announcementId}` 794 794 + } 795 795 + ], 796 796 + [ 797 797 + { 798 798 + text: '⏰ Edit Schedule', 799 799 + callback_data: `edit_announcement_schedule_${announcementId}` 800 800 + } 801 801 + ], 802 802 + [ 803 803 + { 804 804 + text: '🏷️ Edit Name', 805 805 + callback_data: `edit_announcement_name_${announcementId}` 806 806 + } 807 807 + ], 808 808 + [ 809 809 + { 810 810 + text: '🔗 Edit Button', 811 811 + callback_data: `edit_announcement_button_${announcementId}` 812 812 + } 813 813 + ], 814 814 + [ 815 815 + { 816 816 + text: '❌ Cancel', 817 817 + callback_data: 'cancel_edit_announcement' 818 818 + } 819 819 + ] 820 820 + ] 821 821 + } 822 822 + } 823 823 + ); 824 824 + } else if (data[1] === 'announcement' && (data[2] === 'message' || data[2] === 'name' || data[2] === 'schedule' || data[2] === 'button')) { 825 825 + const field = data[2]; 826 826 + const announcementId = data[3]; 827 827 + const announcement = this.announcements.getAnnouncementById(announcementId); 828 828 + 829 829 + if (!announcement) { 830 830 + await this.bot.answerCallbackQuery(query.id, { text: 'Announcement not found.' }); 831 831 + return; 832 832 + } 833 833 + 834 834 + await this.bot.answerCallbackQuery(query.id); 835 835 + 836 836 + // Delete the edit options message 837 837 + await this.bot.deleteMessage(chatId, query.message.message_id); 838 838 + 839 839 + let promptText = ''; 840 840 + switch (field) { 841 841 + case 'message': 842 842 + promptText = `Please enter the new message text for the announcement "${announcement.name}":\n\nCurrent message:\n${announcement.message}`; 843 843 + break; 844 844 + case 'name': 845 845 + promptText = `Please enter the new name for the announcement "${announcement.name}":`; 846 846 + break; 847 847 + case 'schedule': 848 848 + promptText = `Please enter the new cron schedule for the announcement "${announcement.name}" (use https://crontab.guru/ for help):\n\nCurrent schedule: ${announcement.cronSchedule}`; 849 849 + break; 850 850 + case 'button': 851 851 + // For button editing, we'll first ask if they want to add, edit, or remove a button 852 852 + const hasButton = announcement.button && announcement.button.text && announcement.button.url; 853 853 + 854 854 + if (hasButton) { 855 855 + // Show options to edit or remove existing button 856 856 + await this.bot.sendMessage( 857 857 + chatId, 858 858 + `Current button: "${announcement.button.text}" → ${announcement.button.url}\n\nWhat would you like to do?`, 859 859 + { 860 860 + reply_markup: { 861 861 + inline_keyboard: [ 862 862 + [ 863 863 + { 864 864 + text: '✏️ Edit Button', 865 865 + callback_data: `edit_announcement_button_edit_${announcementId}` 866 866 + } 867 867 + ], 868 868 + [ 869 869 + { 870 870 + text: '❌ Remove Button', 871 871 + callback_data: `edit_announcement_button_remove_${announcementId}` 872 872 + } 873 873 + ], 874 874 + [ 875 875 + { 876 876 + text: '↩️ Cancel', 877 877 + callback_data: 'cancel_edit_announcement_button' 878 878 + } 879 879 + ] 880 880 + ] 881 881 + } 882 882 + } 883 883 + ); 884 884 + return; 885 885 + } else { 886 886 + // No existing button, ask if they want to add one 887 887 + await this.bot.sendMessage( 888 888 + chatId, 889 889 + `This announcement doesn't have a button. Would you like to add one?`, 890 890 + { 891 891 + reply_markup: { 892 892 + inline_keyboard: [ 893 893 + [ 894 894 + { 895 895 + text: '➕ Add Button', 896 896 + callback_data: `edit_announcement_button_add_${announcementId}` 897 897 + } 898 898 + ], 899 899 + [ 900 900 + { 901 901 + text: '↩️ Cancel', 902 902 + callback_data: 'cancel_edit_announcement_button' 903 903 + } 904 904 + ] 905 905 + ] 906 906 + } 907 907 + } 908 908 + ); 909 909 + return; 910 910 + } 911 911 + } 912 912 + } 913 913 + break; 914 914 + } 915 915 + 916 916 + case 'new': { 917 917 + if (data[1] === 'announcement') { 918 918 + await this.bot.answerCallbackQuery(query.id); 919 919 + 920 920 + // Delete the announcements list message 921 921 + await this.bot.deleteMessage(chatId, query.message.message_id); 922 922 + 923 923 + // Trigger the /announce command 924 924 + await this.bot.sendMessage( 925 925 + chatId, 926 926 + 'Please use the /announce command followed by your announcement text to create a new announcement.' 927 927 + ); 928 928 + } 929 929 + break; 930 930 + } 931 931 + 932 932 + case 'cancel': { 933 933 + if (data[1] === 'edit' && data[2] === 'announcement') { 934 934 + await this.bot.answerCallbackQuery(query.id, { text: 'Edit cancelled.' }); 935 935 + await this.bot.deleteMessage(chatId, query.message.message_id); 936 936 + 937 937 + // Refresh announcements list 938 938 + await this.bot.onText.handlers.find(h => h.regexp.toString().includes('/announcements'))?._callback({ chat: { id: chatId } }); 939 939 + } 940 940 + break; 941 941 + } 942 942 + 943 943 + case 'edit': { 944 944 + if (data[1] === 'announcement' && data[2] === 'button') { 945 945 + if (data[3] === 'add' || data[3] === 'edit') { 946 946 + // Add or edit a button 947 947 + const announcementId = data[4]; 948 948 + const announcement = this.announcements.getAnnouncementById(announcementId); 949 949 + 950 950 + if (!announcement) { 951 951 + await this.bot.answerCallbackQuery(query.id, { text: 'Announcement not found.' }); 952 952 + return; 953 953 + } 954 954 + 955 955 + await this.bot.answerCallbackQuery(query.id); 956 956 + 957 957 + // Delete the options message 958 958 + await this.bot.deleteMessage(chatId, query.message.message_id); 959 959 + 960 960 + // Store the context for updating 961 961 + this.editingAnnouncementButton = this.editingAnnouncementButton || {}; 962 962 + this.editingAnnouncementButton[query.from.id] = { id: announcementId }; 963 963 + 964 964 + // First ask for button text 965 965 + const buttonTextPrompt = await this.bot.sendMessage( 966 966 + chatId, 967 967 + 'Please enter the button text:', 968 968 + { reply_markup: { force_reply: true } } 969 969 + ); 970 970 + 971 971 + this.bot.onReplyToMessage(chatId, buttonTextPrompt.message_id, async (buttonTextMsg) => { 972 972 + const buttonText = buttonTextMsg.text; 973 973 + 974 974 + // Now ask for the button URL 975 975 + const buttonUrlPrompt = await this.bot.sendMessage( 976 976 + chatId, 977 977 + 'Please enter the button URL:', 978 978 + { reply_markup: { force_reply: true } } 979 979 + ); 980 980 + 981 981 + this.bot.onReplyToMessage(chatId, buttonUrlPrompt.message_id, async (buttonUrlMsg) => { 982 982 + const buttonUrl = buttonUrlMsg.text; 983 983 + 984 984 + // Create the button object 985 985 + const button = { 986 986 + text: buttonText, 987 987 + url: buttonUrl 988 988 + }; 989 989 + 990 990 + try { 991 991 + // Update the announcement with the new button 992 992 + const updated = await this.announcements.updateAnnouncement(announcementId, { button }); 993 993 + 994 994 + if (updated) { 995 995 + await this.bot.sendMessage( 996 996 + chatId, 997 997 + `✅ Button ${data[3] === 'add' ? 'added' : 'updated'} successfully!` 998 998 + ); 999 999 + } else { 1000 1000 + await this.bot.sendMessage( 1001 1001 + chatId, 1002 1002 + `❌ Failed to ${data[3] === 'add' ? 'add' : 'update'} button.` 1003 1003 + ); 1004 1004 + } 1005 1005 + 1006 1006 + // Clean up 1007 1007 + delete this.editingAnnouncementButton[query.from.id]; 1008 1008 + 1009 1009 + // Refresh announcements list 1010 1010 + await this.bot.onText.handlers.find(h => h.regexp.toString().includes('/announcements'))?._callback({ chat: { id: chatId } }); 1011 1011 + } catch (error) { 1012 1012 + await this.bot.sendMessage( 1013 1013 + chatId, 1014 1014 + `❌ Error updating button: ${error.message}` 1015 1015 + ); 1016 1016 + } 1017 1017 + }); 1018 1018 + }); 1019 1019 + } else if (data[3] === 'remove') { 1020 1020 + // Remove a button 1021 1021 + const announcementId = data[4]; 1022 1022 + 1023 1023 + try { 1024 1024 + // Remove the button by setting it to null 1025 1025 + const updated = await this.announcements.updateAnnouncement(announcementId, { button: null }); 1026 1026 + 1027 1027 + if (updated) { 1028 1028 + await this.bot.answerCallbackQuery(query.id, { text: 'Button removed successfully.' }); 1029 1029 + } else { 1030 1030 + await this.bot.answerCallbackQuery(query.id, { text: 'Failed to remove button.' }); 1031 1031 + } 1032 1032 + 1033 1033 + // Delete the options message 1034 1034 + await this.bot.deleteMessage(chatId, query.message.message_id); 1035 1035 + 1036 1036 + // Refresh announcements list 1037 1037 + await this.bot.onText.handlers.find(h => h.regexp.toString().includes('/announcements'))?._callback({ chat: { id: chatId } }); 1038 1038 + } catch (error) { 1039 1039 + await this.bot.answerCallbackQuery(query.id, { text: `Error: ${error.message}` }); 1040 1040 + } 1041 1041 + } 1042 1042 + } 1043 1043 + break; 1044 1044 + } 1045 1045 + 1046 1046 + case 'cancel': { 1047 1047 + if (data[1] === 'edit' && data[2] === 'announcement' && data[3] === 'button') { 1048 1048 + await this.bot.answerCallbackQuery(query.id, { text: 'Button edit cancelled.' }); 1049 1049 + await this.bot.deleteMessage(chatId, query.message.message_id); 1050 1050 + 1051 1051 + // Refresh announcements list 1052 1052 + await this.bot.onText.handlers.find(h => h.regexp.toString().includes('/announcements'))?._callback({ chat: { id: chatId } }); 444 1053 } 445 1054 break; 446 1055 }

+351

utils/announcementManager.js

reviewed

··· 1 1 + /** 2 2 + * Announcement Manager for scheduled text announcements 3 3 + * Handles multiple announcements with individual cron schedules 4 4 + */ 5 5 + 6 6 + const fs = require('fs'); 7 7 + const path = require('path'); 8 8 + const cron = require('node-cron'); 9 9 + const { promisify } = require('util'); 10 10 + const writeFileAsync = promisify(fs.writeFile); 11 11 + const readFileAsync = promisify(fs.readFile); 12 12 + 13 13 + class AnnouncementManager { 14 14 + constructor(telegramBot) { 15 15 + this.telegramBot = telegramBot; 16 16 + this.announcements = []; 17 17 + this.cronJobs = {}; 18 18 + this.filePath = path.join(__dirname, '..', 'announcements.json'); 19 19 + this.initialized = false; 20 20 + } 21 21 + 22 22 + /** 23 23 + * Initialize the announcement manager 24 24 + */ 25 25 + async init() { 26 26 + if (this.initialized) return; 27 27 + 28 28 + try { 29 29 + await this.loadAnnouncementsFromDisk(); 30 30 + this.scheduleAllAnnouncements(); 31 31 + this.initialized = true; 32 32 + console.log('Announcement manager initialized'); 33 33 + } catch (error) { 34 34 + console.error('Error initializing announcement manager:', error); 35 35 + // If loading fails, start with empty announcements 36 36 + this.announcements = []; 37 37 + await this.saveAnnouncementsToDisk(); 38 38 + } 39 39 + } 40 40 + 41 41 + /** 42 42 + * Load announcements from disk 43 43 + */ 44 44 + async loadAnnouncementsFromDisk() { 45 45 + try { 46 46 + // Check if the file exists 47 47 + if (!fs.existsSync(this.filePath)) { 48 48 + // Create an empty announcements file 49 49 + this.announcements = []; 50 50 + await this.saveAnnouncementsToDisk(); 51 51 + return; 52 52 + } 53 53 + 54 54 + // Read and parse the file 55 55 + const data = await readFileAsync(this.filePath, 'utf8'); 56 56 + this.announcements = JSON.parse(data); 57 57 + console.log(`Loaded ${this.announcements.length} announcements from disk`); 58 58 + } catch (error) { 59 59 + console.error('Error loading announcements from disk:', error); 60 60 + throw error; 61 61 + } 62 62 + } 63 63 + 64 64 + /** 65 65 + * Save announcements to disk 66 66 + */ 67 67 + async saveAnnouncementsToDisk() { 68 68 + try { 69 69 + const data = JSON.stringify(this.announcements, null, 2); 70 70 + await writeFileAsync(this.filePath, data, 'utf8'); 71 71 + console.log(`Saved ${this.announcements.length} announcements to disk`); 72 72 + return true; 73 73 + } catch (error) { 74 74 + console.error('Error saving announcements to disk:', error); 75 75 + return false; 76 76 + } 77 77 + } 78 78 + 79 79 + /** 80 80 + * Add a new announcement 81 81 + * @param {string} message - The text message to post 82 82 + * @param {string} cronSchedule - Cron expression for schedule 83 83 + * @param {string} name - Optional name/label for the announcement 84 84 + * @param {Object} button - Optional button configuration {text: string, url: string} 85 85 + * @returns {Promise<Object>} - The added announcement object 86 86 + */ 87 87 + async addAnnouncement(message, cronSchedule, name = '', button = null) { 88 88 + // Validate cron expression 89 89 + if (!this.isValidCronExpression(cronSchedule)) { 90 90 + throw new Error('Invalid cron expression'); 91 91 + } 92 92 + 93 93 + // Validate button if provided 94 94 + if (button && (!button.text || !button.url)) { 95 95 + throw new Error('Button must have both text and url properties'); 96 96 + } 97 97 + 98 98 + // Create new announcement 99 99 + const id = Date.now().toString(); 100 100 + const announcement = { 101 101 + id, 102 102 + message, 103 103 + cronSchedule, 104 104 + name: name || `Announcement ${id.substring(id.length - 4)}`, 105 105 + createdAt: new Date().toISOString(), 106 106 + lastRun: null, 107 107 + button: button 108 108 + }; 109 109 + 110 110 + // Add to our list 111 111 + this.announcements.push(announcement); 112 112 + 113 113 + // Save changes 114 114 + await this.saveAnnouncementsToDisk(); 115 115 + 116 116 + // Schedule it 117 117 + this.scheduleAnnouncement(announcement); 118 118 + 119 119 + return announcement; 120 120 + } 121 121 + 122 122 + /** 123 123 + * Remove an announcement by its ID 124 124 + * @param {string} id - Announcement ID to remove 125 125 + * @returns {Promise<boolean>} - Whether removal was successful 126 126 + */ 127 127 + async removeAnnouncement(id) { 128 128 + const index = this.announcements.findIndex(a => a.id === id); 129 129 + 130 130 + if (index === -1) { 131 131 + return false; 132 132 + } 133 133 + 134 134 + // Stop the cron job 135 135 + if (this.cronJobs[id]) { 136 136 + this.cronJobs[id].stop(); 137 137 + delete this.cronJobs[id]; 138 138 + } 139 139 + 140 140 + // Remove from array 141 141 + this.announcements.splice(index, 1); 142 142 + 143 143 + // Save changes 144 144 + await this.saveAnnouncementsToDisk(); 145 145 + 146 146 + return true; 147 147 + } 148 148 + 149 149 + /** 150 150 + * Update an existing announcement 151 151 + * @param {string} id - ID of the announcement to update 152 152 + * @param {Object} updates - Object with fields to update (message, cronSchedule, or name) 153 153 + * @returns {Promise<Object>} - The updated announcement 154 154 + */ 155 155 + async updateAnnouncement(id, updates) { 156 156 + const index = this.announcements.findIndex(a => a.id === id); 157 157 + 158 158 + if (index === -1) { 159 159 + throw new Error('Announcement not found'); 160 160 + } 161 161 + 162 162 + const announcement = this.announcements[index]; 163 163 + 164 164 + // Update fields 165 165 + if (updates.message !== undefined) { 166 166 + announcement.message = updates.message; 167 167 + } 168 168 + 169 169 + if (updates.name !== undefined) { 170 170 + announcement.name = updates.name; 171 171 + } 172 172 + 173 173 + if (updates.cronSchedule !== undefined) { 174 174 + if (!this.isValidCronExpression(updates.cronSchedule)) { 175 175 + throw new Error('Invalid cron expression'); 176 176 + } 177 177 + 178 178 + announcement.cronSchedule = updates.cronSchedule; 179 179 + 180 180 + // Update the schedule 181 181 + if (this.cronJobs[id]) { 182 182 + this.cronJobs[id].stop(); 183 183 + } 184 184 + this.scheduleAnnouncement(announcement); 185 185 + } 186 186 + 187 187 + // Save changes 188 188 + await this.saveAnnouncementsToDisk(); 189 189 + 190 190 + return announcement; 191 191 + } 192 192 + 193 193 + /** 194 194 + * Get all announcements 195 195 + * @returns {Array} - Array of announcement objects 196 196 + */ 197 197 + getAnnouncements() { 198 198 + return [...this.announcements]; 199 199 + } 200 200 + 201 201 + /** 202 202 + * Get a specific announcement by ID 203 203 + * @param {string} id - The announcement ID 204 204 + * @returns {Object|null} - The announcement object or null if not found 205 205 + */ 206 206 + getAnnouncementById(id) { 207 207 + return this.announcements.find(a => a.id === id) || null; 208 208 + } 209 209 + 210 210 + /** 211 211 + * Schedule a single announcement 212 212 + * @param {Object} announcement - The announcement to schedule 213 213 + */ 214 214 + scheduleAnnouncement(announcement) { 215 215 + // Cancel existing job if it exists 216 216 + if (this.cronJobs[announcement.id]) { 217 217 + this.cronJobs[announcement.id].stop(); 218 218 + } 219 219 + 220 220 + // Create new cron job 221 221 + this.cronJobs[announcement.id] = cron.schedule(announcement.cronSchedule, async () => { 222 222 + try { 223 223 + console.log(`Running scheduled announcement: ${announcement.name}`); 224 224 + 225 225 + // Create message options with markdown support 226 226 + const messageOptions = { parse_mode: 'Markdown' }; 227 227 + 228 228 + // Add inline keyboard if a button is defined 229 229 + if (announcement.button && announcement.button.text && announcement.button.url) { 230 230 + messageOptions.reply_markup = { 231 231 + inline_keyboard: [ 232 232 + [ 233 233 + { 234 234 + text: announcement.button.text, 235 235 + url: announcement.button.url 236 236 + } 237 237 + ] 238 238 + ] 239 239 + }; 240 240 + } 241 241 + 242 242 + // Send the announcement message to the channel 243 243 + const result = await this.telegramBot.bot.sendMessage( 244 244 + this.telegramBot.channelId, 245 245 + announcement.message, 246 246 + messageOptions 247 247 + ); 248 248 + 249 249 + // Update last run time 250 250 + const index = this.announcements.findIndex(a => a.id === announcement.id); 251 251 + if (index !== -1) { 252 252 + this.announcements[index].lastRun = new Date().toISOString(); 253 253 + await this.saveAnnouncementsToDisk(); 254 254 + } 255 255 + 256 256 + console.log(`Successfully sent announcement: ${announcement.name}`); 257 257 + } catch (error) { 258 258 + console.error(`Error sending announcement ${announcement.name}:`, error); 259 259 + } 260 260 + }); 261 261 + 262 262 + console.log(`Scheduled announcement: ${announcement.name} with cron: ${announcement.cronSchedule}`); 263 263 + } 264 264 + 265 265 + /** 266 266 + * Schedule all announcements 267 267 + */ 268 268 + scheduleAllAnnouncements() { 269 269 + // Clear existing jobs 270 270 + Object.values(this.cronJobs).forEach(job => job.stop()); 271 271 + this.cronJobs = {}; 272 272 + 273 273 + // Schedule each announcement 274 274 + this.announcements.forEach(announcement => { 275 275 + this.scheduleAnnouncement(announcement); 276 276 + }); 277 277 + 278 278 + console.log(`Scheduled ${this.announcements.length} announcements`); 279 279 + } 280 280 + 281 281 + /** 282 282 + * Validate a cron expression 283 283 + * @param {string} cronExpression - The cron expression to validate 284 284 + * @returns {boolean} - Whether the expression is valid 285 285 + */ 286 286 + isValidCronExpression(cronExpression) { 287 287 + return cron.validate(cronExpression); 288 288 + } 289 289 + 290 290 + /** 291 291 + * Send an announcement immediately (one-time) 292 292 + * @param {string} id - ID of the announcement to send 293 293 + * @returns {Promise<boolean>} - Whether sending was successful 294 294 + */ 295 295 + async sendAnnouncementNow(id) { 296 296 + const announcement = this.getAnnouncementById(id); 297 297 + 298 298 + if (!announcement) { 299 299 + return false; 300 300 + } 301 301 + 302 302 + try { 303 303 + // Create message options with markdown support 304 304 + const messageOptions = { parse_mode: 'Markdown' }; 305 305 + 306 306 + // Add inline keyboard if a button is defined 307 307 + if (announcement.button && announcement.button.text && announcement.button.url) { 308 308 + messageOptions.reply_markup = { 309 309 + inline_keyboard: [ 310 310 + [ 311 311 + { 312 312 + text: announcement.button.text, 313 313 + url: announcement.button.url 314 314 + } 315 315 + ] 316 316 + ] 317 317 + }; 318 318 + } 319 319 + 320 320 + await this.telegramBot.bot.sendMessage( 321 321 + this.telegramBot.channelId, 322 322 + announcement.message, 323 323 + messageOptions 324 324 + ); 325 325 + 326 326 + // Update last run time 327 327 + const index = this.announcements.findIndex(a => a.id === id); 328 328 + if (index !== -1) { 329 329 + this.announcements[index].lastRun = new Date().toISOString(); 330 330 + await this.saveAnnouncementsToDisk(); 331 331 + } 332 332 + 333 333 + return true; 334 334 + } catch (error) { 335 335 + console.error(`Error sending announcement ${announcement.name}:`, error); 336 336 + return false; 337 337 + } 338 338 + } 339 339 + 340 340 + /** 341 341 + * Shutdown the announcement manager 342 342 + */ 343 343 + shutdown() { 344 344 + // Stop all cron jobs 345 345 + Object.values(this.cronJobs).forEach(job => job.stop()); 346 346 + this.cronJobs = {}; 347 347 + console.log('Announcement manager shutdown complete'); 348 348 + } 349 349 + } 350 350 + 351 351 + module.exports = AnnouncementManager;