buzz/desktop/tests/e2e/nostr-bind.spec.ts

import { expect, test, type Locator, type Page } from "@playwright/test";
import { installMockBridge } from "../helpers/bridge";

type NostrBindPayload = {
  action: string;
  audience: string;
  callbackUrl?: string;
  challengeId: string;
  expiresAt: string;
  nonce: string;
  origin: string;
  protocol: string;
  returnMode: string;
  verificationCode: string;
  version: string;
};

const VALID_REQUEST: NostrBindPayload = {
  action: "bind_nostr_identity",
  audience: "buzz:nostr-identity",
  challengeId: "550e8400-e29b-41d4-a716-446655440000",
  expiresAt: "2099-01-01T00:00:00Z",
  nonce: "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghi01234567",
  origin: "https://admin.example.com",
  protocol: "buzz-nostr-identity",
  returnMode: "clipboard",
  verificationCode: "123456",
  version: "1",
};

async function emitNostrBind(page: Page, payload: NostrBindPayload) {
  await page.evaluate(async (nextPayload) => {
    const internals = (
      window as Window & {
        __TAURI_INTERNALS__?: {
          invoke?: (
            command: string,
            args: Record<string, unknown>,
          ) => Promise<unknown>;
        };
      }
    ).__TAURI_INTERNALS__;
    if (!internals?.invoke) {
      throw new Error("Tauri E2E event bridge is unavailable");
    }
    await internals.invoke("plugin:event|emit", {
      event: "deep-link-nostr-bind",
      payload: nextPayload,
    });
  }, payload);
}

async function openNostrBind(
  page: Page,
  payload: NostrBindPayload = VALID_REQUEST,
  mock?: Parameters<typeof installMockBridge>[1],
) {
  await installMockBridge(page, mock);
  await page.goto("/");
  await page.waitForFunction(
    () =>
      typeof (
        window as Window & {
          __BUZZ_E2E_INVOKE_MOCK_COMMAND__?: unknown;
        }
      ).__BUZZ_E2E_INVOKE_MOCK_COMMAND__ === "function",
  );
  await emitNostrBind(page, payload);
  await expect(page.getByTestId("nostr-bind-page")).toBeVisible();
}

async function pasteCode(input: Locator, code: string) {
  await input.evaluate((element, pastedCode) => {
    const clipboardData = new DataTransfer();
    clipboardData.setData("text", pastedCode);
    element.dispatchEvent(
      new ClipboardEvent("paste", {
        bubbles: true,
        cancelable: true,
        clipboardData,
      }),
    );
  }, code);
}

async function signCommandPayloads(page: Page): Promise<unknown[]> {
  return page.evaluate(() =>
    (
      (
        window as Window & {
          __BUZZ_E2E_COMMAND_LOG__?: Array<{
            command: string;
            payload: unknown;
          }>;
        }
      ).__BUZZ_E2E_COMMAND_LOG__ ?? []
    )
      .filter(({ command }) => command === "sign_nostr_identity_binding")
      .map(({ payload }) => payload),
  );
}

async function installClipboardStub(page: Page, shouldFail: boolean) {
  await page.addInitScript(
    ({ fail }) => {
      const testWindow = window as Window & {
        __BUZZ_E2E_CLIPBOARD_TEXT__?: string;
      };
      Object.defineProperty(navigator, "clipboard", {
        configurable: true,
        value: {
          writeText: async (text: string) => {
            if (fail) {
              throw new Error("clipboard unavailable");
            }
            testWindow.__BUZZ_E2E_CLIPBOARD_TEXT__ = text;
          },
        },
      });
    },
    { fail: shouldFail },
  );
}

