From 534e88f33385137bcc5391f4ea73df517357b1cc Mon Sep 17 00:00:00 2001
From: Neil Macneale V <neil.macneale.v@gmail.com>
Date: Tue, 17 Sep 2024 14:34:00 -0700
Subject: [PATCH] Add fauna schema status (#356)

---
 src/commands/schema/status.ts | 37 ++++++++++++++++++++++++
 test/commands/schema.test.js  | 19 +++++++++++++
 test/integ/schema.test.ts     | 53 +++++++++++++++++++++++++++++++++++
 3 files changed, 109 insertions(+)
 create mode 100644 src/commands/schema/status.ts

diff --git a/src/commands/schema/status.ts b/src/commands/schema/status.ts
new file mode 100644
index 00000000..8803c243
--- /dev/null
+++ b/src/commands/schema/status.ts
@@ -0,0 +1,37 @@
+import SchemaCommand from "../../lib/schema-command";
+
+export default class StatusSchemaCommand extends SchemaCommand {
+  static flags = {
+    ...SchemaCommand.flags,
+  };
+
+  static description = "Print the staged schema status.";
+  static examples = ["$ fauna schema status"];
+
+  async run() {
+    try {
+      const { url, secret } = await this.fetchsetup();
+      const res = await fetch(
+        new URL("/schema/1/staged/status?diff=true", url),
+        {
+          method: "GET",
+          headers: { AUTHORIZATION: `Bearer ${secret}` },
+          // https://github.com/nodejs/node/issues/46221
+          // https://github.com/microsoft/TypeScript-DOM-lib-generator/issues/1483
+          // @ts-expect-error-next-line
+          duplex: "half",
+        }
+      );
+
+      const json = await res.json();
+      if (json.error) {
+        this.error(json.error.message);
+      }
+
+      this.log(json.diff);
+    } catch (err) {
+      console.log(err);
+      this.error(err);
+    }
+  }
+}
diff --git a/test/commands/schema.test.js b/test/commands/schema.test.js
index a174df9f..1cc593d7 100644
--- a/test/commands/schema.test.js
+++ b/test/commands/schema.test.js
@@ -99,6 +99,25 @@ describe("fauna schema push test", () => {
     // Restore the stub after the test
     stubConfirm.restore();
   });
+
+  it("runs schema status", async () => {
+    nock(getEndpoint(), { allowUnmocked: false })
+      .persist()
+      .post("/", matchFqlReq(q.Now()))
+      .reply(200, new Date())
+      .get("/schema/1/staged/status?diff=true")
+      .reply(200, {
+        version: 0,
+        status: "ready",
+        diff: diff.diff,
+      });
+
+    // Stubbing the confirmation to always return true
+    const { stdout } = await runCommand(
+      withOpts(["schema status", "--dir=test/testdata"])
+    );
+    expect(stdout).to.contain(`${diff.diff}`);
+  });
 });
 
 const testdir = "test/testdata";
diff --git a/test/integ/schema.test.ts b/test/integ/schema.test.ts
index 7c83ac3a..7a016cbb 100644
--- a/test/integ/schema.test.ts
+++ b/test/integ/schema.test.ts
@@ -51,4 +51,57 @@ describe.skip("fauna schema staged commands", () => {
       )
     ).to.deep.equal(null);
   });
+
+  const status = async (secret: string) => {
+    const output = await shellOk("fauna schema status --dir .", secret);
+
+    // Remove the "Connected to endpoint" line.
+    return output.split("\n").slice(1).join("\n");
+  };
+
+  it("fauna schema status works", async () => {
+    const secret = await newDB();
+
+    await shellOk(
+      "fauna schema push --dir test/integ/schema/start --force",
+      secret
+    );
+
+    expect(await status(secret)).to.equal(
+      stripMargin(
+        `|Status: none
+         |`
+      )
+    );
+
+    await shellOk(
+      "fauna schema push --dir test/integ/schema/staged_index --force --stage",
+      secret
+    );
+
+    expect(await status(secret)).to.equal(
+      stripMargin(
+        `|Status: ready
+         |The schema is ready to be committed.
+         |
+         |Staged changes:
+         |* Modifying collection \`User\` at 0:90/main.fsl:
+         |  * Summary:
+         |    + added: index \`byName\` (see diff)
+         |
+         |  * Diff:
+         |      collection User {
+         |        name: String
+         |        email: String
+         |    +
+         |    +   index byName {
+         |    +     terms [.name]
+         |    +   }
+         |      }
+         |
+         |
+         |`
+      )
+    );
+  });
 });