Merge pull request #846 from MatrixAI/feature-eng-432-review-and-refactor-rpc-handlers-to-handle-cancellation

Add cancellation to RPC handlers for the `vaults` domain

Author: aryanjassal

src/acl/ACL.ts

@@ -144,11 +144,7 @@ class ACL {
       if (permId in permIds) {
         nodePerm = permIds[permId];
         // Get the first existing perm object
-        let perm: Permission;
-        for (const nodeId_ in nodePerm) {
-          perm = nodePerm[nodeId_];
-          break;
-        }
+        const perm = Object.values(nodePerm)[0];
         // All perm objects are shared
         nodePerm[nodeId] = perm!;
       } else {
@@ -614,8 +610,8 @@ class ACL {
         [...this.aclNodesDbPath, nodeId.toBuffer()],
         true,
       );
-      // Skip if the nodeId doesn't exist
-      // this means that it previously been removed
+      // Skip if the nodeId doesn't exist. This means that it has previously
+      // been removed.
       if (permId == null) {
         continue;
       }

src/client/callers/vaultsSecretsGet.ts

@@ -1,10 +1,10 @@
 import type { HandlerTypes } from '@matrixai/rpc';
 import type VaultsSecretsGet from '../handlers/VaultsSecretsGet';
-import { ServerCaller } from '@matrixai/rpc';
+import { UnaryCaller } from '@matrixai/rpc';
 
 type CallerTypes = HandlerTypes<VaultsSecretsGet>;
 
-const vaultsSecretsGet = new ServerCaller<
+const vaultsSecretsGet = new UnaryCaller<
   CallerTypes['input'],
   CallerTypes['output']
 >();

src/client/handlers/AgentStatus.ts

@@ -5,8 +5,8 @@ import type {
 } from '../types';
 import type PolykeyAgent from '../../PolykeyAgent';
 import { UnaryHandler } from '@matrixai/rpc';
-import * as nodesUtils from '../../nodes/utils';
 import config from '../../config';
+import * as nodesUtils from '../../nodes/utils';
 
 class AgentStatus extends UnaryHandler<
   {

src/client/handlers/AuditEventsGet.ts

@@ -1,4 +1,5 @@
 import type { ContextTimed } from '@matrixai/contexts';
+import type { JSONValue } from '@matrixai/rpc';
 import type { ClientRPCRequestParams, ClientRPCResponseResult } from '../types';
 import type {
   AuditEvent,
@@ -42,8 +43,8 @@ class AuditEventsGet extends ServerHandler<
     }> & {
       paths: Array<TopicSubPath>;
     },
-    _cancel,
-    _meta,
+    _cancel: (reason?: any) => void,
+    _meta: Record<string, JSONValue>,
     ctx: ContextTimed,
   ): AsyncGenerator<ClientRPCResponseResult<AuditEventSerialized>> {
     const { audit }: { audit: Audit } = this.container;

src/client/handlers/GestaltsActionsGetByIdentity.ts

@@ -8,9 +8,9 @@ import type GestaltGraph from '../../gestalts/GestaltGraph';
 import type { GestaltAction } from '../../gestalts/types';
 import type { IdentityId, ProviderId } from '../../ids';
 import { UnaryHandler } from '@matrixai/rpc';
-import * as ids from '../../ids';
 import { validateSync } from '../../validation';
 import { matchSync } from '../../utils';
+import * as ids from '../../ids';
 
 class GestaltsActionsGetByIdentity extends UnaryHandler<
   {

src/client/handlers/IdentitiesAuthenticate.ts

@@ -1,3 +1,5 @@
+import type { ContextTimed } from '@matrixai/contexts';
+import type { JSONValue } from '@matrixai/rpc';
 import type {
   AuthProcessMessage,
   ClientRPCRequestParams,
@@ -21,11 +23,10 @@ class IdentitiesAuthenticate extends ServerHandler<
   public timeout = 120000; // 2 Minutes
   public handle = async function* (
     input: ClientRPCRequestParams<{ providerId: string }>,
-    _cancel,
-    _meta,
-    ctx,
+    _cancel: (reason?: any) => void,
+    _meta: Record<string, JSONValue>,
+    ctx: ContextTimed,
   ): AsyncGenerator<ClientRPCResponseResult<AuthProcessMessage>> {
-    if (ctx.signal.aborted) throw ctx.signal.reason;
     const { identitiesManager }: { identitiesManager: IdentitiesManager } =
       this.container;
     const {
@@ -52,7 +53,7 @@ class IdentitiesAuthenticate extends ServerHandler<
     if (authFlowResult.done) {
       never('authFlow signalled done too soon');
     }
-    if (ctx.signal.aborted) throw ctx.signal.reason;
+    ctx.signal.throwIfAborted();
     yield {
       request: {
         url: authFlowResult.value.url,
@@ -63,7 +64,7 @@ class IdentitiesAuthenticate extends ServerHandler<
     if (!authFlowResult.done) {
       never('authFlow did not signal done when expected');
     }
-    if (ctx.signal.aborted) throw ctx.signal.reason;
+    ctx.signal.throwIfAborted();
     yield {
       response: {
         identityId: authFlowResult.value,

src/client/handlers/IdentitiesAuthenticatedGet.ts

@@ -1,3 +1,5 @@
+import type { ContextTimed } from '@matrixai/contexts';
+import type { JSONValue } from '@matrixai/rpc';
 import type {
   ClientRPCRequestParams,
   ClientRPCResponseResult,
@@ -19,11 +21,10 @@ class IdentitiesAuthenticatedGet extends ServerHandler<
 > {
   public handle = async function* (
     input: ClientRPCRequestParams<{ providerId?: string }>,
-    _cancel,
-    _meta,
-    ctx,
+    _cancel: (reason?: any) => void,
+    _meta: Record<string, JSONValue>,
+    ctx: ContextTimed,
   ): AsyncGenerator<ClientRPCResponseResult<IdentityMessage>> {
-    if (ctx.signal.aborted) throw ctx.signal.reason;
     const { identitiesManager }: { identitiesManager: IdentitiesManager } =
       this.container;
     let providerId: ProviderId | undefined;
@@ -46,12 +47,10 @@ class IdentitiesAuthenticatedGet extends ServerHandler<
         : [providerId];
     for (const providerId of providerIds) {
       const provider = identitiesManager.getProvider(providerId);
-      if (provider == null) {
-        continue;
-      }
+      if (provider == null) continue;
       const identities = await provider.getAuthIdentityIds();
       for (const identityId of identities) {
-        if (ctx.signal.aborted) throw ctx.signal.reason;
+        ctx.signal.throwIfAborted();
         yield {
           providerId: provider.id,
           identityId: identityId,

src/client/handlers/IdentitiesInfoConnectedGet.ts

@@ -1,3 +1,5 @@
+import type { ContextTimed } from '@matrixai/contexts';
+import type { JSONValue } from '@matrixai/rpc';
 import type {
   ClientRPCRequestParams,
   ClientRPCResponseResult,
@@ -22,11 +24,10 @@ class IdentitiesInfoConnectedGet extends ServerHandler<
 > {
   public handle = async function* (
     input: ClientRPCRequestParams<ProviderSearchMessage>,
-    _cancel,
-    _meta,
-    ctx,
+    _cancel: (reason?: any) => void,
+    _meta: Record<string, JSONValue>,
+    ctx: ContextTimed,
   ): AsyncGenerator<ClientRPCResponseResult<IdentityInfoMessage>> {
-    if (ctx.signal.aborted) throw ctx.signal.reason;
     const { identitiesManager }: { identitiesManager: IdentitiesManager } =
       this.container;
     const {
@@ -71,6 +72,7 @@ class IdentitiesInfoConnectedGet extends ServerHandler<
     }
     const identities: Array<AsyncGenerator<IdentityData>> = [];
     for (const providerId of providerIds) {
+      ctx.signal.throwIfAborted();
       // Get provider from id
       const provider = identitiesManager.getProvider(providerId);
       if (provider === undefined) {
@@ -94,7 +96,7 @@ class IdentitiesInfoConnectedGet extends ServerHandler<
     let count = 0;
     for (const gen of identities) {
       for await (const identity of gen) {
-        if (ctx.signal.aborted) throw ctx.signal.reason;
+        ctx.signal.throwIfAborted();
         if (input.limit !== undefined && count >= input.limit) break;
         yield {
           providerId: identity.providerId,

src/client/handlers/IdentitiesInfoGet.ts

@@ -1,3 +1,5 @@
+import type { ContextTimed } from '@matrixai/contexts';
+import type { JSONValue } from '@matrixai/rpc';
 import type {
   ClientRPCRequestParams,
   ClientRPCResponseResult,
@@ -8,11 +10,11 @@ import type { IdentityId, ProviderId } from '../../ids';
 import type IdentitiesManager from '../../identities/IdentitiesManager';
 import type { IdentityData } from '../../identities/types';
 import { ServerHandler } from '@matrixai/rpc';
+import { validateSync } from '../../validation';
+import { matchSync } from '../../utils';
 import * as ids from '../../ids';
 import * as identitiesErrors from '../../identities/errors';
 import * as identitiesUtils from '../../identities/utils';
-import { validateSync } from '../../validation';
-import { matchSync } from '../../utils';
 
 class IdentitiesInfoGet extends ServerHandler<
   {
@@ -23,9 +25,9 @@ class IdentitiesInfoGet extends ServerHandler<
 > {
   public handle = async function* (
     input: ClientRPCRequestParams<ProviderSearchMessage>,
-    _cancel,
-    _meta,
-    ctx,
+    _cancel: (reason?: any) => void,
+    _meta: Record<string, JSONValue>,
+    ctx: ContextTimed,
   ): AsyncGenerator<ClientRPCResponseResult<IdentityInfoMessage>> {
     if (ctx.signal.aborted) throw ctx.signal.reason;
     const { identitiesManager }: { identitiesManager: IdentitiesManager } =
@@ -86,7 +88,7 @@ class IdentitiesInfoGet extends ServerHandler<
       input.limit = identities.length;
     }
     for (let i = 0; i < input.limit; i++) {
-      if (ctx.signal.aborted) throw ctx.signal.reason;
+      ctx.signal.throwIfAborted();
       const identity = identities[i];
       if (identity !== undefined) {
         if (identitiesUtils.matchIdentityData(identity, searchTerms)) {

src/client/handlers/KeysCertsChainGet.ts

@@ -1,3 +1,5 @@
+import type { ContextTimed } from '@matrixai/contexts';
+import type { JSONValue } from '@matrixai/rpc';
 import type {
   CertMessage,
   ClientRPCRequestParams,
@@ -14,17 +16,15 @@ class KeysCertsChainGet extends ServerHandler<
   ClientRPCResponseResult<CertMessage>
 > {
   public handle = async function* (
-    _input,
-    _cancel,
-    _meta,
-    ctx,
+    _input: ClientRPCRequestParams,
+    _cancel: (reason?: any) => void,
+    _meta: Record<string, JSONValue>,
+    ctx: ContextTimed,
   ): AsyncGenerator<ClientRPCResponseResult<CertMessage>> {
     const { certManager }: { certManager: CertManager } = this.container;
     for (const certPEM of await certManager.getCertPEMsChain()) {
-      if (ctx.signal.aborted) throw ctx.signal.reason;
-      yield {
-        cert: certPEM,
-      };
+      ctx.signal.throwIfAborted();
+      yield { cert: certPEM };
     }
   };
 }