Migrate policy Granter

Author: CarsonF

src/components/authorization/policy/builder/granter.decorator.ts

@@ -1,5 +1,4 @@
-import { type DiscoveryService } from '@golevelup/nestjs-discovery';
-import { SetMetadata } from '@nestjs/common';
+import { createMetadataDecorator } from '@seedcompany/nest';
 import { type EnhancedResource, type Many, type ResourceShape } from '~/common';
 import { type ResourceGranter } from './resource-granter';
 
@@ -27,25 +26,12 @@ import { type ResourceGranter } from './resource-granter';
  *   }
  * }
  */
-export const Granter = (
-  resources: GranterMetadata['resources'],
-  factory?: GranterMetadata['factory'],
-): ClassDecorator =>
-  SetMetadata<any, GranterMetadata>(GRANTER_FACTORY_METADATA_KEY, {
-    resources,
-    factory,
-  });
-
-export const discover = (discovery: DiscoveryService) =>
-  discovery.providersWithMetaAtKey<GranterMetadata>(
-    GRANTER_FACTORY_METADATA_KEY,
-  );
-
-const GRANTER_FACTORY_METADATA_KEY = Symbol('GranterFactory');
-
-interface GranterMetadata {
-  resources: Many<ResourceShape<any>>;
-  factory?: <TResourceStatic extends ResourceShape<any>>(
-    resource: EnhancedResource<TResourceStatic>,
-  ) => ResourceGranter<TResourceStatic>;
-}
+export const Granter = createMetadataDecorator({
+  setter: (
+    resources: Many<ResourceShape<any>>,
+    factory?: <TResourceStatic extends ResourceShape<any>>(
+      resource: EnhancedResource<TResourceStatic>,
+    ) => ResourceGranter<TResourceStatic>,
+  ) => ({ resources, factory }),
+  types: ['class'],
+});

src/components/authorization/policy/granters.factory.ts

@@ -1,10 +1,10 @@
-import { DiscoveryService } from '@golevelup/nestjs-discovery';
-import { Injectable } from '@nestjs/common';
+import { Injectable, type Type } from '@nestjs/common';
 import { many, mapEntries } from '@seedcompany/common';
 import { mapValues } from 'lodash';
-import { EnhancedResource } from '~/common';
+import { EnhancedResource, type ResourceShape } from '~/common';
+import { MetadataDiscovery } from '~/core/discovery';
 import { ResourcesHost } from '~/core/resources';
-import { discover } from './builder/granter.decorator';
+import { Granter } from './builder/granter.decorator';
 import {
   DefaultResourceGranter,
   ResourceGranter,
@@ -14,21 +14,23 @@ import { type ResourcesGranter } from './granters';
 @Injectable()
 export class GrantersFactory {
   constructor(
-    private readonly discovery: DiscoveryService,
+    private readonly discovery: MetadataDiscovery,
     private readonly resourcesHost: ResourcesHost,
   ) {}
 
   async makeGranters() {
-    const discoveredGranters = await discover(this.discovery);
+    const discoveredGranters = this.discovery
+      .discover(Granter)
+      .classes<ResourceGranter<ResourceShape<any>>>();
 
     const custom = Object.assign(
       {},
       ...discoveredGranters.map(
-        ({ meta: { resources, factory }, discoveredClass }) =>
+        ({ meta: { resources, factory }, instance }) =>
           mapEntries(many(resources), (raw) => {
             const res = EnhancedResource.of(raw);
-            const granter =
-              factory?.(res) ?? new discoveredClass.dependencyType(res);
+            const GranterImpl = instance.constructor as Type<ResourcesGranter>;
+            const granter = factory?.(res) ?? new GranterImpl(res);
             if (!(granter instanceof ResourceGranter)) {
               throw new Error(
                 `Granter for ${res.name} must extend ResourceGranter class`,