// biome-ignore-all lint/suspicious/noConsole: we use console.log for logging in this file
import {
  ADMIN_EMAIL,
  ADMIN_PASSWORD,
  E2eTestId,
  getIdentityProviderDialogNavButtonTestId,
  getIdpRoleMappingRuleRowTestId,
  KEYCLOAK_OIDC,
  KEYCLOAK_SAML,
  SSO_DOMAIN,
  UI_BASE_URL,
} from "../consts";
import {
  type APIResponse,
  type Browser,
  expect,
  type Page,
  test,
} from "../fixtures";
import {
  clickButton,
  clickTeamActionButton,
  closeOpenDialogs,
  createTeam,
  deleteTeamByName,
  expectAuthenticated,
  extractCertFromMetadata,
  fetchKeycloakSamlMetadata,
  loginViaApi,
  loginViaKeycloak,
  loginViaUi,
} from "../utils";

// Run tests in this file serially to avoid conflicts when both tests
// manipulate identity providers in the same Keycloak realm.
// Also skip webkit and firefox for these tests since they share the same backend
// and running in parallel causes identity provider conflicts.
test.describe.configure({ mode: "serial" });

// =============================================================================
// Shared Test Helpers
// =============================================================================

/**
 * Authenticate as admin via API and navigate to identity providers page.
 * Identity provider tests don't use storage state to avoid session conflicts.
 * Clears existing cookies first to ensure clean authentication state.
 * Uses polling with retry to handle timing issues.
 */
async function ensureAdminAuthenticated(page: Page): Promise<void> {
  // Clear all cookies to ensure no stale session cookies interfere with login
  // This is critical on retries where previous SSO logins may have invalidated sessions
  await page.context().clearCookies();

  // Retry login up to 5 times to handle transient issues and server instability
  let loginSucceeded = false;
  for (let attempt = 1; attempt <= 5; attempt++) {
    loginSucceeded = await loginViaApi(page, ADMIN_EMAIL, ADMIN_PASSWORD);
    if (loginSucceeded) break;
    // Wait before retry
    await page.waitForTimeout(2000);
  }

  if (!loginSucceeded) {
    console.log("Admin API login failed after 5 attempts");
  }

  // Navigate directly to identity providers page
  // The API login should have set session cookies that persist
  await page.goto(`${UI_BASE_URL}/settings/identity-providers`);
  await page.waitForLoadState("domcontentloaded");

  // Wait briefly for any redirects to complete
  await page.waitForTimeout(1000);

  // Check if we got redirected to sign-in (authentication failed)
  if (page.url().includes("/auth/sign-in")) {
    console.log(
      "API login appeared to fail (redirected to sign-in), trying UI fallback...",
    );
    // Try logging in via UI as fallback
    await loginViaUi(page, ADMIN_EMAIL, ADMIN_PASSWORD);
    await page.waitForLoadState("domcontentloaded");

    // Check for error toast or message on the sign-in page
    const errorToast = page.locator('[role="alert"]').first();
    if (await errorToast.isVisible()) {
      const errorText = await errorToast.textContent().catch(() => null);
      if (
        errorText &&
        !errorText.includes("Default Admin Credentials Enabled")
      ) {
        console.log(`UI Login failed with error: ${errorText}`);
      }
    }

    // Wait for login to complete and redirect away from sign-in
    try {
      await expect(page).not.toHaveURL(/\/auth\/sign-in/, { timeout: 45000 });
    } catch {
      // If still on sign-in, try a hard reload and check URL again
      // Sometimes state is stale or cookie needs a nudge
      console.log(
        "Still on sign-in after login, attempting reload to check session...",
      );
      await page.reload();
      await page.waitForLoadState("domcontentloaded");
      await expect(page).not.toHaveURL(/\/auth\/sign-in/, { timeout: 15000 });
    }

    // Navigate to identity providers after UI login
    await page.goto(`${UI_BASE_URL}/settings/identity-providers`);
    await page.waitForLoadState("domcontentloaded");
  }

  await expect(page).toHaveURL(/\/settings\/identity-providers/, {
    timeout: 30000,
  });
  await expect(
    page.getByRole("heading", { name: "Identity Providers" }),
  ).toBeVisible({ timeout: 10000 });
}

/**
 * Fill in the standard OIDC provider form fields.
 */
async function fillOidcProviderForm(
  page: Page,
  providerName: string,
): Promise<void> {
  await page.getByLabel("Provider ID").fill(providerName);
  await page.getByLabel("Issuer").fill(KEYCLOAK_OIDC.issuer);
  const allowedDomainsInput = page.getByLabel("Allowed Email Domains");
  if (await allowedDomainsInput.isVisible().catch(() => false)) {
    await allowedDomainsInput.fill(SSO_DOMAIN);
  }
  await page.getByLabel("Client ID").fill(KEYCLOAK_OIDC.clientId);
  await page.getByLabel("Client Secret").fill(KEYCLOAK_OIDC.clientSecret);
  await page
    .getByLabel("Discovery Endpoint")
    .fill(KEYCLOAK_OIDC.discoveryEndpoint);
  await page
    .getByLabel("Authorization Endpoint")
    .fill(KEYCLOAK_OIDC.authorizationEndpoint);
  await page.getByLabel("Token Endpoint").fill(KEYCLOAK_OIDC.tokenEndpoint);
  await page.getByLabel("JWKS Endpoint").fill(KEYCLOAK_OIDC.jwksEndpoint);
}

