From 160e887b64ee9d7edc2e33f20580ae3613c20dd0 Mon Sep 17 00:00:00 2001
From: Snowflake107 <devsnowflake107@gmail.com>
Date: Thu, 13 May 2021 14:31:36 +0545
Subject: [PATCH] allow extractors to specify source type

---
 src/Player.ts      | 2 +-
 src/types/types.ts | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Player.ts b/src/Player.ts
index d73eb6b..05c773c 100644
--- a/src/Player.ts
+++ b/src/Player.ts
@@ -556,7 +556,7 @@ export class Player extends EventEmitter {
                                 author: data.author,
                                 views: data.views,
                                 engine: data.engine,
-                                source: 'arbitrary',
+                                source: data.source ?? 'arbitrary',
                                 fromPlaylist: false,
                                 requestedBy: message.author,
                                 url: data.url
diff --git a/src/types/types.ts b/src/types/types.ts
index 11d48b2..bc9c541 100644
--- a/src/types/types.ts
+++ b/src/types/types.ts
@@ -92,6 +92,7 @@ export interface ExtractorModelData {
     url: string;
     version?: string;
     important?: boolean;
+    source?: TrackSource;
 }
 
 export interface PlayerProgressbarOptions {