stations.js 34 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070
  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.getSongFromId(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. utils.canUserBeInStation(station, userId, (canBe) => {
  377. if (canBe) return next(null, station);
  378. return next('Insufficient permissions.');
  379. });
  380. },
  381. (station, next) => {
  382. if (!station.currentSong) return next('There is currently no song to skip.');
  383. if (station.currentSong.skipVotes.indexOf(userId) !== -1) return next('You have already voted to skip this song.');
  384. next(null, station);
  385. },
  386. (station, next) => {
  387. db.models.station.update({_id: stationId}, {$push: {"currentSong.skipVotes": userId}}, next)
  388. },
  389. (res, next) => {
  390. stations.updateStation(stationId, next);
  391. },
  392. (station, next) => {
  393. if (!station) return next('Station not found.');
  394. next(null, station);
  395. }
  396. ], (err, station) => {
  397. if (err) {
  398. err = utils.getError(err);
  399. logger.error("STATIONS_VOTE_SKIP", `Vote skipping station "${stationId}" failed. "${err}"`);
  400. return cb({'status': 'failure', 'message': err});
  401. }
  402. logger.success("STATIONS_VOTE_SKIP", `Vote skipping "${stationId}" successful.`);
  403. cache.pub('station.voteSkipSong', stationId);
  404. if (station.currentSong && station.currentSong.skipVotes.length >= 3) stations.skipStation(stationId)();
  405. cb({ status: 'success', message: 'Successfully voted to skip the song.' });
  406. });
  407. }),
  408. /**
  409. * Force skips a station
  410. *
  411. * @param session
  412. * @param stationId - the station id
  413. * @param cb
  414. */
  415. forceSkip: hooks.ownerRequired((session, stationId, cb) => {
  416. async.waterfall([
  417. (next) => {
  418. stations.getStation(stationId, next);
  419. },
  420. (station, next) => {
  421. if (!station) return next('Station not found.');
  422. next();
  423. }
  424. ], (err) => {
  425. if (err) {
  426. err = utils.getError(err);
  427. logger.error("STATIONS_FORCE_SKIP", `Force skipping station "${stationId}" failed. "${err}"`);
  428. return cb({'status': 'failure', 'message': err});
  429. }
  430. notifications.unschedule(`stations.nextSong?id=${stationId}`);
  431. stations.skipStation(stationId)();
  432. logger.success("STATIONS_FORCE_SKIP", `Force skipped station "${stationId}" successfully.`);
  433. return cb({'status': 'success', 'message': 'Successfully skipped station.'});
  434. });
  435. }),
  436. /**
  437. * Leaves the user's current station
  438. *
  439. * @param session
  440. * @param stationId
  441. * @param cb
  442. * @return {{ status: String, userCount: Integer }}
  443. */
  444. leave: (session, stationId, cb) => {
  445. async.waterfall([
  446. (next) => {
  447. stations.getStation(stationId, next);
  448. },
  449. (station, next) => {
  450. if (!station) return next('Station not found.');
  451. next();
  452. }
  453. ], (err, userCount) => {
  454. if (err) {
  455. err = utils.getError(err);
  456. logger.error("STATIONS_LEAVE", `Leaving station "${stationId}" failed. "${err}"`);
  457. return cb({'status': 'failure', 'message': err});
  458. }
  459. logger.success("STATIONS_LEAVE", `Left station "${stationId}" successfully.`);
  460. utils.socketLeaveRooms(session);
  461. delete userList[session.socketId];
  462. return cb({'status': 'success', 'message': 'Successfully left station.', userCount});
  463. });
  464. },
  465. /**
  466. * Updates a station's name
  467. *
  468. * @param session
  469. * @param stationId - the station id
  470. * @param newName - the new station name
  471. * @param cb
  472. */
  473. updateName: hooks.ownerRequired((session, stationId, newName, cb) => {
  474. async.waterfall([
  475. (next) => {
  476. db.models.station.update({_id: stationId}, {$set: {name: newName}}, {runValidators: true}, next);
  477. },
  478. (res, next) => {
  479. stations.updateStation(stationId, next);
  480. }
  481. ], (err) => {
  482. if (err) {
  483. err = utils.getError(err);
  484. logger.error("STATIONS_UPDATE_DISPLAY_NAME", `Updating station "${stationId}" displayName to "${newName}" failed. "${err}"`);
  485. return cb({'status': 'failure', 'message': err});
  486. }
  487. logger.success("STATIONS_UPDATE_DISPLAY_NAME", `Updated station "${stationId}" displayName to "${newName}" successfully.`);
  488. return cb({'status': 'success', 'message': 'Successfully updated the name.'});
  489. });
  490. }),
  491. /**
  492. * Updates a station's display name
  493. *
  494. * @param session
  495. * @param stationId - the station id
  496. * @param newDisplayName - the new station display name
  497. * @param cb
  498. */
  499. updateDisplayName: hooks.ownerRequired((session, stationId, newDisplayName, cb) => {
  500. async.waterfall([
  501. (next) => {
  502. db.models.station.update({_id: stationId}, {$set: {displayName: newDisplayName}}, {runValidators: true}, next);
  503. },
  504. (res, next) => {
  505. stations.updateStation(stationId, next);
  506. }
  507. ], (err) => {
  508. if (err) {
  509. err = utils.getError(err);
  510. logger.error("STATIONS_UPDATE_DISPLAY_NAME", `Updating station "${stationId}" displayName to "${newDisplayName}" failed. "${err}"`);
  511. return cb({'status': 'failure', 'message': err});
  512. }
  513. logger.success("STATIONS_UPDATE_DISPLAY_NAME", `Updated station "${stationId}" displayName to "${newDisplayName}" successfully.`);
  514. return cb({'status': 'success', 'message': 'Successfully updated the display name.'});
  515. });
  516. }),
  517. /**
  518. * Updates a station's description
  519. *
  520. * @param session
  521. * @param stationId - the station id
  522. * @param newDescription - the new station description
  523. * @param cb
  524. */
  525. updateDescription: hooks.ownerRequired((session, stationId, newDescription, cb) => {
  526. async.waterfall([
  527. (next) => {
  528. db.models.station.update({_id: stationId}, {$set: {description: newDescription}}, {runValidators: true}, next);
  529. },
  530. (res, next) => {
  531. stations.updateStation(stationId, next);
  532. }
  533. ], (err) => {
  534. if (err) {
  535. err = utils.getError(err);
  536. logger.error("STATIONS_UPDATE_DESCRIPTION", `Updating station "${stationId}" description to "${newDescription}" failed. "${err}"`);
  537. return cb({'status': 'failure', 'message': err});
  538. }
  539. logger.success("STATIONS_UPDATE_DESCRIPTION", `Updated station "${stationId}" description to "${newDescription}" successfully.`);
  540. return cb({'status': 'success', 'message': 'Successfully updated the description.'});
  541. });
  542. }),
  543. /**
  544. * Updates a station's privacy
  545. *
  546. * @param session
  547. * @param stationId - the station id
  548. * @param newPrivacy - the new station privacy
  549. * @param cb
  550. */
  551. updatePrivacy: hooks.ownerRequired((session, stationId, newPrivacy, cb) => {
  552. async.waterfall([
  553. (next) => {
  554. db.models.station.update({_id: stationId}, {$set: {privacy: newPrivacy}}, {runValidators: true}, next);
  555. },
  556. (res, next) => {
  557. stations.updateStation(stationId, next);
  558. }
  559. ], (err) => {
  560. if (err) {
  561. err = utils.getError(err);
  562. logger.error("STATIONS_UPDATE_PRIVACY", `Updating station "${stationId}" privacy to "${newPrivacy}" failed. "${err}"`);
  563. return cb({'status': 'failure', 'message': err});
  564. }
  565. logger.success("STATIONS_UPDATE_PRIVACY", `Updated station "${stationId}" privacy to "${newPrivacy}" successfully.`);
  566. return cb({'status': 'success', 'message': 'Successfully updated the privacy.'});
  567. });
  568. }),
  569. /**
  570. * Updates a station's party mode
  571. *
  572. * @param session
  573. * @param stationId - the station id
  574. * @param newPartyMode - the new station party mode
  575. * @param cb
  576. */
  577. updatePartyMode: hooks.ownerRequired((session, stationId, newPartyMode, cb) => {
  578. async.waterfall([
  579. (next) => {
  580. stations.getStation(stationId, next);
  581. },
  582. (station, next) => {
  583. if (!station) return next('Station not found.');
  584. if (station.partyMode === newPartyMode) return next('The party mode was already ' + ((newPartyMode) ? 'enabled.' : 'disabled.'));
  585. db.models.station.update({_id: stationId}, {$set: {partyMode: newPartyMode}}, {runValidators: true}, next);
  586. },
  587. (res, next) => {
  588. stations.updateStation(stationId, next);
  589. }
  590. ], (err) => {
  591. if (err) {
  592. err = utils.getError(err);
  593. logger.error("STATIONS_UPDATE_PARTY_MODE", `Updating station "${stationId}" party mode to "${newPartyMode}" failed. "${err}"`);
  594. return cb({'status': 'failure', 'message': err});
  595. }
  596. logger.success("STATIONS_UPDATE_PARTY_MODE", `Updated station "${stationId}" party mode to "${newPartyMode}" successfully.`);
  597. cache.pub('station.updatePartyMode', {stationId: stationId, partyMode: newPartyMode});
  598. stations.skipStation(stationId)();
  599. return cb({'status': 'success', 'message': 'Successfully updated the party mode.'});
  600. });
  601. }),
  602. /**
  603. * Pauses a station
  604. *
  605. * @param session
  606. * @param stationId - the station id
  607. * @param cb
  608. */
  609. pause: hooks.ownerRequired((session, stationId, cb) => {
  610. async.waterfall([
  611. (next) => {
  612. stations.getStation(stationId, next);
  613. },
  614. (station, next) => {
  615. if (!station) return next('Station not found.');
  616. if (station.paused) return next('That station was already paused.');
  617. db.models.station.update({_id: stationId}, {$set: {paused: true, pausedAt: Date.now()}}, next);
  618. },
  619. (res, next) => {
  620. stations.updateStation(stationId, next);
  621. }
  622. ], (err) => {
  623. if (err) {
  624. err = utils.getError(err);
  625. logger.error("STATIONS_PAUSE", `Pausing station "${stationId}" failed. "${err}"`);
  626. return cb({'status': 'failure', 'message': err});
  627. }
  628. logger.success("STATIONS_PAUSE", `Paused station "${stationId}" successfully.`);
  629. cache.pub('station.pause', stationId);
  630. notifications.unschedule(`stations.nextSong?id=${stationId}`);
  631. return cb({'status': 'success', 'message': 'Successfully paused.'});
  632. });
  633. }),
  634. /**
  635. * Resumes a station
  636. *
  637. * @param session
  638. * @param stationId - the station id
  639. * @param cb
  640. */
  641. resume: hooks.ownerRequired((session, stationId, cb) => {
  642. async.waterfall([
  643. (next) => {
  644. stations.getStation(stationId, next);
  645. },
  646. (station, next) => {
  647. if (!station) return next('Station not found.');
  648. if (!station.paused) return next('That station is not paused.');
  649. station.timePaused += (Date.now() - station.pausedAt);
  650. db.models.station.update({_id: stationId}, {$set: {paused: false}, $inc: {timePaused: Date.now() - station.pausedAt}}, next);
  651. },
  652. (res, next) => {
  653. stations.updateStation(stationId, next);
  654. }
  655. ], (err) => {
  656. if (err) {
  657. err = utils.getError(err);
  658. logger.error("STATIONS_RESUME", `Resuming station "${stationId}" failed. "${err}"`);
  659. return cb({'status': 'failure', 'message': err});
  660. }
  661. logger.success("STATIONS_RESUME", `Resuming station "${stationId}" successfully.`);
  662. cache.pub('station.resume', stationId);
  663. return cb({'status': 'success', 'message': 'Successfully resumed.'});
  664. });
  665. }),
  666. /**
  667. * Removes a station
  668. *
  669. * @param session
  670. * @param stationId - the station id
  671. * @param cb
  672. */
  673. remove: hooks.ownerRequired((session, stationId, cb) => {
  674. async.waterfall([
  675. (next) => {
  676. db.models.station.remove({ _id: stationId }, err => next(err));
  677. },
  678. (next) => {
  679. cache.hdel('stations', stationId, err => next(err));
  680. }
  681. ], (err) => {
  682. if (err) {
  683. err = utils.getError(err);
  684. logger.error("STATIONS_REMOVE", `Removing station "${stationId}" failed. "${err}"`);
  685. return cb({ 'status': 'failure', 'message': err });
  686. }
  687. logger.success("STATIONS_REMOVE", `Removing station "${stationId}" successfully.`);
  688. cache.pub('station.remove', stationId);
  689. return cb({ 'status': 'success', 'message': 'Successfully removed.' });
  690. });
  691. }),
  692. /**
  693. * Create a station
  694. *
  695. * @param session
  696. * @param data - the station data
  697. * @param cb
  698. * @param userId
  699. */
  700. create: hooks.loginRequired((session, data, cb, userId) => {
  701. data.name = data.name.toLowerCase();
  702. 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"];
  703. async.waterfall([
  704. (next) => {
  705. if (!data) return next('Invalid data.');
  706. next();
  707. },
  708. (next) => {
  709. db.models.station.findOne({ $or: [{name: data.name}, {displayName: new RegExp(`^${data.displayName}$`, 'i')}] }, next);
  710. },
  711. (station, next) => {
  712. if (station) return next('A station with that name or display name already exists.');
  713. const { name, displayName, description, genres, playlist, type, blacklistedGenres } = data;
  714. if (type === 'official') {
  715. db.models.user.findOne({_id: userId}, (err, user) => {
  716. if (err) return next(err);
  717. if (!user) return next('User not found.');
  718. if (user.role !== 'admin') return next('Admin required.');
  719. db.models.station.create({
  720. name,
  721. displayName,
  722. description,
  723. type,
  724. privacy: 'private',
  725. playlist,
  726. genres,
  727. blacklistedGenres,
  728. currentSong: stations.defaultSong
  729. }, next);
  730. });
  731. } else if (type === 'community') {
  732. if (blacklist.indexOf(name) !== -1) return next('That name is blacklisted. Please use a different name.');
  733. db.models.station.create({
  734. name,
  735. displayName,
  736. description,
  737. type,
  738. privacy: 'private',
  739. owner: userId,
  740. queue: [],
  741. currentSong: null
  742. }, next);
  743. }
  744. }
  745. ], (err, station) => {
  746. if (err) {
  747. err = utils.getError(err);
  748. logger.error("STATIONS_CREATE", `Creating station failed. "${err}"`);
  749. return cb({'status': 'failure', 'message': err});
  750. }
  751. logger.success("STATIONS_CREATE", `Created station "${station._id}" successfully.`);
  752. cache.pub('station.create', station._id);
  753. return cb({'status': 'success', 'message': 'Successfully created station.'});
  754. });
  755. }),
  756. /**
  757. * Adds song to station queue
  758. *
  759. * @param session
  760. * @param stationId - the station id
  761. * @param songId - the song id
  762. * @param cb
  763. * @param userId
  764. */
  765. addToQueue: hooks.loginRequired((session, stationId, songId, cb, userId) => {
  766. async.waterfall([
  767. (next) => {
  768. stations.getStation(stationId, next);
  769. },
  770. (station, next) => {
  771. if (!station) return next('Station not found.');
  772. if (station.type !== 'community') return next('That station is not a community station.');
  773. utils.canUserBeInStation(station, userId, (canBe) => {
  774. if (canBe) return next(null, station);
  775. return next('Insufficient permissions.');
  776. });
  777. },
  778. (station, next) => {
  779. if (station.currentSong && station.currentSong.songId === songId) return next('That song is currently playing.');
  780. async.each(station.queue, (queueSong, next) => {
  781. if (queueSong.songId === songId) return next('That song is already in the queue.');
  782. next();
  783. }, (err) => {
  784. next(err, station);
  785. });
  786. },
  787. (station, next) => {
  788. songs.getSong(songId, (err, song) => {
  789. if (!err && song) return next(null, song, station);
  790. utils.getSongFromYouTube(songId, (song) => {
  791. song.artists = [];
  792. song.skipDuration = 0;
  793. song.likes = -1;
  794. song.dislikes = -1;
  795. song.thumbnail = "empty";
  796. song.explicit = false;
  797. next(null, song, station);
  798. });
  799. });
  800. },
  801. (song, station, next) => {
  802. let queue = station.queue;
  803. song.requestedBy = userId;
  804. queue.push(song);
  805. let totalDuration = 0;
  806. queue.forEach((song) => {
  807. totalDuration += song.duration;
  808. });
  809. if (totalDuration >= 3600 * 3) return next('The max length of the queue is 3 hours.');
  810. next(null, song, station);
  811. },
  812. (song, station, next) => {
  813. let queue = station.queue;
  814. if (queue.length === 0) return next(null, song, station);
  815. let totalDuration = 0;
  816. const userId = queue[queue.length - 1].requestedBy;
  817. station.queue.forEach((song) => {
  818. if (userId === song.requestedBy) {
  819. totalDuration += song.duration;
  820. }
  821. });
  822. if(totalDuration >= 900) return next('The max length of songs per user is 15 minutes.');
  823. next(null, song, station);
  824. },
  825. (song, station, next) => {
  826. let queue = station.queue;
  827. if (queue.length === 0) return next(null, song);
  828. let totalSongs = 0;
  829. const userId = queue[queue.length - 1].requestedBy;
  830. queue.forEach((song) => {
  831. if (userId === song.requestedBy) {
  832. totalSongs++;
  833. }
  834. });
  835. if (totalSongs <= 2) return next(null, song);
  836. if (totalSongs > 3) return next('The max amount of songs per user is 3, and only 2 in a row is allowed.');
  837. if (queue[queue.length - 2].requestedBy !== userId || queue[queue.length - 3] !== userId) return next('The max amount of songs per user is 3, and only 2 in a row is allowed.');
  838. next(null, song);
  839. },
  840. (song, next) => {
  841. db.models.station.update({_id: stationId}, {$push: {queue: song}}, {runValidators: true}, next);
  842. },
  843. (res, next) => {
  844. stations.updateStation(stationId, next);
  845. }
  846. ], (err, station) => {
  847. if (err) {
  848. err = utils.getError(err);
  849. logger.error("STATIONS_ADD_SONG_TO_QUEUE", `Adding song "${songId}" to station "${stationId}" queue failed. "${err}"`);
  850. return cb({'status': 'failure', 'message': err});
  851. }
  852. logger.success("STATIONS_ADD_SONG_TO_QUEUE", `Added song "${songId}" to station "${stationId}" successfully.`);
  853. cache.pub('station.queueUpdate', stationId);
  854. return cb({'status': 'success', 'message': 'Successfully added song to queue.'});
  855. });
  856. }),
  857. /**
  858. * Removes song from station queue
  859. *
  860. * @param session
  861. * @param stationId - the station id
  862. * @param songId - the song id
  863. * @param cb
  864. * @param userId
  865. */
  866. removeFromQueue: hooks.ownerRequired((session, stationId, songId, cb, userId) => {
  867. async.waterfall([
  868. (next) => {
  869. if (!songId) return next('Invalid song id.');
  870. stations.getStation(stationId, next);
  871. },
  872. (station, next) => {
  873. if (!station) return next('Station not found.');
  874. if (station.type !== 'community') return next('Station is not a community station.');
  875. async.each(station.queue, (queueSong, next) => {
  876. if (queueSong.songId === songId) return next(true);
  877. next();
  878. }, (err) => {
  879. if (err === true) return next();
  880. next('Song is not currently in the queue.');
  881. });
  882. },
  883. (next) => {
  884. db.models.update({_id: stationId}, {$pull: {queue: {songId: songId}}}, next);
  885. },
  886. (next) => {
  887. stations.updateStation(stationId, next);
  888. }
  889. ], (err, station) => {
  890. if (err) {
  891. err = utils.getError(err);
  892. logger.error("STATIONS_REMOVE_SONG_TO_QUEUE", `Removing song "${songId}" from station "${stationId}" queue failed. "${err}"`);
  893. return cb({'status': 'failure', 'message': err});
  894. }
  895. logger.success("STATIONS_REMOVE_SONG_TO_QUEUE", `Removed song "${songId}" from station "${stationId}" successfully.`);
  896. cache.pub('station.queueUpdate', stationId);
  897. return cb({'status': 'success', 'message': 'Successfully removed song from queue.'});
  898. });
  899. }),
  900. /**
  901. * Gets the queue from a station
  902. *
  903. * @param session
  904. * @param stationId - the station id
  905. * @param cb
  906. */
  907. getQueue: (session, stationId, cb) => {
  908. async.waterfall([
  909. (next) => {
  910. stations.getStation(stationId, next);
  911. },
  912. (station, next) => {
  913. if (!station) return next('Station not found.');
  914. if (station.type !== 'community') return next('Station is not a community station.');
  915. next(null, station);
  916. },
  917. (station, next) => {
  918. utils.canUserBeInStation(station, session.userId, (canBe) => {
  919. if (canBe) return next(null, station);
  920. return next('Insufficient permissions.');
  921. });
  922. }
  923. ], (err, station) => {
  924. if (err) {
  925. err = utils.getError(err);
  926. logger.error("STATIONS_GET_QUEUE", `Getting queue for station "${stationId}" failed. "${err}"`);
  927. return cb({'status': 'failure', 'message': err});
  928. }
  929. logger.success("STATIONS_GET_QUEUE", `Got queue for station "${stationId}" successfully.`);
  930. return cb({'status': 'success', 'message': 'Successfully got queue.', queue: station.queue});
  931. });
  932. },
  933. /**
  934. * Selects a private playlist for a station
  935. *
  936. * @param session
  937. * @param stationId - the station id
  938. * @param playlistId - the private playlist id
  939. * @param cb
  940. * @param userId
  941. */
  942. selectPrivatePlaylist: hooks.ownerRequired((session, stationId, playlistId, cb, userId) => {
  943. async.waterfall([
  944. (next) => {
  945. stations.getStation(stationId, next);
  946. },
  947. (station, next) => {
  948. if (!station) return next('Station not found.');
  949. if (station.type !== 'community') return next('Station is not a community station.');
  950. if (station.privatePlaylist === playlistId) return next('That private playlist is already selected.');
  951. db.models.playlist.findOne({_id: playlistId}, next);
  952. },
  953. (playlist, next) => {
  954. if (!playlist) return next('Playlist not found.');
  955. let currentSongIndex = (playlist.songs.length > 0) ? playlist.songs.length - 1 : 0;
  956. db.models.station.update({_id: stationId}, {$set: {privatePlaylist: playlistId, currentSongIndex: currentSongIndex}}, {runValidators: true}, next);
  957. },
  958. (res, next) => {
  959. stations.updateStation(stationId, next);
  960. }
  961. ], (err, station) => {
  962. if (err) {
  963. err = utils.getError(err);
  964. logger.error("STATIONS_SELECT_PRIVATE_PLAYLIST", `Selecting private playlist "${playlistId}" for station "${stationId}" failed. "${err}"`);
  965. return cb({'status': 'failure', 'message': err});
  966. }
  967. logger.success("STATIONS_SELECT_PRIVATE_PLAYLIST", `Selected private playlist "${playlistId}" for station "${stationId}" successfully.`);
  968. notifications.unschedule(`stations.nextSong?id${stationId}`);
  969. if (!station.partyMode) stations.skipStation(stationId)();
  970. cache.pub('privatePlaylist.selected', {playlistId, stationId});
  971. return cb({'status': 'success', 'message': 'Successfully selected playlist.'});
  972. });
  973. }),
  974. };