async function installShakeCounter(page: Page) {
  await page.addInitScript(() => {
    const testWindow = window as Window & {
      __BUZZ_E2E_CODE_SHAKE_CALLS__?: number;
    };
    testWindow.__BUZZ_E2E_CODE_SHAKE_CALLS__ = 0;
    const originalAnimate = Element.prototype.animate;
    Element.prototype.animate = function animate(keyframes, options) {
      if (
        this instanceof HTMLElement &&
        this.dataset.testid === "nostr-bind-verification-code"
      ) {
        testWindow.__BUZZ_E2E_CODE_SHAKE_CALLS__ =
          (testWindow.__BUZZ_E2E_CODE_SHAKE_CALLS__ ?? 0) + 1;
      }
      return originalAnimate.call(this, keyframes, options);
    };
  });
}

async function shakeCount(page: Page): Promise<number> {
  return page.evaluate(
    () =>
      (
        window as Window & {
          __BUZZ_E2E_CODE_SHAKE_CALLS__?: number;
        }
      ).__BUZZ_E2E_CODE_SHAKE_CALLS__ ?? 0,
  );
}

test("supports OTP entry and navigation without signing incomplete input", async ({
  page,
}) => {
  await openNostrBind(page);

  await expect(page.getByText("Requesting origin")).toHaveCount(0);
  await expect(page.getByText(VALID_REQUEST.origin)).toHaveCount(0);

  const first = page.getByTestId("nostr-bind-code-digit-1");
  const second = page.getByTestId("nostr-bind-code-digit-2");
  const third = page.getByTestId("nostr-bind-code-digit-3");
  const continueButton = page.getByTestId("nostr-bind-sign-and-copy");

  await first.click();
  await first.press("1");
  await expect(second).toBeFocused();
  await second.press("2");
  await expect(third).toBeFocused();
  await third.press("ArrowLeft");
  await expect(second).toBeFocused();
  await second.press("Backspace");
  await expect(second).toHaveValue("");
  await expect(continueButton).toBeDisabled();
  await expect.poll(() => signCommandPayloads(page)).toEqual([]);
});

test("auto-signs exactly once when the sixth correct digit is typed", async ({
  page,
}) => {
  await openNostrBind(page);

  await page.getByTestId("nostr-bind-code-digit-1").click();
  for (const digit of VALID_REQUEST.verificationCode) {
    await page.keyboard.type(digit);
  }

  await expect(page.getByTestId("nostr-bind-finish-step")).toBeVisible();
  await expect.poll(() => signCommandPayloads(page)).toHaveLength(1);
  await page.waitForTimeout(100);
  await expect.poll(() => signCommandPayloads(page)).toHaveLength(1);
});

test("discards a signature when a newer pairing request arrives", async ({
  page,
}) => {
  await openNostrBind(page, VALID_REQUEST, { nostrBindSignDelayMs: 150 });
  await pasteCode(
    page.getByTestId("nostr-bind-code-digit-1"),
    VALID_REQUEST.verificationCode,
  );
  await expect.poll(() => signCommandPayloads(page)).toHaveLength(1);

  const nextRequest = {
    ...VALID_REQUEST,
    challengeId: "550e8400-e29b-41d4-a716-446655440001",
    verificationCode: "654321",
  };
  await emitNostrBind(page, nextRequest);

  await expect(page.getByTestId("nostr-bind-code-digit-1")).toHaveValue("");
  await page.waitForTimeout(200);
  await expect(page.getByTestId("nostr-bind-code-step")).toBeVisible();
  await expect(page.getByTestId("nostr-bind-finish-step")).toBeHidden();

  await pasteCode(
    page.getByTestId("nostr-bind-code-digit-1"),
    nextRequest.verificationCode,
  );
  await expect(page.getByTestId("nostr-bind-finish-step")).toBeVisible();
  await expect.poll(() => signCommandPayloads(page)).toHaveLength(2);
});

test("repeats mismatch feedback without signing", async ({ page }) => {
  await installShakeCounter(page);
  await openNostrBind(page);

  const first = page.getByTestId("nostr-bind-code-digit-1");
  const continueButton = page.getByTestId("nostr-bind-sign-and-copy");

  await pasteCode(first, "654321");
  await expect(page.getByRole("alert")).toHaveText(
    "That code doesn't match. Check the code and try again.",
  );
  await expect(continueButton).toBeDisabled();
  await expect.poll(() => shakeCount(page)).toBe(1);

  await pasteCode(first, "654321");
  await expect.poll(() => shakeCount(page)).toBe(2);
  await expect.poll(() => signCommandPayloads(page)).toEqual([]);
});

