From 7301cbe76d8c7e84b3720d2f5241ab7acbf96d02 Mon Sep 17 00:00:00 2001
From: Androz2091 <androz2091@gmail.com>
Date: Sat, 11 Jul 2020 11:09:20 +0200
Subject: [PATCH] :sparkles: Add allResults parameter to searchTracks method

---
 src/Player.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/Player.js b/src/Player.js
index 21b5364..52db046 100644
--- a/src/Player.js
+++ b/src/Player.js
@@ -136,6 +136,7 @@ class Player {
     /**
      * Resolve an array of tracks objects from a query string
      * @param {string} query The query
+     * @param {boolean} allResults Whether all the results should be returned, or only the first one
      * @returns {Promise<Track[]>}
      *
      * @example
@@ -164,7 +165,7 @@ class Player {
      *
      * });
      */
-    searchTracks (query) {
+    searchTracks (query, allResults = false) {
         return new Promise(async (resolve, reject) => {
             if (ytpl.validateURL(query)) {
                 const playlistID = await ytpl.getPlaylistID(query).catch(() => {})
@@ -195,7 +196,7 @@ class Player {
                 if (results.items.length < 1) return resolve([])
                 if (err) return resolve([])
                 const resultsVideo = results.items.filter((i) => i.type === 'video')
-                resolve([new Track(resultsVideo[0], null, null)])
+                resolve(allResults ? resultsVideo.map((r) => new Track(r, null, null)) : [new Track(resultsVideo[0], null, null)])
             })
         })
     }