chore: auto generate NOTIFICATION_TYPE_CODE_MAP from NOTIFICATION_TYP…

…E_MAP
pulsate-dev · Dec 17, 2024 · 7ff3fa9 · 7ff3fa9
1 parent 7497cfd
commit 7ff3fa9
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 19 deletions.
diff --git a/pkg/notification/adaptor/repository/dummy/notification.ts b/pkg/notification/adaptor/repository/dummy/notification.ts
@@ -63,21 +63,20 @@ export class InMemoryNotificationRepository implements NotificationRepository {
     const cursor = Option.unwrap(filter.cursor);
 
     switch (cursor.type) {
-      case 'after':
-        return (() => {
-          const afterIndex = res
-            .reverse()
-            .findIndex((n) => n.getID() === cursor.id);
-          return Result.ok(
-            res.slice(afterIndex + 1, afterIndex + limit).reverse(),
-          );
-        })();
-
-      case 'before':
+      case 'after': {
+        const afterIndex = res
+          .reverse()
+          .findIndex((n) => n.getID() === cursor.id);
+        return Result.ok(
+          res.slice(afterIndex + 1, afterIndex + limit).reverse(),
+        );
+      }
+      case 'before': {
         return (() => {
           const beforeIndex = res.findIndex((n) => n.getID() === cursor.id);
           return Result.ok(res.slice(beforeIndex + 1, beforeIndex + 1 + limit));
         })();
+      }
     }
   }
 

diff --git a/pkg/notification/adaptor/repository/prisma/notification.ts b/pkg/notification/adaptor/repository/prisma/notification.ts
@@ -38,17 +38,18 @@ export class PrismaNotificationRepository implements NotificationRepository {
   private readonly NOTIFICATION_TYPE_CODE_MAP: Record<
     number,
     NotificationType
-  > = {
-    1: 'followed',
-    2: 'followRequested',
-    3: 'followAccepted',
-    4: 'mentioned',
-    5: 'renoted',
-    6: 'reacted',
-  } as const;
+  > = [];
 
   constructor(prisma: PrismaClient) {
     this.prisma = prisma;
+
+    for (const [k, v] of Object.entries(this.NOTIFICATION_TYPE_MAP)) {
+      if (v in this.NOTIFICATION_TYPE_CODE_MAP)
+        throw new Error('Duplicate value');
+
+      this.NOTIFICATION_TYPE_CODE_MAP[v] = k as NotificationType;
+    }
+    Object.freeze(this.NOTIFICATION_TYPE_CODE_MAP);
   }
 
   private serialize(