From 5ba21c33375837a1795913417b3f1de560eac105 Mon Sep 17 00:00:00 2001 From: Snowflake107 Date: Fri, 11 Jun 2021 20:42:49 +0545 Subject: [PATCH] feat(Structures): Generator function --- src/Player.ts | 4 ++++ src/Structures/Playlist.ts | 4 ++++ src/Structures/Queue.ts | 4 ++++ 3 files changed, 12 insertions(+) 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 };