// @vitest-environment happy-dom
import { afterEach, beforeEach, describe, expect, it } from "vitest";

import {
  navigateToWorkspaceApp,
  shouldOpenWorkspaceAppInTopWindow,
  workspaceAppDirectHref,
  workspaceAppEmbedTarget,
} from "./workspace-apps";

const defaultUserAgent = navigator.userAgent;

function setUserAgent(userAgent: string) {
  Object.defineProperty(navigator, "userAgent", {
    configurable: true,
    value: userAgent,
  });
}

function setAncestorOrigin(origin: string | null) {
  Object.defineProperty(window.location, "ancestorOrigins", {
    configurable: true,
    value: origin
      ? {
          0: origin,
          length: 1,
          item: (index: number) => (index === 0 ? origin : null),
        }
      : undefined,
  });
}

describe("workspaceAppEmbedTarget", () => {
  it("uses the app URL as the embed root when a mount path is also present", () => {
    expect(
      workspaceAppEmbedTarget({
        path: "/content",
        url: "https://workspace.example.test/content",
      }),
    ).toEqual({ url: "https://workspace.example.test/content" });
  });

  it("falls back to the mounted path when no app URL is available", () => {
    expect(workspaceAppEmbedTarget({ path: "/content", url: null })).toEqual({
      path: "/content",
    });
  });
});

describe("workspaceAppDirectHref", () => {
  it("joins an app-relative route onto an absolute app URL", () => {
    expect(
      workspaceAppDirectHref(
        { path: "/atlas", url: "https://workspace.example.test/atlas" },
        "/emails?status=failed#latest",
      ),
    ).toBe("https://workspace.example.test/atlas/emails?status=failed#latest");
  });

  it("does not duplicate a mount path already present in the target", () => {
    expect(
      workspaceAppDirectHref(
        { path: "/atlas", url: "https://workspace.example.test/atlas" },
        "/atlas",
      ),
    ).toBe("https://workspace.example.test/atlas");
  });

  it("resolves a mounted relative app path", () => {
    expect(workspaceAppDirectHref({ path: "/atlas" }, "/emails")).toBe(
      "/atlas/emails",
    );
  });

  it("falls back to a safe mount path when the app URL is invalid", () => {
    expect(
      workspaceAppDirectHref(
        { path: "/atlas", url: "javascript:alert(1)" },
        "/",
      ),
    ).toBe("/atlas");
  });

  it("rejects an unsafe target path", () => {
    expect(workspaceAppDirectHref({ path: "/atlas" }, "//evil.example")).toBe(
      null,
    );
  });
});

describe("navigateToWorkspaceApp", () => {
  beforeEach(() => {
    window.history.replaceState({}, "", "/");
    setAncestorOrigin(null);
  });

  afterEach(() => {
    window.history.replaceState({}, "", "/");
    setAncestorOrigin(null);
    setUserAgent(defaultUserAgent);
    Object.defineProperty(window, "parent", {
      configurable: true,
      value: window,
    });
    Object.defineProperty(window, "top", {
      configurable: true,
      value: window,
    });
  });

  it("keeps a normal iframe inline", () => {
    Object.defineProperty(window, "parent", {
      configurable: true,
      value: {},
    });

    expect(shouldOpenWorkspaceAppInTopWindow()).toBe(false);
  });

  it("keeps unrelated Electron webviews inline", () => {
    setUserAgent("Mozilla/5.0 Electron/38.0");

    expect(shouldOpenWorkspaceAppInTopWindow()).toBe(false);
  });

  it("ignores a generic Electron preload global", () => {
    setUserAgent("Mozilla/5.0 Electron/38.0");
    Object.defineProperty(window, "electronAPI", {
      configurable: true,
      value: {},
    });

    try {
      expect(shouldOpenWorkspaceAppInTopWindow()).toBe(false);
    } finally {
      delete (window as unknown as { electronAPI?: unknown }).electronAPI;
    }
  });

  it("uses the top window in known native desktop hosts", () => {
    setUserAgent("Mozilla/5.0 Electron/38.0 ChatGPT/1.0");

    expect(shouldOpenWorkspaceAppInTopWindow()).toBe(true);
  });

  it("keeps the hosted Dispatch shell inline", () => {
    Object.defineProperty(window, "parent", {
      configurable: true,
      value: {},
    });
    setAncestorOrigin("https://agent-workspace.builder.io");

    expect(shouldOpenWorkspaceAppInTopWindow()).toBe(false);
  });

  it("uses the top window when Dispatch is inside a Builder webview", () => {
    const topWindow = { location: { href: "" } } as unknown as Window;
    window.history.replaceState({}, "", "/?builder.preview=interact");
    Object.defineProperty(window, "parent", {
      configurable: true,
      value: {},
    });
    Object.defineProperty(window, "top", {
      configurable: true,
      value: topWindow,
    });

    expect(shouldOpenWorkspaceAppInTopWindow()).toBe(true);
    navigateToWorkspaceApp("/mail");

    expect(topWindow.location.href).toBe(
      new URL("/mail", window.location.href).href,
    );
  });

  it("reports when the embedded top window rejects navigation", () => {
    const location = {} as Location;
    Object.defineProperty(location, "href", {
      configurable: true,
      set: () => {
        throw new Error("Top navigation is blocked");
      },
    });
    const topWindow = { location } as unknown as Window;
    window.history.replaceState({}, "", "/?builder.preview=interact");
    Object.defineProperty(window, "parent", {
      configurable: true,
      value: {},
    });
    Object.defineProperty(window, "top", {
      configurable: true,
      value: topWindow,
    });

    expect(navigateToWorkspaceApp("/mail")).toBe(false);
  });

  it("rejects non-http navigation targets", () => {
    const topWindow = { location: { href: "" } } as unknown as Window;
    Object.defineProperty(window, "parent", {
      configurable: true,
      value: {},
    });
    Object.defineProperty(window, "top", {
      configurable: true,
      value: topWindow,
    });

    expect(navigateToWorkspaceApp("javascript:alert(1)")).toBe(false);
    expect(topWindow.location.href).toBe("");
  });
});