import type { WorkspaceDescriptor } from "@/stores/session-store";
import { buildHostRootRoute, buildHostWorkspaceRoute } from "@/utils/host-routes";

function trimNonEmpty(value: string | null | undefined): string | null {
  if (typeof value !== "string") {
    return null;
  }

  const trimmed = value.trim();
  return trimmed.length > 0 ? trimmed : null;
}

export function resolveWorkspaceArchiveRedirectWorkspaceId(input: {
  archivedWorkspaceId: string;
  workspaces: Iterable<WorkspaceDescriptor>;
}): string | null {
  const archivedWorkspaceId = trimNonEmpty(input.archivedWorkspaceId);
  if (!archivedWorkspaceId) {
    return null;
  }

  const workspaces = Array.from(input.workspaces);
  const archivedWorkspace =
    workspaces.find((workspace) => workspace.id === archivedWorkspaceId) ?? null;
  if (!archivedWorkspace) {
    return null;
  }

  const sameProjectWorkspaces = workspaces.filter(
    (workspace) => workspace.projectId === archivedWorkspace.projectId,
  );
  const rootCheckoutWorkspace =
    sameProjectWorkspaces.find(
      (workspace) =>
        workspace.workspaceKind === "local_checkout" && workspace.id !== archivedWorkspace.id,
    ) ?? null;
  if (rootCheckoutWorkspace) {
    return rootCheckoutWorkspace.id;
  }

  const fallbackProjectRootPath = trimNonEmpty(archivedWorkspace.projectRootPath);
  if (fallbackProjectRootPath && fallbackProjectRootPath !== archivedWorkspace.id) {
    return fallbackProjectRootPath;
  }

  const siblingWorkspace =
    sameProjectWorkspaces.find((workspace) => workspace.id !== archivedWorkspace.id) ?? null;
  return siblingWorkspace?.id ?? null;
}

export function buildWorkspaceArchiveRedirectRoute(input: {
  serverId: string;
  archivedWorkspaceId: string;
  workspaces: Iterable<WorkspaceDescriptor>;
}): string {
  const redirectWorkspaceId = resolveWorkspaceArchiveRedirectWorkspaceId({
    archivedWorkspaceId: input.archivedWorkspaceId,
    workspaces: input.workspaces,
  });

  if (!redirectWorkspaceId) {
    return buildHostRootRoute(input.serverId);
  }

  return buildHostWorkspaceRoute(input.serverId, redirectWorkspaceId);
}