Add GGT deploy command

spec/__support__/filesync.ts

@@ -1,6 +1,7 @@
 import fs from "fs-extra";
 import assert from "node:assert";
 import os from "node:os";
+import pMap from "p-map";
 import { expect, vi, type Assertion } from "vitest";
 import { z } from "zod";
 import {
@@ -10,13 +11,16 @@ import {
   type FileSyncDeletedEventInput,
 } from "../../src/__generated__/graphql.js";
 import {
+  FILE_SYNC_FILES_QUERY,
+  FILE_SYNC_HASHES_QUERY,
   PUBLISH_FILE_SYNC_EVENTS_MUTATION,
-  REMOTE_FILES_VERSION_QUERY,
   REMOTE_FILE_SYNC_EVENTS_SUBSCRIPTION,
 } from "../../src/services/app/edit-graphql.js";
-import { Directory, swallowEnoent } from "../../src/services/filesync/directory.js";
+import { Directory, swallowEnoent, type Hashes } from "../../src/services/filesync/directory.js";
 import { FileSync, type File } from "../../src/services/filesync/filesync.js";
+import { isEqualHashes } from "../../src/services/filesync/hashes.js";
 import { noop } from "../../src/services/util/function.js";
+import { isNil } from "../../src/services/util/is.js";
 import { defaults, omit } from "../../src/services/util/object.js";
 import { PromiseSignal } from "../../src/services/util/promise.js";
 import type { PartialExcept } from "../types.js";
@@ -52,11 +56,6 @@ export type SyncScenarioOptions = {
    */
   localFiles: Files;
 
-  /**
-   * The filesVersion Gadget currently has.
-   */
-  gadgetFilesVersion?: 1n | 2n;
-
   /**
    * The files Gadget currently has.
    * @default { ".gadget/": "" }
@@ -122,6 +121,11 @@ export type SyncScenario = {
    * @returns A mock subscription for {@linkcode REMOTE_FILE_SYNC_EVENTS_SUBSCRIPTION}.
    */
   expectGadgetChangesSubscription: () => MockEditGraphQLSubscription<REMOTE_FILE_SYNC_EVENTS_SUBSCRIPTION>;
+
+  /**
+   * Asserts that the local and gadget directories have the same hashes.
+   */
+  expectLocalAndGadgetHashesMatch: () => Promise<void>;
 };
 
 /**
@@ -133,9 +137,9 @@ export type SyncScenario = {
 export const makeSyncScenario = async ({
   filesVersion1Files,
   localFiles,
-  gadgetFilesVersion = 1n,
   gadgetFiles,
 }: Partial<SyncScenarioOptions> = {}): Promise<SyncScenario> => {
+  let gadgetFilesVersion = 1n;
   await writeDir(testDirPath("gadget"), { ".gadget/": "", ...gadgetFiles });
   const gadgetDir = await Directory.init(testDirPath("gadget"));
 
@@ -145,7 +149,8 @@ export const makeSyncScenario = async ({
   const filesVersionDirs = new Map<bigint, Directory>();
   filesVersionDirs.set(1n, filesVersion1Dir);
 
-  if (gadgetFilesVersion === 2n) {
+  if (!isEqualHashes(await gadgetDir.hashes(), await filesVersion1Dir.hashes())) {
+    gadgetFilesVersion = 2n;
     await fs.copy(gadgetDir.path, testDirPath("fv-2"));
     filesVersionDirs.set(2n, await Directory.init(testDirPath("fv-2")));
   }
@@ -155,8 +160,10 @@ export const makeSyncScenario = async ({
     await writeDir(testDirPath("local"), localFiles);
     await localDir.loadIgnoreFile();
 
-    const syncJson: SyncJson = { app: testApp.slug, filesVersion: "1", mtime: Date.now() };
-    await fs.outputJSON(localDir.absolute(".gadget/sync.json"), syncJson, { spaces: 2 });
+    if (!localFiles[".gadget/sync.json"]) {
+      const syncJson: SyncJson = { app: testApp.slug, filesVersion: "1", mtime: Date.now() };
+      await fs.outputJSON(localDir.absolute(".gadget/sync.json"), syncJson, { spaces: 2 });
+    }
   }
 
   FileSync.init.mockRestore?.();
@@ -191,21 +198,84 @@ export const makeSyncScenario = async ({
       await fs.chmod(gadgetDir.absolute(file.path), file.mode & 0o777);
     }
 
-    gadgetFilesVersion += 1n;
-    const newFilesVersionDir = await Directory.init(testDirPath(`fv-${gadgetFilesVersion}`));
-    await fs.copy(gadgetDir.path, newFilesVersionDir.path);
-    filesVersionDirs.set(gadgetFilesVersion, newFilesVersionDir);
-    log.trace("new files version", { gadgetFilesVersion });
+    const gadgetFilesVersionDir = filesVersionDirs.get(gadgetFilesVersion);
+    assert(gadgetFilesVersionDir, `filesVersionDir ${gadgetFilesVersion} doesn't exist`);
+    if (!isEqualHashes(await gadgetDir.hashes(), await gadgetFilesVersionDir.hashes())) {
+      gadgetFilesVersion += 1n;
+      const newFilesVersionDir = await Directory.init(testDirPath(`fv-${gadgetFilesVersion}`));
+      await fs.copy(gadgetDir.path, newFilesVersionDir.path);
+      filesVersionDirs.set(gadgetFilesVersion, newFilesVersionDir);
+      log.trace("new files version", { gadgetFilesVersion });
+    }
   };
 
   void nockEditGraphQLResponse({
     optional: true,
     persist: true,
-    query: REMOTE_FILES_VERSION_QUERY,
-    result: () => {
+    query: FILE_SYNC_HASHES_QUERY,
+    expectVariables: z.object({ filesVersion: z.string().optional() }).optional(),
+    result: async (variables) => {
+      let filesVersion: bigint;
+      let hashes: Hashes;
+
+      if (isNil(variables?.filesVersion)) {
+        log.trace("sending gadget hashes", { gadgetFilesVersion, variables });
+        filesVersion = gadgetFilesVersion;
+        hashes = await gadgetDir.hashes();
+      } else {
+        filesVersion = BigInt(variables.filesVersion);
+        log.trace("sending files version hashes", { filesVersion, variables });
+        const filesVersionDir = filesVersionDirs.get(filesVersion);
+        assert(filesVersionDir, `filesVersionDir ${filesync.filesVersion} doesn't exist`);
+        hashes = await filesVersionDir.hashes();
+      }
+
       return {
         data: {
-          remoteFilesVersion: String(gadgetFilesVersion),
+          fileSyncHashes: {
+            filesVersion: String(filesVersion),
+            hashes,
+          },
+        },
+      };
+    },
+  });
+
+  void nockEditGraphQLResponse({
+    optional: true,
+    persist: true,
+    query: FILE_SYNC_FILES_QUERY,
+    expectVariables: z.object({
+      filesVersion: z.string().optional(),
+      paths: z.array(z.string()),
+      encoding: z.nativeEnum(FileSyncEncoding).optional(),
+    }),
+    result: async ({ filesVersion, paths, encoding }) => {
+      filesVersion ??= String(gadgetFilesVersion);
+      encoding ??= FileSyncEncoding.Base64;
+
+      const filesVersionDir = filesVersionDirs.get(BigInt(filesVersion));
+      assert(filesVersionDir, `filesVersionDir ${filesync.filesVersion} doesn't exist`);
+
+      return {
+        data: {
+          fileSyncFiles: {
+            filesVersion: filesVersion,
+            files: await pMap(paths, async (filepath) => {
+              const stats = await fs.stat(filesVersionDir.absolute(filepath));
+              let content = "";
+              if (stats.isFile()) {
+                content = (await fs.readFile(filesVersionDir.absolute(filepath), { encoding })) as string;
+              }
+
+              return {
+                path: filepath,
+                mode: stats.mode,
+                content,
+                encoding: FileSyncEncoding.Base64,
+              };
+            }),
+          },
         },
       };
     },
@@ -359,9 +429,46 @@ export const makeSyncScenario = async ({
     },
 
     expectGadgetChangesSubscription: () => mockEditGraphQLSubs.expectSubscription(REMOTE_FILE_SYNC_EVENTS_SUBSCRIPTION),
+
+    expectLocalAndGadgetHashesMatch: async () => {
+      const localHashes = await localDir.hashes();
+      const gadgetHashes = await gadgetDir.hashes();
+      expect(localHashes).toEqual(gadgetHashes);
+    },
   };
 };
 
+/**
+ * Creates hashes of the given files.
+ */
+export const makeHashes = async ({
+  filesVersionFiles,
+  localFiles,
+  gadgetFiles,
+}: {
+  filesVersionFiles: Files;
+  localFiles: Files;
+  gadgetFiles?: Files;
+}): Promise<{ filesVersionHashes: Hashes; localHashes: Hashes; gadgetHashes: Hashes }> => {
+  const [filesVersionHashes, localHashes, gadgetHashes] = await Promise.all([
+    writeDir(testDirPath("filesVersion"), filesVersionFiles)
+      .then(() => Directory.init(testDirPath("filesVersion")))
+      .then((dir) => dir.hashes()),
+
+    writeDir(testDirPath("local"), localFiles)
+      .then(() => Directory.init(testDirPath("local")))
+      .then((dir) => dir.hashes()),
+
+    !gadgetFiles
+      ? Promise.resolve({})
+      : writeDir(testDirPath("gadget"), gadgetFiles)
+          .then(() => Directory.init(testDirPath("gadget")))
+          .then((dir) => dir.hashes()),
+  ]);
+
+  return { filesVersionHashes, localHashes, gadgetHashes };
+};
+
 export const defaultFileMode = os.platform() === "win32" ? 0o100666 : 0o100644;
 export const defaultDirMode = os.platform() === "win32" ? 0o40666 : 0o40755;