fixes #46 Sanitize mattermost username to match openai regex.

yGuy · Jul 20, 2023 · dab85a8 · dab85a8
1 parent c6203eb
commit dab85a8
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "chatgpt-mattermost-bot",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "private": true,
   "scripts": {
     "start": "ts-node ./src/botservice.ts",

diff --git a/src/botservice.ts b/src/botservice.ts
@@ -18,7 +18,7 @@ if(!global.FormData) {
 }
 
 const name = process.env['MATTERMOST_BOTNAME'] || '@chatgpt'
-const contextMsgCount = Number(process.env['BOT_CONTEXT_MSG'] ?? 7)
+const contextMsgCount = Number(process.env['BOT_CONTEXT_MSG'] ?? 100)
 
 /* List of all registered plugins */
 const plugins: PluginBase<any>[] = [
@@ -185,6 +185,15 @@ async function userIdToName(userId: string): Promise<string> {
     } else {
         // username not in cache our outdated
         username = (await mmClient.getUser(userId)).username
+
+        if(!/^[a-zA-Z0-9_-]{1,64}$/.test(username)) {
+            username = username.replace(/[.@!?]/g,'_').slice(0, 64)
+        }
+
+        if(!/^[a-zA-Z0-9_-]{1,64}$/.test(username)) {
+            username = [...username.matchAll(/[a-zA-Z0-9_-]/g)].join('').slice(0, 64)
+        }
+
         usernameCache[userId] = {
             username: username,
             expireTime: Date.now() + 1000 * 60 * 5