feat: make completion fragments user-specific

src/entity/base/BaseCommand.ts

@@ -35,11 +35,15 @@ export abstract class BaseCommand extends DataEntity<Array<string>> {
   }
 
   public getHeadOrNumber(key: string, defaultValue: number): number {
-    const value = Number(this.getHead(key));
-    if (isNaN(value)) {
-      return defaultValue;
+    if (this.has(key)) {
+      const value = Number(this.get(key));
+      if (isNaN(value)) {
+        return defaultValue;
+      } else {
+        return value;
+      }
     } else {
-      return value;
+      return defaultValue;
     }
   }
 

src/listener/ExpressListener.ts

@@ -9,10 +9,9 @@ import { Counter, Registry } from 'prom-client';
 import { Connection, Repository } from 'typeorm';
 
 import { BotServiceOptions } from 'src/BotService';
-import { Token, JwtFields } from 'src/entity/auth/Token';
+import { JwtFields, Token } from 'src/entity/auth/Token';
 import { Context, ContextOptions } from 'src/entity/Context';
 import { Message } from 'src/entity/Message';
-import { NotInitializedError } from 'src/error/NotInitializedError';
 import { Listener, ListenerData } from 'src/listener/Listener';
 import { SessionListener } from 'src/listener/SessionListener';
 import { ServiceModule } from 'src/module/ServiceModule';

src/listener/SessionListener.ts

@@ -1,5 +1,3 @@
-import { isNil } from 'lodash';
-
 import { BotServiceOptions } from 'src/BotService';
 import { User } from 'src/entity/auth/User';
 import { Message } from 'src/entity/Message';
@@ -40,13 +38,6 @@ export abstract class SessionListener<TData extends ListenerData> extends BaseLi
 
   public async getSession(uid: string): Promise<Session | undefined> {
     const now = this.clock.getSeconds();
-    const session = this.sessions.get(uid);
-
-    // session must have been created in the past and expire in the future
-    if (isNil(session) || session.createdAt.valueOf() >= now || session.expiresAt.valueOf() <= now) {
-      return;
-    } else {
-      return session;
-    }
+    return this.sessions.get(uid);
   }
 }

src/migration/0001546283532-FragmentUser.ts

@@ -0,0 +1,19 @@
+import { MigrationInterface, QueryRunner, TableColumn } from 'typeorm';
+
+import { TABLE_FRAGMENT } from 'src/entity/Fragment';
+
+const COLUMN_USER = new TableColumn({
+  name: 'userId',
+  type: 'varchar',
+});
+
+export class FragmentUser0001546283532 implements MigrationInterface {
+  public async up(query: QueryRunner): Promise<void> {
+    await query.clearTable(TABLE_FRAGMENT);
+    await query.addColumn(TABLE_FRAGMENT, COLUMN_USER);
+  }
+
+  public async down(query: QueryRunner): Promise<void> {
+    await query.dropColumn(TABLE_FRAGMENT, COLUMN_USER);
+  }
+}

src/module/MigrationModule.ts

@@ -13,6 +13,7 @@ import { CreateToken0001544317462 } from 'src/migration/0001544317462-CreateToke
 import { KeywordCommand0001545509108 } from 'src/migration/0001545509108-KeywordCommand';
 import { CreateTick0001546063195 } from 'src/migration/0001546063195-CreateTick';
 import { Dates0001546236755 } from 'src/migration/0001546236755-Dates';
+import { FragmentUser0001546283532 } from 'src/migration/0001546283532-FragmentUser';
 
 export class MigrationModule extends Module {
   public async configure(options: ModuleOptions): Promise<void> {
@@ -31,6 +32,7 @@ export class MigrationModule extends Module {
       KeywordCommand0001545509108,
       CreateTick0001546063195,
       Dates0001546236755,
+      FragmentUser0001546283532,
     ]);
   }
 }