monkeytype/public/js/db.js

const db = firebase.firestore();

let dbSnapshot = null;

async function db_getUserSnapshot() {
  let user = firebase.auth().currentUser;
  if (user == null) return false;
  let snap = {
    results: undefined,
    personalBests: {},
    tags: [],
    favouriteThemes: [],
  };
  // await db.collection('results')
  //     .orderBy('timestamp', 'desc')
  //     .where('uid', '==', user.uid)
  //     .get()
  //     .then(data => {
  //         // console.log('getting data from db!');
  //         data.docs.forEach(doc => {
  //             ret.push(doc.data());
  //         })
  //     })
  try {
    // await db
    //   .collection(`users/${user.uid}/results/`)
    //   .orderBy("timestamp", "desc")
    //   .get()
    //   .then((data) => {
    //     // console.log('getting data from db!');
    //     data.docs.forEach((doc) => {
    //       let result = doc.data();
    //       result.id = doc.id;
    //       snap.results.push(result);
    //     });
    //   })
    //   .catch((e) => {
    //     throw e;
    //   });
    await db
      .collection(`users/${user.uid}/tags/`)
      .get()
      .then((data) => {
        // console.log('getting data from db!');
        data.docs.forEach((doc) => {
          let tag = doc.data();
          tag.id = doc.id;
          snap.tags.push(tag);
        });
      })
      .catch((e) => {
        throw e;
      });
    await db
      .collection("users")
      .doc(user.uid)
      .get()
      .then((res) => {
        // console.log('getting data from db!');
        let data = res.data();
        try {
          if (data.personalBests !== undefined) {
            snap.personalBests = data.personalBests;
          }
          snap.discordId = data.discordId;
          snap.pairingCode = data.discordPairingCode;
          snap.config = data.config;
          snap.favouriteThemes =
            data.favouriteThemes === undefined ? [] : data.favouriteThemes;
          snap.globalStats = {
            time: data.timeTyping,
            started: data.startedTests,
            completed: data.completedTests
          }
        } catch (e) {
          //
        }
      })
      .catch((e) => {
        throw e;
      });
  } catch (e) {
    console.error(e);
    showNotification("Error getting account data.", 3000);
  }
  dbSnapshot = snap;
  return dbSnapshot;
}

async function db_getUserResults() {
  let user = firebase.auth().currentUser;
  if (user == null) return false;
  if (dbSnapshot === null) return false;
  if (dbSnapshot.results !== undefined) {
    return true;
  } else {
    try {
      return await db
        .collection(`users/${user.uid}/results/`)
        .orderBy("timestamp", "desc")
        .limit(1000)
        .get()
        .then((data) => {
          dbSnapshot.results = [];
          let len = data.docs.length;
          data.docs.forEach((doc, index) => {
            let result = doc.data();
            result.id = doc.id;
            dbSnapshot.results.push(result);
          });
          return true;
        })
        .catch((e) => {
          throw e;
        });
    } catch (e) {
      console.error(e);
      return false;
    }
  }
}

async function db_getUserHighestWpm(
  mode,
  mode2,
  punctuation,
  language,
  difficulty
) {
  function cont() {
    let topWpm = 0;
    dbSnapshot.results.forEach((result) => {
      if (
        result.mode == mode &&
        result.mode2 == mode2 &&
        result.punctuation == punctuation &&
        result.language == language &&
        result.difficulty == difficulty
      ) {
        if (result.wpm > topWpm) {
          topWpm = result.wpm;
        }
      }
    });
    return topWpm;
  }

  let retval;
  if (dbSnapshot == null || dbSnapshot.results === undefined) {
    // await db_getUserResults().then(data => {
    //     retval = cont();
    // });
    retval = 0;
  } else {
    retval = cont();
  }
  return retval;
}

async function db_getLocalPB(mode, mode2, punctuation, language, difficulty) {
  function cont() {
    let ret = 0;
    try {
      dbSnapshot.personalBests[mode][mode2].forEach((pb) => {
        if (
          pb.punctuation == punctuation &&
          pb.difficulty == difficulty &&
          pb.language == language
        ) {
          ret = pb.wpm;
        }
      });
      return ret;
    } catch (e) {
      return ret;
    }
  }

  let retval;
  if (dbSnapshot == null) {
    // await db_getUserResults().then(data => {
    //     retval = cont();
    // });
  } else {
    retval = cont();
  }
  return retval;
}

async function db_saveLocalPB(
  mode,
  mode2,
  punctuation,
  language,
  difficulty,
  wpm,
  acc,
  raw
) {
  function cont() {
    try {
      let found = false;
      if (dbSnapshot.personalBests[mode][mode2] === undefined) {
        dbSnapshot.personalBests[mode][mode2] = [];
      }
      dbSnapshot.personalBests[mode][mode2].forEach((pb) => {
        if (
          pb.punctuation == punctuation &&
          pb.difficulty == difficulty &&
          pb.language == language
        ) {
          found = true;
          pb.wpm = wpm;
          pb.acc = acc;
          pb.raw = raw;
        }
      });
      if (!found) {
        //nothing found
        dbSnapshot.personalBests[mode][mode2].push({
          language: language,
          difficulty: difficulty,
          punctuation: punctuation,
          wpm: wpm,
          acc: acc,
          raw: raw,
        });
      }
    } catch (e) {
      //that mode or mode2 is not found
      dbSnapshot.personalBests[mode] = {};
      dbSnapshot.personalBests[mode][mode2] = [
        {
          language: language,
          difficulty: difficulty,
          punctuation: punctuation,
          wpm: wpm,
          acc: acc,
          raw: raw,
        },
      ];
    }
  }

  let retval;
  if (dbSnapshot == null) {
    // await db_getUserResults().then(data => {
    //     retval = cont();
    // });
  } else {
    cont();
  }
}