scoresaber-reloadedv3/projects/website/src/common/database/database.ts

import Dexie, { EntityTable } from "dexie";
import BeatSaverMap from "./types/beatsaver-map";
import Settings from "./types/settings";
import { Friend } from "@/common/database/types/friends";
import ScoreSaberPlayerToken from "@ssr/common/types/token/scoresaber/score-saber-player-token";
import { scoresaberService } from "@ssr/common/service/impl/scoresaber";
import { setCookieValue } from "@ssr/common/utils/cookie-utils";

const SETTINGS_ID = "SSR"; // DO NOT CHANGE

export default class Database extends Dexie {
  /**
   * The settings for the website.
   */
  settings!: EntityTable<Settings, "id">;

  /**
   * Cached BeatSaver maps
   */
  beatSaverMaps!: EntityTable<BeatSaverMap, "hash">;

  /**
   * The added friends
   */
  friends!: EntityTable<Friend, "id">;

  constructor() {
    super("ScoreSaberReloaded");

    // Stores
    this.version(1).stores({
      settings: "id",
      beatSaverMaps: "hash",
      friends: "id",
    });

    // Mapped tables
    this.settings.mapToClass(Settings);
    this.beatSaverMaps.mapToClass(BeatSaverMap);

    // Populate default settings if the table is empty
    this.on("populate", () => this.populateDefaults());

    this.on("ready", async () => {
      const settings = await this.getSettings();
      // If the settings are not found, return
      if (settings == undefined || settings.playerId == undefined) {
        return;
      }
      await setCookieValue("playerId", settings.playerId);
    });
  }

  /**
   * Populates the default settings
   */
  async populateDefaults() {
    await this.resetSettings();
  }

  /**
   * Gets the settings from the database
   *
   * @returns the settings
   */
  async getSettings(): Promise<Settings | undefined> {
    return this.settings.get(SETTINGS_ID);
  }

  /**
   * Sets the settings in the database
   *
   * @param settings the settings to set
   * @returns the settings
   */
  async setSettings(settings: Settings) {
    return this.settings.update(SETTINGS_ID, settings);
  }

  /**
   * Adds a friend
   *
   * @param id the id of the friend
   */
  public async addFriend(id: string) {
    await this.friends.add({ id });
  }

  /**
   * Removes a friend
   *
   * @param id the id of the friend
   */
  public async removeFriend(id: string) {
    await this.friends.delete(id);
  }

  /**
   * Checks if this player is a friend
   *
   * @param id the id of the player
   */
  public async isFriend(id: string): Promise<boolean> {
    const friend = await this.friends.get(id);
    return friend != undefined;
  }

  /**
   * Gets all friends as {@link ScoreSaberPlayerToken}'s
   *
   * @returns the friends
   */
  public async getFriends(): Promise<ScoreSaberPlayerToken[]> {
    const friends = await this.friends.toArray();
    const players = await Promise.all(
      friends.map(async ({ id }) => {
        return await scoresaberService.lookupPlayer(id, true);
      })
    );
    return players.filter(player => player !== undefined) as ScoreSaberPlayerToken[];
  }

  /**
   * Resets the settings in the database
   */
  async resetSettings() {
    this.settings.delete(SETTINGS_ID);
    this.settings.add({
      id: SETTINGS_ID, // Fixed ID for the single settings object
      backgroundCover: "/assets/background.jpg",
    });

    return this.getSettings();
  }
}

export const db = new Database();