async function createOidcProviderViaApi(
  page: Page,
  providerName: string,
): Promise<void> {
  const response = await page.request.post(
    `${UI_BASE_URL}/api/identity-providers`,
    {
      data: {
        providerId: providerName,
        issuer: KEYCLOAK_OIDC.issuer,
        domain: SSO_DOMAIN,
        oidcConfig: {
          issuer: KEYCLOAK_OIDC.issuer,
          pkce: true,
          enableRpInitiatedLogout: true,
          clientId: KEYCLOAK_OIDC.clientId,
          clientSecret: KEYCLOAK_OIDC.clientSecret,
          discoveryEndpoint: KEYCLOAK_OIDC.discoveryEndpoint,
          authorizationEndpoint: KEYCLOAK_OIDC.authorizationEndpoint,
          tokenEndpoint: KEYCLOAK_OIDC.tokenEndpoint,
          jwksEndpoint: KEYCLOAK_OIDC.jwksEndpoint,
          scopes: ["openid", "email", "profile"],
          mapping: {
            id: "sub",
            email: "email",
            name: "name",
          },
          overrideUserInfo: true,
        },
        teamSyncConfig: {
          enabled: true,
          groupsExpression: "{{#each groups}}{{this}},{{/each}}",
        },
      },
    },
  );

  await expectApiResponseOk(response, "create identity provider");

  const createdProvider = (await response.json()) as {
    providerId: string;
    teamSyncConfig?: {
      enabled?: boolean;
      groupsExpression?: string;
    };
  };

  expect(createdProvider.providerId).toBe(providerName);
  expect(createdProvider.teamSyncConfig?.enabled).toBe(true);
  expect(createdProvider.teamSyncConfig?.groupsExpression).toBe(
    "{{#each groups}}{{this}},{{/each}}",
  );
}

async function deleteProviderByProviderIdViaApi(
  page: Page,
  providerName: string,
): Promise<void> {
  const providersResponse = await page.request.get(
    `${UI_BASE_URL}/api/identity-providers`,
  );
  await expectApiResponseOk(providersResponse, "list identity providers");

  const providers = (await providersResponse.json()) as Array<{
    id: string;
    providerId: string;
  }>;

  const provider = providers.find((item) => item.providerId === providerName);
  if (!provider) {
    return;
  }

  const deleteResponse = await page.request.delete(
    `${UI_BASE_URL}/api/identity-providers/${provider.id}`,
  );
  await expectApiResponseOk(deleteResponse, "delete identity provider");
}

async function expectApiResponseOk(
  response: APIResponse,
  action: string,
): Promise<void> {
  const bodyText = await response.text();
  expect(
    response.ok(),
    `${action} failed with ${response.status()}: ${bodyText}`,
  ).toBe(true);
}

async function getTeamIdByNameViaApi(
  page: Page,
  teamName: string,
): Promise<string> {
  const response = await page.request.get(
    `${UI_BASE_URL}/api/teams?limit=10&offset=0&name=${encodeURIComponent(teamName)}`,
  );
  await expectApiResponseOk(response, "list teams");

  const body = (await response.json()) as {
    data: Array<{ id: string; name: string }>;
  };
  const team = body.data.find((item) => item.name === teamName);

  expect(team, `Expected team "${teamName}" to exist`).toBeDefined();
  return team?.id ?? "";
}

function getRoleMappingRuleRow(page: Page, index: number) {
  return page.getByTestId(getIdpRoleMappingRuleRowTestId(index));
}

async function openIdentityProviderDialogSection(
  page: Page,
  section: string,
): Promise<void> {
  await page
    .getByTestId(getIdentityProviderDialogNavButtonTestId(section))
    .click();
}

function getIdentityProviderConfigId(
  providerType: "Generic OIDC" | "Generic SAML",
): string {
  return providerType === "Generic OIDC" ? "generic-oidc" : "generic-saml";
}

async function openIdentityProviderDialog(
  page: Page,
  providerType: "Generic OIDC" | "Generic SAML",
): Promise<void> {
  const configId = getIdentityProviderConfigId(providerType);
  const openButton = page.getByTestId(
    `${E2eTestId.IdentityProviderOpenDialogButton}-${configId}`,
  );
  await openButton.waitFor({ state: "visible", timeout: 20000 });
  await openButton.click();
  await expect(page.getByRole("dialog")).toBeVisible({ timeout: 10000 });
}

async function waitForIdentityProviderDialogToClose(page: Page): Promise<void> {
  await expect
    .poll(
      async () => {
        const createVisible = await page
          .getByTestId(E2eTestId.IdentityProviderCreateButton)
          .isVisible()
          .catch(() => false);
        const updateVisible = await page
          .getByTestId(E2eTestId.IdentityProviderUpdateButton)
          .isVisible()
          .catch(() => false);
        return createVisible || updateVisible ? "open" : "closed";
      },
      { timeout: 10_000, intervals: [250, 500, 1000] },
    )
    .toBe("closed");
}

