const functions = require("firebase-functions"); const admin = require("firebase-admin"); let key = "./serviceAccountKey.json"; if (process.env.GCLOUD_PROJECT === "monkey-type") { key = "./serviceAccountKey_live.json"; } var serviceAccount = require(key); admin.initializeApp({ credential: admin.credential.cert(serviceAccount), }); const db = admin.firestore(); exports.moveResults = functions .runWith({ timeoutSeconds: 540, memory: "2GB" }) .https.onCall((request, response) => { return db .collection("results") .orderBy("timestamp", "desc") .limit(2000) .get() .then((data) => { data.docs.forEach((doc) => { let result = doc.data(); if (result.moved === undefined || result.moved === false) { db.collection(`results`).doc(doc.id).update({ moved: true }); db.collection(`users/${result.uid}/results`).add(result); console.log(`moving doc ${doc.id}`); } }); return; }); }); function getAllNames() { return admin .auth() .listUsers() .then((data) => { let names = []; data.users.forEach((user) => { names.push(user.displayName); }); return names; }); } function getAllUsers() { return admin .auth() .listUsers() .then((data) => { return data.users; }); } function isUsernameValid(name) { if (name === null || name === undefined || name === "") return false; if (/miodec/.test(name)) return false; if (name.length > 12) return false; return /^[0-9a-zA-Z_.-]+$/.test(name); } exports.checkNameAvailability = functions.https.onCall((request, response) => { // 1 - available // -1 - unavailable (taken) // -2 - not valid name // -999 - unknown error try { if (!isUsernameValid(request.name)) return -2; return getAllNames().then((data) => { let available = 1; data.forEach((name) => { try { if (name.toLowerCase() === request.name.toLowerCase()) available = -1; } catch (e) { // } }); return available; }); } catch (e) { return -999; } }); exports.changeName = functions.https.onCall((request, response) => { try { if (!isUsernameValid(request.name)) { console.warn( `${request.uid} tried to change their name to ${request.name} - not valid` ); return 0; } return getAllNames().then((data) => { let available = 1; data.forEach((name) => { try { if (name.toLowerCase() === request.name.toLowerCase()) available = 0; } catch (e) { // } }); if (available === 1) { return admin .auth() .updateUser(request.uid, { displayName: request.name, }) .then((d) => { console.log( `${request.uid} changed their name to ${request.name} - done` ); return 1; }) .catch((e) => { console.error( `${request.uid} tried to change their name to ${request.name} - ${e.message}` ); return -1; }); } else { console.warn( `${request.uid} tried to change their name to ${request.name} - already taken` ); return 0; } }); } catch (e) { console.error( `${request.uid} tried to change their name to ${request.name} - ${e}` ); return -1; } }); exports.checkIfNeedsToChangeName = functions.https.onCall( (request, response) => { try { return admin .auth() .getUser(request.uid) .then((requestUser) => { if (!isUsernameValid(requestUser.displayName)) { //invalid name, needs to change console.log( `user ${requestUser.uid} ${requestUser.displayName} needs to change name` ); return 1; } else { //valid name, but need to change if not duplicate return getAllUsers().then((users) => { let sameName = []; //look for name names users.forEach((user) => { if (user.uid !== requestUser.uid) { try { if ( user.displayName.toLowerCase() === requestUser.displayName.toLowerCase() ) { sameName.push(user); } } catch (e) { // } } }); if (sameName.length === 0) { return 0; } else { //check when the request user made the account compared to others let earliestTimestamp = 999999999999999; sameName.forEach((sn) => { let ts = new Date(sn.metadata.creationTime).getTime() / 1000; if (ts <= earliestTimestamp) { earliestTimestamp = ts; } }); if ( new Date(requestUser.metadata.creationTime).getTime() / 1000 > earliestTimestamp ) { console.log( `user ${requestUser.uid} ${requestUser.displayName} needs to change name` ); return 2; } else { return 0; } } }); } }); } catch (e) { return -1; } } ); function checkIfPB(uid, obj) { return db .collection(`users`) .doc(uid) .get() .then((data) => { let pbs = null; try { pbs = data.data().personalBests; if (pbs === undefined) { throw new Error("pb is undefined"); } } catch (e) { return db .collection("users") .doc(uid) .update({ personalBests: { [obj.mode]: { [obj.mode2]: [ { language: obj.language, difficulty: obj.difficulty, punctuation: obj.punctuation, wpm: obj.wpm, }, ], }, }, }) .then((e) => { return true; }) .catch((e) => { return db .collection("users") .doc(uid) .set({ personalBests: { [obj.mode]: { [obj.mode2]: [ { language: obj.language, difficulty: obj.difficulty, punctuation: obj.punctuation, wpm: obj.wpm, }, ], }, }, }) .then((e) => { return true; }); }); } // //check mode, mode2, punctuation, language and difficulty let toUpdate = false; let found = false; try { pbs[obj.mode][obj.mode2].forEach((pb) => { if ( pb.punctuation === obj.punctuation && pb.difficulty === obj.difficulty && pb.language === obj.language ) { //entry like this already exists, compare wpm found = true; if (pb.wpm < obj.wpm) { //new pb pb.wpm = obj.wpm; toUpdate = true; } else { //no pb return false; } } }); //checked all pbs, nothing found - meaning this is a new pb if (!found) { pbs[obj.mode][obj.mode2].push({ language: obj.language, difficulty: obj.difficulty, punctuation: obj.punctuation, wpm: obj.wpm, }); toUpdate = true; } } catch (e) { // console.log(e); pbs[obj.mode] = {}; pbs[obj.mode][obj.mode2] = [ { language: obj.language, difficulty: obj.difficulty, punctuation: obj.punctuation, wpm: obj.wpm, }, ]; toUpdate = true; } if (toUpdate) { return db .collection("users") .doc(uid) .update({ personalBests: pbs }) .then((e) => { return true; }); } else { return false; } }); } exports.testCompleted = functions.https.onCall((request, response) => { try { if (request.uid === undefined || request.obj === undefined) { console.error(`error saving result for ${request.uid} - missing input`); return -1; } let obj = request.obj; let err = false; Object.keys(obj).forEach((key) => { let val = obj[key]; if (Array.isArray(val)) { val.forEach((valarr) => { if (!/^[0-9a-zA-Z._]+$/.test(valarr)) err = true; }); } else { if (val === undefined || !/^[0-9a-zA-Z._]+$/.test(val)) err = true; } }); if (err) { console.error( `error saving result for ${request.uid} - bad input - ${JSON.stringify( request.obj )}` ); return -1; } if (obj.wpm <= 0 || obj.wpm > 350 || obj.acc < 50 || obj.acc > 100) { return -1; } db.collection("users") .doc(request.uid) .get() .then((ret) => { let userdata = ret.data(); return db .collection(`users/${request.uid}/results`) .add(obj) .then((e) => { return checkIfPB(request.uid, request.obj).then((e) => { if (e) { console.log( `saved result for ${request.uid} (new PB) - ${JSON.stringify( request.obj )}` ); if (obj.mode === "time" && String(obj.mode2) === "60") { let besttime60 = 0; try { userdata.personalBests.time[60].forEach((result) => { if (result.wpm > besttime60) besttime60 = result.wpm; }); } catch (e) { besttime60 = 0; } if (obj.wpm >= besttime60) { console.log( `sending command to the bot to update the role for user ${request.uid} with wpm ${obj.wpm}` ); updateDiscordRole(userdata.discordId, obj.wpm); } return; } return 2; } else { console.log( `saved result for ${request.uid} - ${JSON.stringify( request.obj )}` ); return 1; } }); }) .catch((e) => { console.error( `error saving result when checking for PB for ${request.uid} - ${e.message}` ); return -1; }); }) .catch((e) => { console.error( `error saving result when getting user info ${request.uid} - ${e.message}` ); return -1; }); } catch (e) { console.error(`error saving result for ${request.uid} - ${e}`); return -1; } }); function updateDiscordRole(discordId, wpm) { db.collection("bot-commands").add({ command: "updateRole", arguments: [discordId, wpm], executed: false, requestTimestamp: Date.now(), }); } function isTagValid(name) { if (name === null || name === undefined || name === "") return false; if (name.length > 16) return false; return /^[0-9a-zA-Z_.-]+$/.test(name); } exports.addTag = functions.https.onCall((request, response) => { try { if (!isTagValid(request.name)) { return { resultCode: -1 }; } else { return db .collection(`users/${request.uid}/tags`) .add({ name: request.name, }) .then((e) => { console.log(`user ${request.uid} created a tag: ${request.name}`); return { resultCode: 1, id: e.id, }; }) .catch((e) => { console.error( `error while creating tag for user ${request.uid}: ${e.message}` ); return { resultCode: -999 }; }); } } catch (e) { console.error(`error adding tag for ${request.uid} - ${e}`); return { resultCode: -999 }; } }); exports.editTag = functions.https.onCall((request, response) => { try { if (!isTagValid(request.name)) { return { resultCode: -1 }; } else { return db .collection(`users/${request.uid}/tags`) .doc(request.tagid) .update({ name: request.name, }) .then((e) => { console.log(`user ${request.uid} updated a tag: ${request.name}`); return { resultCode: 1, }; }) .catch((e) => { console.error( `error while updating tag for user ${request.uid}: ${e.message}` ); return { resultCode: -999 }; }); } } catch (e) { console.error(`error updating tag for ${request.uid} - ${e}`); return { resultCode: -999 }; } }); exports.removeTag = functions.https.onCall((request, response) => { try { return db .collection(`users/${request.uid}/tags`) .doc(request.tagid) .delete() .then((e) => { console.log(`user ${request.uid} deleted a tag`); return { resultCode: 1, }; }) .catch((e) => { console.error( `error deleting tag for user ${request.uid}: ${e.message}` ); return { resultCode: -999 }; }); } catch (e) { console.error(`error deleting tag for ${request.uid} - ${e}`); return { resultCode: -999 }; } }); exports.updateResultTags = functions.https.onCall((request, response) => { try { let validTags = true; request.tags.forEach((tag) => { if (!/^[0-9a-zA-Z]+$/.test(tag)) validTags = false; }); if (validTags) { return db .collection(`users/${request.uid}/results`) .doc(request.resultid) .update({ tags: request.tags, }) .then((e) => { console.log( `user ${request.uid} updated tags for result ${request.resultid}` ); return { resultCode: 1, }; }) .catch((e) => { console.error( `error while updating tags for result by user ${request.uid}: ${e.message}` ); return { resultCode: -999 }; }); } else { console.error(`invalid tags for user ${request.uid}: ${request.tags}`); return { resultCode: -1 }; } } catch (e) { console.error(`error updating tags by ${request.uid} - ${e}`); return { resultCode: -999 }; } }); function isConfigKeyValid(name) { if (name === null || name === undefined || name === "") return false; if (name.length > 30) return false; return /^[0-9a-zA-Z_.\-#]+$/.test(name); } exports.saveConfig = functions.https.onCall((request, response) => { try { if (request.uid === undefined || request.obj === undefined) { console.error(`error saving config for ${request.uid} - missing input`); return -1; } let obj = request.obj; let err = false; Object.keys(obj).forEach((key) => { let val = obj[key]; if (Array.isArray(val)) { val.forEach((valarr) => { if (!isConfigKeyValid(valarr)) { err = true; console.error(`${key}: ${valarr} failed regex check`); } }); } else { if (!isConfigKeyValid(val)) { err = true; console.error(`${key}: ${val} failed regex check`); } } }); if (err) { console.error( `error saving config for ${request.uid} - bad input - ${JSON.stringify( request.obj )}` ); return -1; } return db .collection(`users`) .doc(request.uid) .set( { config: obj, }, { merge: true } ) .then((e) => { return 1; }) .catch((e) => { console.error( `error saving config to DB for ${request.uid} - ${e.message}` ); return -1; }); } catch (e) { console.error(`error saving config for ${request.uid} - ${e}`); return { resultCode: -999 }; } }); function generate(n) { var add = 1, max = 12 - add; if (n > max) { return generate(max) + generate(n - max); } max = Math.pow(10, n + add); var min = max / 10; // Math.pow(10, n) basically var number = Math.floor(Math.random() * (max - min + 1)) + min; return ("" + number).substring(add); } exports.generatePairingCode = functions.https.onCall((request, response) => { try { if (request === null) { console.error( `error while trying to generate discord pairing code - no input` ); return { status: -999, }; } return db .collection("users") .doc(request.uid) .get() .then((userDoc) => { userDocData = userDoc.data(); if (userDocData.discordPairingCode !== undefined) { console.log( `user ${request.uid} already has code ${userDocData.discordPairingCode}` ); return { status: 2, pairingCode: userDocData.discordPairingCode, }; } else { return db .collection("users") .get() .then((res) => { let existingCodes = []; res.docs.forEach((doc) => { let docData = doc.data(); if (docData.discordPairingCode !== undefined) { existingCodes.push(docData.discordPairingCode); } }); console.log(`existing codes ${JSON.stringify(existingCodes)}`); let randomCode = generate(9); while (existingCodes.includes(randomCode)) { randomCode = generate(9); } return db .collection("users") .doc(request.uid) .update( { discordPairingCode: randomCode, }, { merge: true } ) .then((res) => { console.log( `generated ${randomCode} for user ${request.uid}` ); return { status: 1, pairingCode: randomCode, }; }) .catch((e) => { console.error( `error while trying to set discord pairing code ${randomCode} for user ${request.uid} - ${e}` ); return { status: -999, }; }); }); } }); } catch (e) { console.error( `error while trying to generate discord pairing code for user ${request.uid} - ${e}` ); return { status: -999, }; } }); // exports.getConfig = functions.https.onCall((request,response) => { // try{ // if(request.uid === undefined){ // console.error(`error getting config for ${request.uid} - missing input`); // return -1; // } // return admin.firestore().collection(`users`).doc(request.uid).get().then(e => { // return e.data().config; // }).catch(e => { // console.error(`error getting config from DB for ${request.uid} - ${e.message}`); // return -1; // }); // }catch(e){ // console.error(`error getting config for ${request.uid} - ${e}`); // return {resultCode:-999}; // } // })