monkeytype/functions/index.js
2020-07-22 14:42:15 +01:00

1357 lines
38 KiB
JavaScript

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.toLowerCase())) return false;
if (/bitly/.test(name.toLowerCase())) 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 db
.collection("users")
.doc(request.uid)
.get()
.then((doc) => {
if (
doc.data().name === undefined ||
doc.data().name === null ||
doc.data().name === ""
) {
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) {
db.collection("users")
.doc(request.uid)
.update({ name: requestUser.displayName })
.then(() => {
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 {
db.collection("users")
.doc(request.uid)
.update({ name: requestUser.displayName })
.then(() => {
return 0;
});
}
}
})
.catch((e) => {
console.error(`error getting all users - ${e}`);
});
}
});
} else {
console.log("name is good");
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 {
if (pbs[obj.mode][obj.mode2] === undefined) {
pbs[obj.mode][obj.mode2] = [];
}
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;
}
});
}
function stdDev(array) {
const n = array.length;
const mean = array.reduce((a, b) => a + b) / n;
return Math.sqrt(
array.map((x) => Math.pow(x - mean, 2)).reduce((a, b) => a + b) / n
);
}
exports.testCompleted = functions.https.onCall(async (request, response) => {
try {
if (request.uid === undefined || request.obj === undefined) {
console.error(`error saving result for ${request.uid} - missing input`);
return { resultCode: -999 };
}
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 { resultCode: -1 };
}
if (obj.wpm <= 0 || obj.wpm > 350 || obj.acc < 50 || obj.acc > 100) {
return { resultCode: -1 };
}
let keySpacing = null;
let keyDuration = null;
try {
keySpacing = {
average:
obj.keySpacing.reduce((previous, current) => (current += previous)) /
obj.keySpacing.length,
sd: stdDev(obj.keySpacing),
};
keyDuration = {
average:
obj.keyDuration.reduce((previous, current) => (current += previous)) /
obj.keyDuration.length,
sd: stdDev(obj.keyDuration),
};
} catch (e) {
console.error(
`cant verify key spacing or duration for user ${request.uid}! - ${e} - ${obj.keySpacing} ${obj.keyDuration}`
);
}
obj.keySpacingStats = keySpacing;
obj.keyDurationStats = keyDuration;
if (obj.mode == "time" && (obj.mode2 == 15 || obj.mode2 == 60)) {
} else {
obj.keySpacing = "removed";
obj.keyDuration = "removed";
}
emailVerified = await admin
.auth()
.getUser(request.uid)
.then((user) => {
return user.emailVerified;
});
return db
.collection("users")
.doc(request.uid)
.get()
.then((ret) => {
let userdata = ret.data();
let name = userdata.name === undefined ? false : userdata.name;
let banned = userdata.banned === undefined ? false : userdata.banned;
let verified = userdata.verified;
request.obj.name = name;
//check keyspacing and duration here
if (obj.mode === "time") {
if (verified === false || verified === undefined) {
if (keySpacing !== null && keyDuration !== null) {
if (
keySpacing.sd <= 15 ||
keyDuration.sd <= 10 ||
keyDuration.average < 15
) {
console.error(
`possible bot detected by user (${obj.wpm} ${obj.rawWpm} ${
obj.acc
}) ${request.uid} ${name} - spacing ${JSON.stringify(
keySpacing
)} duration ${JSON.stringify(keyDuration)}`
);
return { resultCode: -2 };
}
if (
(keySpacing.sd > 15 && keySpacing.sd <= 25) ||
(keyDuration.sd > 10 && keyDuration.sd <= 15) ||
(keyDuration.average > 15 && keyDuration.average <= 20)
) {
console.error(
`very close to bot threshold by user (${obj.wpm} ${
obj.rawWpm
} ${obj.acc}) ${
request.uid
} ${name} - spacing ${JSON.stringify(
keySpacing
)} duration ${JSON.stringify(keyDuration)}`
);
}
} else {
return { resultCode: -3 };
}
}
}
return db
.collection(`users/${request.uid}/results`)
.add(obj)
.then((e) => {
let createdDocId = e.id;
return Promise.all([
checkLeaderboards(
request.obj,
"global",
banned,
name,
verified,
emailVerified
),
checkLeaderboards(
request.obj,
"daily",
banned,
name,
verified,
emailVerified
),
checkIfPB(request.uid, request.obj),
])
.then((values) => {
let globallb = values[0].insertedAt;
let dailylb = values[1].insertedAt;
let ispb = values[2];
// console.log(values);
let usr =
userdata.discordId !== undefined
? userdata.discordId
: userdata.name;
if (
globallb !== null &&
globallb.insertedAt >= 0 &&
globallb.insertedAt <= 9 &&
globallb.newBest
) {
let lbstring = `${obj.mode} ${obj.mode2} global`;
console.log(
`sending command to the bot to announce lb update ${
userdata.discordId
} ${globallb + 1} ${lbstring} ${obj.wpm}`
);
announceLbUpdate(
usr,
globallb.insertedAt + 1,
lbstring,
obj.wpm
);
}
let returnobj = {
resultCode: null,
globalLeaderboard: globallb,
dailyLeaderboard: dailylb,
lbBanned: banned,
name: name,
createdId: createdDocId,
needsToVerify: values[0].needsToVerify,
needsToVerifyEmail: values[0].needsToVerifyEmail,
};
if (ispb) {
console.log(
`saved result for ${
request.uid
} (new PB) - ${JSON.stringify(request.obj)}`
);
if (
obj.mode === "time" &&
String(obj.mode2) === "60" &&
userdata.discordId !== null &&
userdata.discordId !== undefined
) {
console.log(
`sending command to the bot to update the role for user ${request.uid} with wpm ${obj.wpm}`
);
updateDiscordRole(userdata.discordId, Math.round(obj.wpm));
}
returnobj.resultCode = 2;
} else {
console.log(
`saved result for ${request.uid} - ${JSON.stringify(
request.obj
)}`
);
returnobj.resultCode = 1;
}
return returnobj;
})
.catch((e) => {
console.error(
`error saving result when checking for PB / checking leaderboards for ${request.uid} - ${e.message}`
);
return { resultCode: -999, message: e.message };
});
})
.catch((e) => {
console.error(
`error saving result when adding result to the db for ${request.uid} - ${e.message}`
);
return { resultCode: -999, message: e.message };
});
})
.catch((e) => {
console.error(
`error saving result when getting user data for ${request.uid} - ${e.message}`
);
return { resultCode: -999, message: e.message };
});
} catch (e) {
console.error(
`error saving result for ${request.uid} - ${JSON.stringify(
request.obj
)} - ${e}`
);
return { resultCode: -999, message: e.message };
}
});
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);
}
class Leaderboard {
constructor(size, mode, mode2, type, starting) {
this.size = size;
this.board = [];
this.mode = mode;
this.mode2 = parseInt(mode2);
this.type = type;
if (starting !== undefined && starting !== null) {
starting.forEach((entry) => {
if (
entry.mode == this.mode &&
parseInt(entry.mode2) === parseInt(this.mode2)
) {
let hid = entry.hidden === undefined ? false : entry.hidden;
this.board.push({
uid: entry.uid,
name: entry.name,
wpm: parseFloat(entry.wpm),
raw: parseFloat(entry.raw),
acc: parseFloat(entry.acc),
mode: entry.mode,
mode2: parseInt(entry.mode2),
timestamp: entry.timestamp,
hidden: hid,
});
}
});
}
this.sortBoard();
this.clipBoard();
}
sortBoard() {
this.board.sort((a, b) => {
if (a.wpm === b.wpm) {
if (a.acc === b.acc) {
return a.timestamp - b.timestamp;
} else {
return b.acc - a.acc;
}
} else {
return b.wpm - a.wpm;
}
});
}
clipBoard() {
let boardLength = this.board.length;
if (boardLength > this.size) {
while (this.board.length !== this.size) {
this.board.pop();
}
}
}
logBoard() {
console.log(this.board);
}
removeDuplicates(insertedAt, uid) {
//return true if a better result is found
let found = false;
// let ret;
let foundAt = null;
if (this.board !== undefined) {
this.board.forEach((entry, index) => {
if (entry.uid === uid) {
if (found) {
this.board.splice(index, 1);
// if (index > insertedAt) {
// //removed old result
// ret = false;
// } else {
// ret = true;
// }
} else {
found = true;
foundAt = index;
}
}
});
}
// console.log(ret);
// return ret;
return foundAt;
}
insert(a) {
let insertedAt = -1;
if (a.mode == this.mode && parseInt(a.mode2) === parseInt(this.mode2)) {
this.board.forEach((b, index) => {
if (insertedAt !== -1) return;
if (a.wpm === b.wpm) {
if (a.acc === b.acc) {
if (a.timestamp < b.timestamp) {
this.board.splice(index, 0, {
uid: a.uid,
name: a.name,
wpm: parseFloat(a.wpm),
raw: parseFloat(a.rawWpm),
acc: parseFloat(a.acc),
mode: a.mode,
mode2: parseInt(a.mode2),
timestamp: a.timestamp,
hidden: a.hidden === undefined ? false : a.hidden,
});
insertedAt = index;
}
} else {
if (a.acc > b.acc) {
this.board.splice(index, 0, {
uid: a.uid,
name: a.name,
wpm: parseFloat(a.wpm),
raw: parseFloat(a.rawWpm),
acc: parseFloat(a.acc),
mode: a.mode,
mode2: parseInt(a.mode2),
timestamp: a.timestamp,
hidden: a.hidden === undefined ? false : a.hidden,
});
insertedAt = index;
}
}
} else {
if (a.wpm > b.wpm) {
this.board.splice(index, 0, {
uid: a.uid,
name: a.name,
wpm: parseFloat(a.wpm),
raw: parseFloat(a.rawWpm),
acc: parseFloat(a.acc),
mode: a.mode,
mode2: parseInt(a.mode2),
timestamp: a.timestamp,
hidden: a.hidden === undefined ? false : a.hidden,
});
insertedAt = index;
}
}
});
if (this.board.length < this.size && insertedAt === -1) {
this.board.push({
uid: a.uid,
name: a.name,
wpm: parseFloat(a.wpm),
raw: parseFloat(a.rawWpm),
acc: parseFloat(a.acc),
mode: a.mode,
mode2: parseInt(a.mode2),
timestamp: a.timestamp,
hidden: a.hidden === undefined ? false : a.hidden,
});
insertedAt = this.board.length - 1;
}
// console.log("before duplicate remove");
// console.log(this.board);
let newBest = false;
let foundAt = null;
if (insertedAt >= 0) {
// if (this.removeDuplicates(insertedAt, a.uid)) {
// insertedAt = -2;
// }
foundAt = this.removeDuplicates(insertedAt, a.uid);
if (foundAt >= insertedAt) {
//new better result
newBest = true;
}
}
// console.log(this.board);
this.clipBoard();
return {
insertedAt: insertedAt,
newBest: newBest,
foundAt: foundAt,
};
} else {
return {
insertedAt: -999,
};
}
}
}
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,
};
}
});
async function checkLeaderboards(
resultObj,
type,
banned,
name,
verified,
emailVerified
) {
// return {
// insertedAt: null,
// };
try {
if (emailVerified === false)
return {
insertedAt: null,
needsToVerifyEmail: true,
};
if (!name)
return {
insertedAt: null,
noName: true,
};
if (banned)
return {
insertedAt: null,
banned: true,
};
if (verified === false)
return {
insertedAt: null,
needsToVerify: true,
};
if (
resultObj.mode === "time" &&
["15", "60"].includes(String(resultObj.mode2)) &&
resultObj.language === "english"
) {
return db
.collection("leaderboards")
.where("mode", "==", String(resultObj.mode))
.where("mode2", "==", String(resultObj.mode2))
.where("type", "==", type)
.get()
.then((ret) => {
if (ret.docs.length === 0) {
//no lb found, create
console.log(
`no ${resultObj.mode} ${resultObj.mode2} ${type} leaderboard found - creating`
);
let toAdd = {
size: 20,
mode: String(resultObj.mode),
mode2: String(resultObj.mode2),
type: type,
};
return db
.collection("leaderboards")
.doc(
`${String(resultObj.mode)}_${String(resultObj.mode2)}_${type}`
)
.set(toAdd)
.then((ret) => {
return cont(
`${String(resultObj.mode)}_${String(
resultObj.mode2
)}_${type}`,
toAdd
);
});
} else {
//continue
return cont(
`${String(resultObj.mode)}_${String(resultObj.mode2)}_${type}`,
ret.docs[0].data()
);
}
});
function cont(docid, documentData) {
let boardInfo = documentData;
let boardData = boardInfo.board;
// console.log(`info ${JSON.stringify(boardInfo)}`);
// console.log(`data ${JSON.stringify(boardData)}`);
let lb = new Leaderboard(
boardInfo.size,
resultObj.mode,
resultObj.mode2,
boardInfo.type,
boardData
);
// console.log("board created");
// lb.logBoard();
console.log(
`inserting result by user ${resultObj.uid} ${resultObj.wpm} ${resultObj.rawWpm} ${resultObj.acc} into leaderboard ${resultObj.mode} ${resultObj.mode2} ${type}`
);
let insertResult = lb.insert(resultObj);
// console.log("board after inseft");
// lb.logBoard();
// console.log(lb);
if (insertResult.insertedAt >= 0) {
//update the database here
console.log(
`leaderboard changed ${resultObj.mode} ${
resultObj.mode2
} ${type} - ${JSON.stringify(lb.board)}`
);
db.collection("leaderboards").doc(docid).set(
{
size: lb.size,
type: lb.type,
board: lb.board,
},
{ merge: true }
);
} else {
// console.log("board is the same");
}
return {
insertedAt: insertResult,
};
}
} else {
return {
insertedAt: null,
};
}
} catch (e) {
console.error(
`error while checking leaderboards - ${e} - ${type} ${resultObj}`
);
return {
insertedAt: null,
};
}
}
exports.getLeaderboard = functions.https.onCall((request, response) => {
return db
.collection("leaderboards")
.where("mode", "==", String(request.mode))
.where("mode2", "==", String(request.mode2))
.where("type", "==", String(request.type))
.get()
.then(async (data) => {
// console.log("got data");
if (data.docs.length === 0) return null;
let lbdata = data.docs[0].data();
if (lbdata.board !== undefined) {
// console.log("replacing users");
// for (let i = 0; i < lbdata.board.length; i++) {
// await db
// .collection("users")
// .doc(lbdata.board[i].uid)
// .get()
// .then((doc) => {
// if (
// lbdata.board[i].uid !== null &&
// lbdata.board[i].uid === request.uid
// ) {
// lbdata.board[i].currentUser = true;
// }
// lbdata.board[i].name = doc.data().name;
// lbdata.board[i].uid = null;
// });
// }
lbdata.board.forEach((boardentry) => {
if (boardentry.uid !== null && boardentry.uid === request.uid) {
boardentry.currentUser = true;
}
boardentry.uid = null;
});
// console.log(lbdata);
if (request.type === "daily") {
let resetTime = new Date(Date.now());
resetTime.setHours(0, 0, 0, 0);
resetTime.setDate(resetTime.getUTCDate() + 1);
resetTime = resetTime.valueOf();
lbdata.resetTime = resetTime;
}
return lbdata;
} else {
if (
lbdata.board === undefined ||
lbdata.board === [] ||
lbdata.board.length === 0
) {
return lbdata;
} else {
return [];
}
}
});
});
exports.scheduledFunctionCrontab = functions.pubsub
.schedule("00 00 * * *")
.timeZone("Africa/Abidjan")
.onRun((context) => {
try {
console.log("moving daily leaderboards to history");
db.collection("leaderboards")
.where("type", "==", "daily")
.get()
.then((res) => {
res.docs.forEach((doc) => {
let lbdata = doc.data();
announceDailyLbResult(lbdata);
t = new Date();
db.collection("leaderboards_history")
.doc(
`${t.getUTCDate()}_${t.getUTCMonth()}_${t.getUTCFullYear()}_${
lbdata.mode
}_${lbdata.mode2}`
)
.set(lbdata);
db.collection("leaderboards").doc(doc.id).set(
{
board: [],
},
{ merge: true }
);
});
});
return null;
} catch (e) {
console.error(`error while moving daily leaderboards to history - ${e}`);
}
});
async function announceLbUpdate(discordId, pos, lb, wpm) {
db.collection("bot-commands").add({
command: "sayLbUpdate",
arguments: [discordId, pos, lb, wpm],
executed: false,
requestTimestamp: Date.now(),
});
}
async function announceDailyLbResult(lbdata) {
db.collection("bot-commands").add({
command: "announceDailyLbResult",
arguments: [lbdata],
executed: false,
requestTimestamp: Date.now(),
});
}