async function settleIdentityProviderDialog(page: Page): Promise<void> {
  const createVisible = await page
    .getByTestId(E2eTestId.IdentityProviderCreateButton)
    .isVisible()
    .catch(() => false);
  const updateVisible = await page
    .getByTestId(E2eTestId.IdentityProviderUpdateButton)
    .isVisible()
    .catch(() => false);

  if (createVisible || updateVisible) {
    await closeOpenDialogs(page, { timeoutMs: 10_000 });
  }
}

async function expectIdentityProviderToExistViaApi(
  page: Page,
  providerName: string,
): Promise<void> {
  await expect(async () => {
    const response = await page.request.get(
      `${UI_BASE_URL}/api/identity-providers`,
    );
    await expectApiResponseOk(response, "list identity providers");

    const providers = (await response.json()) as Array<{ providerId: string }>;
    expect(
      providers.some((provider) => provider.providerId === providerName),
    ).toBe(true);
  }).toPass({ timeout: 15_000, intervals: [500, 1000, 2000] });
}

/**
 * Delete an identity provider via the UI dialog.
 */
async function deleteProviderViaDialog(page: Page): Promise<void> {
  await page.getByTestId(E2eTestId.IdentityProviderDeleteButton).click();
  await expect(page.getByText(/Are you sure/i)).toBeVisible();
  await clickButton({ page, options: { name: "Delete", exact: true } });
  await waitForIdentityProviderDialogToClose(page);
}

/**
 * Ensure a clean slate by deleting any existing identity provider of the given type.
 * This makes tests idempotent - they can be retried or re-run without manual cleanup.
 *
 * @param page - The Playwright page (logged in as admin, on identity providers page)
 * @param providerType - Either "Generic OIDC" or "Generic SAML"
 */
async function deleteExistingProviderIfExists(
  page: Page,
  providerType: "Generic OIDC" | "Generic SAML",
): Promise<void> {
  // Verify we're on the identity providers page before proceeding
  // This handles cases where previous test left page on a different route
  await expect(page).toHaveURL(/\/settings\/identity-providers/, {
    timeout: 10000,
  });

  // Wait for the Identity Providers heading to be visible (page content loaded)
  await expect(
    page.getByRole("heading", { name: "Identity Providers" }),
  ).toBeVisible({ timeout: 15000 });

  const createButton = page.getByTestId(E2eTestId.IdentityProviderCreateButton);
  const updateButton = page.getByTestId(E2eTestId.IdentityProviderUpdateButton);

  // Multiple generic providers can exist in the local/CI test environment.
  // Keep deleting any matching provider until the card opens a true create dialog.
  await expect(async () => {
    await page.goto(`${UI_BASE_URL}/settings/identity-providers`);
    await page.waitForLoadState("domcontentloaded");
    await expect(
      page.getByRole("heading", { name: "Identity Providers" }),
    ).toBeVisible({ timeout: 10000 });

    await openIdentityProviderDialog(page, providerType);

    if (await createButton.isVisible().catch(() => false)) {
      return;
    }

    await expect(updateButton).toBeVisible({ timeout: 5000 });
    await deleteProviderViaDialog(page);
    throw new Error(
      `Deleted existing ${providerType}; retrying until create dialog is available`,
    );
  }).toPass({ timeout: 60_000, intervals: [1000, 2000, 5000] });
}

async function signInViaIdentityProvider(params: {
  browser: Browser;
  providerName: string;
}): Promise<{
  context: Awaited<ReturnType<Browser["newContext"]>>;
  page: Page;
}> {
  const context = await params.browser.newContext({
    storageState: undefined,
  });
  const page = await context.newPage();

  await page.goto(`${UI_BASE_URL}/auth/sign-in`);
  await page.waitForLoadState("domcontentloaded");

  const ssoButton = page.getByRole("button", {
    name: new RegExp(params.providerName, "i"),
  });
  await expect(ssoButton).toBeVisible({ timeout: 10_000 });

  await clickButton({
    page,
    options: { name: new RegExp(params.providerName, "i") },
  });

  const loginSucceeded = await loginViaKeycloak(page);
  expect(loginSucceeded).toBe(true);
  await expectAuthenticated(page, 15_000);

  return { context, page };
}

async function expectRolesPageAfterSsoLogin(
  browser: Browser,
  providerName: string,
): Promise<void> {
  let lastError: unknown;

  for (let attempt = 1; attempt <= 3; attempt++) {
    const ssoContext = await browser.newContext({
      storageState: undefined,
    });
    const ssoPage = await ssoContext.newPage();

    try {
      await ssoPage.goto(`${UI_BASE_URL}/auth/sign-in`);
      await ssoPage.waitForLoadState("domcontentloaded");

      const ssoButton = ssoPage.getByRole("button", {
        name: new RegExp(providerName, "i"),
      });
      await expect(ssoButton).toBeVisible({ timeout: 10000 });

      await clickButton({
        page: ssoPage,
        options: { name: new RegExp(providerName, "i") },
      });

      const loginSucceeded = await loginViaKeycloak(ssoPage);
      expect(loginSucceeded).toBe(true);

      await expectAuthenticated(ssoPage, 15000);
      await expectActiveSession(ssoPage);

      await expect(async () => {
        await ssoPage.goto(`${UI_BASE_URL}/settings/roles`);
        await ssoPage.waitForLoadState("domcontentloaded");
        await ssoPage.waitForLoadState("networkidle").catch(() => {});

        await expect(ssoPage).toHaveURL(/\/settings\/roles/, { timeout: 5000 });
        await expectActiveSession(ssoPage);
        await expect(
          ssoPage.getByTestId(E2eTestId.SidebarUserProfile),
        ).toBeVisible({
          timeout: 5000,
        });
        await expect(
          ssoPage.getByRole("heading", { name: "Roles" }),
        ).toBeVisible({
          timeout: 10000,
        });
      }).toPass({ timeout: 30_000, intervals: [1000, 2000, 5000] });

      await ssoContext.close();
      return;
    } catch (error) {
      lastError = error;
      console.log(
        `OIDC login attempt ${attempt}/3 failed for ${providerName}: ${
          error instanceof Error ? error.message : String(error)
        }`,
      );
      await ssoContext.close();
      if (attempt < 3) {
        await new Promise((resolve) => setTimeout(resolve, 2000));
      }
    }
  }

  throw lastError;
}

