utils.js 26 KB

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