test("honors reduced motion while rejecting a mismatched code", async ({
  page,
}) => {
  await page.emulateMedia({ reducedMotion: "reduce" });
  await installShakeCounter(page);
  await openNostrBind(page);

  await pasteCode(page.getByTestId("nostr-bind-code-digit-1"), "654321");
  await expect(page.getByRole("alert")).toBeVisible();
  await expect.poll(() => shakeCount(page)).toBe(0);
  await expect.poll(() => signCommandPayloads(page)).toEqual([]);
});

test("rejects an expired request without signing", async ({ page }) => {
  await openNostrBind(page, {
    ...VALID_REQUEST,
    expiresAt: "2000-01-01T00:00:00Z",
  });

  await expect(
    page.getByText(
      "This binding link has expired. Request a new one from the requesting app.",
    ),
  ).toBeVisible();
  await pasteCode(
    page.getByTestId("nostr-bind-code-digit-1"),
    VALID_REQUEST.verificationCode,
  );
  await expect(page.getByTestId("nostr-bind-sign-and-copy")).toBeDisabled();
  await expect.poll(() => signCommandPayloads(page)).toEqual([]);
});

test("cancels a request without signing", async ({ page }) => {
  await openNostrBind(page);

  await page.getByRole("button", { name: "Cancel" }).click();
  await expect(page.getByTestId("nostr-bind-page")).toBeHidden();
  await expect.poll(() => signCommandPayloads(page)).toEqual([]);
});

test("signs a valid request, shows the response, and copies it", async ({
  page,
}) => {
  await installClipboardStub(page, false);
  await openNostrBind(page);
  await pasteCode(
    page.getByTestId("nostr-bind-code-digit-1"),
    VALID_REQUEST.verificationCode,
  );

  await expect(page.getByTestId("nostr-bind-finish-step")).toBeVisible();
  const response = page.getByTestId("nostr-bind-signed-response");
  await expect(response).toContainText("e2e-signed-nostr-binding");
  await expect
    .poll(() => signCommandPayloads(page))
    .toEqual([
      {
        challengeId: VALID_REQUEST.challengeId,
        expiresAt: VALID_REQUEST.expiresAt,
        nonce: VALID_REQUEST.nonce,
        origin: VALID_REQUEST.origin,
        verificationCode: VALID_REQUEST.verificationCode,
      },
    ]);

  const signedResponse = await response.textContent();
  await page.getByTestId("nostr-bind-copy-response").click();
  await expect(
    page.getByTestId("nostr-bind-copy-response"),
  ).toHaveAccessibleName("Copied");
  await expect
    .poll(() =>
      page.evaluate(
        () =>
          (
            window as Window & {
              __BUZZ_E2E_CLIPBOARD_TEXT__?: string;
            }
          ).__BUZZ_E2E_CLIPBOARD_TEXT__,
      ),
    )
    .toBe(signedResponse);
  await expect.poll(() => signCommandPayloads(page)).toHaveLength(1);

  await page.getByRole("button", { name: "Continue" }).click();
  await expect(page.getByTestId("nostr-bind-page")).toBeHidden();
});