async function expectActiveSession(page: Page): Promise<void> {
  await expect
    .poll(
      async () => {
        return page.evaluate(async () => {
          const response = await fetch("/api/auth/get-session", {
            credentials: "include",
            cache: "no-store",
          });

          if (!response.ok) {
            return null;
          }

          const session = await response.json();

          if (session?.user?.email) {
            return session.user.email;
          }

          const sidebarEmail =
            document
              .querySelector('[data-testid="sidebar-user-profile"]')
              ?.textContent?.match(
                /[A-Z0-9._%+-]+@[A-Z0-9.-]+\.[A-Z]{2,}/i,
              )?.[0] ?? null;

          return sidebarEmail;
        });
      },
      {
        timeout: 20_000,
        intervals: [500, 1000, 2000, 5000],
      },
    )
    .toBe(ADMIN_EMAIL);
}

test.describe("Identity Provider Team Sync E2E", () => {
  test("should sync user to team based on SSO group membership", async ({
    page,
    browser,
    goToPage,
    makeRandomString,
  }) => {
    test.slow();
    test.setTimeout(180_000);

    const providerName = `TeamSyncOIDC${Date.now()}`;
    const teamName = makeRandomString(8, "SyncTeam");
    const externalGroup = "archestra-admins"; // Matches Keycloak admin user's group

    // STEP 1: Authenticate and create OIDC provider
    await ensureAdminAuthenticated(page);
    await deleteProviderByProviderIdViaApi(page, providerName);
    await createOidcProviderViaApi(page, providerName);

    // STEP 2: Navigate to teams page and create a team
    await ensureAdminAuthenticated(page);
    await goToPage(page, "/settings/teams");
    await page.waitForLoadState("domcontentloaded");
    await createTeam({
      page,
      name: teamName,
      description: "Team for testing SSO group sync",
    });

    // STEP 3: Link external group to the team. External group sync lives in
    // the tabbed team management dialog (opened via the row's Edit action),
    // under the "External Group Sync" section.
    await clickTeamActionButton({
      page,
      teamName,
      actionName: "Edit",
    });

    // Wait for the dialog and switch to the External Group Sync section
    const teamDialog = page.getByRole("dialog");
    await expect(teamDialog).toBeVisible();
    await teamDialog
      .getByRole("button", { name: "External Group Sync" })
      .click();

    // Add the external group mapping
    await page.getByPlaceholder(/archestra-admins/).fill(externalGroup);
    await clickButton({ page, options: { name: "Add" } });

    // Wait for the success toast to confirm the API call completed
    // This is critical - the group must be saved to the database before SSO login
    await expect(page.getByText("External group mapping added")).toBeVisible({
      timeout: 10000,
    });

    // Also verify the group appears in the current mappings list (not just the input)
    await expect(page.getByRole("dialog").getByText(externalGroup)).toBeVisible(
      { timeout: 5000 },
    );

    // Close the dialog - use first() to target the text button, not the X icon
    await clickButton({
      page,
      options: { name: "Close", exact: true },
      first: true,
    });
    await expect(page.getByRole("dialog")).not.toBeVisible({ timeout: 5000 });

    // STEP 4: Test SSO login with admin user (in archestra-admins group)
    const { context: ssoContext } = await signInViaIdentityProvider({
      browser,
      providerName,
    });

    try {
      // STEP 5: Verify user was automatically added to the team
      // Re-authenticate the admin page before polling. In CI, the SSO flow can
      // invalidate or age out the pre-existing admin session used for cleanup
      // and verification.
      await ensureAdminAuthenticated(page);
      const teamId = await getTeamIdByNameViaApi(page, teamName);

      // Poll the teams API until the synced membership row exists. This is a
      // more direct assertion than reopening the members dialog and lets us
      // verify the actual SSO-synced row shape.
      await expect(async () => {
        const response = await page.request.get(
          `${UI_BASE_URL}/api/teams/${teamId}/members`,
        );
        await expectApiResponseOk(response, "get team members");

        const members = (await response.json()) as Array<{
          email: string;
          syncedFromSso: boolean;
        }>;

        const syncedMember = members.find(
          (member) => member.email === ADMIN_EMAIL,
        );
        expect(
          syncedMember,
          `Expected ${ADMIN_EMAIL} to appear in synced members for ${teamName}`,
        ).toBeDefined();
        expect(syncedMember?.syncedFromSso).toBe(true);
      }).toPass({ timeout: 120_000, intervals: [3000, 5000, 7000, 10000] });

      // Success! The SSO user was automatically synced to the team.
    } finally {
      await ssoContext.close();
    }

    // STEP 6: Cleanup
    // Delete the team
    await goToPage(page, "/settings/teams");
    await page.waitForLoadState("domcontentloaded");
    await deleteTeamByName(page, teamName);

    await deleteProviderByProviderIdViaApi(page, providerName);
  });
});

