kandev/apps/web/components/workspace-source-picker/executor-capabilities.ts

import type { ExecutorType } from "@/lib/types/http";

const KNOWN_EXECUTOR_TYPES = new Set<ExecutorType>([
  "local",
  "local_pc",
  "local_docker",
  "sprites",
  "ssh",
  "remote_docker",
  "remote_vps",
  "k8s",
  "worktree",
]);

type SavedRepositoryIdentity = {
  remote_url?: string;
  provider?: string;
  provider_repo_id?: string;
  provider_host?: string;
  provider_owner?: string;
  provider_name?: string;
};

export function getWorkspaceSourceCapabilities(
  executorType: ExecutorType | string | null | undefined,
): {
  canAddFolders: boolean;
  canChooseCheckoutBranch: boolean;
  requiresCloneableLocalRepository: boolean;
  /**
   * False while the primary executor binding has not resolved yet or the type
   * is unknown. Callers must not treat that the same as a known executor that
   * cannot host folders — an unresolved capability means "not yet known," not "no."
   */
  executorCapabilitiesKnown: boolean;
} {
  const usesLiveCheckout = executorType === "local" || executorType === "local_pc";
  const requiresClone =
    executorType === "local_docker" ||
    executorType === "remote_docker" ||
    executorType === "ssh" ||
    executorType === "sprites";
  return {
    canAddFolders: usesLiveCheckout || executorType === "worktree",
    canChooseCheckoutBranch: !usesLiveCheckout,
    requiresCloneableLocalRepository: requiresClone,
    executorCapabilitiesKnown: KNOWN_EXECUTOR_TYPES.has(executorType as ExecutorType),
  };
}

/** Mirrors the backend remoteRepositoryLocator eligibility check. */
export function hasCloneableSavedRepository(repository: SavedRepositoryIdentity): boolean {
  return hasSafeRemoteURL(repository.remote_url) || hasProviderRepositoryLocator(repository);
}

function hasProviderRepositoryLocator(repository: SavedRepositoryIdentity): boolean {
  const provider = repository.provider?.trim().toLowerCase();
  if (!provider || !repository.provider_owner?.trim() || !repository.provider_name?.trim()) {
    return false;
  }
  if (repository.provider_host?.trim()) return true;
  return provider === "github" || provider === "gitlab";
}

function hasSafeRemoteURL(remoteURL: string | undefined): boolean {
  const locator = remoteURL?.trim() ?? "";
  if (!locator || locator.startsWith("/") || locator.startsWith("file:")) return false;
  if (locator.startsWith("git@")) return true;
  try {
    const parsed = new URL(locator);
    return Boolean(parsed.host) && ["https:", "http:", "ssh:", "git:"].includes(parsed.protocol);
  } catch {
    return false;
  }
}

export function canChooseCheckoutBranchForSource(
  sourceType: "saved_repository" | "local_repository" | "remote_repository" | "folder",
  executorType: ExecutorType | string | null | undefined,
): boolean {
  return (
    sourceType !== "folder" && getWorkspaceSourceCapabilities(executorType).canChooseCheckoutBranch
  );
}