stations.js 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011
  1. 'use strict';
  2. const async = require('async'),
  3. request = require('request'),
  4. config = require('config'),
  5. _ = require('underscore')._;
  6. const io = require('../io');
  7. const db = require('../db');
  8. const cache = require('../cache');
  9. const notifications = require('../notifications');
  10. const utils = require('../utils');
  11. const logger = require('../logger');
  12. const stations = require('../stations');
  13. const songs = require('../songs');
  14. const hooks = require('./hooks');
  15. let userList = {};
  16. let usersPerStation = {};
  17. let usersPerStationCount = {};
  18. setInterval(() => {
  19. let stationsCountUpdated = [];
  20. let stationsUpdated = [];
  21. let oldUsersPerStation = usersPerStation;
  22. usersPerStation = {};
  23. let oldUsersPerStationCount = usersPerStationCount;
  24. usersPerStationCount = {};
  25. async.each(Object.keys(userList), function(socketId, next) {
  26. let socket = utils.socketFromSession(socketId);
  27. let stationId = userList[socketId];
  28. if (!socket || Object.keys(socket.rooms).indexOf(`station.${stationId}`) === -1) {
  29. if (stationsCountUpdated.indexOf(stationId) === -1) stationsCountUpdated.push(stationId);
  30. if (stationsUpdated.indexOf(stationId) === -1) stationsUpdated.push(stationId);
  31. delete userList[socketId];
  32. return next();
  33. }
  34. if (!usersPerStationCount[stationId]) usersPerStationCount[stationId] = 0;
  35. usersPerStationCount[stationId]++;
  36. if (!usersPerStation[stationId]) usersPerStation[stationId] = [];
  37. async.waterfall([
  38. (next) => {
  39. if (!socket.session || !socket.session.sessionId) return next('No session found.');
  40. cache.hget('sessions', socket.session.sessionId, next);
  41. },
  42. (session, next) => {
  43. if (!session) return next('Session not found.');
  44. db.models.user.findOne({_id: session.userId}, next);
  45. },
  46. (user, next) => {
  47. if (!user) return next('User not found.');
  48. if (usersPerStation[stationId].indexOf(user.username) !== -1) return next('User already in the list.');
  49. next(null, user.username);
  50. }
  51. ], (err, username) => {
  52. if (!err) {
  53. usersPerStation[stationId].push(username);
  54. }
  55. next();
  56. });
  57. //TODO Code to show users
  58. }, (err) => {
  59. for (let stationId in usersPerStationCount) {
  60. if (oldUsersPerStationCount[stationId] !== usersPerStationCount[stationId]) {
  61. if (stationsCountUpdated.indexOf(stationId) === -1) stationsCountUpdated.push(stationId);
  62. }
  63. }
  64. for (let stationId in usersPerStation) {
  65. if (_.difference(usersPerStation[stationId], oldUsersPerStation[stationId]).length > 0 || _.difference(oldUsersPerStation[stationId], usersPerStation[stationId]).length > 0) {
  66. if (stationsUpdated.indexOf(stationId) === -1) stationsUpdated.push(stationId);
  67. }
  68. }
  69. stationsCountUpdated.forEach((stationId) => {
  70. console.log("Updating count of ", stationId);
  71. cache.pub('station.updateUserCount', stationId);
  72. });
  73. stationsUpdated.forEach((stationId) => {
  74. console.log("Updating ", stationId);
  75. cache.pub('station.updateUsers', stationId);
  76. });
  77. //console.log("Userlist", usersPerStation);
  78. });
  79. }, 3000);
  80. cache.sub('station.updateUsers', stationId => {
  81. let list = usersPerStation[stationId] || [];
  82. utils.emitToRoom(`station.${stationId}`, "event:users.updated", list);
  83. });
  84. cache.sub('station.updateUserCount', stationId => {
  85. let count = usersPerStationCount[stationId] || 0;
  86. utils.emitToRoom(`station.${stationId}`, "event:userCount.updated", count);
  87. stations.getStation(stationId, (err, station) => {
  88. if (station.privacy === 'public') utils.emitToRoom('home', "event:userCount.updated", stationId, count);
  89. else {
  90. let sockets = utils.getRoomSockets('home');
  91. for (let socketId in sockets) {
  92. let socket = sockets[socketId];
  93. let session = sockets[socketId].session;
  94. if (session.sessionId) {
  95. cache.hget('sessions', session.sessionId, (err, session) => {
  96. if (!err && session) {
  97. db.models.user.findOne({_id: session.userId}, (err, user) => {
  98. if (user.role === 'admin') socket.emit("event:userCount.updated", stationId, count);
  99. else if (station.type === "community" && station.owner === session.userId) socket.emit("event:userCount.updated", stationId, count);
  100. });
  101. }
  102. });
  103. }
  104. }
  105. }
  106. })
  107. });
  108. cache.sub('station.updatePartyMode', data => {
  109. utils.emitToRoom(`station.${data.stationId}`, "event:partyMode.updated", data.partyMode);
  110. });
  111. cache.sub('privatePlaylist.selected', data => {
  112. utils.emitToRoom(`station.${data.stationId}`, "event:privatePlaylist.selected", data.playlistId);
  113. });
  114. cache.sub('station.pause', stationId => {
  115. utils.emitToRoom(`station.${stationId}`, "event:stations.pause");
  116. });
  117. cache.sub('station.resume', stationId => {
  118. stations.getStation(stationId, (err, station) => {
  119. utils.emitToRoom(`station.${stationId}`, "event:stations.resume", { timePaused: station.timePaused });
  120. });
  121. });
  122. cache.sub('station.queueUpdate', stationId => {
  123. stations.getStation(stationId, (err, station) => {
  124. if (!err) utils.emitToRoom(`station.${stationId}`, "event:queue.update", station.queue);
  125. });
  126. });
  127. cache.sub('station.voteSkipSong', stationId => {
  128. utils.emitToRoom(`station.${stationId}`, "event:song.voteSkipSong");
  129. });
  130. cache.sub('station.remove', stationId => {
  131. utils.emitToRoom(`station.${stationId}`, 'event:stations.remove');
  132. utils.emitToRoom('admin.stations', 'event:admin.station.removed', stationId);
  133. });
  134. cache.sub('station.create', stationId => {
  135. stations.initializeStation(stationId, (err, station) => {
  136. station.userCount = usersPerStationCount[stationId] || 0;
  137. if (err) console.error(err);
  138. utils.emitToRoom('admin.stations', 'event:admin.station.added', station);
  139. // TODO If community, check if on whitelist
  140. if (station.privacy === 'public') utils.emitToRoom('home', "event:stations.created", station);
  141. else {
  142. let sockets = utils.getRoomSockets('home');
  143. for (let socketId in sockets) {
  144. let socket = sockets[socketId];
  145. let session = sockets[socketId].session;
  146. if (session.sessionId) {
  147. cache.hget('sessions', session.sessionId, (err, session) => {
  148. if (!err && session) {
  149. db.models.user.findOne({_id: session.userId}, (err, user) => {
  150. if (user.role === 'admin') socket.emit("event:stations.created", station);
  151. else if (station.type === "community" && station.owner === session.userId) socket.emit("event:stations.created", station);
  152. });
  153. }
  154. });
  155. }
  156. }
  157. }
  158. });
  159. });
  160. module.exports = {
  161. /**
  162. * Get a list of all the stations
  163. *
  164. * @param session
  165. * @param cb
  166. * @return {{ status: String, stations: Array }}
  167. */
  168. index: (session, cb) => {
  169. async.waterfall([
  170. (next) => {
  171. cache.hgetall('stations', next);
  172. },
  173. (stations, next) => {
  174. let resultStations = [];
  175. for (let id in stations) {
  176. resultStations.push(stations[id]);
  177. }
  178. next(null, stations);
  179. },
  180. (stations, next) => {
  181. let resultStations = [];
  182. async.each(stations, (station, next) => {
  183. async.waterfall([
  184. (next) => {
  185. if (station.privacy === 'public') return next(true);
  186. if (!session.sessionId) return next(`Insufficient permissions.`);
  187. cache.hget('sessions', session.sessionId, next);
  188. },
  189. (session, next) => {
  190. if (!session) return next(`Insufficient permissions.`);
  191. db.models.user.findOne({_id: session.userId}, next);
  192. },
  193. (user, next) => {
  194. if (!user) return next(`Insufficient permissions.`);
  195. if (user.role === 'admin') return next(true);
  196. if (station.type === 'official') return next(`Insufficient permissions.`);
  197. if (station.owner === session.userId) return next(true);
  198. next(`Insufficient permissions.`);
  199. }
  200. ], (err) => {
  201. station.userCount = usersPerStationCount[station._id] || 0;
  202. if (err === true) resultStations.push(station);
  203. next();
  204. });
  205. }, () => {
  206. next(null, resultStations);
  207. });
  208. }
  209. ], (err, stations) => {
  210. if (err) {
  211. err = utils.getError(err);
  212. logger.error("STATIONS_INDEX", `Indexing stations failed. "${err}"`);
  213. return cb({'status': 'failure', 'message': err});
  214. }
  215. logger.success("STATIONS_INDEX", `Indexing stations successful.`);
  216. return cb({'status': 'success', 'stations': stations});
  217. });
  218. },
  219. /**
  220. * Finds a station by name
  221. *
  222. * @param session
  223. * @param stationName - the station name
  224. * @param cb
  225. */
  226. findByName: (session, stationName, cb) => {
  227. async.waterfall([
  228. (next) => {
  229. stations.getStationByName(stationName, next);
  230. },
  231. (station, next) => {
  232. if (!station) return next('Station not found.');
  233. next(null, station);
  234. }
  235. ], (err, station) => {
  236. if (err) {
  237. err = utils.getError(err);
  238. logger.error("STATIONS_FIND_BY_NAME", `Finding station "${stationName}" failed. "${err}"`);
  239. return cb({'status': 'failure', 'message': err});
  240. }
  241. logger.success("STATIONS_FIND_BY_NAME", `Found station "${stationName}" successfully.`);
  242. cb({status: 'success', data: station});
  243. });
  244. },
  245. /**
  246. * Gets the official playlist for a station
  247. *
  248. * @param session
  249. * @param stationId - the station id
  250. * @param cb
  251. */
  252. getPlaylist: (session, stationId, cb) => {
  253. async.waterfall([
  254. (next) => {
  255. stations.getStation(stationId, next);
  256. },
  257. (station, next) => {
  258. if (!station) return next('Station not found.');
  259. if (station.type !== 'official') return next('This is not an official station.');
  260. next();
  261. },
  262. (next) => {
  263. cache.hget("officialPlaylists", stationId, next);
  264. },
  265. (playlist, next) => {
  266. if (!playlist) return next('Playlist not found.');
  267. next(null, playlist);
  268. }
  269. ], (err, playlist) => {
  270. if (err) {
  271. err = utils.getError(err);
  272. logger.error("STATIONS_GET_PLAYLIST", `Getting playlist for station "${stationId}" failed. "${err}"`);
  273. return cb({'status': 'failure', 'message': err});
  274. }
  275. logger.success("STATIONS_GET_PLAYLIST", `Got playlist for station "${stationId}" successfully.`);
  276. cb({status: 'success', data: playlist.songs})
  277. });
  278. },
  279. /**
  280. * Joins the station by its name
  281. *
  282. * @param session
  283. * @param stationName - the station name
  284. * @param cb
  285. * @return {{ status: String, userCount: Integer }}
  286. */
  287. join: (session, stationName, cb) => {
  288. async.waterfall([
  289. (next) => {
  290. stations.getStationByName(stationName, next);
  291. },
  292. (station, next) => {
  293. if (!station) return next('Station not found.');
  294. async.waterfall([
  295. (next) => {
  296. if (station.privacy !== 'private') return next(true);
  297. if (!session.userId) return next('An error occurred while joining the station.');
  298. next();
  299. },
  300. (next) => {
  301. db.models.user.findOne({_id: session.userId}, next);
  302. },
  303. (user, next) => {
  304. if (!user) return next('An error occurred while joining the station.');
  305. if (user.role === 'admin') return next(true);
  306. if (station.type === 'official') return next('An error occurred while joining the station.');
  307. if (station.owner === session.userId) return next(true);
  308. next('An error occurred while joining the station.');
  309. }
  310. ], (err) => {
  311. if (err === true) return next(null, station);
  312. next(utils.getError(err));
  313. });
  314. },
  315. (station, next) => {
  316. utils.socketJoinRoom(session.socketId, `station.${station._id}`);
  317. let data = {
  318. _id: station._id,
  319. type: station.type,
  320. currentSong: station.currentSong,
  321. startedAt: station.startedAt,
  322. paused: station.paused,
  323. timePaused: station.timePaused,
  324. description: station.description,
  325. displayName: station.displayName,
  326. privacy: station.privacy,
  327. partyMode: station.partyMode,
  328. owner: station.owner,
  329. privatePlaylist: station.privatePlaylist
  330. };
  331. userList[session.socketId] = station._id;
  332. next(null, data);
  333. },
  334. (data, next) => {
  335. data.userCount = usersPerStationCount[data._id] || 0;
  336. data.users = usersPerStation[data._id] || [];
  337. if (!data.currentSong || !data.currentSong.title) return next(null, data);
  338. utils.socketJoinSongRoom(session.socketId, `song.${data.currentSong.songId}`);
  339. data.currentSong.skipVotes = data.currentSong.skipVotes.length;
  340. songs.getSong(data.currentSong.songId, (err, song) => {
  341. if (!err && song) {
  342. data.currentSong.likes = song.likes;
  343. data.currentSong.dislikes = song.dislikes;
  344. } else {
  345. data.currentSong.likes = -1;
  346. data.currentSong.dislikes = -1;
  347. }
  348. next(null, data);
  349. });
  350. }
  351. ], (err, data) => {
  352. if (err) {
  353. err = utils.getError(err);
  354. logger.error("STATIONS_JOIN", `Joining station "${stationName}" failed. "${err}"`);
  355. return cb({'status': 'failure', 'message': err});
  356. }
  357. logger.success("STATIONS_JOIN", `Joined station "${data._id}" successfully.`);
  358. cb({status: 'success', data});
  359. });
  360. },
  361. /**
  362. * Votes to skip a station
  363. *
  364. * @param session
  365. * @param stationId - the station id
  366. * @param cb
  367. * @param userId
  368. */
  369. voteSkip: hooks.loginRequired((session, stationId, cb, userId) => {
  370. async.waterfall([
  371. (next) => {
  372. stations.getStation(stationId, next);
  373. },
  374. (station, next) => {
  375. if (!station) return next('Station not found.');
  376. if (!station.currentSong) return next('There is currently no song to skip.');
  377. if (station.currentSong.skipVotes.indexOf(userId) !== -1) return next('You have already voted to skip this song.');
  378. next(null, station);
  379. },
  380. (station, next) => {
  381. db.models.station.update({_id: stationId}, {$push: {"currentSong.skipVotes": userId}}, next)
  382. },
  383. (res, next) => {
  384. stations.updateStation(stationId, next);
  385. },
  386. (station, next) => {
  387. if (!station) return next('Station not found.');
  388. next(null, station);
  389. }
  390. ], (err, station) => {
  391. if (err) {
  392. err = utils.getError(err);
  393. logger.error("STATIONS_VOTE_SKIP", `Vote skipping station "${stationId}" failed. "${err}"`);
  394. return cb({'status': 'failure', 'message': err});
  395. }
  396. logger.success("STATIONS_VOTE_SKIP", `Vote skipping "${stationId}" successful.`);
  397. cache.pub('station.voteSkipSong', stationId);
  398. if (station.currentSong && station.currentSong.skipVotes.length >= 3) stations.skipStation(stationId)();
  399. cb({ status: 'success', message: 'Successfully voted to skip the song.' });
  400. });
  401. }),
  402. /**
  403. * Force skips a station
  404. *
  405. * @param session
  406. * @param stationId - the station id
  407. * @param cb
  408. */
  409. forceSkip: hooks.ownerRequired((session, stationId, cb) => {
  410. async.waterfall([
  411. (next) => {
  412. stations.getStation(stationId, next);
  413. },
  414. (station, next) => {
  415. if (!station) return next('Station not found.');
  416. next();
  417. }
  418. ], (err) => {
  419. if (err) {
  420. err = utils.getError(err);
  421. logger.error("STATIONS_FORCE_SKIP", `Force skipping station "${stationId}" failed. "${err}"`);
  422. return cb({'status': 'failure', 'message': err});
  423. }
  424. notifications.unschedule(`stations.nextSong?id=${stationId}`);
  425. stations.skipStation(stationId)();
  426. logger.success("STATIONS_FORCE_SKIP", `Force skipped station "${stationId}" successfully.`);
  427. return cb({'status': 'success', 'message': 'Successfully skipped station.'});
  428. });
  429. }),
  430. /**
  431. * Leaves the user's current station
  432. *
  433. * @param session
  434. * @param stationId
  435. * @param cb
  436. * @return {{ status: String, userCount: Integer }}
  437. */
  438. leave: (session, stationId, cb) => {
  439. async.waterfall([
  440. (next) => {
  441. stations.getStation(stationId, next);
  442. },
  443. (station, next) => {
  444. if (!station) return next('Station not found.');
  445. next();
  446. },
  447. (next) => {
  448. cache.client.hincrby('station.userCounts', stationId, -1, next);
  449. }
  450. ], (err, userCount) => {
  451. if (err) {
  452. err = utils.getError(err);
  453. logger.error("STATIONS_LEAVE", `Leaving station "${stationId}" failed. "${err}"`);
  454. return cb({'status': 'failure', 'message': err});
  455. }
  456. logger.success("STATIONS_LEAVE", `Left station "${stationId}" successfully.`);
  457. utils.socketLeaveRooms(session);
  458. delete userList[session.socketId];
  459. return cb({'status': 'success', 'message': 'Successfully left station.', userCount});
  460. });
  461. },
  462. /**
  463. * Updates a station's name
  464. *
  465. * @param session
  466. * @param stationId - the station id
  467. * @param newName - the new station name
  468. * @param cb
  469. */
  470. updateName: hooks.ownerRequired((session, stationId, newName, cb) => {
  471. async.waterfall([
  472. (next) => {
  473. db.models.station.update({_id: stationId}, {$set: {name: newName}}, next);
  474. },
  475. (res, next) => {
  476. stations.updateStation(stationId, next);
  477. }
  478. ], (err) => {
  479. if (err) {
  480. err = utils.getError(err);
  481. logger.error("STATIONS_UPDATE_DISPLAY_NAME", `Updating station "${stationId}" displayName to "${newName}" failed. "${err}"`);
  482. return cb({'status': 'failure', 'message': err});
  483. }
  484. logger.success("STATIONS_UPDATE_DISPLAY_NAME", `Updated station "${stationId}" displayName to "${newName}" successfully.`);
  485. return cb({'status': 'success', 'message': 'Successfully updated the name.'});
  486. });
  487. }),
  488. /**
  489. * Updates a station's display name
  490. *
  491. * @param session
  492. * @param stationId - the station id
  493. * @param newDisplayName - the new station display name
  494. * @param cb
  495. */
  496. updateDisplayName: hooks.ownerRequired((session, stationId, newDisplayName, cb) => {
  497. async.waterfall([
  498. (next) => {
  499. db.models.station.update({_id: stationId}, {$set: {displayName: newDisplayName}}, next);
  500. },
  501. (res, next) => {
  502. stations.updateStation(stationId, next);
  503. }
  504. ], (err) => {
  505. if (err) {
  506. err = utils.getError(err);
  507. logger.error("STATIONS_UPDATE_DISPLAY_NAME", `Updating station "${stationId}" displayName to "${newDisplayName}" failed. "${err}"`);
  508. return cb({'status': 'failure', 'message': err});
  509. }
  510. logger.success("STATIONS_UPDATE_DISPLAY_NAME", `Updated station "${stationId}" displayName to "${newDisplayName}" successfully.`);
  511. return cb({'status': 'success', 'message': 'Successfully updated the display name.'});
  512. });
  513. }),
  514. /**
  515. * Updates a station's description
  516. *
  517. * @param session
  518. * @param stationId - the station id
  519. * @param newDescription - the new station description
  520. * @param cb
  521. */
  522. updateDescription: hooks.ownerRequired((session, stationId, newDescription, cb) => {
  523. async.waterfall([
  524. (next) => {
  525. db.models.station.update({_id: stationId}, {$set: {description: newDescription}}, next);
  526. },
  527. (res, next) => {
  528. stations.updateStation(stationId, next);
  529. }
  530. ], (err) => {
  531. if (err) {
  532. err = utils.getError(err);
  533. logger.error("STATIONS_UPDATE_DESCRIPTION", `Updating station "${stationId}" description to "${newDescription}" failed. "${err}"`);
  534. return cb({'status': 'failure', 'message': err});
  535. }
  536. logger.success("STATIONS_UPDATE_DESCRIPTION", `Updated station "${stationId}" description to "${newDescription}" successfully.`);
  537. return cb({'status': 'success', 'message': 'Successfully updated the description.'});
  538. });
  539. }),
  540. /**
  541. * Updates a station's privacy
  542. *
  543. * @param session
  544. * @param stationId - the station id
  545. * @param newPrivacy - the new station privacy
  546. * @param cb
  547. */
  548. updatePrivacy: hooks.ownerRequired((session, stationId, newPrivacy, cb) => {
  549. async.waterfall([
  550. (next) => {
  551. db.models.station.update({_id: stationId}, {$set: {privacy: newPrivacy}}, next);
  552. },
  553. (res, next) => {
  554. stations.updateStation(stationId, next);
  555. }
  556. ], (err) => {
  557. if (err) {
  558. err = utils.getError(err);
  559. logger.error("STATIONS_UPDATE_PRIVACY", `Updating station "${stationId}" privacy to "${newPrivacy}" failed. "${err}"`);
  560. return cb({'status': 'failure', 'message': err});
  561. }
  562. logger.success("STATIONS_UPDATE_PRIVACY", `Updated station "${stationId}" privacy to "${newPrivacy}" successfully.`);
  563. return cb({'status': 'success', 'message': 'Successfully updated the privacy.'});
  564. });
  565. }),
  566. /**
  567. * Updates a station's party mode
  568. *
  569. * @param session
  570. * @param stationId - the station id
  571. * @param newPartyMode - the new station party mode
  572. * @param cb
  573. */
  574. updatePartyMode: hooks.ownerRequired((session, stationId, newPartyMode, cb) => {
  575. async.waterfall([
  576. (next) => {
  577. stations.getStation(stationId, next);
  578. },
  579. (station, next) => {
  580. if (!station) return next('Station not found.');
  581. if (station.partyMode === newPartyMode) return next('The party mode was already ' + ((newPartyMode) ? 'enabled.' : 'disabled.'));
  582. db.models.station.update({_id: stationId}, {$set: {partyMode: newPartyMode}}, next);
  583. },
  584. (res, next) => {
  585. stations.updateStation(stationId, next);
  586. }
  587. ], (err) => {
  588. if (err) {
  589. err = utils.getError(err);
  590. logger.error("STATIONS_UPDATE_PARTY_MODE", `Updating station "${stationId}" party mode to "${newPartyMode}" failed. "${err}"`);
  591. return cb({'status': 'failure', 'message': err});
  592. }
  593. logger.success("STATIONS_UPDATE_PARTY_MODE", `Updated station "${stationId}" party mode to "${newPartyMode}" successfully.`);
  594. cache.pub('station.updatePartyMode', {stationId: stationId, partyMode: newPartyMode});
  595. stations.skipStation(stationId)();
  596. return cb({'status': 'success', 'message': 'Successfully updated the party mode.'});
  597. });
  598. }),
  599. /**
  600. * Pauses a station
  601. *
  602. * @param session
  603. * @param stationId - the station id
  604. * @param cb
  605. */
  606. pause: hooks.ownerRequired((session, stationId, cb) => {
  607. async.waterfall([
  608. (next) => {
  609. stations.getStation(stationId, next);
  610. },
  611. (station, next) => {
  612. if (!station) return next('Station not found.');
  613. if (station.paused) return next('That station was already paused.');
  614. db.models.station.update({_id: stationId}, {$set: {paused: true, pausedAt: Date.now()}}, next);
  615. },
  616. (res, next) => {
  617. stations.updateStation(stationId, next);
  618. }
  619. ], (err) => {
  620. if (err) {
  621. err = utils.getError(err);
  622. logger.error("STATIONS_PAUSE", `Pausing station "${stationId}" failed. "${err}"`);
  623. return cb({'status': 'failure', 'message': err});
  624. }
  625. logger.success("STATIONS_PAUSE", `Paused station "${stationId}" successfully.`);
  626. cache.pub('station.pause', stationId);
  627. notifications.unschedule(`stations.nextSong?id=${stationId}`);
  628. return cb({'status': 'success', 'message': 'Successfully paused.'});
  629. });
  630. }),
  631. /**
  632. * Resumes a station
  633. *
  634. * @param session
  635. * @param stationId - the station id
  636. * @param cb
  637. */
  638. resume: hooks.ownerRequired((session, stationId, cb) => {
  639. async.waterfall([
  640. (next) => {
  641. stations.getStation(stationId, next);
  642. },
  643. (station, next) => {
  644. if (!station) return next('Station not found.');
  645. if (!station.paused) return next('That station is not paused.');
  646. station.timePaused += (Date.now() - station.pausedAt);
  647. db.models.station.update({_id: stationId}, {$set: {paused: false}, $inc: {timePaused: Date.now() - station.pausedAt}}, next);
  648. },
  649. (next) => {
  650. stations.updateStation(stationId, next);
  651. }
  652. ], (err) => {
  653. if (err) {
  654. err = utils.getError(err);
  655. logger.error("STATIONS_RESUME", `Resuming station "${stationId}" failed. "${err}"`);
  656. return cb({'status': 'failure', 'message': err});
  657. }
  658. logger.success("STATIONS_RESUME", `Resuming station "${stationId}" successfully.`);
  659. cache.pub('station.resume', stationId);
  660. return cb({'status': 'success', 'message': 'Successfully resumed.'});
  661. });
  662. }),
  663. /**
  664. * Removes a station
  665. *
  666. * @param session
  667. * @param stationId - the station id
  668. * @param cb
  669. */
  670. remove: hooks.ownerRequired((session, stationId, cb) => {
  671. async.waterfall([
  672. (next) => {
  673. db.models.station.remove({ _id: stationId }, err => next(err));
  674. },
  675. (next) => {
  676. cache.hdel('stations', stationId, err => next(err));
  677. }
  678. ], (err) => {
  679. if (err) {
  680. err = utils.getError(err);
  681. logger.error("STATIONS_REMOVE", `Removing station "${stationId}" failed. "${err}"`);
  682. return cb({ 'status': 'failure', 'message': err });
  683. }
  684. logger.success("STATIONS_REMOVE", `Removing station "${stationId}" successfully.`);
  685. cache.pub('station.remove', stationId);
  686. return cb({ 'status': 'success', 'message': 'Successfully removed.' });
  687. });
  688. }),
  689. /**
  690. * Create a station
  691. *
  692. * @param session
  693. * @param data - the station data
  694. * @param cb
  695. * @param userId
  696. */
  697. create: hooks.loginRequired((session, data, cb, userId) => {
  698. console.log(data);
  699. data.name = data.name.toLowerCase();
  700. let blacklist = ["country", "edm", "musare", "hip-hop", "rap", "top-hits", "todays-hits", "old-school", "christmas", "about", "support", "staff", "help", "news", "terms", "privacy", "profile", "c", "community", "tos", "login", "register", "p", "official", "o", "trap", "faq", "team", "donate", "buy", "shop", "forums", "explore", "settings", "admin", "auth", "reset_password"];
  701. async.waterfall([
  702. (next) => {
  703. if (!data) return next('Invalid data.');
  704. next();
  705. },
  706. (next) => {
  707. db.models.station.findOne({ $or: [{name: data.name}, {displayName: new RegExp(`^${data.displayName}$`, 'i')}] }, next);
  708. },
  709. (station, next) => {
  710. if (station) return next('A station with that name or display name already exists.');
  711. const { name, displayName, description, genres, playlist, type, blacklistedGenres } = data;
  712. if (type === 'official') {
  713. db.models.user.findOne({_id: userId}, (err, user) => {
  714. if (err) return next(err);
  715. if (!user) return next('User not found.');
  716. if (user.role !== 'admin') return next('Admin required.');
  717. db.models.station.create({
  718. name,
  719. displayName,
  720. description,
  721. type,
  722. privacy: 'private',
  723. playlist,
  724. genres,
  725. blacklistedGenres,
  726. currentSong: stations.defaultSong
  727. }, next);
  728. });
  729. } else if (type === 'community') {
  730. if (blacklist.indexOf(name) !== -1) return next('That name is blacklisted. Please use a different name.');
  731. db.models.station.create({
  732. name,
  733. displayName,
  734. description,
  735. type,
  736. privacy: 'private',
  737. owner: userId,
  738. queue: [],
  739. currentSong: null
  740. }, next);
  741. }
  742. }
  743. ], (err, station) => {
  744. if (err) {
  745. console.log(err);
  746. err = utils.getError(err);
  747. logger.error("STATIONS_CREATE", `Creating station failed. "${err}"`);
  748. return cb({'status': 'failure', 'message': err});
  749. }
  750. logger.success("STATIONS_CREATE", `Created station "${station._id}" successfully.`);
  751. cache.pub('station.create', station._id);
  752. return cb({'status': 'success', 'message': 'Successfully created station.'});
  753. });
  754. }),
  755. /**
  756. * Adds song to station queue
  757. *
  758. * @param session
  759. * @param stationId - the station id
  760. * @param songId - the song id
  761. * @param cb
  762. * @param userId
  763. */
  764. addToQueue: hooks.loginRequired((session, stationId, songId, cb, userId) => {
  765. async.waterfall([
  766. (next) => {
  767. stations.getStation(stationId, next);
  768. },
  769. (station, next) => {
  770. if (!station) return next('Station not found.');
  771. if (station.type !== 'community') return next('That station is not a community station.');
  772. if (station.currentSong && station.currentSong.songId === songId) return next('That song is currently playing.');
  773. async.each(station.queue, (queueSong, next) => {
  774. if (queueSong.songId === songId) return next('That song is already in the queue.');
  775. next();
  776. }, (err) => {
  777. next(err, station);
  778. });
  779. },
  780. (station, next) => {
  781. songs.getSong(songId, (err, song) => {
  782. if (!err && song) return next(null, song);
  783. console.log(53, songId);
  784. utils.getSongFromYouTube(songId, (song) => {
  785. song.artists = [];
  786. song.skipDuration = 0;
  787. song.likes = -1;
  788. song.dislikes = -1;
  789. song.thumbnail = "empty";
  790. song.explicit = false;
  791. next(null, song);
  792. });
  793. });
  794. },
  795. (song, next) => {
  796. song.requestedBy = userId;
  797. db.models.station.update({_id: stationId}, {$push: {queue: song}}, next);
  798. },
  799. (res, next) => {
  800. stations.updateStation(stationId, next);
  801. }
  802. ], (err, station) => {
  803. if (err) {
  804. err = utils.getError(err);
  805. logger.error("STATIONS_ADD_SONG_TO_QUEUE", `Adding song "${songId}" to station "${stationId}" queue failed. "${err}"`);
  806. return cb({'status': 'failure', 'message': err});
  807. }
  808. logger.success("STATIONS_ADD_SONG_TO_QUEUE", `Added song "${songId}" to station "${stationId}" successfully.`);
  809. cache.pub('station.queueUpdate', stationId);
  810. return cb({'status': 'success', 'message': 'Successfully added song to queue.'});
  811. });
  812. }),
  813. /**
  814. * Removes song from station queue
  815. *
  816. * @param session
  817. * @param stationId - the station id
  818. * @param songId - the song id
  819. * @param cb
  820. * @param userId
  821. */
  822. removeFromQueue: hooks.ownerRequired((session, stationId, songId, cb, userId) => {
  823. async.waterfall([
  824. (next) => {
  825. if (!songId) return next('Invalid song id.');
  826. stations.getStation(stationId, next);
  827. },
  828. (station, next) => {
  829. if (!station) return next('Station not found.');
  830. if (station.type !== 'community') return next('Station is not a community station.');
  831. async.each(station.queue, (queueSong, next) => {
  832. if (queueSong.songId === songId) return next(true);
  833. next();
  834. }, (err) => {
  835. if (err === true) return next();
  836. next('Song is not currently in the queue.');
  837. });
  838. },
  839. (next) => {
  840. db.models.update({_id: stationId}, {$pull: {queue: {songId: songId}}}, next);
  841. },
  842. (next) => {
  843. stations.updateStation(stationId, next);
  844. }
  845. ], (err, station) => {
  846. if (err) {
  847. err = utils.getError(err);
  848. logger.error("STATIONS_REMOVE_SONG_TO_QUEUE", `Removing song "${songId}" from station "${stationId}" queue failed. "${err}"`);
  849. return cb({'status': 'failure', 'message': err});
  850. }
  851. logger.success("STATIONS_REMOVE_SONG_TO_QUEUE", `Removed song "${songId}" from station "${stationId}" successfully.`);
  852. cache.pub('station.queueUpdate', stationId);
  853. return cb({'status': 'success', 'message': 'Successfully removed song from queue.'});
  854. });
  855. }),
  856. /**
  857. * Gets the queue from a station
  858. *
  859. * @param session
  860. * @param stationId - the station id
  861. * @param cb
  862. */
  863. getQueue: hooks.adminRequired((session, stationId, cb) => {
  864. async.waterfall([
  865. (next) => {
  866. stations.getStation(stationId, next);
  867. },
  868. (station, next) => {
  869. if (!station) return next('Station not found.');
  870. if (station.type !== 'community') return next('Station is not a community station.');
  871. next(null, station);
  872. }
  873. ], (err, station) => {
  874. if (err) {
  875. err = utils.getError(err);
  876. logger.error("STATIONS_GET_QUEUE", `Getting queue for station "${stationId}" failed. "${err}"`);
  877. return cb({'status': 'failure', 'message': err});
  878. }
  879. logger.success("STATIONS_GET_QUEUE", `Got queue for station "${stationId}" successfully.`);
  880. return cb({'status': 'success', 'message': 'Successfully got queue.', queue: station.queue});
  881. });
  882. }),
  883. /**
  884. * Selects a private playlist for a station
  885. *
  886. * @param session
  887. * @param stationId - the station id
  888. * @param playlistId - the private playlist id
  889. * @param cb
  890. * @param userId
  891. */
  892. selectPrivatePlaylist: hooks.ownerRequired((session, stationId, playlistId, cb, userId) => {
  893. async.waterfall([
  894. (next) => {
  895. stations.getStation(stationId, next);
  896. },
  897. (station, next) => {
  898. if (!station) return next('Station not found.');
  899. if (station.type !== 'community') return next('Station is not a community station.');
  900. if (station.privatePlaylist === playlistId) return next('That private playlist is already selected.');
  901. db.models.playlist.findOne({_id: playlistId}, next);
  902. },
  903. (playlist, next) => {
  904. if (!playlist) return next('Playlist not found.');
  905. let currentSongIndex = (playlist.songs.length > 0) ? playlist.songs.length - 1 : 0;
  906. db.models.station.update({_id: stationId}, {$set: {privatePlaylist: playlistId, currentSongIndex: currentSongIndex}}, next);
  907. },
  908. (res, next) => {
  909. stations.updateStation(stationId, next);
  910. }
  911. ], (err, station) => {
  912. if (err) {
  913. err = utils.getError(err);
  914. logger.error("STATIONS_SELECT_PRIVATE_PLAYLIST", `Selecting private playlist "${playlistId}" for station "${stationId}" failed. "${err}"`);
  915. return cb({'status': 'failure', 'message': err});
  916. }
  917. logger.success("STATIONS_SELECT_PRIVATE_PLAYLIST", `Selected private playlist "${playlistId}" for station "${stationId}" successfully.`);
  918. if (!station.partyMode) stations.skipStation(stationId)();
  919. cache.pub('privatePlaylist.selected', {playlistId, stationId});
  920. return cb({'status': 'success', 'message': 'Successfully selected playlist.'});
  921. });
  922. }),
  923. };