import config from "config"; import async from "async"; import crypto from "crypto"; import CoreClass from "../core"; let AppModule; let DBModule; let PlaylistsModule; let UtilsModule; let PunishmentsModule; let CacheModule; let NotificationsModule; class _APIModule extends CoreClass { // eslint-disable-next-line require-jsdoc constructor() { super("api"); } /** * Initialises the api module * * @returns {Promise} - returns promise (reject, resolve) */ initialize() { return new Promise((resolve, reject) => { AppModule = this.moduleManager.modules.app; DBModule = this.moduleManager.modules.db; PlaylistsModule = this.moduleManager.modules.playlists; UtilsModule = this.moduleManager.modules.utils; PunishmentsModule = this.moduleManager.modules.punishments; CacheModule = this.moduleManager.modules.cache; NotificationsModule = this.moduleManager.modules.notifications; const SIDname = config.get("cookie.SIDname"); const isLoggedIn = (req, res, next) => { let SID; async.waterfall( [ next => { UtilsModule.runJob("PARSE_COOKIES", { cookieString: req.headers.cookie }) .then(res => { SID = res[SIDname]; next(null); }) .catch(next); }, next => { if (!SID) return next("No SID."); return next(); }, next => { CacheModule.runJob("HGET", { table: "sessions", key: SID }).then(session => next(null, session) ); }, (session, next) => { if (!session) return next("No session found."); session.refreshDate = Date.now(); req.session = session; return CacheModule.runJob("HSET", { table: "sessions", key: SID, value: session }).then(session => { next(null, session); }); }, (res, next) => { // check if a session's user / IP is banned PunishmentsModule.runJob("GET_PUNISHMENTS", {}) .then(punishments => { const isLoggedIn = !!(req.session && req.session.refreshDate); const userId = isLoggedIn ? req.session.userId : null; const banishment = { banned: false, ban: 0 }; punishments.forEach(punishment => { if (punishment.expiresAt > banishment.ban) banishment.ban = punishment; if ( punishment.type === "banUserId" && isLoggedIn && punishment.value === userId ) banishment.banned = true; if (punishment.type === "banUserIp" && punishment.value === req.ip) banishment.banned = true; }); req.banishment = banishment; next(); }) .catch(() => { next(); }); } ], err => { if (err) return res.json({ status: "error", message: "You are not logged in" }); return next(); } ); }; AppModule.runJob("GET_APP", {}) .then(response => { response.app.get("/", (req, res) => { res.json({ status: "success", message: "Coming Soon" }); }); response.app.get("/export/playlist/:playlistId", async (req, res) => { const { playlistId } = req.params; const userModel = await DBModule.runJob("GET_MODEL", { modelName: "user" }); PlaylistsModule.runJob("GET_PLAYLIST", { playlistId }) .then(playlist => { if (playlist.privacy === "public") res.json({ status: "success", playlist }); else { isLoggedIn(req, res, () => { if (playlist.createdBy === req.session.userId) res.json({ status: "success", playlist }); else { userModel.findOne({ _id: req.session.userId }, (err, user) => { if (err) res.json({ status: "error", message: err.message }); else if (user.role === "admin") res.json({ status: "success", playlist }); else res.json({ status: "error", message: "You're not allowed to download this playlist." }); }); } }); } }) .catch(err => { res.json({ status: "error", message: err.message }); }); }); if (config.get("debug.stationIssue")) { response.app.get("/debug_station", async (req, res) => { const responseObject = {}; const stationModel = await DBModule.runJob("GET_MODEL", { modelName: "station" }); async.waterfall( [ next => { stationModel.find({}, next); }, (stations, next) => { responseObject.mongo = { stations }; next(); }, next => { CacheModule.runJob("HGETALL", { table: "stations" }) .then(stations => { next(null, stations); }) .catch(err => { console.log(err); next(err); }); }, (stations, next) => { responseObject.redis = { stations }; next(); }, next => { responseObject.cryptoExamples = {}; responseObject.mongo.stations.forEach(station => { const payloadName = `stations.nextSong?id=${station._id}`; responseObject.cryptoExamples[station._id] = crypto .createHash("md5") .update(`_notification:${payloadName}_`) .digest("hex"); }); next(); }, next => { NotificationsModule.pub .KEYS("*") .then(redisKeys => next(null, redisKeys)) .catch(next); }, (redisKeys, next) => { responseObject.redis = { ...redisKeys, ttl: {} }; async.eachLimit( redisKeys, 1, (redisKey, next) => { NotificationsModule.pub .TTL(redisKey) .then(ttl => { responseObject.redis.ttl[redisKey] = ttl; next(); }) .catch(next); }, next ); }, next => { responseObject.debugLogs = this.moduleManager.debugLogs.stationIssue; next(); }, next => { responseObject.debugJobs = this.moduleManager.debugJobs; next(); } ], err => { if (err) { console.log(err); return res.json({ error: err, objectSoFar: responseObject }); } const responseJson = JSON.stringify(responseObject, (key, value) => { if ( key === "module" || key === "task" || key === "onFinish" || key === "server" || key === "nsp" || key === "socket" || key === "res" || key === "client" || key === "_idleNext" || key === "_idlePrev" ) { return undefined; } if (key === "parentJob" && value) return value.toString(); return value; }); return res.end(responseJson); } ); }); } resolve(); }) .catch(err => { reject(err); }); }); } } export default new _APIModule();