test.describe("Identity Provider OIDC E2E Flow with Keycloak", () => {
  test("should configure OIDC provider, login via SSO, update, and delete", async ({
    page,
    browser,
    goToPage,
  }) => {
    test.slow();
    const providerName = `KeycloakOIDC${Date.now()}`;

    // STEP 1: Authenticate and clean up any existing provider
    await ensureAdminAuthenticated(page);
    await deleteExistingProviderIfExists(page, "Generic OIDC");

    // STEP 2: Create the provider via API, then continue exercising the UI for
    // SSO login, update, and delete. The create dialog has become flaky in CI
    // after recent UI changes, but the persisted provider behavior is the real
    // contract this flow depends on.
    await createOidcProviderViaApi(page, providerName);
    await settleIdentityProviderDialog(page);

    // Verify the provider is now shown as "Enabled"
    await page.reload();
    await page.waitForLoadState("domcontentloaded");

    // STEP 3: Verify SSO button appears on login page and test SSO login
    // Use a fresh browser context (not logged in) to test the SSO flow
    const { context: ssoContext } = await signInViaIdentityProvider({
      browser,
      providerName,
    });

    try {
      // SSO login successful - user is now logged in
    } finally {
      await ssoContext.close();
    }

    // STEP 5: Use the original admin page context to update the provider
    // (the original page context is still logged in as admin)
    await goToPage(page, "/settings/identity-providers");
    await page.waitForLoadState("domcontentloaded");

    // Click on Generic OIDC card to edit (our provider)
    await openIdentityProviderDialog(page, "Generic OIDC");
    await expect(
      page.getByTestId(E2eTestId.IdentityProviderUpdateButton),
    ).toBeVisible({ timeout: 10_000 });
    await expect(page.getByLabel("Client ID")).toBeVisible({
      timeout: 10_000,
    });

    // Update a Generic OIDC field that is rendered in the edit dialog.
    await page.getByLabel("Client ID").clear();
    await page
      .getByLabel("Client ID")
      .fill(`${KEYCLOAK_OIDC.clientId}-updated`);

    // Save changes
    await page.getByTestId(E2eTestId.IdentityProviderUpdateButton).click();
    await expectIdentityProviderToExistViaApi(page, providerName);
    await settleIdentityProviderDialog(page);

    // STEP 6: Delete the provider
    await openIdentityProviderDialog(page, "Generic OIDC");
    await deleteProviderViaDialog(page);

    // STEP 7: Verify SSO button no longer appears on login page
    // Use a fresh context to check the sign-in page
    const verifyContext = await browser.newContext({
      storageState: undefined,
    });
    const verifyPage = await verifyContext.newPage();

    try {
      await verifyPage.goto(`${UI_BASE_URL}/auth/sign-in`);
      await verifyPage.waitForLoadState("domcontentloaded");

      // SSO button for our provider should no longer be visible
      await expect(
        verifyPage.getByRole("button", {
          name: new RegExp(providerName, "i"),
        }),
      ).not.toBeVisible({ timeout: 5000 });
    } finally {
      await verifyContext.close();
    }
  });
});

test.describe("Identity Provider IdP Logout (RP-Initiated Logout)", () => {
  test("should terminate IdP session on Archestra sign-out", async ({
    page,
    browser,
    goToPage,
  }) => {
    test.slow();
    const providerName = `IdPLogoutOIDC${Date.now()}`;

    // STEP 1: Authenticate as admin and create OIDC provider
    await ensureAdminAuthenticated(page);
    await deleteExistingProviderIfExists(page, "Generic OIDC");
    await fillOidcProviderForm(page, providerName);
    await page.getByTestId(E2eTestId.IdentityProviderCreateButton).click();
    await expect(page.getByRole("dialog")).not.toBeVisible({ timeout: 10000 });

    // STEP 2: Login via SSO in a fresh context
    const { context: ssoContext, page: ssoPage } =
      await signInViaIdentityProvider({
        browser,
        providerName,
      });

    try {
      // STEP 3: Sign out from Archestra
      // Navigate to sign-out which should redirect to Keycloak logout, then back to sign-in
      await ssoPage.goto(`${UI_BASE_URL}/auth/sign-out`);

      // Wait for the redirect chain:
      // Archestra sign-out -> Keycloak end_session_endpoint -> post_logout_redirect_uri (/auth/sign-in)
      // The URL should eventually land back on the sign-in page
      await ssoPage.waitForURL(/\/auth\/sign-in/, { timeout: 30000 });
      await ssoPage.waitForLoadState("domcontentloaded");

      // STEP 4: Verify IdP session was terminated
      // Click SSO button again - Keycloak should require re-authentication (not auto-login)
      const ssoButtonAgain = ssoPage.getByRole("button", {
        name: new RegExp(providerName, "i"),
      });
      await expect(ssoButtonAgain).toBeVisible({ timeout: 10000 });
      await clickButton({
        page: ssoPage,
        options: { name: new RegExp(providerName, "i") },
      });

      // Should redirect to Keycloak login form (not auto-login)
      await ssoPage.waitForURL(/.*localhost:30081.*|.*keycloak.*/, {
        timeout: 30000,
      });
      await ssoPage.waitForLoadState("domcontentloaded");

      // Verify Keycloak is showing the login form (not auto-redirecting)
      const usernameField = ssoPage.getByLabel("Username or email");
      await expect(usernameField).toBeVisible({ timeout: 10000 });

      // IdP session was terminated - Keycloak requires re-authentication
    } finally {
      await ssoContext.close();
    }

    // STEP 5: Cleanup - delete the identity provider
    await goToPage(page, "/settings/identity-providers");
    await page.waitForLoadState("domcontentloaded");
    await openIdentityProviderDialog(page, "Generic OIDC");
    await deleteProviderViaDialog(page);
  });
});

