Finish ToolUsage module

Author: CarsonF

dbschema/resource.gel

@@ -1,3 +1,5 @@
 module default {
-  abstract type Resource extending Mixin::Audited;
+  abstract type Resource extending Mixin::Audited {
+    tools := .<container[is Tool::Usage];
+  }
 }

src/components/authorization/policies/by-role/consultant.policy.ts

@@ -26,6 +26,7 @@ export const projectTransitions = () =>
     r.ProgressReportTeamNews,
     r.ProgressReportMedia,
     r.ProgressReportVarianceExplanation,
+    r.ToolUsage,
     r.StepProgress,
   ].map((it) => it.when(member).read),
 

src/components/authorization/policies/by-role/field-operations-director.policy.ts

@@ -21,6 +21,7 @@ import { Policy, Role } from '../util';
     'Field Ops Rejects Proposal',
   ).execute,
   r.PeriodicReport.edit,
+  r.ToolUsage.edit.create.delete,
   r.StepProgress.edit,
 ])
 export class FieldOperationsDirectorPolicy {}

src/components/authorization/policies/by-role/project-manager.policy.ts

@@ -163,6 +163,7 @@ export const momentumProjectsTransitions = () =>
       ])
       .children((c) => c.posts.read.create),
     r.ProjectMember.read.when(member).edit.create.delete,
+    r.ToolUsage.read.when(member).edit.create.delete,
     [r.StepProgress].flatMap((it) => [
       it.whenAll(member, variant('partner')).read,
       it.whenAll(member, variant('official')).edit,

src/components/tools/tool-usage/dto/create-tool-usage.dto.ts

@@ -1,5 +1,5 @@
 import { Field, InputType, ObjectType } from '@nestjs/graphql';
-import { CalendarDate, DateField, type ID, IdField } from '~/common';
+import { type CalendarDate, DateField, type ID, IdField } from '~/common';
 import { ToolUsage } from './tool-usage.dto';
 
 @InputType()
@@ -15,7 +15,7 @@ export abstract class CreateToolUsage {
   readonly tool: ID<'Tool'>;
 
   @DateField({ nullable: true })
-  readonly startDate?: CalendarDate;
+  readonly startDate?: CalendarDate | null;
 }
 
 @ObjectType()

src/components/tools/tool-usage/dto/tool-usage.dto.ts

@@ -3,7 +3,6 @@ import {
   Resource,
   type Secured,
   SecuredDateNullable,
-  SecuredProperty,
   type SetUnsecuredType,
   type UnsecuredDto,
 } from '~/common';
@@ -27,12 +26,9 @@ export class ToolUsage extends Resource {
   @Field()
   readonly startDate: SecuredDateNullable;
 
-  readonly creator: LinkTo<'User'>;
+  readonly creator: LinkTo<'Actor'>;
 }
 
-@ObjectType()
-export class SecuredToolUsage extends SecuredProperty(ToolUsage) {}
-
 declare module '~/core/resources/map' {
   interface ResourceMap {
     ToolUsage: typeof ToolUsage;

src/components/tools/tool-usage/dto/update-tool-usage.dto.ts

@@ -5,18 +5,12 @@ import { ToolUsage } from './tool-usage.dto';
 @InputType()
 export abstract class UpdateToolUsage {
   @IdField()
-  readonly id: ID;
+  readonly id: ID<ToolUsage>;
 
   @DateField({ nullable: true })
   readonly startDate?: CalendarDate | null;
 }
 
-@InputType()
-export abstract class UpdateToolUsageInput {
-  @Field()
-  readonly usage: UpdateToolUsage;
-}
-
 @ObjectType()
 export abstract class UpdateToolUsageOutput {
   @Field()

src/components/tools/tool-usage/resource-tools.resolver.ts

@@ -3,17 +3,14 @@ import { Resource } from '~/common';
 import { ToolUsage } from './dto';
 import { ToolUsageService } from './tool-usage.service';
 
-@Resolver(() => Resource, { isAbstract: true })
+@Resolver(() => Resource)
 export class ResourceToolsResolver {
   constructor(private readonly toolUsageService: ToolUsageService) {}
 
   @ResolveField(() => [ToolUsage], {
-    description: 'Tool usages for this resource',
+    description: 'Tools used in this resource',
   })
-  async tools(@Parent() resource: Resource): Promise<ToolUsage[]> {
-    if (resource.__typename === 'ToolUsage' || resource.__typename === 'Tool') {
-      return [];
-    }
-    return await this.toolUsageService.findByContainerId(resource.id);
+  async tools(@Parent() resource: Resource): Promise<readonly ToolUsage[]> {
+    return await this.toolUsageService.readByContainer(resource);
   }
 }

src/components/tools/tool-usage/tool-usage.gel.repository.ts

@@ -0,0 +1,38 @@
+import { Injectable } from '@nestjs/common';
+import { type ID, type PublicOf } from '~/common';
+import { e, RepoFor } from '~/core/gel';
+import { ToolUsage } from './dto';
+import { type ToolUsageRepository as Neo4jRepository } from './tool-usage.neo4j.repository';
+
+const resAsBaseNode = e.shape(e.Resource, (res) => ({
+  identity: res.id,
+  labels: e.array_agg(e.set(res.__type__.name)),
+  properties: e.select({
+    id: res.id,
+    createdAt: res.createdAt,
+  }),
+}));
+
+@Injectable()
+export class ToolUsageRepository
+  extends RepoFor(ToolUsage, {
+    hydrate: (usage) => ({
+      ...usage['*'],
+      container: resAsBaseNode(usage.container),
+      tool: usage.tool['*'],
+      creator: usage.createdBy,
+    }),
+  })
+  implements PublicOf<Neo4jRepository>
+{
+  async listForContainer(container: ID) {
+    return await this.db.run(this.listForContainerQuery, { container });
+  }
+  private readonly listForContainerQuery = e.params(
+    { container: e.uuid },
+    ($) => {
+      const container = e.cast(e.Resource, $.container);
+      return e.select(container.tools, this.hydrate);
+    },
+  );
+}

src/components/tools/tool-usage/tool-usage.module.ts

@@ -1,20 +1,22 @@
 import { Module } from '@nestjs/common';
+import { splitDb } from '~/core/database';
 import { ToolCoreModule } from '../tool/tool.module';
 import { ResourceToolsResolver } from './resource-tools.resolver';
+import { ToolUsageRepository as GelRepository } from './tool-usage.gel.repository';
 import { ToolUsageLoader } from './tool-usage.loader';
-import { ToolUsageRepository } from './tool-usage.repository';
+import { ToolUsageRepository as Neo4jRepository } from './tool-usage.neo4j.repository';
 import { ToolUsageResolver } from './tool-usage.resolver';
 import { ToolUsageService } from './tool-usage.service';
 
 @Module({
   imports: [ToolCoreModule],
   providers: [
-    ToolUsageService,
-    ToolUsageRepository,
     ToolUsageResolver,
-    ToolUsageLoader,
     ResourceToolsResolver,
+    ToolUsageLoader,
+    ToolUsageService,
+    splitDb(Neo4jRepository, GelRepository),
   ],
-  exports: [ToolUsageService, ResourceToolsResolver],
+  exports: [ToolUsageService],
 })
 export class ToolUsageModule {}