fix: build

Author: aryanjassal

src/git/http.ts

@@ -1,3 +1,4 @@
+import type { ContextCancellable } from '@matrixai/contexts';
 import type {
   CapabilityList,
   Reference,
@@ -10,7 +11,6 @@ import { Buffer } from 'buffer';
 import git from 'isomorphic-git';
 import * as gitUtils from './utils';
 import * as utils from '../utils';
-import {ContextCancellable} from "@matrixai/contexts";
 
 /**
  * Reference discovery
@@ -111,15 +111,18 @@ import {ContextCancellable} from "@matrixai/contexts";
  *
  * `referenceList` is called for generating the `ref_list` stage.
  */
-async function* advertiseRefGenerator({
-  efs,
-  dir,
-  gitDir,
-}: {
-  efs: EncryptedFS;
-  dir: string;
-  gitDir: string;
-}, ctx: ContextCancellable): AsyncGenerator<Buffer, void, void> {
+async function* advertiseRefGenerator(
+  {
+    efs,
+    dir,
+    gitDir,
+  }: {
+    efs: EncryptedFS;
+    dir: string;
+    gitDir: string;
+  },
+  ctx: ContextCancellable,
+): AsyncGenerator<Buffer, void, void> {
   // Providing side-band-64, symref for the HEAD and agent name capabilities
   const capabilityList = [
     gitUtils.SIDE_BAND_64_CAPABILITY,
@@ -131,11 +134,14 @@ async function* advertiseRefGenerator({
     }),
     gitUtils.AGENT_CAPABILITY,
   ];
-  const objectGenerator = gitUtils.listReferencesGenerator({
-    efs,
-    dir,
-    gitDir,
-  }, ctx );
+  const objectGenerator = gitUtils.listReferencesGenerator(
+    {
+      efs,
+      dir,
+      gitDir,
+    },
+    ctx,
+  );
 
   // PKT-LINE("# service=$servicename" LF)
   yield packetLineBuffer(gitUtils.REFERENCE_DISCOVERY_HEADER);
@@ -344,34 +350,43 @@ async function parsePackRequest(
  * It will respond with the `PKT-LINE(NAK_BUFFER)` and then the `packFile` data chunked into lines for the stream.
  *
  */
-async function* generatePackRequest({
-  efs,
-  dir,
-  gitDir,
-  body,
-}: {
-  efs: EncryptedFS;
-  dir: string;
-  gitDir: string;
-  body: Array<Buffer>;
-}, ctx: ContextCancellable): AsyncGenerator<Buffer, void, void> {
-  const [wants, haves, _capabilities] = await parsePackRequest(body);
-  const objectIds = await gitUtils.listObjects({
-    efs: efs,
+async function* generatePackRequest(
+  {
+    efs,
     dir,
-    gitDir: gitDir,
-    wants,
-    haves,
-  }, ctx);
+    gitDir,
+    body,
+  }: {
+    efs: EncryptedFS;
+    dir: string;
+    gitDir: string;
+    body: Array<Buffer>;
+  },
+  ctx: ContextCancellable,
+): AsyncGenerator<Buffer, void, void> {
+  const [wants, haves, _capabilities] = await parsePackRequest(body);
+  const objectIds = await gitUtils.listObjects(
+    {
+      efs: efs,
+      dir,
+      gitDir: gitDir,
+      wants,
+      haves,
+    },
+    // ctx,
+  );
   // Reply that we have no common history and that we need to send everything
   yield packetLineBuffer(gitUtils.NAK_BUFFER);
   // Send everything over in pack format
-  yield* generatePackData({
-    efs: efs,
-    dir,
-    gitDir,
-    objectIds,
-  }, ctx);
+  yield* generatePackData(
+    {
+      efs: efs,
+      dir,
+      gitDir,
+      objectIds,
+    },
+    ctx,
+  );
   // Send dummy progress data
   yield packetLineBuffer(
     gitUtils.DUMMY_PROGRESS_BUFFER,
@@ -387,19 +402,22 @@ async function* generatePackRequest({
  * The `packFile` is chunked into the `packetLineBuffer` with the size defined by `chunkSize`.
  *
  */
-async function* generatePackData({
-  efs,
-  dir,
-  gitDir,
-  objectIds,
-  chunkSize = gitUtils.PACK_CHUNK_SIZE,
-}: {
-  efs: EncryptedFS;
-  dir: string;
-  gitDir: string;
-  objectIds: Array<ObjectId>;
-  chunkSize?: number;
-}, ctx: ContextCancellable): AsyncGenerator<Buffer, void, void> {
+async function* generatePackData(
+  {
+    efs,
+    dir,
+    gitDir,
+    objectIds,
+    chunkSize = gitUtils.PACK_CHUNK_SIZE,
+  }: {
+    efs: EncryptedFS;
+    dir: string;
+    gitDir: string;
+    objectIds: Array<ObjectId>;
+    chunkSize?: number;
+  },
+  ctx: ContextCancellable,
+): AsyncGenerator<Buffer, void, void> {
   let packFile: PackObjectsResult;
   // In case of errors we don't want to throw them. This will result in the error being thrown into `isometric-git`
   // when it consumes the response. It handles this by logging out the error which we don't want to happen.

src/git/utils.ts

@@ -158,15 +158,16 @@ async function listObjects({
   gitDir: string;
   wants: ObjectIdList;
   haves: ObjectIdList;
-}, ctx: ContextCancellable): Promise<ObjectIdList> {
+}/*, ctx: ContextCancellable*/): Promise<ObjectIdList> {
+  // TODO: add support for ctx
   const commits = new Set<string>();
   const trees = new Set<string>();
   const blobs = new Set<string>();
   const tags = new Set<string>();
   const havesSet: Set<string> = new Set(haves);
 
   async function walk(objectId: ObjectId, type: ObjectType): Promise<void> {
-    ctx.signal.throwIfAborted();
+    // ctx.signal.throwIfAborted();
     // If object was listed as a have then we don't need to walk over it
     if (havesSet.has(objectId)) return;
     switch (type) {

src/nodes/agent/handlers/VaultsGitInfoGet.ts

@@ -25,9 +25,9 @@ class VaultsGitInfoGet extends RawHandler<{
 }> {
   public handle = async (
     input: [JSONRPCRequest, ReadableStream<Uint8Array>],
-    _cancel,
+    _cancel: (reason?: any) => void,
     meta: Record<string, JSONValue> | undefined,
-    _ctx: ContextTimed, // TODO: use
+    ctx: ContextTimed,
   ): Promise<[JSONObject, ReadableStream<Uint8Array>]> => {
     const { db, vaultManager, acl } = this.container;
     const [headerMessage, inputStream] = input;
@@ -91,7 +91,10 @@ class VaultsGitInfoGet extends RawHandler<{
     let handleInfoRequestGen: AsyncGenerator<Buffer>;
     const stream = new ReadableStream({
       start: async () => {
-        handleInfoRequestGen = vaultManager.handleInfoRequest(data.vaultId);
+        handleInfoRequestGen = vaultManager.handleInfoRequest(
+          data.vaultId,
+          ctx,
+        );
       },
       pull: async (controller) => {
         const result = await handleInfoRequestGen.next();

src/nodes/agent/handlers/VaultsGitPackGet.ts

@@ -1,6 +1,6 @@
 import type { DB } from '@matrixai/db';
-import type { JSONObject, JSONRPCRequest } from '@matrixai/rpc';
-import type {ContextTimed} from '@matrixai/contexts';
+import type { JSONObject, JSONRPCRequest, JSONValue } from '@matrixai/rpc';
+import type { ContextTimed } from '@matrixai/contexts';
 import type { VaultName } from '../../../vaults/types';
 import type ACL from '../../../acl/ACL';
 import type VaultManager from '../../../vaults/VaultManager';
@@ -23,8 +23,8 @@ class VaultsGitPackGet extends RawHandler<{
 }> {
   public handle = async (
     input: [JSONRPCRequest, ReadableStream<Uint8Array>],
-    _cancel,
-    meta,
+    _cancel: (reason: any) => void,
+    meta: Record<string, JSONValue>,
     ctx: ContextTimed,
   ): Promise<[JSONObject, ReadableStream<Uint8Array>]> => {
     const { vaultManager, acl, db } = this.container;
@@ -79,7 +79,7 @@ class VaultsGitPackGet extends RawHandler<{
         for await (const message of inputStream) {
           body.push(Buffer.from(message));
         }
-        packRequestGen = vaultManager.handlePackRequest(vaultId, body);
+        packRequestGen = vaultManager.handlePackRequest(vaultId, body, ctx);
       },
       pull: async (controller) => {
         const next = await packRequestGen.next();

src/nodes/agent/handlers/VaultsScan.ts

@@ -1,11 +1,12 @@
 import type { DB } from '@matrixai/db';
-import type {ContextTimed} from '@matrixai/contexts';
+import type { ContextTimed } from '@matrixai/contexts';
 import type {
   AgentRPCRequestParams,
   AgentRPCResponseResult,
   VaultsScanMessage,
 } from '../types';
 import type VaultManager from '../../../vaults/VaultManager';
+import type { JSONValue } from '@matrixai/rpc';
 import { ServerHandler } from '@matrixai/rpc';
 import * as agentErrors from '../errors';
 import * as agentUtils from '../utils';
@@ -23,12 +24,13 @@ class VaultsScan extends ServerHandler<
   AgentRPCResponseResult<VaultsScanMessage>
 > {
   public handle = async function* (
-    input: AgentRPCRequestParams,
-    _cancel,
-    meta,
+    _input: AgentRPCRequestParams,
+    _cancel: (reason?: any) => void,
+    meta: Record<string, JSONValue> | undefined,
     ctx: ContextTimed,
   ): AsyncGenerator<AgentRPCResponseResult<VaultsScanMessage>> {
-    const { vaultManager, db } = this.container;
+    const { vaultManager, db }: { vaultManager: VaultManager; db: DB } =
+      this.container;
     const requestingNodeId = agentUtils.nodeIdFromMeta(meta);
     if (requestingNodeId == null) {
       throw new agentErrors.ErrorAgentNodeIdMissing();

src/vaults/VaultManager.ts

@@ -40,7 +40,7 @@ import * as nodesUtils from '../nodes/utils';
 import * as keysUtils from '../keys/utils';
 import config from '../config';
 import { mkdirExists } from '../utils/utils';
-import {ContextCancellable} from "@matrixai/contexts";
+import {ContextCancellable, ContextTimed} from "@matrixai/contexts";
 
 /**
  * Object map pattern for each vault
@@ -804,11 +804,12 @@ class VaultManager {
   @ready(new vaultsErrors.ErrorVaultManagerNotRunning())
   public async *handleInfoRequest(
     vaultId: VaultId,
+    ctx: ContextTimed,
     tran?: DBTransaction,
   ): AsyncGenerator<Buffer, void, void> {
     if (tran == null) {
       const handleInfoRequest = (tran: DBTransaction) =>
-        this.handleInfoRequest(vaultId, tran);
+        this.handleInfoRequest(vaultId, ctx, tran);
       return yield* this.db.withTransactionG(async function* (tran) {
         return yield* handleInfoRequest(tran);
       });
@@ -826,7 +827,7 @@ class VaultManager {
           efs,
           dir: path.join(vaultsUtils.encodeVaultId(vaultId), 'contents'),
           gitDir: path.join(vaultsUtils.encodeVaultId(vaultId), '.git'),
-        });
+        }, ctx);
       },
     );
   }