test("returns a signed response in the callback fragment after consent", async ({
  page,
}) => {
  await installClipboardStub(page, false);
  await openNostrBind(page, {
    ...VALID_REQUEST,
    callbackUrl: "https://admin.example.com/buzz?source=bind#stale",
    returnMode: "browser_fragment_v1",
  });

  await expect
    .poll(async () =>
      page.evaluate(
        () =>
          (
            (
              window as Window & {
                __BUZZ_E2E_COMMAND_LOG__?: Array<{ command: string }>;
              }
            ).__BUZZ_E2E_COMMAND_LOG__ ?? []
          ).filter(({ command }) => command === "plugin:opener|open_url")
            .length,
      ),
    )
    .toBe(0);
  await pasteCode(
    page.getByTestId("nostr-bind-code-digit-1"),
    VALID_REQUEST.verificationCode,
  );
  await expect(page.getByTestId("nostr-bind-finish-step")).toBeVisible();

  await expect(
    page.getByRole("heading", { name: "Continue in your browser" }),
  ).toBeVisible();
  const manualFallback = page.getByTestId("nostr-bind-manual-fallback");
  await expect(manualFallback).not.toHaveAttribute("open", "");
  await expect(page.getByTestId("nostr-bind-signed-response")).toBeHidden();
  await expect(page.getByTestId("nostr-bind-copy-response")).toBeHidden();

  const fallbackSummary = manualFallback.locator("summary");
  await fallbackSummary.focus();
  await fallbackSummary.press("Enter");
  await expect(manualFallback).toHaveAttribute("open", "");
  const signedResponse = page.getByTestId("nostr-bind-signed-response");
  await expect(signedResponse).toContainText("e2e-signed-nostr-binding");
  const copyResponse = page.getByTestId("nostr-bind-copy-response");
  await expect(copyResponse).toBeVisible();
  await copyResponse.click();
  await expect
    .poll(() =>
      page.evaluate(
        () =>
          (
            window as Window & {
              __BUZZ_E2E_CLIPBOARD_TEXT__?: string;
            }
          ).__BUZZ_E2E_CLIPBOARD_TEXT__,
      ),
    )
    .toBe(await signedResponse.textContent());

  const callback = await page.evaluate(() => {
    const command = (
      (
        window as Window & {
          __BUZZ_E2E_COMMAND_LOG__?: Array<{
            command: string;
            payload: { url?: string };
          }>;
        }
      ).__BUZZ_E2E_COMMAND_LOG__ ?? []
    ).find(({ command }) => command === "plugin:opener|open_url");
    return command?.payload.url;
  });
  const callbackUrl = new URL(callback ?? "");
  expect(callbackUrl.origin).toBe("https://admin.example.com");
  expect(callbackUrl.pathname).toBe("/buzz");
  expect(callbackUrl.search).toBe("?source=bind");
  expect(callbackUrl.searchParams.has("buzz_bind")).toBe(false);
  expect(callbackUrl.hash).toMatch(/^#buzz_bind=v1\.[A-Za-z0-9_-]+$/);
});

test("opens the manual fallback when returning to the browser fails", async ({
  page,
}) => {
  await openNostrBind(
    page,
    {
      ...VALID_REQUEST,
      callbackUrl: "https://admin.example.com/buzz",
      returnMode: "browser_fragment_v1",
    },
    { openerError: "browser unavailable" },
  );

  await pasteCode(
    page.getByTestId("nostr-bind-code-digit-1"),
    VALID_REQUEST.verificationCode,
  );

  await expect(
    page.getByText(
      "Could not open the browser. Copy the response below to finish manually.",
    ),
  ).toBeVisible();
  await expect(page.getByTestId("nostr-bind-manual-fallback")).toHaveAttribute(
    "open",
    "",
  );
  await expect(page.getByTestId("nostr-bind-copy-response")).toBeVisible();
});

test("keeps the signed response available when clipboard access fails", async ({
  page,
}) => {
  await installClipboardStub(page, true);
  await openNostrBind(page);
  await pasteCode(
    page.getByTestId("nostr-bind-code-digit-1"),
    VALID_REQUEST.verificationCode,
  );
  await expect(page.getByTestId("nostr-bind-finish-step")).toBeVisible();

  await page.getByTestId("nostr-bind-copy-response").click();
  await expect(
    page
      .getByTestId("nostr-bind-manual-fallback-content")
      .getByText("Buzz couldn't access the clipboard. Try again."),
  ).toBeVisible();
  await expect(page.getByTestId("nostr-bind-signed-response")).toContainText(
    "e2e-signed-nostr-binding",
  );
  await expect(
    page.getByTestId("nostr-bind-copy-response"),
  ).toHaveAccessibleName("Copy response");
});