mirror of
https://github.com/danbulant/discord.js
synced 2026-05-27 14:02:02 +00:00
feat(Collector): add idle time for a Collector to stop itself (#2942)
* Implement idle feature * Add typings * Minimal fixes * Make everything in Collector and not attached to ReactionCollector * set this._idletimeout to null when collector ends * also set this._timeout to null when collector ends
This commit is contained in:
parent
adb082305d
commit
f1433a2d97
2 changed files with 24 additions and 1 deletions
|
|
@ -16,6 +16,7 @@ const EventEmitter = require('events');
|
|||
* Options to be applied to the collector.
|
||||
* @typedef {Object} CollectorOptions
|
||||
* @property {number} [time] How long to run the collector for in milliseconds
|
||||
* @property {number} [idle] How long to stop the collector after inactivity in milliseconds
|
||||
* @property {boolean} [dispose=false] Whether to dispose data when it's deleted
|
||||
*/
|
||||
|
||||
|
|
@ -66,10 +67,18 @@ class Collector extends EventEmitter {
|
|||
*/
|
||||
this._timeout = null;
|
||||
|
||||
/**
|
||||
* Timeout for cleanup due to inactivity
|
||||
* @type {?Timeout}
|
||||
* @private
|
||||
*/
|
||||
this._idletimeout = null;
|
||||
|
||||
this.handleCollect = this.handleCollect.bind(this);
|
||||
this.handleDispose = this.handleDispose.bind(this);
|
||||
|
||||
if (options.time) this._timeout = this.client.setTimeout(() => this.stop('time'), options.time);
|
||||
if (options.idle) this._idletimeout = this.client.setTimeout(() => this.stop('idle'), options.idle);
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
@ -89,6 +98,11 @@ class Collector extends EventEmitter {
|
|||
* @param {...*} args The arguments emitted by the listener
|
||||
*/
|
||||
this.emit('collect', ...args);
|
||||
|
||||
if (this._idletimeout) {
|
||||
this.client.clearTimeout(this._idletimeout);
|
||||
this._idletimeout = this.client.setTimeout(() => this.stop('idle'), this.options.idle);
|
||||
}
|
||||
}
|
||||
this.checkEnd();
|
||||
}
|
||||
|
|
@ -155,7 +169,14 @@ class Collector extends EventEmitter {
|
|||
stop(reason = 'user') {
|
||||
if (this.ended) return;
|
||||
|
||||
if (this._timeout) this.client.clearTimeout(this._timeout);
|
||||
if (this._timeout) {
|
||||
this.client.clearTimeout(this._timeout);
|
||||
this._timeout = null;
|
||||
}
|
||||
if (this._idletimeout) {
|
||||
this.client.clearTimeout(this._idletimeout);
|
||||
this._idletimeout = null;
|
||||
}
|
||||
this.ended = true;
|
||||
|
||||
/**
|
||||
|
|
|
|||
2
typings/index.d.ts
vendored
2
typings/index.d.ts
vendored
|
|
@ -350,6 +350,7 @@ declare module 'discord.js' {
|
|||
export abstract class Collector<K, V> extends EventEmitter {
|
||||
constructor(client: Client, filter: CollectorFilter, options?: CollectorOptions);
|
||||
private _timeout: NodeJS.Timer | null;
|
||||
private _idletimeout: NodeJS.Timer | null;
|
||||
|
||||
public readonly client: Client;
|
||||
public collected: Collection<K, V>;
|
||||
|
|
@ -1759,6 +1760,7 @@ declare module 'discord.js' {
|
|||
|
||||
interface CollectorOptions {
|
||||
time?: number;
|
||||
idle?: number;
|
||||
dispose?: boolean;
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue