1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009 |
- import config from "config";
- import async from "async";
- import crypto from "crypto";
- import request from "request";
- import CoreClass from "../core";
- let UtilsModule;
- let IOModule;
- let SpotifyModule;
- let CacheModule;
- class _UtilsModule extends CoreClass {
-
- constructor() {
- super("utils");
- this.youtubeRequestCallbacks = [];
- this.youtubeRequestsPending = 0;
- this.youtubeRequestsActive = false;
- UtilsModule = this;
- }
-
- initialize() {
- return new Promise(resolve => {
- IOModule = this.moduleManager.modules.io;
- SpotifyModule = this.moduleManager.modules.spotify;
- CacheModule = this.moduleManager.modules.cache;
- resolve();
- });
- }
-
- PARSE_COOKIES(payload) {
- return new Promise((resolve, reject) => {
- const cookies = {};
- if (typeof payload.cookieString !== "string") return reject(new Error("Cookie string is not a string"));
-
- payload.cookieString.split("; ").map(cookie => {
- cookies[cookie.substring(0, cookie.indexOf("="))] = cookie.substring(
- cookie.indexOf("=") + 1,
- cookie.length
- );
- });
- return resolve(cookies);
- });
- }
-
-
-
-
-
-
-
-
-
-
- REMOVE_COOKIE(payload) {
- return new Promise((resolve, reject) => {
- let cookies;
- try {
- cookies = UtilsModule.runJob(
- "PARSE_COOKIES",
- {
- cookieString: payload.cookieString
- },
- this
- );
- } catch (err) {
- return reject(err);
- }
- delete cookies[payload.cookieName];
- return resolve(this.toString(cookies));
- });
- }
-
- HTML_ENTITIES(payload) {
- return new Promise(resolve => {
- resolve(
- String(payload.str)
- .replace(/&/g, "&")
- .replace(/</g, "<")
- .replace(/>/g, ">")
- .replace(/"/g, """)
- );
- });
- }
-
- async GENERATE_RANDOM_STRING(payload) {
- const chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789".split("");
- const promises = [];
- for (let i = 0; i < payload.length; i += 1) {
- this.log(
- "ERROR",
- "CHECK THIS!?!?!?!??!?!?!?!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"
- );
- promises.push(
- UtilsModule.runJob(
- "GET_RANDOM_NUMBER",
- {
- min: 0,
- max: chars.length - 1
- },
- this
- )
- );
- }
- const randomNums = await Promise.all(promises);
- const randomChars = [];
- for (let i = 0; i < payload.length; i += 1) {
- randomChars.push(chars[randomNums[i]]);
- }
- return new Promise(resolve => resolve(randomChars.join("")));
- }
-
- async GET_SOCKET_FROM_ID(payload) {
-
- const io = await IOModule.runJob("IO", {}, this);
- return new Promise(resolve => resolve(io.sockets.sockets[payload.socketId]));
- }
-
- GET_RANDOM_NUMBER(payload) {
-
- return new Promise(resolve =>
- resolve(Math.floor(Math.random() * (payload.max - payload.min + 1)) + payload.min)
- );
- }
-
- CONVERT_TIME(payload) {
-
- return new Promise(resolve => {
- let { duration } = payload;
- let a = duration.match(/\d+/g);
- if (duration.indexOf("M") >= 0 && duration.indexOf("H") === -1 && duration.indexOf("S") === -1) {
- a = [0, a[0], 0];
- }
- if (duration.indexOf("H") >= 0 && duration.indexOf("M") === -1) {
- a = [a[0], 0, a[1]];
- }
- if (duration.indexOf("H") >= 0 && duration.indexOf("M") === -1 && duration.indexOf("S") === -1) {
- a = [a[0], 0, 0];
- }
- duration = 0;
- if (a.length === 3) {
- duration += parseInt(a[0]) * 3600;
- duration += parseInt(a[1]) * 60;
- duration += parseInt(a[2]);
- }
- if (a.length === 2) {
- duration += parseInt(a[0]) * 60;
- duration += parseInt(a[1]);
- }
- if (a.length === 1) {
- duration += parseInt(a[0]);
- }
- const hours = Math.floor(duration / 3600);
- const minutes = Math.floor((duration % 3600) / 60);
- const seconds = Math.floor((duration % 3600) % 60);
- resolve(
- (hours < 10 ? `0${hours}:` : `${hours}:`) +
- (minutes < 10 ? `0${minutes}:` : `${minutes}:`) +
- (seconds < 10 ? `0${seconds}` : seconds)
- );
- });
- }
-
- GUID() {
- return new Promise(resolve => {
- resolve(
- [1, 1, 0, 1, 0, 1, 0, 1, 0, 1, 1, 1]
- .map(b =>
- b
- ? Math.floor((1 + Math.random()) * 0x10000)
- .toString(16)
- .substring(1)
- : "-"
- )
- .join("")
- );
- });
- }
-
-
- async SOCKET_FROM_SESSION(payload) {
-
- const io = await IOModule.runJob("IO", {}, this);
- return new Promise((resolve, reject) => {
- const ns = io.of("/");
- if (ns) {
- return resolve(ns.connected[payload.socketId]);
- }
- return reject();
- });
- }
-
- async SOCKETS_FROM_SESSION_ID(payload) {
- const io = await IOModule.runJob("IO", {}, this);
- return new Promise(resolve => {
- const ns = io.of("/");
- const sockets = [];
- if (ns) {
- return async.each(
- Object.keys(ns.connected),
- (id, next) => {
- const { session } = ns.connected[id];
- if (session.sessionId === payload.sessionId) sockets.push(session.sessionId);
- next();
- },
- () => {
- resolve({ sockets });
- }
- );
- }
- return resolve();
- });
- }
-
- async SOCKETS_FROM_USER(payload) {
- const io = await IOModule.runJob("IO", {}, this);
- return new Promise((resolve, reject) => {
- const ns = io.of("/");
- const sockets = [];
- if (ns) {
- return async.each(
- Object.keys(ns.connected),
- (id, next) => {
- const { session } = ns.connected[id];
- CacheModule.runJob(
- "HGET",
- {
- table: "sessions",
- key: session.sessionId
- },
- this
- )
- .then(session => {
- if (session && session.userId === payload.userId) sockets.push(ns.connected[id]);
- next();
- })
- .catch(err => {
- next(err);
- });
- },
- err => {
- if (err) return reject(err);
- return resolve({ sockets });
- }
- );
- }
- return resolve();
- });
- }
-
- async SOCKETS_FROM_IP(payload) {
- const io = await IOModule.runJob("IO", {}, this);
- return new Promise(resolve => {
- const ns = io.of("/");
- const sockets = [];
- if (ns) {
- return async.each(
- Object.keys(ns.connected),
- (id, next) => {
- const { session } = ns.connected[id];
- CacheModule.runJob(
- "HGET",
- {
- table: "sessions",
- key: session.sessionId
- },
- this
- )
- .then(session => {
- if (session && ns.connected[id].ip === payload.ip) sockets.push(ns.connected[id]);
- next();
- })
- .catch(() => next());
- },
- () => {
- resolve({ sockets });
- }
- );
- }
- return resolve();
- });
- }
-
- async SOCKETS_FROM_USER_WITHOUT_CACHE(payload) {
- const io = await IOModule.runJob("IO", {}, this);
- return new Promise(resolve => {
- const ns = io.of("/");
- const sockets = [];
- if (ns) {
- return async.each(
- Object.keys(ns.connected),
- (id, next) => {
- const { session } = ns.connected[id];
- if (session.userId === payload.userId) sockets.push(ns.connected[id]);
- next();
- },
- () => {
- resolve({ sockets });
- }
- );
- }
- return resolve();
- });
- }
-
- async SOCKET_LEAVE_ROOMS(payload) {
- const socket = await UtilsModule.runJob(
- "SOCKET_FROM_SESSION",
- {
- socketId: payload.socketId
- },
- this
- );
- return new Promise(resolve => {
- const { rooms } = socket;
- for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
- socket.leave(room);
- }
- return resolve();
- });
- }
-
- async SOCKET_JOIN_ROOM(payload) {
- const socket = await UtilsModule.runJob(
- "SOCKET_FROM_SESSION",
- {
- socketId: payload.socketId
- },
- this
- );
- return new Promise(resolve => {
- const { rooms } = socket;
- for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
- socket.leave(room);
- }
- socket.join(payload.room);
- return resolve();
- });
- }
-
-
- async SOCKET_JOIN_SONG_ROOM(payload) {
-
- const socket = await UtilsModule.runJob(
- "SOCKET_FROM_SESSION",
- {
- socketId: payload.socketId
- },
- this
- );
- return new Promise(resolve => {
- const { rooms } = socket;
- for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
- if (room.indexOf("song.") !== -1) socket.leave(rooms);
- }
- socket.join(payload.room);
- return resolve();
- });
- }
-
-
- SOCKETS_JOIN_SONG_ROOM(payload) {
-
- return new Promise(resolve => {
- for (let id = 0, socketKeys = Object.keys(payload.sockets); id < socketKeys.length; id += 1) {
- const socket = payload.sockets[socketKeys[id]];
- const { rooms } = socket;
- for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
- if (room.indexOf("song.") !== -1) socket.leave(room);
- }
- socket.join(payload.room);
- }
- return resolve();
- });
- }
-
-
- SOCKETS_LEAVE_SONG_ROOMS(payload) {
-
- return new Promise(resolve => {
- for (let id = 0, socketKeys = Object.keys(payload.sockets); id < socketKeys.length; id += 1) {
- const socket = payload.sockets[socketKeys[id]];
- const { rooms } = socket;
- for (let room = 0, roomKeys = Object.keys(rooms); room < roomKeys.length; room += 1) {
- if (room.indexOf("song.") !== -1) socket.leave(room);
- }
- }
- resolve();
- });
- }
-
- async EMIT_TO_ROOM(payload) {
- const io = await IOModule.runJob("IO", {}, this);
- return new Promise(resolve => {
- const { sockets } = io.sockets;
- for (let id = 0, socketKeys = Object.keys(sockets); id < socketKeys.length; id += 1) {
- const socket = sockets[socketKeys[id]];
- if (socket.rooms[payload.room]) {
- socket.emit(...payload.args);
- }
- }
- return resolve();
- });
- }
-
- async GET_ROOM_SOCKETS(payload) {
- const io = await IOModule.runJob("IO", {}, this);
- return new Promise(resolve => {
- const { sockets } = io.sockets;
- const roomSockets = [];
- for (let id = 0, socketKeys = Object.keys(sockets); id < socketKeys.length; id += 1) {
- const socket = sockets[socketKeys[id]];
- if (socket.rooms[payload.room]) roomSockets.push(socket);
- }
- return resolve(roomSockets);
- });
- }
-
- GET_SONG_FROM_YOUTUBE(payload) {
-
- return new Promise((resolve, reject) => {
- this.youtubeRequestCallbacks.push({
- cb: () => {
- this.youtubeRequestsActive = true;
- const youtubeParams = [
- "part=snippet,contentDetails,statistics,status",
- `id=${encodeURIComponent(payload.songId)}`,
- `key=${config.get("apis.youtube.key")}`
- ].join("&");
- request(`https://www.googleapis.com/youtube/v3/videos?${youtubeParams}`, (err, res, body) => {
- this.youtubeRequestCallbacks.splice(0, 1);
- if (this.youtubeRequestCallbacks.length > 0) {
- this.youtubeRequestCallbacks[0].cb(this.youtubeRequestCallbacks[0].songId);
- } else this.youtubeRequestsActive = false;
- if (err) {
- console.error(err);
- return null;
- }
- body = JSON.parse(body);
- if (body.error) {
- console.log("ERROR", "GET_SONG_FROM_YOUTUBE", `${body.error.message}`);
- return reject(new Error("An error has occured. Please try again later."));
- }
- if (body.items[0] === undefined)
- return reject(
- new Error("The specified video does not exist or cannot be publicly accessed.")
- );
-
- let dur = body.items[0].contentDetails.duration;
- dur = dur.replace("PT", "");
- let duration = 0;
- dur = dur.replace(/([\d]*)H/, (v, v2) => {
- v2 = Number(v2);
- duration = v2 * 60 * 60;
- return "";
- });
- dur = dur.replace(/([\d]*)M/, (v, v2) => {
- v2 = Number(v2);
- duration += v2 * 60;
- return "";
- });
-
- dur = dur.replace(/([\d]*)S/, (v, v2) => {
- v2 = Number(v2);
- duration += v2;
- return "";
- });
- const song = {
- songId: body.items[0].id,
- title: body.items[0].snippet.title,
- duration
- };
- return resolve({ song });
- });
- },
- songId: payload.songId
- });
- if (!this.youtubeRequestsActive) {
- this.youtubeRequestCallbacks[0].cb(this.youtubeRequestCallbacks[0].songId);
- }
- });
- }
-
- FILTER_MUSIC_VIDEOS_YOUTUBE(payload) {
-
- return new Promise((resolve, reject) => {
-
- function getNextPage(cb2) {
- const localVideoIds = payload.videoIds.splice(0, 50);
- const youtubeParams = [
- "part=topicDetails",
- `id=${encodeURIComponent(localVideoIds.join(","))}`,
- `maxResults=50`,
- `key=${config.get("apis.youtube.key")}`
- ].join("&");
- request(`https://www.googleapis.com/youtube/v3/videos?${youtubeParams}`, (err, res, body) => {
- if (err) {
- console.error(err);
- return reject(new Error("Failed to find playlist from YouTube"));
- }
- body = JSON.parse(body);
- if (body.error) {
- console.log("ERROR", "FILTER_MUSIC_VIDEOS_YOUTUBE", `${body.error.message}`);
- return reject(new Error("An error has occured. Please try again later."));
- }
- const songIds = [];
- body.items.forEach(item => {
- const songId = item.id;
- if (!item.topicDetails) return;
- if (item.topicDetails.relevantTopicIds.indexOf("/m/04rlf") !== -1) {
- songIds.push(songId);
- }
- });
- if (payload.videoIds.length > 0) {
- return getNextPage(newSongIds => {
- cb2(songIds.concat(newSongIds));
- });
- }
- return cb2(songIds);
- });
- }
- if (payload.videoIds.length === 0) resolve({ songIds: [] });
- else getNextPage(songIds => resolve({ songIds }));
- });
- }
-
- GET_PLAYLIST_FROM_YOUTUBE(payload) {
-
- return new Promise((resolve, reject) => {
- const local = this;
- const name = "list".replace(/[\\[]/, "\\[").replace(/[\]]/, "\\]");
- const regex = new RegExp(`[\\?&]${name}=([^&#]*)`);
- const splitQuery = regex.exec(payload.url);
- if (!splitQuery) {
- console.log("ERROR", "GET_PLAYLIST_FROM_YOUTUBE", "Invalid YouTube playlist URL query.");
- return reject(new Error("An error has occured. Please try again later."));
- }
- const playlistId = splitQuery[1];
-
- function getPage(pageToken, songs) {
- const nextPageToken = pageToken ? `pageToken=${pageToken}` : "";
- const youtubeParams = [
- "part=contentDetails",
- `playlistId=${encodeURIComponent(playlistId)}`,
- `maxResults=50`,
- `key=${config.get("apis.youtube.key")}`,
- nextPageToken
- ].join("&");
- request(
- `https://www.googleapis.com/youtube/v3/playlistItems?${youtubeParams}`,
- async (err, res, body) => {
- if (err) {
- console.error(err);
- return reject(new Error("Failed to find playlist from YouTube"));
- }
- body = JSON.parse(body);
- if (body.error) {
- console.log("ERROR", "GET_PLAYLIST_FROM_YOUTUBE", `${body.error.message}`);
- return reject(new Error("An error has occured. Please try again later."));
- }
- songs = songs.concat(body.items);
- if (body.nextPageToken) return getPage(body.nextPageToken, songs);
- songs = songs.map(song => song.contentDetails.videoId);
- if (!payload.musicOnly) return resolve({ songs });
- return local
- .runJob(
- "FILTER_MUSIC_VIDEOS_YOUTUBE",
- {
- videoIds: songs.slice()
- },
- this
- )
- .then(filteredSongs => {
- resolve({ filteredSongs, songs });
- });
- }
- );
- }
- return getPage(null, []);
- });
- }
-
- async GET_SONG_FROM_SPOTIFY(payload) {
-
- const token = await SpotifyModule.runJob("GET_TOKEN", {}, this);
- return new Promise((resolve, reject) => {
- if (!config.get("apis.spotify.enabled")) return reject(new Error("Spotify is not enabled."));
- const song = { ...payload.song };
- const spotifyParams = [`q=${encodeURIComponent(payload.song.title)}`, `type=track`].join("&");
- const options = {
- url: `https://api.spotify.com/v1/search?${spotifyParams}`,
- headers: {
- Authorization: `Bearer ${token}`
- }
- };
- return request(options, (err, res, body) => {
- if (err) console.error(err);
- body = JSON.parse(body);
- if (body.error) console.error(body.error);
- for (let i = 0, bodyKeys = Object.keys(body); i < bodyKeys.length; i += 1) {
- const { items } = body[i];
- for (let j = 0, itemKeys = Object.keys(body); j < itemKeys.length; j += 1) {
- const item = items[j];
- let hasArtist = false;
- for (let k = 0; k < item.artists.length; k += 1) {
- const artist = item.artists[k];
- if (song.title.indexOf(artist.name) !== -1) hasArtist = true;
- }
- if (hasArtist && song.title.indexOf(item.name) !== -1) {
- song.duration = item.duration_ms / 1000;
- song.artists = item.artists.map(artist => artist.name);
- song.title = item.name;
- song.explicit = item.explicit;
- song.thumbnail = item.album.images[1].url;
- break;
- }
- }
- }
- resolve({ song });
- });
- });
- }
-
- async GET_SONGS_FROM_SPOTIFY(payload) {
-
- const token = await SpotifyModule.runJob("GET_TOKEN", {}, this);
- return new Promise((resolve, reject) => {
- if (!config.get("apis.spotify.enabled")) return reject(new Error("Spotify is not enabled."));
- const spotifyParams = [`q=${encodeURIComponent(payload.title)}`, `type=track`].join("&");
- const options = {
- url: `https://api.spotify.com/v1/search?${spotifyParams}`,
- headers: {
- Authorization: `Bearer ${token}`
- }
- };
- return request(options, (err, res, body) => {
- if (err) return console.error(err);
- body = JSON.parse(body);
- if (body.error) return console.error(body.error);
- const songs = [];
- for (let i = 0, bodyKeys = Object.keys(body); i < bodyKeys.length; i += 1) {
- const { items } = body[i];
- for (let j = 0, itemKeys = Object.keys(body); j < itemKeys.length; j += 1) {
- const item = items[j];
- let hasArtist = false;
- for (let k = 0; k < item.artists.length; k += 1) {
- const localArtist = item.artists[k];
- if (payload.artist.toLowerCase() === localArtist.name.toLowerCase()) hasArtist = true;
- }
- if (
- hasArtist &&
- (payload.title.indexOf(item.name) !== -1 || item.name.indexOf(payload.title) !== -1)
- ) {
- const song = {};
- song.duration = item.duration_ms / 1000;
- song.artists = item.artists.map(artist => artist.name);
- song.title = item.name;
- song.explicit = item.explicit;
- song.thumbnail = item.album.images[1].url;
- songs.push(song);
- }
- }
- }
- return resolve({ songs });
- });
- });
- }
-
- SHUFFLE(payload) {
-
- return new Promise(resolve => {
- const array = payload.array.slice();
- let currentIndex = payload.array.length;
- let temporaryValue;
- let randomIndex;
-
- while (currentIndex !== 0) {
-
- randomIndex = Math.floor(Math.random() * currentIndex);
- currentIndex -= 1;
-
- temporaryValue = array[currentIndex];
- array[currentIndex] = array[randomIndex];
- array[randomIndex] = temporaryValue;
- }
- resolve({ array });
- });
- }
-
- GET_ERROR(payload) {
- return new Promise(resolve => {
- let error = "An error occurred.";
- if (typeof payload.error === "string") error = payload.error;
- else if (payload.error.message) {
- if (payload.error.message !== "Validation failed") error = payload.error.message;
- else error = payload.error.errors[Object.keys(payload.error.errors)].message;
- }
- resolve(error);
- });
- }
-
- CREATE_GRAVATAR(payload) {
- return new Promise(resolve => {
- const hash = crypto.createHash("md5").update(payload.email).digest("hex");
- resolve(`https://www.gravatar.com/avatar/${hash}`);
- });
- }
-
- DEBUG() {
- return new Promise(resolve => resolve());
- }
- }
- export default new _UtilsModule();
|