[EdgeDB] Pin Queries (#3237)

Author: CarsonF

src/components/pin/pin.edgedb.repository.ts

@@ -0,0 +1,32 @@
+import { Injectable } from '@nestjs/common';
+import { ID, PublicOf } from '~/common';
+import { CommonRepository, e } from '~/core/edgedb';
+import { PinRepository } from './pin.repository';
+
+@Injectable()
+export class PinEdgeDBRepository
+  extends CommonRepository
+  implements PublicOf<PinRepository>
+{
+  async isPinned(id: ID) {
+    const resource = e.cast(e.Mixin.Pinnable, e.uuid(id));
+    const query = e.op(resource, 'in', e.global.currentUser.pins);
+    return await this.db.run(query);
+  }
+
+  async add(id: ID) {
+    const resource = e.cast(e.Mixin.Pinnable, e.uuid(id));
+    const query = e.update(e.global.currentUser, () => ({
+      set: { pins: { '+=': resource } },
+    }));
+    await this.db.run(query);
+  }
+
+  async remove(id: ID) {
+    const resource = e.cast(e.Mixin.Pinnable, e.uuid(id));
+    const query = e.update(e.global.currentUser, () => ({
+      set: { pins: { '-=': resource } },
+    }));
+    await this.db.run(query);
+  }
+}

src/components/pin/pin.module.ts

@@ -1,10 +1,16 @@
 import { Module } from '@nestjs/common';
+import { splitDb } from '~/core';
+import { PinEdgeDBRepository } from './pin.edgedb.repository';
 import { PinRepository } from './pin.repository';
 import { PinResolver } from './pin.resolver';
 import { PinService } from './pin.service';
 
 @Module({
-  providers: [PinResolver, PinService, PinRepository],
+  providers: [
+    PinResolver,
+    PinService,
+    splitDb(PinRepository, PinEdgeDBRepository),
+  ],
   exports: [PinService],
 })
 export class PinModule {}