aslobot-matrix/src/helpers.ts
2025-12-28 18:06:07 +01:00

100 lines
2.3 KiB
TypeScript

import { config } from "./config.js";
import { state } from "./store/store.js";
import { type IUser, type TRole } from "./store/types.js";
import type { ILevel } from "./types.js";
const fixUserData = (user: IUser) => {
if (!user.experience) {
user.experience = 0;
}
if (!user.money) {
user.money = 0;
}
if (!user.aiCost) {
user.aiCost = 0;
}
};
const getUserById = (userId: string): IUser => {
const user = state.users.find((user) => user.id === userId);
if (!user) {
return {
id: ":",
role: "NONE",
experience: 0,
money: 0,
aiCost: 0,
lastMessageTimestamp: 0,
lastExperienceGainTimestamp: 0,
lastMoneyGainTimestamp: 0,
};
}
fixUserData(user);
return user;
};
const checkRoles = (roles: TRole[], userId: string) => {
const user = getUserById(userId);
return roles.includes(user.role);
};
const getLevel = (experience: number): ILevel => {
let tmpExperience = experience;
let expToNextLevel = config.app.experience.startingRequirement;
let level = 0;
while (tmpExperience >= expToNextLevel) {
level++;
tmpExperience -= expToNextLevel;
expToNextLevel = expToNextLevel *= config.app.experience.multiplier;
}
return {
level: level,
experience: Math.floor(experience),
experienceInLevel: Math.floor(tmpExperience),
expToNextLevel: Math.floor(expToNextLevel),
};
};
const getUserName = (user: IUser): string => {
const userPattern = /@[a-zA-Z0-9]*/;
const match = user.id.match(userPattern)?.at(0);
if (!match) {
return "";
}
return match.replaceAll("@", "");
};
const getUserFromMention = (mention: string | undefined): IUser | undefined => {
if (!mention) {
return undefined;
}
const regex = /\[\@\S*:aslan2142.space\]/;
const match = mention
.match(regex)
?.at(0)
?.replaceAll("[", "")
.replaceAll("]", "");
const user = state.users.find((user) => user.id === match);
if (!user) {
return undefined;
}
fixUserData(user);
return user;
};
export {
fixUserData,
getUserById,
checkRoles,
getLevel,
getUserName,
getUserFromMention,
};