zkldi_Tachi/bot/src/utils/utils.ts

import { APIApplicationCommandOptionChoice } from "discord-api-types";
import { FormatGame, Game, Playtypes } from "tachi-common";
import { IDStrings, UGSRatingsLookup } from "tachi-common/js/types";
import { ProcessEnv } from "../setup";
import { PrependTachiUrl } from "./fetch-tachi";

export interface SimpleGameType<T extends Game> {
	game: T;
	playtype: Playtypes[T];
}

export const simpleGameTypeToString = <T extends Game>(game: SimpleGameType<T>): IDStrings =>
	<IDStrings>`${game.game}:${game.playtype}`;

export const stringToSimpleGameType = (game: string): SimpleGameType<Game> => {
	return { game: <Game>game.split(":")[0], playtype: <Playtypes[never]>game.split(":")[1] };
};

export const formatGameWrapper = (game: SimpleGameType<Game>): string => FormatGame(game.game, game.playtype);

export const capitalise = (s: string): string => (s && s[0].toUpperCase() + s.slice(1)) || "";

export const prettyRatingString = <I extends IDStrings = IDStrings>(rating: UGSRatingsLookup[I]): string => {
	return capitalise(rating);
};

export const getPfpUrl = (userId: number): string => {
	if (ProcessEnv.ENV !== "prod") {
		return `https://kamaitachi.xyz/static/images/users/${userId}-pfp.png`;
	}
	return PrependTachiUrl(`/users/${userId}/pfp`);
};

/** @TODO @zkldi is there somewhere this exists already? */
export const gameIdentifierStrings = [
	"iidx:SP",
	"iidx:DP",
	"sdvx:Single",
	"usc:Single",
	"ddr:SP",
	"ddr:DP",
	"maimai:Single",
	"museca:Single",
	"bms:7K",
	"bms:14K",
	"chunithm:Single",
	"gitadora:Gita",
	"gitadora:Dora"
];
export const gamesToChoicesObject = (): [name: string, value: string][] => {
	return gameIdentifierStrings.map((identifier) => {
		return [formatGameWrapper(stringToSimpleGameType(identifier)), identifier];
	});
};