monkeytype/backend/dao/ape-keys.ts

import _ from "lodash";
import db from "../init/db";
import { Filter, ObjectId, MatchKeysAndValues } from "mongodb";
import MonkeyError from "../utils/error";

const COLLECTION_NAME = "ape-keys";

function getApeKeyFilter(
  uid: string,
  keyId: string
): Filter<MonkeyTypes.ApeKey> {
  return {
    _id: new ObjectId(keyId),
    uid,
  };
}

class ApeKeysDAO {
  static async getApeKeys(uid: string): Promise<MonkeyTypes.ApeKey[]> {
    return await db
      .collection<MonkeyTypes.ApeKey>(COLLECTION_NAME)
      .find({ uid })
      .toArray();
  }

  static async getApeKey(keyId: string): Promise<MonkeyTypes.ApeKey | null> {
    return await db
      .collection<MonkeyTypes.ApeKey>(COLLECTION_NAME)
      .findOne({ _id: new ObjectId(keyId) });
  }

  static async countApeKeysForUser(uid: string): Promise<number> {
    const apeKeys = await this.getApeKeys(uid);
    return _.size(apeKeys);
  }

  static async addApeKey(apeKey: MonkeyTypes.ApeKey): Promise<string> {
    const insertionResult = await db
      .collection<MonkeyTypes.ApeKey>(COLLECTION_NAME)
      .insertOne(apeKey);
    return insertionResult.insertedId.toHexString();
  }

  private static async updateApeKey(
    uid: string,
    keyId: string,
    updates: MatchKeysAndValues<MonkeyTypes.ApeKey>
  ): Promise<void> {
    const updateResult = await db
      .collection<MonkeyTypes.ApeKey>(COLLECTION_NAME)
      .updateOne(getApeKeyFilter(uid, keyId), {
        $inc: { useCount: _.has(updates, "lastUsedOn") ? 1 : 0 },
        $set: _.pickBy(updates, (value) => !_.isNil(value)),
      });

    if (updateResult.modifiedCount === 0) {
      throw new MonkeyError(404, "ApeKey not found");
    }
  }

  static async editApeKey(
    uid: string,
    keyId: string,
    name: string,
    enabled: boolean
  ): Promise<void> {
    const apeKeyUpdates = {
      name,
      enabled,
      modifiedOn: Date.now(),
    };

    await this.updateApeKey(uid, keyId, apeKeyUpdates);
  }

  static async updateLastUsedOn(uid: string, keyId: string): Promise<void> {
    const apeKeyUpdates = {
      lastUsedOn: Date.now(),
    };

    await this.updateApeKey(uid, keyId, apeKeyUpdates);
  }

  static async deleteApeKey(uid: string, keyId: string): Promise<void> {
    const deletionResult = await db
      .collection<MonkeyTypes.ApeKey>(COLLECTION_NAME)
      .deleteOne(getApeKeyFilter(uid, keyId));

    if (deletionResult.deletedCount === 0) {
      throw new MonkeyError(404, "ApeKey not found");
    }
  }
}

export default ApeKeysDAO;