Create unavailability edgedb repo

Co-authored-by: Carson Full <[email protected]>

Author: Andre Turner

src/components/user/unavailability/unavailability.edgedb.repository.ts

@@ -0,0 +1,70 @@
+import { Injectable } from '@nestjs/common';
+import { Range } from 'edgedb';
+import { ID, PublicOf } from '~/common';
+import { e, RepoFor } from '~/core/edgedb';
+import {
+  CreateUnavailability,
+  Unavailability,
+  UpdateUnavailability,
+} from './dto';
+import { UnavailabilityRepository } from './unavailability.repository';
+
+@Injectable()
+export class UnavailabilityEdgeDBRepository
+  extends RepoFor(Unavailability, {
+    hydrate: (unavailability) => ({
+      ...unavailability['*'],
+      start: e.assert_exists(e.range_get_lower(unavailability.dates)),
+      end: e.assert_exists(e.range_get_upper(unavailability.dates)),
+    }),
+  }).customize((cls) => {
+    return class extends cls {
+      async create(input: CreateUnavailability) {
+        const user = e.cast(e.User, e.uuid(input.userId));
+        const inserted = e.insert(e.User.Unavailability, {
+          description: input.description,
+          dates: new Range(input.start, input.end),
+        });
+        const updatedUser = e.update(user, () => ({
+          set: {
+            unavailabilities: { '+=': inserted },
+          },
+        }));
+        const query = e.select(inserted, (u) => ({
+          ...this.hydrate(u),
+          updatedUser, // Attach to query, so it is executed.
+        }));
+        return await this.db.run(query);
+      }
+
+      async update(input: UpdateUnavailability) {
+        const unavailability = e.cast(e.User.Unavailability, e.uuid(input.id));
+        const updated = e.update(unavailability, () => ({
+          set: {
+            description: input.description,
+            dates: e.cast(
+              e.range(e.datetime),
+              e.range(
+                input.start ?? e.range_get_lower(unavailability.dates),
+                input.end ?? e.range_get_upper(unavailability.dates),
+              ),
+            ),
+          },
+        }));
+        const query = e.select(updated, this.hydrate);
+        return await this.db.run(query);
+      }
+    };
+  })
+  implements PublicOf<UnavailabilityRepository>
+{
+  async getUserIdByUnavailability(id: ID) {
+    const unavailability = e.cast(e.User.Unavailability, e.uuid(id));
+    const query = e.assert_exists(
+      e.select(e.User, (user) => ({
+        filter_single: e.op(unavailability, 'in', user.unavailabilities),
+      })),
+    );
+    return await this.db.run(query);
+  }
+}

src/components/user/unavailability/unavailability.module.ts

@@ -1,5 +1,7 @@
 import { forwardRef, Module } from '@nestjs/common';
+import { splitDb } from '~/core/database';
 import { AuthorizationModule } from '../../authorization/authorization.module';
+import { UnavailabilityEdgeDBRepository } from './unavailability.edgedb.repository';
 import { UnavailabilityLoader } from './unavailability.loader';
 import { UnavailabilityRepository } from './unavailability.repository';
 import { UnavailabilityResolver } from './unavailability.resolver';
@@ -10,7 +12,7 @@ import { UnavailabilityService } from './unavailability.service';
   providers: [
     UnavailabilityResolver,
     UnavailabilityService,
-    UnavailabilityRepository,
+    splitDb(UnavailabilityRepository, UnavailabilityEdgeDBRepository),
     UnavailabilityLoader,
   ],
   exports: [UnavailabilityService],