diff --git a/src/Player.ts b/src/Player.ts index f1d3f49..642cbe8 100644 --- a/src/Player.ts +++ b/src/Player.ts @@ -25,6 +25,10 @@ class DiscordPlayer extends EventEmitter { getQueue(guild: Snowflake) { return this.queues.get(guild); } + + *[Symbol.iterator]() { + yield* Array.from(this.queues.values()); + } } export { DiscordPlayer as Player }; diff --git a/src/Structures/Playlist.ts b/src/Structures/Playlist.ts index 44f1643..04a81c2 100644 --- a/src/Structures/Playlist.ts +++ b/src/Structures/Playlist.ts @@ -9,6 +9,10 @@ class Playlist { this.player = player; this.tracks = tracks ?? []; } + + *[Symbol.iterator]() { + yield* this.tracks; + } } export { Playlist }; diff --git a/src/Structures/Queue.ts b/src/Structures/Queue.ts index b86b96b..095c1f2 100644 --- a/src/Structures/Queue.ts +++ b/src/Structures/Queue.ts @@ -57,6 +57,10 @@ class Queue { play() { throw new Error("Not implemented"); } + + *[Symbol.iterator]() { + yield* this.tracks; + } } export { Queue };