test.describe("Identity Provider Role Mapping E2E", () => {
  test("should evaluate second rule when first rule does not match", async ({
    page,
    browser,
    goToPage,
  }) => {
    test.slow();
    const providerName = `MultiRuleOIDC${Date.now()}`;

    // STEP 1: Authenticate and clean up any existing provider
    await ensureAdminAuthenticated(page);
    await deleteExistingProviderIfExists(page, "Generic OIDC");

    // STEP 2: Fill in OIDC provider form
    await fillOidcProviderForm(page, providerName);

    // STEP 3: Configure Role Mapping with TWO rules
    // The first rule will NOT match (looks for a non-existent group)
    // The second rule WILL match (looks for archestra-admins group)
    await openIdentityProviderDialogSection(page, "role-mapping");

    const addRuleButton = page.getByTestId(E2eTestId.IdpRoleMappingAddRule);
    await expect(addRuleButton).toBeVisible();

    // Add FIRST rule - will NOT match (non-existent group -> editor role)
    await addRuleButton.click();
    await getRoleMappingRuleRow(page, 0)
      .getByTestId(E2eTestId.IdpRoleMappingRuleTemplate)
      .fill('{{#includes groups "non-existent-group"}}true{{/includes}}');
    await getRoleMappingRuleRow(page, 0)
      .getByTestId(E2eTestId.IdpRoleMappingRuleRole)
      .click();
    await page.getByRole("option", { name: "Editor" }).click();

    // Add SECOND rule - WILL match (archestra-admins group -> admin role)
    await addRuleButton.click();
    await getRoleMappingRuleRow(page, 1)
      .getByTestId(E2eTestId.IdpRoleMappingRuleTemplate)
      .fill('{{#includes groups "archestra-admins"}}true{{/includes}}');
    await getRoleMappingRuleRow(page, 1)
      .getByTestId(E2eTestId.IdpRoleMappingRuleRole)
      .click();
    await page.getByRole("option", { name: "Admin" }).click();

    // Set default role to member (so we can verify role mapping works, not just fallback)
    const defaultRoleSelect = page.getByTestId(
      E2eTestId.IdpRoleMappingDefaultRole,
    );
    if (await defaultRoleSelect.isVisible()) {
      await defaultRoleSelect.click();
      await page.getByRole("option", { name: "Member" }).click();
    }

    // Submit the form
    await page.getByTestId(E2eTestId.IdentityProviderCreateButton).click();
    await expect(page.getByRole("dialog")).not.toBeVisible({ timeout: 10000 });

    // STEP 4: Test SSO login with admin user (in archestra-admins group)
    // The first rule should NOT match, but the second rule SHOULD match
    // STEP 5: Verify the user has admin role (from second rule, not editor from first)
    // The OIDC callback can hit transient Keycloak connection errors in CI, so retry
    // the whole fresh-context login flow before considering this a real failure.
    await expectRolesPageAfterSsoLogin(browser, providerName);

    // STEP 6: Cleanup - delete the provider
    await goToPage(page, "/settings/identity-providers");
    await page.waitForLoadState("domcontentloaded");
    await openIdentityProviderDialog(page, "Generic OIDC");
    await deleteProviderViaDialog(page);
  });

  test("should map admin group to admin role via OIDC", async ({
    page,
    browser,
    goToPage,
  }) => {
    test.slow();
    const providerName = `RoleMappingOIDC${Date.now()}`;

    // STEP 1: Authenticate and clean up any existing provider
    await ensureAdminAuthenticated(page);
    await deleteExistingProviderIfExists(page, "Generic OIDC");

    // STEP 2: Fill in OIDC provider form
    await fillOidcProviderForm(page, providerName);

    // STEP 2: Configure Role Mapping
    await openIdentityProviderDialogSection(page, "role-mapping");

    // Wait for accordion to expand - look for the Add Rule button
    const addRuleButton = page.getByTestId(E2eTestId.IdpRoleMappingAddRule);
    await expect(addRuleButton).toBeVisible();

    // Add a rule to map archestra-admins group to admin role
    await addRuleButton.click();

    // Fill in the Handlebars template using data-testid
    // Keycloak sends groups as an array, so we check if 'archestra-admins' is in it
    await page
      .getByTestId(E2eTestId.IdpRoleMappingRuleTemplate)
      .fill('{{#includes groups "archestra-admins"}}true{{/includes}}');

    // Select admin role using data-testid
    const roleSelect = page.getByTestId(E2eTestId.IdpRoleMappingRuleRole);
    await roleSelect.click();
    await page.getByRole("option", { name: "Admin" }).click();

    // Set default role to member (so we can verify role mapping works)
    const defaultRoleSelect = page.getByTestId(
      E2eTestId.IdpRoleMappingDefaultRole,
    );
    if (await defaultRoleSelect.isVisible()) {
      await defaultRoleSelect.click();
      await page.getByRole("option", { name: "Member" }).click();
    }

    // Submit the form
    await page.getByTestId(E2eTestId.IdentityProviderCreateButton).click();
    await expect(page.getByRole("dialog")).not.toBeVisible({ timeout: 10000 });

    // STEP 3: Test SSO login with admin user (in archestra-admins group)
    // The admin user is configured in Keycloak with the archestra-admins group
    const ssoContext = await browser.newContext({
      storageState: undefined,
    });
    const ssoPage = await ssoContext.newPage();

    try {
      await ssoPage.goto(`${UI_BASE_URL}/auth/sign-in`);
      await ssoPage.waitForLoadState("domcontentloaded");

      // Wait for SSO button to appear (provider was just created)
      const ssoButton = ssoPage.getByRole("button", {
        name: new RegExp(providerName, "i"),
      });
      await expect(ssoButton).toBeVisible({ timeout: 10000 });

      // Click SSO button and login via Keycloak
      await clickButton({
        page: ssoPage,
        options: { name: new RegExp(providerName, "i") },
      });

      // Login via Keycloak (admin user is in archestra-admins group)
      const loginSucceeded = await loginViaKeycloak(ssoPage);
      expect(loginSucceeded).toBe(true);

      // Verify we're logged in
      await expectAuthenticated(ssoPage, 15000);

      // Verify the user has admin role by checking they can access admin-only pages
      // The Roles settings page is only accessible to admins
      await ssoPage.goto(`${UI_BASE_URL}/settings/roles`);
      await ssoPage.waitForLoadState("domcontentloaded");

      // If user has admin role, they should see the Roles page
      // If not, they would be redirected or see an error
      await expect(
        ssoPage.getByText("Roles", { exact: true }).first(),
      ).toBeVisible({ timeout: 10000 });

      // Success! The admin user was mapped to admin role via Handlebars template
      // Note: The syncSsoRole function (for subsequent logins) is covered by unit tests
    } finally {
      await ssoContext.close();
    }

    // STEP 4: Cleanup - delete the provider
    await goToPage(page, "/settings/identity-providers");
    await page.waitForLoadState("domcontentloaded");
    await openIdentityProviderDialog(page, "Generic OIDC");
    await deleteProviderViaDialog(page);
  });
});

