File size: 3,395 Bytes
c120a1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 |
import { warn } from '../index.js';
import { QuickReply } from './QuickReply.js';
import { QuickReplySettings } from './QuickReplySettings.js';
export class AutoExecuteHandler {
/** @type {QuickReplySettings} */ settings;
/** @type {Boolean[]}*/ preventAutoExecuteStack = [];
constructor(/** @type {QuickReplySettings} */settings) {
this.settings = settings;
}
checkExecute() {
return this.settings.isEnabled && !this.preventAutoExecuteStack.slice(-1)[0];
}
async performAutoExecute(/** @type {QuickReply[]} */qrList) {
for (const qr of qrList) {
this.preventAutoExecuteStack.push(qr.preventAutoExecute);
try {
await qr.execute({ isAutoExecute:true });
} catch (ex) {
warn(ex);
} finally {
this.preventAutoExecuteStack.pop();
}
}
}
getCommands(eventName) {
const getFromConfig = (config) => {
// This safely handles cases where a link exists but the set hasn't been loaded (link.set is null)
return config?.setList?.map(link => link.set ? link.set.qrList.filter(qr => qr[eventName]) : [])?.flat() ?? [];
};
return [
...getFromConfig(this.settings.config),
...getFromConfig(this.settings.chatConfig),
...getFromConfig(this.settings.charConfig),
];
}
async handleStartup() {
if (!this.checkExecute()) return;
await this.performAutoExecute(this.getCommands('executeOnStartup'));
}
async handleUser() {
if (!this.checkExecute()) return;
await this.performAutoExecute(this.getCommands('executeOnUser'));
}
async handleAi() {
if (!this.checkExecute()) return;
await this.performAutoExecute(this.getCommands('executeOnAi'));
}
async handleChatChanged() {
if (!this.checkExecute()) return;
await this.performAutoExecute(this.getCommands('executeOnChatChange'));
}
async handleGroupMemberDraft() {
if (!this.checkExecute()) return;
await this.performAutoExecute(this.getCommands('executeOnGroupMemberDraft'));
}
async handleNewChat() {
if (!this.checkExecute()) return;
await this.performAutoExecute(this.getCommands('executeOnNewChat'));
}
async handleBeforeGeneration() {
if (!this.checkExecute()) return;
await this.performAutoExecute(this.getCommands('executeBeforeGeneration'));
}
/**
* @param {any[]} entries Set of activated entries
*/
async handleWIActivation(entries) {
if (!this.checkExecute() || !Array.isArray(entries) || entries.length === 0) return;
const automationIds = entries.map(entry => entry.automationId).filter(Boolean);
if (automationIds.length === 0) return;
const getFromConfig = (config) => {
return config?.setList
?.map(link => link.set ? link.set.qrList.filter(qr => qr.automationId && automationIds.includes(qr.automationId)) : [])
?.flat() ?? [];
};
const qrList = [
...getFromConfig(this.settings.config),
...getFromConfig(this.settings.chatConfig),
...getFromConfig(this.settings.charConfig),
];
await this.performAutoExecute(qrList);
}
}
|