// @effect-diagnostics nodeBuiltinImport:off -- Real temporary files exercise the native portal URI boundary.import * as NodeEvents from "node:events";import * as NodeFSP from "node:fs/promises";import * as NodeOS from "node:os";import * as NodePath from "node:path";import * as NodeURL from "node:url";import {  DBusError,  Message,  MessageType,  RequestNameReply,  Variant,  type MessageBus,} from "dbus-next";import { afterEach, beforeEach, expect, it, vi } from "vite-plus/test"; const { connect, resize, imageSize } = vi.hoisted(() => ({  connect: vi.fn(),  resize: vi.fn(() => ({ toPNG: () => Buffer.from("resized") })),  imageSize: { width: 800, height: 600 },}));vi.mock("dbus-next", async (original) => ({  ...(await original<typeof import("dbus-next")>()),  sessionBus: connect,}));vi.mock("electron", () => ({  nativeImage: {    createFromBuffer: () => ({      isEmpty: () => false,      getSize: () => imageSize,      resize,    }),  },})); import {  captureLinuxWindow,  getLinuxCaptureSupport,  readPortalPng,  resizeLinuxCapture,} from "./LinuxSnapShot.ts"; const png = Buffer.from([137, 80, 78, 71, 13, 10, 26, 10, 0]);const appId = "com.t3tools.T3Code";const metadata = {  title: "Editor",  appName: "Text Editor",  appIdentifier: "org.gnome.TextEditor.desktop",  processId: 42,  bounds: { x: -100, y: 20, width: 800, height: 600 },}; class FakeBus extends NodeEvents.EventEmitter {  calls: Message[] = [];  portalVersion = 3;  targets: number | undefined = 8;  extensionVersion: number | undefined;  kdeVersion: number | undefined = 5;  kdeError: Error | undefined;  status = 0;  uri = "file:///missing.png";  registryError: Error | undefined;  screenshotError: Error | undefined;  extensionError: Error | undefined;  activationError: Error | undefined;  animation: Promise<void> | undefined;  sendResponse = true;  differentHandle = false;  wrongSender = false;  requestName = vi.fn(async () => RequestNameReply.PRIMARY_OWNER);  disconnect = vi.fn();  send = vi.fn();   signal(path: string, body: unknown[], sender = ":1.2") {    this.emit(      "message",      new Message({        type: MessageType.SIGNAL,        sender,        path,        interface: "org.freedesktop.portal.Request",        member: "Response",        body,      }),    );  }   async call(message: Message) {    this.calls.push(message);    const reply = (body: unknown[]) =>      new Message({        type: MessageType.METHOD_RETURN,        replySerial: "1",        destination: ":1.23",        body,      });    switch (message.member) {      case "Register":        if (this.registryError) throw this.registryError;        return reply([]);      case "GetAll":        if (message.destination === "org.kde.KWin.ScreenShot2") {          if (this.kdeError) throw this.kdeError;          if (this.kdeVersion === undefined)            throw new DBusError("org.freedesktop.DBus.Error.ServiceUnknown", "Absent");          return reply([{ Version: new Variant("u", this.kdeVersion) }]);        }        if (message.destination === "org.freedesktop.portal.Desktop")          return reply([            {              version: new Variant("u", this.portalVersion),              ...(this.targets === undefined                ? {}                : { AvailableTargets: new Variant("u", this.targets) }),            },          ]);        if (this.extensionVersion === undefined)          throw new DBusError("org.freedesktop.DBus.Error.ServiceUnknown", "Absent");        return reply([{ Version: new Variant("u", this.extensionVersion) }]);      case "GetNameOwner":        return reply([":1.2"]);      case "AddMatch":        return reply([]);      case "Screenshot": {        if (this.screenshotError) throw this.screenshotError;        const options = message.body[1] as Record<string, Variant<unknown>>;        const handle = `/org/freedesktop/portal/desktop/request/1_23/${this.differentHandle ? "actual" : options.handle_token?.value}`;        if (this.sendResponse)          this.signal(            handle,            [this.status, { uri: new Variant("s", this.uri) }],            this.wrongSender ? ":1.99" : ":1.2",          );        return reply([handle]);      }      case "Capture":        if (this.extensionError) throw this.extensionError;        return reply([png, JSON.stringify(metadata)]);      case "CaptureWithFeedback":        return reply([png, JSON.stringify(metadata), message.body[1]]);      case "Activate":        if (this.activationError) throw this.activationError;        return reply([]);      case "Animate":        await this.animation;        return reply([]);      default:        throw new Error(`Unexpected method ${message.member}`);    }  }} let bus: FakeBus;let directory: string;beforeEach(async () => {  bus = new FakeBus();  connect.mockImplementation(() => bus as unknown as MessageBus);  directory = await NodeFSP.mkdtemp(NodePath.join(NodeOS.tmpdir(), "t3-portal-test-"));  bus.uri = NodeURL.pathToFileURL(NodePath.join(directory, "image.png")).href;  await NodeFSP.writeFile(NodePath.join(directory, "image.png"), png);  imageSize.width = 800;  imageSize.height = 600;  vi.stubEnv("FLATPAK_ID", "");  vi.stubEnv("SNAP", "");  vi.stubEnv("NIRI_SOCKET", "");  vi.stubEnv("XDG_CURRENT_DESKTOP", "GNOME");}); it("uses native KWin only in an unsandboxed KDE session", async () => {  vi.stubEnv("XDG_CURRENT_DESKTOP", "KDE");  expect((await getLinuxCaptureSupport(appId)).linuxBackend).toBe("kde");  vi.stubEnv("FLATPAK_ID", "com.example.App");  expect((await getLinuxCaptureSupport(appId)).linuxBackend).not.toBe("kde");}); it("redetects the desktop without using another session's extension or stale Niri socket", async () => {  bus.portalVersion = 2;  bus.extensionVersion = 2;  vi.stubEnv("NIRI_SOCKET", "/previous-session/niri.sock");  expect(await getLinuxCaptureSupport(appId)).toEqual({    linuxBackend: "gnome-extension",    linuxFeedbackAvailable: true,  });   vi.stubEnv("XDG_CURRENT_DESKTOP", "KDE");  expect(await getLinuxCaptureSupport(appId)).toEqual({    linuxBackend: "kde",    linuxFeedbackAvailable: false,  });  bus.kdeVersion = undefined;  expect((await getLinuxCaptureSupport(appId)).linuxBackend).toBe("picker");   vi.stubEnv("XDG_CURRENT_DESKTOP", "GNOME");  expect(await getLinuxCaptureSupport(appId)).toEqual({    linuxBackend: "gnome-extension",    linuxFeedbackAvailable: true,  });}); it("uses the standard path when KWin's native capability is missing, but not when denied", async () => {  vi.stubEnv("XDG_CURRENT_DESKTOP", "KDE");  bus.kdeVersion = undefined;  expect((await getLinuxCaptureSupport(appId)).linuxBackend).toBe("screenshot-portal");  bus.kdeVersion = 1;  bus.portalVersion = 2;  expect((await getLinuxCaptureSupport(appId)).linuxBackend).toBe("picker");  bus.kdeError = new DBusError("org.freedesktop.DBus.Error.AccessDenied", "Denied");  await expect(getLinuxCaptureSupport(appId)).rejects.toThrow("Denied");}); it("does not fall through to the picker when KDE needs helper setup", async () => {  vi.stubEnv("XDG_CURRENT_DESKTOP", "KDE");  await expect(captureLinuxWindow(appId)).rejects.toThrow("KDE capture setup");  expect(bus.calls.some((call) => call.member === "Screenshot")).toBe(false);  expect(bus.disconnect).toHaveBeenCalledOnce();});it("selects Hyprland without probing a different desktop or falling back on missing setup", async () => {  vi.stubEnv("XDG_CURRENT_DESKTOP", "Hyprland");  expect((await getLinuxCaptureSupport(appId)).linuxBackend).toBe("hyprland");  await expect(captureLinuxWindow(appId)).rejects.toThrow("Hyprland capture setup");  expect(bus.calls).toEqual([]);});afterEach(async () => {  vi.useRealTimers();  vi.unstubAllEnvs();  await NodeFSP.rm(directory, { recursive: true });}); it.each([  [2, 8, 1, "gnome-extension"],  [3, undefined, 1, "gnome-extension"],  [3, 7, undefined, "picker"],  [3, 8, 1, "screenshot-portal"],  [4, 15, undefined, "screenshot-portal"],  [2, undefined, 2, "gnome-extension"],  [2, undefined, 99, "picker"],] as const)(  "selects capabilities v%s / targets %s / extension %s",  async (version, targets, extension, expected) => {    bus.portalVersion = version;    bus.targets = targets;    bus.extensionVersion = extension;    expect((await getLinuxCaptureSupport(appId)).linuxBackend).toBe(expected);    expect(bus.calls[0]?.member).toBe("Register");    expect(bus.disconnect).toHaveBeenCalledOnce();  },); it.each([1, 2])(  "reports feedback separately from capture support for extension v%s",  async (version) => {    bus.portalVersion = 2;    bus.extensionVersion = version;    expect(await getLinuxCaptureSupport(appId)).toEqual({      linuxBackend: "gnome-extension",      linuxFeedbackAvailable: version === 2,    });  },); it("retains the authenticated connection until the compositor flight has landed", async () => {  bus.portalVersion = 2;  bus.extensionVersion = 2;  const snapshot = await captureLinuxWindow(appId, { flash: false, animate: true });  const feedback = snapshot!.feedback!;  expect(snapshot).toMatchObject({ png, window: metadata, feedback: { animationStarted: true } });  expect(bus.disconnect).not.toHaveBeenCalled();  await feedback.activate("T3 Code");  const landed = Promise.withResolvers<void>();  bus.animation = landed.promise;  const flight = feedback.animateTo({ x: 0.1, y: 0.8, width: 0.2, height: 0.1 });  const complete = feedback.complete();  expect(bus.disconnect).not.toHaveBeenCalled();  landed.resolve();  await Promise.all([flight, complete]);  expect(    bus.calls      .filter((call) => ["CaptureWithFeedback", "Activate", "Animate"].includes(call.member))      .map((call) => call.body),  ).toEqual([[false, true], ["T3 Code"], [0.1, 0.8, 0.2, 0.1]]);  expect(bus.disconnect).toHaveBeenCalledOnce();  feedback.close();  expect(bus.disconnect).toHaveBeenCalledOnce();}); it("keeps capture working with v1 and does not request unavailable effects", async () => {  bus.portalVersion = 2;  bus.extensionVersion = 1;  expect(await captureLinuxWindow(appId, { flash: true, animate: true })).toEqual({    png,    window: metadata,  });  expect(bus.calls.some((call) => call.member === "CaptureWithFeedback")).toBe(false);  expect(bus.disconnect).toHaveBeenCalledOnce();}); it("can activate without animation, and expires a renderer-abandoned capture", async () => {  vi.useFakeTimers();  bus.portalVersion = 2;  bus.extensionVersion = 2;  const snapshot = await captureLinuxWindow(appId, { flash: false, animate: false });  expect(snapshot!.feedback!.animationStarted).toBe(false);  bus.activationError = new Error("No window");  await expect(snapshot!.feedback!.activate("T3 Code")).rejects.toThrow("No window");  expect(snapshot!.png).toEqual(png);  await vi.advanceTimersByTimeAsync(15_000);  expect(bus.disconnect).toHaveBeenCalledOnce();}); it.each([false, true])(  "accepts an early Response with a changed request path: %s",  async (differentHandle) => {    bus.differentHandle = differentHandle;    expect(await captureLinuxWindow(appId)).toEqual({ png });    const screenshot = bus.calls.find((call) => call.member === "Screenshot");    expect(screenshot?.body[1]).toMatchObject({      target: { signature: "u", value: 8 },      interactive: { value: false },    });    expect(bus.calls.findIndex((call) => call.member === "AddMatch")).toBeLessThan(      bus.calls.indexOf(screenshot!),    );    expect(bus.listenerCount("message")).toBe(0);    expect(bus.disconnect).toHaveBeenCalledOnce();  },); it.each([1, 2])(  "does not try the extension or picker after portal rejection %s",  async (status) => {    bus.extensionVersion = 1;    bus.status = status;    await expect(captureLinuxWindow(appId)).rejects.toThrow(      status === 1 ? "cancelled" : "did not allow",    );    expect(bus.calls.some((call) => call.member === "Capture")).toBe(false);    expect(bus.disconnect).toHaveBeenCalledOnce();  },); it("does not downgrade a failed screenshot request", async () => {  bus.screenshotError = new DBusError("org.freedesktop.DBus.Error.AccessDenied", "Denied");  await expect(captureLinuxWindow(appId)).rejects.toThrow("Denied");  expect(bus.send).toHaveBeenCalledWith(expect.objectContaining({ member: "Close" }));  expect(bus.disconnect).toHaveBeenCalledOnce();}); it.each([false, true])(  "times out and closes requests, ignoring forged responses: %s",  async (wrongSender) => {    vi.useFakeTimers();    bus.sendResponse = wrongSender;    bus.wrongSender = wrongSender;    const capture = expect(captureLinuxWindow(appId)).rejects.toThrow("timed out");    await vi.advanceTimersByTimeAsync(120_001);    await capture;    expect(bus.send).toHaveBeenCalledWith(expect.objectContaining({ member: "Close" }));    expect(bus.listenerCount("message")).toBe(0);    expect(bus.disconnect).toHaveBeenCalledOnce();  },); it("uses the GNOME extension and preserves its window identity", async () => {  bus.portalVersion = 2;  bus.extensionVersion = 1;  expect(await captureLinuxWindow(appId)).toEqual({ png, window: metadata });  expect(bus.requestName).toHaveBeenCalledWith(`${appId}.SnapShot`, expect.any(Number));  expect(bus.calls.some((call) => call.member === "Screenshot")).toBe(false);}); it("fails instead of opening a picker after an extension denial", async () => {  bus.portalVersion = 2;  bus.extensionVersion = 1;  bus.extensionError = new Error("Locked session");  await expect(captureLinuxWindow(appId)).rejects.toThrow("Locked session");  expect(bus.disconnect).toHaveBeenCalledOnce();}); it("allows the picker only when both automatic backends are absent", async () => {  bus.portalVersion = 2;  expect(await captureLinuxWindow(appId)).toBeUndefined();  expect(bus.calls.some((call) => call.member === "Screenshot" || call.member === "Capture")).toBe(    false,  );}); it("tolerates an older portal without Registry, but not registration denial", async () => {  bus.registryError = new DBusError("org.freedesktop.DBus.Error.UnknownMethod", "Old portal");  expect((await getLinuxCaptureSupport(appId)).linuxBackend).toBe("screenshot-portal");  bus.registryError = new DBusError("org.freedesktop.DBus.Error.AccessDenied", "Denied identity");  await expect(getLinuxCaptureSupport(appId)).rejects.toThrow("Denied identity");}); it("does not register a host identity from inside a sandbox", async () => {  vi.stubEnv("FLATPAK_ID", appId);  await getLinuxCaptureSupport(appId);  expect(bus.calls.some((call) => call.member === "Register")).toBe(false);}); it("reads only local PNG files without removing the portal file", async () => {  expect(await readPortalPng(bus.uri)).toEqual(png);  expect(await readPortalPng(bus.uri)).toEqual(png);  await expect(readPortalPng("https://example.com/image.png")).rejects.toThrow();  await expect(readPortalPng("file://remote/image.png")).rejects.toThrow();  await NodeFSP.writeFile(NodePath.join(directory, "image.png"), "not a PNG");  await expect(readPortalPng(bus.uri)).rejects.toThrow("Invalid");}); it("bounds large images without distorting their aspect ratio", () => {  imageSize.width = 4000;  imageSize.height = 2000;  resizeLinuxCapture(png);  expect(resize).toHaveBeenLastCalledWith({ width: 2560, height: 1280, quality: "best" });});