Declare EdgeDB types for resources

Author: CarsonF

src/components/comments/mention-notification/comment-via-mention-notification.dto.ts

@@ -1,9 +1,11 @@
 import { ObjectType } from '@nestjs/graphql';
 import { keys as keysOf } from 'ts-transformer-keys';
 import { SecuredProps } from '~/common';
-import { LinkTo } from '~/core/resources';
+import { e } from '~/core/edgedb';
+import { LinkTo, RegisterResource } from '~/core/resources';
 import { Notification } from '../../notifications';
 
+@RegisterResource({ db: e.Notification.CommentViaMention })
 @ObjectType({
   implements: [Notification],
 })
@@ -14,3 +16,12 @@ export class CommentViaMentionNotification extends Notification {
 
   readonly comment: LinkTo<'Comment'>;
 }
+
+declare module '~/core/resources/map' {
+  interface ResourceMap {
+    CommentMentionedNotification: typeof CommentViaMentionNotification;
+  }
+  interface ResourceDBMap {
+    CommentMentionedNotification: typeof e.Notification.CommentViaMention;
+  }
+}

src/components/notification-system/system-notification.dto.ts

@@ -2,8 +2,11 @@ import { Field, ObjectType } from '@nestjs/graphql';
 import { keys as keysOf } from 'ts-transformer-keys';
 import { SecuredProps } from '~/common';
 import { MarkdownScalar } from '~/common/markdown.scalar';
+import { RegisterResource } from '~/core';
+import { e } from '~/core/edgedb';
 import { Notification } from '../notifications';
 
+@RegisterResource({ db: e.Notification.System })
 @ObjectType({
   implements: [Notification],
 })
@@ -14,3 +17,12 @@ export class SystemNotification extends Notification {
   @Field(() => MarkdownScalar)
   readonly message: string;
 }
+
+declare module '~/core/resources/map' {
+  interface ResourceMap {
+    SystemNotification: typeof SystemNotification;
+  }
+  interface ResourceDBMap {
+    SystemNotification: typeof e.Notification.System;
+  }
+}

src/components/notifications/dto/notification.dto.ts

@@ -2,9 +2,10 @@ import { Field, InterfaceType } from '@nestjs/graphql';
 import { DateTime } from 'luxon';
 import { keys as keysOf } from 'ts-transformer-keys';
 import { DateTimeField, Resource, SecuredProps } from '~/common';
+import { e } from '~/core/edgedb';
 import { RegisterResource } from '~/core/resources';
 
-@RegisterResource()
+@RegisterResource({ db: e.default.Notification })
 @InterfaceType({
   implements: [Resource],
 })
@@ -28,7 +29,7 @@ declare module '~/core/resources/map' {
   interface ResourceMap {
     Notification: typeof Notification;
   }
-  // interface ResourceDBMap {
-  //   Notification: typeof e.Notification;
-  // }
+  interface ResourceDBMap {
+    Notification: typeof e.default.Notification;
+  }
 }