ws.js 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773
  1. /**
  2. * @file
  3. */
  4. import config from "config";
  5. import async from "async";
  6. import WebSocket from "ws";
  7. import { EventEmitter } from "events";
  8. import CoreClass from "../core";
  9. let WSModule;
  10. let AppModule;
  11. let CacheModule;
  12. let UtilsModule;
  13. let DBModule;
  14. let PunishmentsModule;
  15. class _WSModule extends CoreClass {
  16. // eslint-disable-next-line require-jsdoc
  17. constructor() {
  18. super("ws");
  19. WSModule = this;
  20. }
  21. /**
  22. * Initialises the ws module
  23. *
  24. * @returns {Promise} - returns promise (reject, resolve)
  25. */
  26. async initialize() {
  27. this.setStage(1);
  28. AppModule = this.moduleManager.modules.app;
  29. CacheModule = this.moduleManager.modules.cache;
  30. UtilsModule = this.moduleManager.modules.utils;
  31. DBModule = this.moduleManager.modules.db;
  32. PunishmentsModule = this.moduleManager.modules.punishments;
  33. this.actions = (await import("./actions")).default;
  34. this.userModel = await DBModule.runJob("GET_MODEL", { modelName: "user" });
  35. this.setStage(2);
  36. this.SIDname = config.get("cookie.SIDname");
  37. // TODO: Check every 30s/, for all sockets, if they are still allowed to be in the rooms they are in, and on socket at all (permission changing/banning)
  38. const server = await AppModule.runJob("SERVER");
  39. // this._io.origins(config.get("cors.origin"));
  40. this._io = new WebSocket.Server({ server, path: "/ws" });
  41. this.rooms = {};
  42. return new Promise(resolve => {
  43. this.setStage(3);
  44. this._io.on("connection", async (socket, req) => {
  45. socket.dispatch = (...args) => socket.send(JSON.stringify(args));
  46. socket.actions = new EventEmitter();
  47. socket.actions.setMaxListeners(0);
  48. socket.listen = (target, cb) => socket.actions.addListener(target, args => cb(args));
  49. WSModule.runJob("HANDLE_WS_USE", { socket, req }).then(socket =>
  50. WSModule.runJob("HANDLE_WS_CONNECTION", { socket })
  51. );
  52. socket.isAlive = true;
  53. socket.on("pong", function heartbeat() {
  54. this.isAlive = true;
  55. });
  56. });
  57. const keepAliveInterval = setInterval(() => {
  58. this._io.clients.forEach(socket => {
  59. if (socket.isAlive === false) return socket.terminate();
  60. socket.isAlive = false;
  61. return socket.ping(() => {});
  62. });
  63. }, 45000);
  64. this._io.on("close", () => clearInterval(keepAliveInterval));
  65. this.setStage(4);
  66. return resolve();
  67. });
  68. }
  69. /**
  70. * Returns the websockets variable
  71. *
  72. * @returns {Promise} - returns a promise (resolve, reject)
  73. */
  74. WS() {
  75. return new Promise(resolve => resolve(WSModule._io));
  76. }
  77. /**
  78. * Obtains socket object for a specified socket id
  79. *
  80. * @param {object} payload - object containing the payload
  81. * @param {string} payload.socketId - the id of the socket
  82. * @returns {Promise} - returns promise (reject, resolve)
  83. */
  84. async SOCKET_FROM_SOCKET_ID(payload) {
  85. return new Promise(resolve => {
  86. const { clients } = WSModule._io;
  87. if (clients)
  88. // eslint-disable-next-line consistent-return
  89. clients.forEach(socket => {
  90. if (socket.session.socketId === payload.socketId) return resolve(socket);
  91. });
  92. // socket doesn't exist
  93. return resolve();
  94. });
  95. }
  96. /**
  97. * Gets all sockets for a specified session id
  98. *
  99. * @param {object} payload - object containing the payload
  100. * @param {string} payload.sessionId - user session id
  101. * @returns {Promise} - returns promise (reject, resolve)
  102. */
  103. async SOCKETS_FROM_SESSION_ID(payload) {
  104. return new Promise(resolve => {
  105. const { clients } = WSModule._io;
  106. const sockets = [];
  107. if (clients) {
  108. return async.each(
  109. Object.keys(clients),
  110. (id, next) => {
  111. const { session } = clients[id];
  112. if (session.sessionId === payload.sessionId) sockets.push(session.sessionId);
  113. next();
  114. },
  115. () => resolve(sockets)
  116. );
  117. }
  118. return resolve();
  119. });
  120. }
  121. /**
  122. * Returns any sockets for a specific user
  123. *
  124. * @param {object} payload - object that contains the payload
  125. * @param {string} payload.userId - the user id
  126. * @returns {Promise} - returns promise (reject, resolve)
  127. */
  128. async SOCKETS_FROM_USER(payload) {
  129. return new Promise((resolve, reject) => {
  130. const sockets = [];
  131. return async.eachLimit(
  132. WSModule._io.clients,
  133. 1,
  134. (socket, next) => {
  135. const { sessionId } = socket.session;
  136. if (sessionId) {
  137. return CacheModule.runJob("HGET", { table: "sessions", key: sessionId }, this)
  138. .then(session => {
  139. if (session && session.userId === payload.userId) sockets.push(socket);
  140. next();
  141. })
  142. .catch(err => next(err));
  143. }
  144. return next();
  145. },
  146. err => {
  147. if (err) return reject(err);
  148. return resolve(sockets);
  149. }
  150. );
  151. });
  152. }
  153. /**
  154. * Returns any sockets from a specific ip address
  155. *
  156. * @param {object} payload - object that contains the payload
  157. * @param {string} payload.ip - the ip address in question
  158. * @returns {Promise} - returns promise (reject, resolve)
  159. */
  160. async SOCKETS_FROM_IP(payload) {
  161. return new Promise(resolve => {
  162. const { clients } = WSModule._io;
  163. const sockets = [];
  164. return async.each(
  165. Object.keys(clients),
  166. (id, next) => {
  167. const { session } = clients[id];
  168. CacheModule.runJob("HGET", { table: "sessions", key: session.sessionId }, this)
  169. .then(session => {
  170. if (session && clients[id].ip === payload.ip) sockets.push(clients[id]);
  171. next();
  172. })
  173. .catch(() => next());
  174. },
  175. () => resolve(sockets)
  176. );
  177. });
  178. }
  179. /**
  180. * Returns any sockets from a specific user without using redis/cache
  181. *
  182. * @param {object} payload - object that contains the payload
  183. * @param {string} payload.userId - the id of the user in question
  184. * @returns {Promise} - returns promise (reject, resolve)
  185. */
  186. async SOCKETS_FROM_USER_WITHOUT_CACHE(payload) {
  187. return new Promise(resolve => {
  188. const { clients } = WSModule._io;
  189. const sockets = [];
  190. if (clients) {
  191. return async.each(
  192. Object.keys(clients),
  193. (id, next) => {
  194. const { session } = clients[id];
  195. if (session.userId === payload.userId) sockets.push(clients[id]);
  196. next();
  197. },
  198. () => resolve(sockets)
  199. );
  200. }
  201. return resolve();
  202. });
  203. }
  204. /**
  205. * Allows a socket to leave any rooms they are connected to
  206. *
  207. * @param {object} payload - object that contains the payload
  208. * @param {string} payload.socketId - the id of the socket which should leave all their rooms
  209. * @returns {Promise} - returns promise (reject, resolve)
  210. */
  211. async SOCKET_LEAVE_ROOMS(payload) {
  212. return new Promise(resolve => {
  213. // filter out rooms that the user is in
  214. Object.keys(WSModule.rooms).forEach(room => {
  215. WSModule.rooms[room] = WSModule.rooms[room].filter(participant => participant !== payload.socketId);
  216. });
  217. return resolve();
  218. });
  219. }
  220. /**
  221. * Allows a socket to leave a specific room they are connected to
  222. *
  223. * @param {object} payload - object that contains the payload
  224. * @param {string} payload.socketId - the id of the socket which should leave a room
  225. * @param {string} payload.room - the room
  226. * @returns {Promise} - returns promise (reject, resolve)
  227. */
  228. async SOCKET_LEAVE_ROOM(payload) {
  229. return new Promise(resolve => {
  230. // filter out rooms that the user is in
  231. if (WSModule.rooms[payload.room])
  232. WSModule.rooms[payload.room] = WSModule.rooms[payload.room].filter(
  233. participant => participant !== payload.socketId
  234. );
  235. return resolve();
  236. });
  237. }
  238. /**
  239. * Allows a socket to join a specified room (this will remove them from any rooms they are currently in)
  240. *
  241. * @param {object} payload - object that contains the payload
  242. * @param {string} payload.socketId - the id of the socket which should join the room
  243. * @param {string} payload.room - the name of the room
  244. * @returns {Promise} - returns promise (reject, resolve)
  245. */
  246. async SOCKET_JOIN_ROOM(payload) {
  247. const { room, socketId } = payload;
  248. return new Promise(resolve => {
  249. // create room if it doesn't exist, and add socketId to array
  250. if (WSModule.rooms[room]) {
  251. if (!(socketId in WSModule.rooms[room])) WSModule.rooms[room].push(socketId);
  252. } else WSModule.rooms[room] = [socketId];
  253. return resolve();
  254. });
  255. }
  256. /**
  257. * Emits arguments to any sockets that are in a specified a room
  258. *
  259. * @param {object} payload - object that contains the payload
  260. * @param {string} payload.room - the name of the room to emit arguments
  261. * @param {object} payload.args - any arguments to be emitted to the sockets in the specific room
  262. * @returns {Promise} - returns promise (reject, resolve)
  263. */
  264. async EMIT_TO_ROOM(payload) {
  265. return new Promise(resolve => {
  266. // if the room exists
  267. if (WSModule.rooms[payload.room] && WSModule.rooms[payload.room].length > 0)
  268. return WSModule.rooms[payload.room].forEach(async socketId => {
  269. // get every socketId (and thus every socket) in the room, and dispatch to each
  270. const socket = await WSModule.runJob("SOCKET_FROM_SOCKET_ID", { socketId }, this);
  271. if (socket) socket.dispatch(...payload.args);
  272. return resolve();
  273. });
  274. return resolve();
  275. });
  276. }
  277. /**
  278. * Emits arguments to any sockets that are in specified rooms
  279. *
  280. * @param {object} payload - object that contains the payload
  281. * @param {Array} payload.rooms - array of strings with the name of each room e.g. ["station-page", "song.1234"]
  282. * @param {object} payload.args - any arguments to be emitted to the sockets in the specific room
  283. * @returns {Promise} - returns promise (reject, resolve)
  284. */
  285. async EMIT_TO_ROOMS(payload) {
  286. return new Promise(resolve =>
  287. async.each(
  288. payload.rooms,
  289. (room, next) => {
  290. WSModule.runJob("EMIT_TO_ROOM", { room, args: payload.args });
  291. return next();
  292. },
  293. () => resolve()
  294. )
  295. );
  296. }
  297. /**
  298. * Allows a socket to join a 'song' room
  299. *
  300. * @param {object} payload - object that contains the payload
  301. * @param {string} payload.socketId - the id of the socket which should join the room
  302. * @param {string} payload.room - the name of the room
  303. * @returns {Promise} - returns promise (reject, resolve)
  304. */
  305. async SOCKET_JOIN_SONG_ROOM(payload) {
  306. const { room, socketId } = payload;
  307. // leave any other song rooms the user is in
  308. await WSModule.runJob("SOCKETS_LEAVE_SONG_ROOMS", { sockets: [socketId] }, this);
  309. return new Promise(resolve => {
  310. // join the room
  311. if (WSModule.rooms[room]) WSModule.rooms[room].push(socketId);
  312. else WSModule.rooms[room] = [socketId];
  313. return resolve();
  314. });
  315. }
  316. /**
  317. * Allows multiple sockets to join a 'song' room
  318. *
  319. * @param {object} payload - object that contains the payload
  320. * @param {Array} payload.sockets - array of socketIds
  321. * @param {object} payload.room - the name of the room
  322. * @returns {Promise} - returns promise (reject, resolve)
  323. */
  324. SOCKETS_JOIN_SONG_ROOM(payload) {
  325. return new Promise(resolve => {
  326. Promise.allSettled(
  327. payload.sockets.map(async socketId => {
  328. await WSModule.runJob("SOCKET_JOIN_SONG_ROOM", { socketId, room: payload.room }, this);
  329. })
  330. ).then(() => resolve());
  331. });
  332. }
  333. /**
  334. * Allows multiple sockets to leave any 'song' rooms they are in
  335. *
  336. * @param {object} payload - object that contains the payload
  337. * @param {Array} payload.sockets - array of socketIds
  338. * @returns {Promise} - returns promise (reject, resolve)
  339. */
  340. SOCKETS_LEAVE_SONG_ROOMS(payload) {
  341. return new Promise(resolve =>
  342. Promise.allSettled(
  343. payload.sockets.map(async socketId => {
  344. const rooms = await WSModule.runJob("GET_ROOMS_FOR_SOCKET", { socketId }, this);
  345. rooms.forEach(room => {
  346. if (room.indexOf("song.") !== -1)
  347. WSModule.rooms[room] = WSModule.rooms[room].filter(participant => participant !== socketId);
  348. });
  349. })
  350. ).then(() => resolve())
  351. );
  352. }
  353. /**
  354. * Gets any sockets connected to a room
  355. *
  356. * @param {object} payload - object that contains the payload
  357. * @param {string} payload.room - the name of the room
  358. * @returns {Promise} - returns promise (reject, resolve)
  359. */
  360. async GET_SOCKETS_FOR_ROOM(payload) {
  361. return new Promise(resolve => {
  362. if (WSModule.rooms[payload.room]) return resolve(WSModule.rooms[payload.room]);
  363. return resolve([]);
  364. });
  365. }
  366. /**
  367. * Gets any rooms a socket is connected to
  368. *
  369. * @param {object} payload - object that contains the payload
  370. * @param {string} payload.socketId - the id of the socket to check the rooms for
  371. * @returns {Promise} - returns promise (reject, resolve)
  372. */
  373. async GET_ROOMS_FOR_SOCKET(payload) {
  374. return new Promise(resolve => {
  375. const rooms = [];
  376. Object.keys(WSModule.rooms).forEach(room => {
  377. if (WSModule.rooms[room].includes(payload.socketId)) rooms.push(room);
  378. });
  379. return resolve(rooms);
  380. });
  381. }
  382. /**
  383. * Handles use of websockets
  384. *
  385. * @param {object} payload - object that contains the payload
  386. * @returns {Promise} - returns promise (reject, resolve)
  387. */
  388. async HANDLE_WS_USE(payload) {
  389. return new Promise(resolve => {
  390. const { socket, req } = payload;
  391. let SID = "";
  392. socket.ip = req.headers["x-forwarded-for"] || "0..0.0";
  393. return async.waterfall(
  394. [
  395. next => {
  396. if (!req.headers.cookie) return next("No cookie exists yet.");
  397. return UtilsModule.runJob("PARSE_COOKIES", { cookieString: req.headers.cookie }, this).then(
  398. res => {
  399. SID = res[WSModule.SIDname];
  400. next(null);
  401. }
  402. );
  403. },
  404. next => {
  405. if (!SID) return next("No SID.");
  406. return next();
  407. },
  408. // see if session exists for cookie
  409. next => {
  410. CacheModule.runJob("HGET", { table: "sessions", key: SID }, this)
  411. .then(session => next(null, session))
  412. .catch(next);
  413. },
  414. (session, next) => {
  415. if (!session) return next("No session found.");
  416. session.refreshDate = Date.now();
  417. socket.session = session;
  418. return CacheModule.runJob(
  419. "HSET",
  420. {
  421. table: "sessions",
  422. key: SID,
  423. value: session
  424. },
  425. this
  426. ).then(session => next(null, session));
  427. },
  428. (res, next) => {
  429. // check if a session's user / IP is banned
  430. PunishmentsModule.runJob("GET_PUNISHMENTS", {}, this)
  431. .then(punishments => {
  432. const isLoggedIn = !!(socket.session && socket.session.refreshDate);
  433. const userId = isLoggedIn ? socket.session.userId : null;
  434. const banishment = {
  435. banned: false,
  436. ban: 0
  437. };
  438. punishments.forEach(punishment => {
  439. if (punishment.expiresAt > banishment.ban) banishment.ban = punishment;
  440. if (punishment.type === "banUserId" && isLoggedIn && punishment.value === userId)
  441. banishment.banned = true;
  442. if (punishment.type === "banUserIp" && punishment.value === socket.ip)
  443. banishment.banned = true;
  444. });
  445. socket.banishment = banishment;
  446. next();
  447. })
  448. .catch(() => next());
  449. }
  450. ],
  451. () => {
  452. if (!socket.session) socket.session = { socketId: req.headers["sec-websocket-key"] };
  453. else socket.session.socketId = req.headers["sec-websocket-key"];
  454. resolve(socket);
  455. }
  456. );
  457. });
  458. }
  459. /**
  460. * Handles a websocket connection
  461. *
  462. * @param {object} payload - object that contains the payload
  463. * @param {object} payload.socket - socket itself
  464. * @returns {Promise} - returns promise (reject, resolve)
  465. */
  466. async HANDLE_WS_CONNECTION(payload) {
  467. return new Promise(resolve => {
  468. const { socket } = payload;
  469. let sessionInfo = "";
  470. if (socket.session.sessionId) sessionInfo = ` UserID: ${socket.session.userId}.`;
  471. // if session is banned
  472. if (socket.banishment && socket.banishment.banned) {
  473. WSModule.log(
  474. "INFO",
  475. "IO_BANNED_CONNECTION",
  476. `A user tried to connect, but is currently banned. IP: ${socket.ip}.${sessionInfo}`
  477. );
  478. socket.dispatch("keep.event:banned", { data: { ban: socket.banishment.ban } });
  479. return socket.close(); // close socket connection
  480. }
  481. WSModule.log("INFO", "IO_CONNECTION", `User connected. IP: ${socket.ip}.${sessionInfo}`);
  482. // catch when the socket has been disconnected
  483. socket.on("close", async () => {
  484. if (socket.session.sessionId) sessionInfo = ` UserID: ${socket.session.userId}.`;
  485. WSModule.log("INFO", "IO_DISCONNECTION", `User disconnected. IP: ${socket.ip}.${sessionInfo}`);
  486. // leave all rooms when a socket connection is closed (to prevent rooms object building up)
  487. await WSModule.runJob("SOCKET_LEAVE_ROOMS", { socketId: socket.session.socketId });
  488. });
  489. // catch errors on the socket
  490. socket.onerror = error => {
  491. console.error("SOCKET ERROR: ", error);
  492. };
  493. if (socket.session.sessionId) {
  494. CacheModule.runJob("HGET", {
  495. table: "sessions",
  496. key: socket.session.sessionId
  497. })
  498. .then(session => {
  499. if (session && session.userId) {
  500. WSModule.userModel.findOne({ _id: session.userId }, (err, user) => {
  501. if (err || !user) return socket.dispatch("ready", { data: { loggedIn: false } });
  502. let role = "";
  503. let username = "";
  504. let userId = "";
  505. let email = "";
  506. if (user) {
  507. role = user.role;
  508. username = user.username;
  509. email = user.email.address;
  510. userId = session.userId;
  511. }
  512. return socket.dispatch("ready", {
  513. data: { loggedIn: true, role, username, userId, email }
  514. });
  515. });
  516. } else socket.dispatch("ready", { data: { loggedIn: false } });
  517. })
  518. .catch(() => socket.dispatch("ready", { data: { loggedIn: false } }));
  519. } else socket.dispatch("ready", { data: { loggedIn: false } });
  520. socket.onmessage = message => {
  521. const data = JSON.parse(message.data);
  522. if (data.length === 0) return socket.dispatch("ERROR", "Not enough arguments specified.");
  523. if (typeof data[0] !== "string") return socket.dispatch("ERROR", "First argument must be a string.");
  524. const namespaceAction = data[0];
  525. if (
  526. !namespaceAction ||
  527. namespaceAction.indexOf(".") === -1 ||
  528. namespaceAction.indexOf(".") !== namespaceAction.lastIndexOf(".")
  529. )
  530. return socket.dispatch("ERROR", "Invalid first argument");
  531. const namespace = data[0].split(".")[0];
  532. const action = data[0].split(".")[1];
  533. if (!namespace) return socket.dispatch("ERROR", "Invalid namespace.");
  534. if (!action) return socket.dispatch("ERROR", "Invalid action.");
  535. if (!WSModule.actions[namespace]) return socket.dispatch("ERROR", "Namespace not found.");
  536. if (!WSModule.actions[namespace][action]) return socket.dispatch("ERROR", "Action not found.");
  537. if (data[data.length - 1].CB_REF) {
  538. const { CB_REF } = data[data.length - 1];
  539. data.pop();
  540. return socket.actions.emit(data.shift(0), [...data, res => socket.dispatch("CB_REF", CB_REF, res)]);
  541. }
  542. return socket.actions.emit(data.shift(0), data);
  543. };
  544. // have the socket listen for each action
  545. Object.keys(WSModule.actions).forEach(namespace => {
  546. Object.keys(WSModule.actions[namespace]).forEach(action => {
  547. // the full name of the action
  548. const name = `${namespace}.${action}`;
  549. // listen for this action to be called
  550. socket.listen(name, async args =>
  551. WSModule.runJob("RUN_ACTION", { socket, namespace, action, args })
  552. );
  553. });
  554. });
  555. return resolve();
  556. });
  557. }
  558. /**
  559. * Runs an action
  560. *
  561. * @param {object} payload - object that contains the payload
  562. * @returns {Promise} - returns promise (reject, resolve)
  563. */
  564. async RUN_ACTION(payload) {
  565. return new Promise((resolve, reject) => {
  566. const { socket, namespace, action, args } = payload;
  567. // the full name of the action
  568. const name = `${namespace}.${action}`;
  569. let cb = args[args.length - 1];
  570. if (typeof cb !== "function")
  571. cb = () => {
  572. WSModule.log("INFO", "IO_MODULE", `There was no callback provided for ${name}.`);
  573. };
  574. else args.pop();
  575. WSModule.log("INFO", "IO_ACTION", `A user executed an action. Action: ${namespace}.${action}.`);
  576. // load the session from the cache
  577. new Promise(resolve => {
  578. if (socket.session.sessionId)
  579. CacheModule.runJob("HGET", {
  580. table: "sessions",
  581. key: socket.session.sessionId
  582. })
  583. .then(session => {
  584. // make sure the sockets sessionId isn't set if there is no session
  585. if (socket.session.sessionId && session === null) delete socket.session.sessionId;
  586. resolve();
  587. })
  588. .catch(() => {
  589. if (typeof cb === "function")
  590. cb({
  591. status: "error",
  592. message: "An error occurred while obtaining your session"
  593. });
  594. reject(new Error("An error occurred while obtaining the session"));
  595. });
  596. else resolve();
  597. })
  598. .then(() => {
  599. // call the job that calls the action, passing it the session, and the arguments the websocket passed us
  600. WSModule.runJob("RUN_ACTION2", { session: socket.session, namespace, action, args }, this)
  601. .then(response => {
  602. cb(response);
  603. resolve();
  604. })
  605. .catch(err => {
  606. if (typeof cb === "function")
  607. cb({
  608. status: "error",
  609. message: "An error occurred while executing the specified action."
  610. });
  611. reject(err);
  612. WSModule.log(
  613. "ERROR",
  614. "IO_ACTION_ERROR",
  615. `Some type of exception occurred in the action ${namespace}.${action}. Error message: ${err.message}`
  616. );
  617. });
  618. })
  619. .catch(reject);
  620. });
  621. }
  622. /**
  623. * Runs an action
  624. *
  625. * @param {object} payload - object that contains the payload
  626. * @returns {Promise} - returns promise (reject, resolve)
  627. */
  628. async RUN_ACTION2(payload) {
  629. return new Promise((resolve, reject) => {
  630. const { session, namespace, action, args } = payload;
  631. try {
  632. // call the the action, passing it the session, and the arguments the websocket passed us
  633. WSModule.actions[namespace][action].apply(
  634. this,
  635. [session].concat(args).concat([
  636. result => {
  637. WSModule.log(
  638. "INFO",
  639. "RUN_ACTION2",
  640. `Response to action. Action: ${namespace}.${action}. Response status: ${result.status}`
  641. );
  642. resolve(result);
  643. }
  644. ])
  645. );
  646. } catch (err) {
  647. reject(err);
  648. WSModule.log(
  649. "ERROR",
  650. "IO_ACTION_ERROR",
  651. `Some type of exception occurred in the action ${namespace}.${action}. Error message: ${err.message}`
  652. );
  653. }
  654. });
  655. }
  656. }
  657. export default new _WSModule();