Skip to content

Commit

Permalink
feat(Collector): add idle time for a Collector to stop itself (#2942)
Browse files Browse the repository at this point in the history
* 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
  • Loading branch information
nirewen authored and SpaceEEC committed Jul 11, 2019
1 parent adb0823 commit f1433a2
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 1 deletion.
23 changes: 22 additions & 1 deletion src/structures/interfaces/Collector.js
Original file line number Diff line number Diff line change
Expand Up @@ -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
*/

Expand Down Expand Up @@ -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);
}

/**
Expand All @@ -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();
}
Expand Down Expand Up @@ -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;

/**
Expand Down
2 changes: 2 additions & 0 deletions typings/index.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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>;
Expand Down Expand Up @@ -1759,6 +1760,7 @@ declare module 'discord.js' {

interface CollectorOptions {
time?: number;
idle?: number;
dispose?: boolean;
}

Expand Down

0 comments on commit f1433a2

Please sign in to comment.