test.describe("Identity Provider SAML E2E Flow with Keycloak", () => {
  test("should configure SAML provider, login via SSO, update, and delete", async ({
    page,
    browser,
    goToPage,
  }) => {
    test.skip(
      true,
      "Currently failing in CI (identity-providers.ee.spec.ts:968 SAML/Keycloak flow)",
    );
    test.slow();

    // Fetch IdP metadata dynamically (Keycloak regenerates certs on restart)
    const idpMetadata = await fetchKeycloakSamlMetadata();
    const idpCert = extractCertFromMetadata(idpMetadata);
    const providerName = `KeycloakSAML${Date.now()}`;

    // STEP 1: Authenticate and clean up any existing provider
    await ensureAdminAuthenticated(page);
    await deleteExistingProviderIfExists(page, "Generic SAML");

    // STEP 2: Fill in SAML provider form
    await page.getByLabel("Provider ID").fill(providerName);
    await page
      .getByLabel("Issuer", { exact: true })
      .fill(KEYCLOAK_SAML.entityId);
    await page.getByLabel("Domain").fill(SSO_DOMAIN);
    await page
      .getByLabel("SAML Issuer / Entity ID")
      .fill(KEYCLOAK_SAML.entityId);
    await page.getByLabel("SSO Entry Point URL").fill(KEYCLOAK_SAML.ssoUrl);
    await page.getByLabel("IdP Certificate").fill(idpCert);

    // IdP Metadata XML is required to avoid ERR_IDP_METADATA_MISSING_SINGLE_SIGN_ON_SERVICE error
    // The field is nested as samlConfig.idpMetadata.metadata in the schema
    await page.getByLabel("IdP Metadata XML (Recommended)").fill(idpMetadata);

    await page
      .getByLabel("Callback URL (ACS URL)")
      .fill(`http://localhost:3000/api/auth/sso/saml2/sp/acs/${providerName}`);
    // Audience should match what Keycloak sends in the SAML assertion
    await page.getByLabel("Audience (Optional)").fill("http://localhost:3000");
    // SP Entity ID is required for Better Auth to generate proper SP metadata
    // See: https://github.com/better-auth/better-auth/issues/4833
    await page.getByLabel("SP Entity ID").fill("http://localhost:3000");

    // IMPORTANT: Due to a bug in Better Auth's SSO plugin (saml.SPMetadata is not a function),
    // we must provide full SP metadata XML to bypass the broken auto-generation.
    // See: https://github.com/better-auth/better-auth/issues/4833
    // NOTE: AuthnRequestsSigned must match the IdP's WantAuthnRequestsSigned setting
    // For testing purposes, we set both to false to avoid signing complexity
    const spMetadataXml = `<?xml version="1.0" encoding="UTF-8"?>
<md:EntityDescriptor xmlns:md="urn:oasis:names:tc:SAML:2.0:metadata" entityID="http://localhost:3000">
  <md:SPSSODescriptor AuthnRequestsSigned="false" WantAssertionsSigned="true" protocolSupportEnumeration="urn:oasis:names:tc:SAML:2.0:protocol">
    <md:NameIDFormat>urn:oasis:names:tc:SAML:1.1:nameid-format:emailAddress</md:NameIDFormat>
    <md:AssertionConsumerService Binding="urn:oasis:names:tc:SAML:2.0:bindings:HTTP-POST" Location="http://localhost:3000/api/auth/sso/saml2/sp/acs/${providerName}" index="0" isDefault="true"/>
  </md:SPSSODescriptor>
</md:EntityDescriptor>`;
    await page.getByLabel("SP Metadata XML (Optional)").fill(spMetadataXml);

    // Configure attribute mapping to match Keycloak's SAML attribute names
    // These match the simple attribute names configured in helm/e2e-tests/values.yaml
    // Keycloak sends: email, firstName, lastName, name
    await page.getByLabel("Email Attribute").fill("email");
    await page.getByLabel("Display Name Attribute").fill("name");
    await page.getByLabel("First Name Attribute (Optional)").fill("firstName");
    await page.getByLabel("Last Name Attribute (Optional)").fill("lastName");

    // Submit the form
    await page.getByTestId(E2eTestId.IdentityProviderCreateButton).click();

    // Wait for dialog to close and provider to be created
    // Also wait for network to be idle to ensure the provider is fully created
    await expect(page.getByRole("dialog")).not.toBeVisible({ timeout: 10000 });
    await page.waitForLoadState("domcontentloaded");

    // Verify the provider is now shown as "Enabled"
    await page.reload();
    await page.waitForLoadState("domcontentloaded");

    // STEP 3: Verify SSO button appears on login page and test SSO login
    // NOTE: SAML account linking works because the backend automatically sets
    // `domainVerified: true` for SAML providers as a workaround for:
    // https://github.com/better-auth/better-auth/issues/6481
    const ssoContext = await browser.newContext({
      storageState: undefined,
    });
    const ssoPage = await ssoContext.newPage();

    try {
      await ssoPage.goto(`${UI_BASE_URL}/auth/sign-in`);
      await ssoPage.waitForLoadState("domcontentloaded");

      // Verify SSO button for our provider appears
      const ssoButton = ssoPage.getByRole("button", {
        name: new RegExp(providerName, "i"),
      });
      await expect(ssoButton).toBeVisible({ timeout: 10000 });

      // STEP 4: Click SSO button and login via Keycloak SAML
      await clickButton({
        page: ssoPage,
        options: { name: new RegExp(providerName, "i") },
      });

      // Login via Keycloak and wait for redirect back to Archestra
      // SAML flows can be slower due to XML processing, so we increased timeout in loginViaKeycloak
      const loginSucceeded = await loginViaKeycloak(ssoPage);
      expect(loginSucceeded).toBe(true);

      // Verify we're logged in
      await expectAuthenticated(ssoPage, 15000);

      // SSO login successful - user is now logged in
    } finally {
      await ssoContext.close();
    }

    // STEP 5: Use the original admin page context to update the provider
    // (the original page context is still logged in as admin)
    await goToPage(page, "/settings/identity-providers");
    await page.waitForLoadState("domcontentloaded");

    // Click on Generic SAML card to edit (our provider)
    await openIdentityProviderDialog(page, "Generic SAML");

    // Update the domain (use a subdomain to keep it valid for the same email domain)
    await page.getByLabel("Domain").clear();
    await page.getByLabel("Domain").fill(`updated.${SSO_DOMAIN}`);

    // Save changes
    await page.getByTestId(E2eTestId.IdentityProviderUpdateButton).click();
    await expect(page.getByRole("dialog")).not.toBeVisible({ timeout: 10000 });
    await page.waitForLoadState("domcontentloaded");

    // STEP 6: Delete the provider
    await openIdentityProviderDialog(page, "Generic SAML");
    await deleteProviderViaDialog(page);
    await page.waitForLoadState("domcontentloaded");

    // STEP 7: Verify SSO button no longer appears on login page
    // Use a fresh context to check the sign-in page
    const verifyContext = await browser.newContext({
      storageState: undefined,
    });
    const verifyPage = await verifyContext.newPage();

    try {
      await verifyPage.goto(`${UI_BASE_URL}/auth/sign-in`);
      await verifyPage.waitForLoadState("domcontentloaded");

      // SSO button for our provider should no longer be visible
      await expect(
        verifyPage.getByRole("button", { name: new RegExp(providerName, "i") }),
      ).not.toBeVisible({ timeout: 10000 });
    } finally {
      await verifyContext.close();
    }
  });
});
