utils.js 27 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009
  1. import config from "config";
  2. import async from "async";
  3. import crypto from "crypto";
  4. import request from "request";
  5. import CoreClass from "../core";
  6. let UtilsModule;
  7. let IOModule;
  8. let SpotifyModule;
  9. let CacheModule;
  10. class _UtilsModule extends CoreClass {
  11. // eslint-disable-next-line require-jsdoc
  12. constructor() {
  13. super("utils");
  14. this.youtubeRequestCallbacks = [];
  15. this.youtubeRequestsPending = 0;
  16. this.youtubeRequestsActive = false;
  17. UtilsModule = this;
  18. }
  19. /**
  20. * Initialises the utils module
  21. *
  22. * @returns {Promise} - returns promise (reject, resolve)
  23. */
  24. initialize() {
  25. return new Promise(resolve => {
  26. IOModule = this.moduleManager.modules.io;
  27. SpotifyModule = this.moduleManager.modules.spotify;
  28. CacheModule = this.moduleManager.modules.cache;
  29. resolve();
  30. });
  31. }
  32. /**
  33. * Parses the cookie into a readable object
  34. *
  35. * @param {object} payload - object that contains the payload
  36. * @param {string} payload.cookieString - the cookie string
  37. * @returns {Promise} - returns promise (reject, resolve)
  38. */
  39. PARSE_COOKIES(payload) {
  40. return new Promise((resolve, reject) => {
  41. const cookies = {};
  42. if (typeof payload.cookieString !== "string") return reject(new Error("Cookie string is not a string"));
  43. // eslint-disable-next-line array-callback-return
  44. payload.cookieString.split("; ").map(cookie => {
  45. cookies[cookie.substring(0, cookie.indexOf("="))] = cookie.substring(
  46. cookie.indexOf("=") + 1,
  47. cookie.length
  48. );
  49. });
  50. return resolve(cookies);
  51. });
  52. }
  53. // COOKIES_TO_STRING() {//cookies
  54. // return new Promise((resolve, reject) => {
  55. // let newCookie = [];
  56. // for (let prop in cookie) {
  57. // newCookie.push(prop + "=" + cookie[prop]);
  58. // }
  59. // return newCookie.join("; ");
  60. // });
  61. // }
  62. /**
  63. * Removes a cookie by name
  64. *
  65. * @param {object} payload - object that contains the payload
  66. * @param {object} payload.cookieString - the cookie string
  67. * @param {string} payload.cookieName - the unique name of the cookie
  68. * @returns {Promise} - returns promise (reject, resolve)
  69. */
  70. REMOVE_COOKIE(payload) {
  71. return new Promise((resolve, reject) => {
  72. let cookies;
  73. try {
  74. cookies = UtilsModule.runJob(
  75. "PARSE_COOKIES",
  76. {
  77. cookieString: payload.cookieString
  78. },
  79. this
  80. );
  81. } catch (err) {
  82. return reject(err);
  83. }
  84. delete cookies[payload.cookieName];
  85. return resolve(this.toString(cookies));
  86. });
  87. }
  88. /**
  89. * Replaces any html reserved characters in a string with html entities
  90. *
  91. * @param {object} payload - object that contains the payload
  92. * @param {string} payload.str - the string to replace characters with html entities
  93. * @returns {Promise} - returns promise (reject, resolve)
  94. */
  95. HTML_ENTITIES(payload) {
  96. return new Promise(resolve => {
  97. resolve(
  98. String(payload.str)
  99. .replace(/&/g, "&")
  100. .replace(/</g, "&lt;")
  101. .replace(/>/g, "&gt;")
  102. .replace(/"/g, "&quot;")
  103. );
  104. });
  105. }
  106. /**
  107. * Generates a random string of a specified length
  108. *
  109. * @param {object} payload - object that contains the payload
  110. * @param {number} payload.length - the length the random string should be
  111. * @returns {Promise} - returns promise (reject, resolve)
  112. */
  113. async GENERATE_RANDOM_STRING(payload) {
  114. const chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789".split("");
  115. const promises = [];
  116. for (let i = 0; i < payload.length; i += 1) {
  117. this.log(
  118. "ERROR",
  119. "CHECK THIS!?!?!?!??!?!?!?!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"
  120. );
  121. promises.push(
  122. UtilsModule.runJob(
  123. "GET_RANDOM_NUMBER",
  124. {
  125. min: 0,
  126. max: chars.length - 1
  127. },
  128. this
  129. )
  130. );
  131. }
  132. const randomNums = await Promise.all(promises);
  133. const randomChars = [];
  134. for (let i = 0; i < payload.length; i += 1) {
  135. randomChars.push(chars[randomNums[i]]);
  136. }
  137. return new Promise(resolve => resolve(randomChars.join("")));
  138. }
  139. /**
  140. * Returns a socket object from a socket identifier
  141. *
  142. * @param {object} payload - object that contains the payload
  143. * @param {string} payload.socketId - the socket id
  144. * @returns {Promise} - returns promise (reject, resolve)
  145. */
  146. async GET_SOCKET_FROM_ID(payload) {
  147. // socketId
  148. const io = await IOModule.runJob("IO", {}, this);
  149. return new Promise(resolve => resolve(io.sockets.sockets[payload.socketId]));
  150. }
  151. /**
  152. * Creates a random number within a range
  153. *
  154. * @param {object} payload - object that contains the payload
  155. * @param {number} payload.min - the minimum number the result should be
  156. * @param {number} payload.max - the maximum number the result should be
  157. * @returns {Promise} - returns promise (reject, resolve)
  158. */
  159. GET_RANDOM_NUMBER(payload) {
  160. // min, max
  161. return new Promise(resolve =>
  162. resolve(Math.floor(Math.random() * (payload.max - payload.min + 1)) + payload.min)
  163. );
  164. }
  165. /**
  166. * Converts ISO8601 time format (YouTube API) to HH:MM:SS
  167. *
  168. * @param {object} payload - object contaiing the payload
  169. * @param {string} payload.duration - string in the format of ISO8601
  170. * @returns {Promise} - returns a promise (resolve, reject)
  171. */
  172. CONVERT_TIME(payload) {
  173. // duration
  174. return new Promise(resolve => {
  175. let { duration } = payload;
  176. let a = duration.match(/\d+/g);
  177. if (duration.indexOf("M") >= 0 && duration.indexOf("H") === -1 && duration.indexOf("S") === -1) {
  178. a = [0, a[0], 0];
  179. }
  180. if (duration.indexOf("H") >= 0 && duration.indexOf("M") === -1) {
  181. a = [a[0], 0, a[1]];
  182. }
  183. if (duration.indexOf("H") >= 0 && duration.indexOf("M") === -1 && duration.indexOf("S") === -1) {
  184. a = [a[0], 0, 0];
  185. }
  186. duration = 0;
  187. if (a.length === 3) {
  188. duration += parseInt(a[0]) * 3600;
  189. duration += parseInt(a[1]) * 60;
  190. duration += parseInt(a[2]);
  191. }
  192. if (a.length === 2) {
  193. duration += parseInt(a[0]) * 60;
  194. duration += parseInt(a[1]);
  195. }
  196. if (a.length === 1) {
  197. duration += parseInt(a[0]);
  198. }
  199. const hours = Math.floor(duration / 3600);
  200. const minutes = Math.floor((duration % 3600) / 60);
  201. const seconds = Math.floor((duration % 3600) % 60);
  202. resolve(
  203. (hours < 10 ? `0${hours}:` : `${hours}:`) +
  204. (minutes < 10 ? `0${minutes}:` : `${minutes}:`) +
  205. (seconds < 10 ? `0${seconds}` : seconds)
  206. );
  207. });
  208. }
  209. /**
  210. * Creates a random identifier for e.g. sessionId
  211. *
  212. * @returns {Promise} - returns promise (reject, resolve)
  213. */
  214. GUID() {
  215. return new Promise(resolve => {
  216. resolve(
  217. [1, 1, 0, 1, 0, 1, 0, 1, 0, 1, 1, 1]
  218. .map(b =>
  219. b
  220. ? Math.floor((1 + Math.random()) * 0x10000)
  221. .toString(16)
  222. .substring(1)
  223. : "-"
  224. )
  225. .join("")
  226. );
  227. });
  228. }
  229. // UNKNOWN
  230. // eslint-disable-next-line require-jsdoc
  231. async SOCKET_FROM_SESSION(payload) {
  232. // socketId
  233. const io = await IOModule.runJob("IO", {}, this);
  234. return new Promise((resolve, reject) => {
  235. const ns = io.of("/");
  236. if (ns) {
  237. return resolve(ns.connected[payload.socketId]);
  238. }
  239. return reject();
  240. });
  241. }
  242. /**
  243. * Gets all sockets for a specified session id
  244. *
  245. * @param {object} payload - object containing the payload
  246. * @param {string} payload.sessionId - user session id
  247. * @returns {Promise} - returns promise (reject, resolve)
  248. */
  249. async SOCKETS_FROM_SESSION_ID(payload) {
  250. const io = await IOModule.runJob("IO", {}, this);
  251. return new Promise(resolve => {
  252. const ns = io.of("/");
  253. const sockets = [];
  254. if (ns) {
  255. return async.each(
  256. Object.keys(ns.connected),
  257. (id, next) => {
  258. const { session } = ns.connected[id];
  259. if (session.sessionId === payload.sessionId) sockets.push(session.sessionId);
  260. next();
  261. },
  262. () => {
  263. resolve({ sockets });
  264. }
  265. );
  266. }
  267. return resolve();
  268. });
  269. }
  270. /**
  271. * Returns any sockets for a specific user
  272. *
  273. * @param {object} payload - object that contains the payload
  274. * @param {string} payload.userId - the user id
  275. * @returns {Promise} - returns promise (reject, resolve)
  276. */
  277. async SOCKETS_FROM_USER(payload) {
  278. const io = await IOModule.runJob("IO", {}, this);
  279. return new Promise((resolve, reject) => {
  280. const ns = io.of("/");
  281. const sockets = [];
  282. if (ns) {
  283. return async.each(
  284. Object.keys(ns.connected),
  285. (id, next) => {
  286. const { session } = ns.connected[id];
  287. CacheModule.runJob(
  288. "HGET",
  289. {
  290. table: "sessions",
  291. key: session.sessionId
  292. },
  293. this
  294. )
  295. .then(session => {
  296. if (session && session.userId === payload.userId) sockets.push(ns.connected[id]);
  297. next();
  298. })
  299. .catch(err => {
  300. next(err);
  301. });
  302. },
  303. err => {
  304. if (err) return reject(err);
  305. return resolve({ sockets });
  306. }
  307. );
  308. }
  309. return resolve();
  310. });
  311. }
  312. /**
  313. * Returns any sockets from a specific ip address
  314. *
  315. * @param {object} payload - object that contains the payload
  316. * @param {string} payload.ip - the ip address in question
  317. * @returns {Promise} - returns promise (reject, resolve)
  318. */
  319. async SOCKETS_FROM_IP(payload) {
  320. const io = await IOModule.runJob("IO", {}, this);
  321. return new Promise(resolve => {
  322. const ns = io.of("/");
  323. const sockets = [];
  324. if (ns) {
  325. return async.each(
  326. Object.keys(ns.connected),
  327. (id, next) => {
  328. const { session } = ns.connected[id];
  329. CacheModule.runJob(
  330. "HGET",
  331. {
  332. table: "sessions",
  333. key: session.sessionId
  334. },
  335. this
  336. )
  337. .then(session => {
  338. if (session && ns.connected[id].ip === payload.ip) sockets.push(ns.connected[id]);
  339. next();
  340. })
  341. .catch(() => next());
  342. },
  343. () => {
  344. resolve({ sockets });
  345. }
  346. );
  347. }
  348. return resolve();
  349. });
  350. }
  351. /**
  352. * Returns any sockets from a specific user without using redis/cache
  353. *
  354. * @param {object} payload - object that contains the payload
  355. * @param {string} payload.userId - the id of the user in question
  356. * @returns {Promise} - returns promise (reject, resolve)
  357. */
  358. async SOCKETS_FROM_USER_WITHOUT_CACHE(payload) {
  359. const io = await IOModule.runJob("IO", {}, this);
  360. return new Promise(resolve => {
  361. const ns = io.of("/");
  362. const sockets = [];
  363. if (ns) {
  364. return async.each(
  365. Object.keys(ns.connected),
  366. (id, next) => {
  367. const { session } = ns.connected[id];
  368. if (session.userId === payload.userId) sockets.push(ns.connected[id]);
  369. next();
  370. },
  371. () => {
  372. resolve({ sockets });
  373. }
  374. );
  375. }
  376. return resolve();
  377. });
  378. }
  379. /**
  380. * Allows a socket to leave any rooms they are connected to
  381. *
  382. * @param {object} payload - object that contains the payload
  383. * @param {string} payload.socketId - the id of the socket which should leave all their rooms
  384. * @returns {Promise} - returns promise (reject, resolve)
  385. */
  386. async SOCKET_LEAVE_ROOMS(payload) {
  387. const socket = await UtilsModule.runJob(
  388. "SOCKET_FROM_SESSION",
  389. {
  390. socketId: payload.socketId
  391. },
  392. this
  393. );
  394. return new Promise(resolve => {
  395. const { rooms } = socket;
  396. for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
  397. socket.leave(room);
  398. }
  399. return resolve();
  400. });
  401. }
  402. /**
  403. * Allows a socket to join a specified room
  404. *
  405. * @param {object} payload - object that contains the payload
  406. * @param {string} payload.socketId - the id of the socket which should join the room
  407. * @param {object} payload.room - the object representing the room the socket should join
  408. * @returns {Promise} - returns promise (reject, resolve)
  409. */
  410. async SOCKET_JOIN_ROOM(payload) {
  411. const socket = await UtilsModule.runJob(
  412. "SOCKET_FROM_SESSION",
  413. {
  414. socketId: payload.socketId
  415. },
  416. this
  417. );
  418. return new Promise(resolve => {
  419. const { rooms } = socket;
  420. for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
  421. socket.leave(room);
  422. }
  423. socket.join(payload.room);
  424. return resolve();
  425. });
  426. }
  427. // UNKNOWN
  428. // eslint-disable-next-line require-jsdoc
  429. async SOCKET_JOIN_SONG_ROOM(payload) {
  430. // socketId, room
  431. const socket = await UtilsModule.runJob(
  432. "SOCKET_FROM_SESSION",
  433. {
  434. socketId: payload.socketId
  435. },
  436. this
  437. );
  438. return new Promise(resolve => {
  439. const { rooms } = socket;
  440. for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
  441. if (room.indexOf("song.") !== -1) socket.leave(rooms);
  442. }
  443. socket.join(payload.room);
  444. return resolve();
  445. });
  446. }
  447. // UNKNOWN
  448. // eslint-disable-next-line require-jsdoc
  449. SOCKETS_JOIN_SONG_ROOM(payload) {
  450. // sockets, room
  451. return new Promise(resolve => {
  452. for (let id = 0, socketKeys = Object.keys(payload.sockets); id < socketKeys.length; id += 1) {
  453. const socket = payload.sockets[socketKeys[id]];
  454. const { rooms } = socket;
  455. for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
  456. if (room.indexOf("song.") !== -1) socket.leave(room);
  457. }
  458. socket.join(payload.room);
  459. }
  460. return resolve();
  461. });
  462. }
  463. // UNKNOWN
  464. // eslint-disable-next-line require-jsdoc
  465. SOCKETS_LEAVE_SONG_ROOMS(payload) {
  466. // sockets
  467. return new Promise(resolve => {
  468. for (let id = 0, socketKeys = Object.keys(payload.sockets); id < socketKeys.length; id += 1) {
  469. const socket = payload.sockets[socketKeys[id]];
  470. const { rooms } = socket;
  471. for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
  472. if (room.indexOf("song.") !== -1) socket.leave(room);
  473. }
  474. }
  475. resolve();
  476. });
  477. }
  478. /**
  479. * Emits arguments to any sockets that are in a specified a room
  480. *
  481. * @param {object} payload - object that contains the payload
  482. * @param {string} payload.room - the name of the room to emit arguments
  483. * @param {object} payload.args - any arguments to be emitted to the sockets in the specific room
  484. * @returns {Promise} - returns promise (reject, resolve)
  485. */
  486. async EMIT_TO_ROOM(payload) {
  487. const io = await IOModule.runJob("IO", {}, this);
  488. return new Promise(resolve => {
  489. const { sockets } = io.sockets;
  490. for (let id = 0, socketKeys = Object.keys(sockets); id < socketKeys.length; id += 1) {
  491. const socket = sockets[socketKeys[id]];
  492. if (socket.rooms[payload.room]) {
  493. socket.emit(...payload.args);
  494. }
  495. }
  496. return resolve();
  497. });
  498. }
  499. /**
  500. * Gets any sockets connected to a room
  501. *
  502. * @param {object} payload - object that contains the payload
  503. * @param {string} payload.room - the name of the room
  504. * @returns {Promise} - returns promise (reject, resolve)
  505. */
  506. async GET_ROOM_SOCKETS(payload) {
  507. const io = await IOModule.runJob("IO", {}, this);
  508. return new Promise(resolve => {
  509. const { sockets } = io.sockets;
  510. const roomSockets = [];
  511. for (let id = 0, socketKeys = Object.keys(sockets); id < socketKeys.length; id += 1) {
  512. const socket = sockets[socketKeys[id]];
  513. if (socket.rooms[payload.room]) roomSockets.push(socket);
  514. }
  515. return resolve(roomSockets);
  516. });
  517. }
  518. /**
  519. * Gets the details of a song using the YouTube API
  520. *
  521. * @param {object} payload - object that contains the payload
  522. * @param {string} payload.songId - the YouTube API id of the song
  523. * @returns {Promise} - returns promise (reject, resolve)
  524. */
  525. GET_SONG_FROM_YOUTUBE(payload) {
  526. // songId, cb
  527. return new Promise((resolve, reject) => {
  528. this.youtubeRequestCallbacks.push({
  529. cb: () => {
  530. this.youtubeRequestsActive = true;
  531. const youtubeParams = [
  532. "part=snippet,contentDetails,statistics,status",
  533. `id=${encodeURIComponent(payload.songId)}`,
  534. `key=${config.get("apis.youtube.key")}`
  535. ].join("&");
  536. request(`https://www.googleapis.com/youtube/v3/videos?${youtubeParams}`, (err, res, body) => {
  537. this.youtubeRequestCallbacks.splice(0, 1);
  538. if (this.youtubeRequestCallbacks.length > 0) {
  539. this.youtubeRequestCallbacks[0].cb(this.youtubeRequestCallbacks[0].songId);
  540. } else this.youtubeRequestsActive = false;
  541. if (err) {
  542. console.error(err);
  543. return null;
  544. }
  545. body = JSON.parse(body);
  546. if (body.error) {
  547. console.log("ERROR", "GET_SONG_FROM_YOUTUBE", `${body.error.message}`);
  548. return reject(new Error("An error has occured. Please try again later."));
  549. }
  550. if (body.items[0] === undefined)
  551. return reject(
  552. new Error("The specified video does not exist or cannot be publicly accessed.")
  553. );
  554. // TODO Clean up duration converter
  555. let dur = body.items[0].contentDetails.duration;
  556. dur = dur.replace("PT", "");
  557. let duration = 0;
  558. dur = dur.replace(/([\d]*)H/, (v, v2) => {
  559. v2 = Number(v2);
  560. duration = v2 * 60 * 60;
  561. return "";
  562. });
  563. dur = dur.replace(/([\d]*)M/, (v, v2) => {
  564. v2 = Number(v2);
  565. duration += v2 * 60;
  566. return "";
  567. });
  568. // eslint-disable-next-line no-unused-vars
  569. dur = dur.replace(/([\d]*)S/, (v, v2) => {
  570. v2 = Number(v2);
  571. duration += v2;
  572. return "";
  573. });
  574. const song = {
  575. songId: body.items[0].id,
  576. title: body.items[0].snippet.title,
  577. duration
  578. };
  579. return resolve({ song });
  580. });
  581. },
  582. songId: payload.songId
  583. });
  584. if (!this.youtubeRequestsActive) {
  585. this.youtubeRequestCallbacks[0].cb(this.youtubeRequestCallbacks[0].songId);
  586. }
  587. });
  588. }
  589. /**
  590. * Filters a list of YouTube videos so that they only contains videos with music
  591. *
  592. * @param {object} payload - object that contains the payload
  593. * @param {Array} payload.videoIds - an array of YouTube videoIds to filter through
  594. * @returns {Promise} - returns promise (reject, resolve)
  595. */
  596. FILTER_MUSIC_VIDEOS_YOUTUBE(payload) {
  597. // videoIds, cb
  598. return new Promise((resolve, reject) => {
  599. /**
  600. * @param {Function} cb2 - callback
  601. */
  602. function getNextPage(cb2) {
  603. const localVideoIds = payload.videoIds.splice(0, 50);
  604. const youtubeParams = [
  605. "part=topicDetails",
  606. `id=${encodeURIComponent(localVideoIds.join(","))}`,
  607. `maxResults=50`,
  608. `key=${config.get("apis.youtube.key")}`
  609. ].join("&");
  610. request(`https://www.googleapis.com/youtube/v3/videos?${youtubeParams}`, (err, res, body) => {
  611. if (err) {
  612. console.error(err);
  613. return reject(new Error("Failed to find playlist from YouTube"));
  614. }
  615. body = JSON.parse(body);
  616. if (body.error) {
  617. console.log("ERROR", "FILTER_MUSIC_VIDEOS_YOUTUBE", `${body.error.message}`);
  618. return reject(new Error("An error has occured. Please try again later."));
  619. }
  620. const songIds = [];
  621. body.items.forEach(item => {
  622. const songId = item.id;
  623. if (!item.topicDetails) return;
  624. if (item.topicDetails.relevantTopicIds.indexOf("/m/04rlf") !== -1) {
  625. songIds.push(songId);
  626. }
  627. });
  628. if (payload.videoIds.length > 0) {
  629. return getNextPage(newSongIds => {
  630. cb2(songIds.concat(newSongIds));
  631. });
  632. }
  633. return cb2(songIds);
  634. });
  635. }
  636. if (payload.videoIds.length === 0) resolve({ songIds: [] });
  637. else getNextPage(songIds => resolve({ songIds }));
  638. });
  639. }
  640. /**
  641. * Returns an array of songs taken from a YouTube playlist
  642. *
  643. * @param {object} payload - object that contains the payload
  644. * @param {boolean} payload.musicOnly - whether to return music videos or all videos in the playlist
  645. * @param {string} payload.url - the url of the YouTube playlist
  646. * @returns {Promise} - returns promise (reject, resolve)
  647. */
  648. GET_PLAYLIST_FROM_YOUTUBE(payload) {
  649. // payload includes: url, musicOnly
  650. return new Promise((resolve, reject) => {
  651. const local = this;
  652. const name = "list".replace(/[\\[]/, "\\[").replace(/[\]]/, "\\]");
  653. const regex = new RegExp(`[\\?&]${name}=([^&#]*)`);
  654. const splitQuery = regex.exec(payload.url);
  655. if (!splitQuery) {
  656. console.log("ERROR", "GET_PLAYLIST_FROM_YOUTUBE", "Invalid YouTube playlist URL query.");
  657. return reject(new Error("An error has occured. Please try again later."));
  658. }
  659. const playlistId = splitQuery[1];
  660. /**
  661. * @param {string} pageToken - page token for YouTube API
  662. * @param {Array} songs - array of songs
  663. */
  664. function getPage(pageToken, songs) {
  665. const nextPageToken = pageToken ? `pageToken=${pageToken}` : "";
  666. const youtubeParams = [
  667. "part=contentDetails",
  668. `playlistId=${encodeURIComponent(playlistId)}`,
  669. `maxResults=50`,
  670. `key=${config.get("apis.youtube.key")}`,
  671. nextPageToken
  672. ].join("&");
  673. request(
  674. `https://www.googleapis.com/youtube/v3/playlistItems?${youtubeParams}`,
  675. async (err, res, body) => {
  676. if (err) {
  677. console.error(err);
  678. return reject(new Error("Failed to find playlist from YouTube"));
  679. }
  680. body = JSON.parse(body);
  681. if (body.error) {
  682. console.log("ERROR", "GET_PLAYLIST_FROM_YOUTUBE", `${body.error.message}`);
  683. return reject(new Error("An error has occured. Please try again later."));
  684. }
  685. songs = songs.concat(body.items);
  686. if (body.nextPageToken) return getPage(body.nextPageToken, songs);
  687. songs = songs.map(song => song.contentDetails.videoId);
  688. if (!payload.musicOnly) return resolve({ songs });
  689. return local
  690. .runJob(
  691. "FILTER_MUSIC_VIDEOS_YOUTUBE",
  692. {
  693. videoIds: songs.slice()
  694. },
  695. this
  696. )
  697. .then(filteredSongs => {
  698. resolve({ filteredSongs, songs });
  699. });
  700. }
  701. );
  702. }
  703. return getPage(null, []);
  704. });
  705. }
  706. /**
  707. * Gets the details of a song from the Spotify API
  708. *
  709. * @param {object} payload - object that contains the payload
  710. * @param {object} payload.song - the song object (song.title etc.)
  711. * @returns {Promise} - returns promise (reject, resolve)
  712. */
  713. async GET_SONG_FROM_SPOTIFY(payload) {
  714. // song
  715. const token = await SpotifyModule.runJob("GET_TOKEN", {}, this);
  716. return new Promise((resolve, reject) => {
  717. if (!config.get("apis.spotify.enabled")) return reject(new Error("Spotify is not enabled."));
  718. const song = { ...payload.song };
  719. const spotifyParams = [`q=${encodeURIComponent(payload.song.title)}`, `type=track`].join("&");
  720. const options = {
  721. url: `https://api.spotify.com/v1/search?${spotifyParams}`,
  722. headers: {
  723. Authorization: `Bearer ${token}`
  724. }
  725. };
  726. return request(options, (err, res, body) => {
  727. if (err) console.error(err);
  728. body = JSON.parse(body);
  729. if (body.error) console.error(body.error);
  730. for (let i = 0, bodyKeys = Object.keys(body); i < bodyKeys.length; i += 1) {
  731. const { items } = body[i];
  732. for (let j = 0, itemKeys = Object.keys(body); j < itemKeys.length; j += 1) {
  733. const item = items[j];
  734. let hasArtist = false;
  735. for (let k = 0; k < item.artists.length; k += 1) {
  736. const artist = item.artists[k];
  737. if (song.title.indexOf(artist.name) !== -1) hasArtist = true;
  738. }
  739. if (hasArtist && song.title.indexOf(item.name) !== -1) {
  740. song.duration = item.duration_ms / 1000;
  741. song.artists = item.artists.map(artist => artist.name);
  742. song.title = item.name;
  743. song.explicit = item.explicit;
  744. song.thumbnail = item.album.images[1].url;
  745. break;
  746. }
  747. }
  748. }
  749. resolve({ song });
  750. });
  751. });
  752. }
  753. /**
  754. * Returns the details of multiple songs from the Spotify API
  755. *
  756. * @param {object} payload - object that contains the payload
  757. * @param {object} payload.title - the query/title of a song to search the API with
  758. * @returns {Promise} - returns promise (reject, resolve)
  759. */
  760. async GET_SONGS_FROM_SPOTIFY(payload) {
  761. // title, artist
  762. const token = await SpotifyModule.runJob("GET_TOKEN", {}, this);
  763. return new Promise((resolve, reject) => {
  764. if (!config.get("apis.spotify.enabled")) return reject(new Error("Spotify is not enabled."));
  765. const spotifyParams = [`q=${encodeURIComponent(payload.title)}`, `type=track`].join("&");
  766. const options = {
  767. url: `https://api.spotify.com/v1/search?${spotifyParams}`,
  768. headers: {
  769. Authorization: `Bearer ${token}`
  770. }
  771. };
  772. return request(options, (err, res, body) => {
  773. if (err) return console.error(err);
  774. body = JSON.parse(body);
  775. if (body.error) return console.error(body.error);
  776. const songs = [];
  777. for (let i = 0, bodyKeys = Object.keys(body); i < bodyKeys.length; i += 1) {
  778. const { items } = body[i];
  779. for (let j = 0, itemKeys = Object.keys(body); j < itemKeys.length; j += 1) {
  780. const item = items[j];
  781. let hasArtist = false;
  782. for (let k = 0; k < item.artists.length; k += 1) {
  783. const localArtist = item.artists[k];
  784. if (payload.artist.toLowerCase() === localArtist.name.toLowerCase()) hasArtist = true;
  785. }
  786. if (
  787. hasArtist &&
  788. (payload.title.indexOf(item.name) !== -1 || item.name.indexOf(payload.title) !== -1)
  789. ) {
  790. const song = {};
  791. song.duration = item.duration_ms / 1000;
  792. song.artists = item.artists.map(artist => artist.name);
  793. song.title = item.name;
  794. song.explicit = item.explicit;
  795. song.thumbnail = item.album.images[1].url;
  796. songs.push(song);
  797. }
  798. }
  799. }
  800. return resolve({ songs });
  801. });
  802. });
  803. }
  804. /**
  805. * Shuffles an array of songs
  806. *
  807. * @param {object} payload - object that contains the payload
  808. * @param {object} payload.array - an array of songs that should be shuffled
  809. * @returns {Promise} - returns promise (reject, resolve)
  810. */
  811. SHUFFLE(payload) {
  812. // array
  813. return new Promise(resolve => {
  814. const array = payload.array.slice();
  815. let currentIndex = payload.array.length;
  816. let temporaryValue;
  817. let randomIndex;
  818. // While there remain elements to shuffle...
  819. while (currentIndex !== 0) {
  820. // Pick a remaining element...
  821. randomIndex = Math.floor(Math.random() * currentIndex);
  822. currentIndex -= 1;
  823. // And swap it with the current element.
  824. temporaryValue = array[currentIndex];
  825. array[currentIndex] = array[randomIndex];
  826. array[randomIndex] = temporaryValue;
  827. }
  828. resolve({ array });
  829. });
  830. }
  831. /**
  832. * Creates an error
  833. *
  834. * @param {object} payload - object that contains the payload
  835. * @param {object} payload.error - object that contains the error
  836. * @param {string} payload.message - possible error message
  837. * @param {object} payload.errors - possible object that contains multiple errors
  838. * @returns {Promise} - returns promise (reject, resolve)
  839. */
  840. GET_ERROR(payload) {
  841. return new Promise(resolve => {
  842. let error = "An error occurred.";
  843. if (typeof payload.error === "string") error = payload.error;
  844. else if (payload.error.message) {
  845. if (payload.error.message !== "Validation failed") error = payload.error.message;
  846. else error = payload.error.errors[Object.keys(payload.error.errors)].message;
  847. }
  848. resolve(error);
  849. });
  850. }
  851. /**
  852. * Creates the gravatar url for a specified email address
  853. *
  854. * @param {object} payload - object that contains the payload
  855. * @param {string} payload.email - the email address
  856. * @returns {Promise} - returns promise (reject, resolve)
  857. */
  858. CREATE_GRAVATAR(payload) {
  859. return new Promise(resolve => {
  860. const hash = crypto.createHash("md5").update(payload.email).digest("hex");
  861. resolve(`https://www.gravatar.com/avatar/${hash}`);
  862. });
  863. }
  864. /**
  865. * @returns {Promise} - returns promise (reject, resolve)
  866. */
  867. DEBUG() {
  868. return new Promise(resolve => resolve());
  869. }
  870. }
  871. export default new _UtilsModule();