jonny_bro/discord-player-play-dl
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Player#use & Player#unuse should return the extractor

Browse source
This commit is contained in:
Snowflake107 2021-06-18 00:02:55 +05:45
parent 233a24dd34
commit 0619462f2c
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/Player.ts
View file

@ -399,12 +399,12 @@ class DiscordPlayer extends EventEmitter<PlayerEvents> {
} }
} }
use(extractorName: string, extractor: ExtractorModel | any, force = false) { use(extractorName: string, extractor: ExtractorModel | any, force = false): ExtractorModel {
if (!extractorName) throw new Error("Cannot use unknown extractor!"); if (!extractorName) throw new Error("Cannot use unknown extractor!");
if (this.extractors.has(extractorName) && !force) return this; if (this.extractors.has(extractorName) && !force) return this.extractors.get(extractorName);
if (extractor instanceof ExtractorModel) { if (extractor instanceof ExtractorModel) {
this.extractors.set(extractorName, extractor); this.extractors.set(extractorName, extractor);
return this; return extractor;
} }
for (const method of ["validate", "getInfo"]) { for (const method of ["validate", "getInfo"]) {
@ -414,12 +414,14 @@ class DiscordPlayer extends EventEmitter<PlayerEvents> {
const model = new ExtractorModel(extractorName, extractor); const model = new ExtractorModel(extractorName, extractor);
this.extractors.set(model.name, model); this.extractors.set(model.name, model);
return this; return model;
} }
unuse(extractorName: string) { unuse(extractorName: string) {
if (!this.extractors.has(extractorName)) throw new Error(`Cannot find extractor "${extractorName}"`); if (!this.extractors.has(extractorName)) throw new Error(`Cannot find extractor "${extractorName}"`);
const prev = this.extractors.get(extractorName);
this.extractors.delete(extractorName); this.extractors.delete(extractorName);
return prev;
} }
*[Symbol.iterator]() { *[Symbol.iterator]() {