rename bot directory to src, add chatwidget

2025-12-19 08:41:39 +01:00 · 2025-07-09 16:50:16 +02:00
parent 8fd889856b
commit 3e025a586a
59 changed files with 417 additions and 1 deletions
--- a/src/user.ts
+++ b/src/user.ts
@@ -0,0 +1,100 @@
+import { redis } from "bun";
+import { chatterApi } from ".";
+import { HelixUser } from "@twurple/api"
+
+const EXPIRETIME = 60 * 60 // 60 minutes
+
+// The objective of this class is to:
+// store displayname, username and id to reduce api calls
+// keep track of temporary user specific flags (vulnerable to explosives, locked from using items)
+//
+// The userlookup key is used to find id's based on the username.
+//
+// The vulnchatters and userlookup look similar, but they're not the same
+// userlookup expiration gets set when user chats or is targeted by another user
+// vulnchatters only gets set when user chats
+
+export class User {
+  public username!: string;
+  public id!: string;
+  public displayName!: string;
+
+  static async initUsername(username: string): Promise<User | null> {
+    const userObj = new User();
+    userObj.username = username;
+    const userid = await redis.get(`userlookup:${username}`);
+    if (!userid) {
+      const userdata = await chatterApi.users.getUserByName(username);
+      if (!userdata) return null;
+      userObj._setCache(userdata);
+      userObj.id = userdata.id;
+      userObj.displayName = userdata.displayName;
+    } else {
+      const displayname = await redis.get(`user:${userid}:displayName`);
+      userObj._setExpire(userid, username);
+      userObj.id = userid;
+      userObj.displayName = displayname!;
+    };
+    return userObj;
+  };
+
+  static async initUserId(userId: string): Promise<User | null> {
+    const userObj = new User();
+    userObj.id = userId;
+    if (!await redis.exists(`user:${userId}:displayName`)) {
+      const userdata = await chatterApi.users.getUserById(userId);
+      if (!userdata) return null;
+      userObj._setCache(userdata);
+      userObj.username = userdata.name;
+      userObj.displayName = userdata.displayName;
+    } else {
+      const [displayName, username] = await Promise.all([
+        redis.get(`user:${userId}:displayName`),
+        redis.get(`user:${userId}:username`)
+      ]);
+      userObj._setExpire(userId, username!);
+      userObj.username = username!;
+      userObj.displayName = displayName!;
+    };
+    return userObj;
+  };
+
+  private async _setCache(userdata: HelixUser) {
+    await Promise.all([
+      redis.set(`user:${userdata.id}:displayName`, userdata.displayName),
+      redis.set(`user:${userdata.id}:username`, userdata.name),
+      redis.set(`userlookup:${userdata.name}`, userdata.id)
+    ]);
+    await this._setExpire(userdata.id, userdata.name);
+  };
+
+  private async _setExpire(userId: string, userName: string) {
+    await Promise.all([
+      redis.expire(`user:${userId}:displayName`, EXPIRETIME),
+      redis.expire(`user:${userId}:username`, EXPIRETIME),
+      redis.expire(`userlookup:${userName}`, EXPIRETIME)
+    ]);
+  };
+
+  public async itemLock(): Promise<boolean> {
+    const lock = await redis.get(`user:${this.id}:itemlock`);
+    return lock === '1';
+  };
+
+  public async setLock(): Promise<void> {
+    await redis.set(`user:${this.id}:itemlock`, '1');
+  };
+
+  public async clearLock(): Promise<void> {
+    await redis.set(`user:${this.id}:itemlock`, '0');
+  };
+
+  public async makeVulnerable(): Promise<void> {
+    await redis.set(`vulnchatters:${this.id}`, this.displayName);
+    await redis.expire(`vulnchatters:${this.id}`, Math.floor(EXPIRETIME / 2)); // Vulnerable chatter gets removed from the pool after 30 minutes
+  };
+
+  public async makeInvulnerable(): Promise<void> {
+    await redis.del(`vulnchatters:${this.id}`);
+  };
+};