eventSource: Add autofire on emit for APP_READY

aadae85a2a2f5e5fa9e34a2cb9815c1d2a0c066a

Cohee <18619528+Cohee1207@users.noreply.github.com>

public/lib/eventemitter.js+43 -2 Viewed

 /* Polyfill EventEmitter. */
-var EventEmitter = function () {
+/**
+ * Creates an event emitter.
+ * @param {string[]} autoFireAfterEmit Auto-fire event names
+ */
+var EventEmitter = function (autoFireAfterEmit = []) {
     this.events = {};
+    this.autoFireLastArgs = new Map();
+    this.autoFireAfterEmit = new Set(autoFireAfterEmit);
 };
+/**
+ * Adds a listener to an event.
+ * @param {string} event Event name
+ * @param {function} listener Event listener
+ * @returns
+ */
 EventEmitter.prototype.on = function (event, listener) {
     // Unknown event used by external libraries?
     if (event === undefined) {

+    }
     this.events[event].push(listener);
+    if (this.autoFireAfterEmit.has(event) && this.autoFireLastArgs.has(event)) {
+        listener.apply(this, this.autoFireLastArgs.get(event));
+    }
 };
 /**

@@ -60,6 +76,10 @@ EventEmitter.prototype.makeLast = function (event, listener) {
60	}	76	}
61		77
62	events.push(listener);	78	events.push(listener);
		79
		80	if (this.autoFireAfterEmit.has(event) && this.autoFireLastArgs.has(event)) {
		81	listener.apply(this, this.autoFireLastArgs.get(event));
		82	}
63	}	83	}
64		84
65	/**	85	/**

+    }
     events.unshift(listener);
+    if (this.autoFireAfterEmit.has(event) && this.autoFireLastArgs.has(event)) {
+        listener.apply(this, this.autoFireLastArgs.get(event));
+    }
+}
+/**
+ * Removes a listener from an event.
+ * @param {string} event Event name
+ * @param {function} listener Event listener
+ */
 EventEmitter.prototype.removeListener = function (event, listener) {
     var idx;

+    }
 };
+/**
+ * Emits an event with optional arguments.
+ * @param {string} event Event name
+ */
 EventEmitter.prototype.emit = async function (event) {
     let args = [].slice.call(arguments, 1);
     if (localStorage.getItem('eventTracing') === 'true') {

@@ -118,6 +151,10 @@ EventEmitter.prototype.emit = async function (event) {
118	}	151	}
119	}	152	}
120	}	153	}
		154
		155	if (this.autoFireAfterEmit.has(event)) {
		156	this.autoFireLastArgs.set(event, args);
		157	}
121	};	158	};
122		159
123	EventEmitter.prototype.emitAndWait = function (event) {	160	EventEmitter.prototype.emitAndWait = function (event) {

@@ -144,10 +181,14 @@ EventEmitter.prototype.emitAndWait = function (event) {
144	}	181	}
145	}	182	}
146	}	183	}
		184
		185	if (this.autoFireAfterEmit.has(event)) {
		186	this.autoFireLastArgs.set(event, args);
		187	}
147	};	188	};
148		189
149	EventEmitter.prototype.once = function (event, listener) {	190	EventEmitter.prototype.once = function (event, listener) {
150	this.on(event, function g () {	191	this.on(event, function g() {
151	this.removeListener(event, g);	192	this.removeListener(event, g);
152	listener.apply(this, arguments);	193	listener.apply(this, arguments);
153	});	194	});

public/script.js+1 -1 Viewed

     TOOL_CALLS_RENDERED: 'tool_calls_rendered',
 };
-export const eventSource = new EventEmitter();
+export const eventSource = new EventEmitter([event_types.APP_READY]);
 eventSource.on(event_types.CHAT_CHANGED, processChatSlashCommands);