kandev/apps/web/components/task/changes-panel-helpers.ts

"use client";

import {
  hashDiff,
  normalizeDiffContent,
  reviewFileKey,
  splitReviewFileKey,
} from "@/components/review/types";
import type { FileInfo } from "@/lib/state/store";
import type { PRDiffFile } from "@/lib/types/github";
import { normalizeFileChangeStatus, type FileChangeStatus } from "@/lib/utils/file-change-status";
import type { PRChangedFile } from "./changes-panel-timeline";
import type { CommitDetailTarget } from "./changes-diff-target";

export type ChangedFile = {
  path: string;
  status: FileChangeStatus;
  staged: boolean;
  plus: number | undefined;
  minus: number | undefined;
  oldPath: string | undefined;
  /** Repository this file belongs to in multi-repo workspaces; empty for single-repo. */
  repositoryName?: string;
};

/**
 * Picks the group label stamped onto each PR's files so the changes panel
 * can show one section per PR for multi-branch tasks.
 *
 * Rules:
 * - Only one PR on the task -> no stamp (the section header is enough).
 * - Multi-repo, one PR per repo -> stamp the repo name (legacy behavior).
 * - Any repo has multiple PRs -> append the branch / PR number so the
 *   group label disambiguates which PR each file belongs to.
 */
export function computePRGroupStamp(args: {
  needsStamp: boolean;
  taskHasMultipleRepos: boolean;
  anyRepoMultiPR: boolean;
  repoName: string;
  branch: string;
  prNumber: number;
}): string {
  if (!args.needsStamp) return "";
  const label = args.branch || `PR #${args.prNumber}`;
  if (args.anyRepoMultiPR && args.taskHasMultipleRepos) {
    return `${args.repoName} · ${label}`;
  }
  if (args.anyRepoMultiPR) {
    return label;
  }
  return args.repoName;
}

export function mapToChangedFiles(files: FileInfo[]): ChangedFile[] {
  return files.map((file) => ({
    path: file.path,
    status: file.status,
    staged: file.staged,
    plus: file.additions,
    minus: file.deletions,
    oldPath: file.old_path,
    repositoryName: file.repository_name,
  }));
}

type CumulativeDiffFiles = Record<
  string,
  {
    path?: string;
    repository_name?: string;
    diff?: string;
    status?: string;
    additions?: number;
    deletions?: number;
  }
>;

function cumulativeFileIdentity(
  mapKey: string,
  file: CumulativeDiffFiles[string],
  useRepositoryKeys: boolean,
) {
  const path = file.path ?? splitReviewFileKey(mapKey).path;
  const repositoryName = useRepositoryKeys ? file.repository_name : undefined;
  return {
    key: reviewFileKey({ path, repository_name: repositoryName }),
    path,
    repositoryName,
  };
}

export type ReviewProgressPRFile = PRDiffFile & { repository_name?: string };

export type ReviewProgressPRSource = {
  repositoryName: string;
  files: PRDiffFile[];
};

export function selectPRFilesForReviewProgress(
  sourcesByPRID: ReadonlyMap<string, ReviewProgressPRSource>,
  primaryPRID: string | undefined,
  useRepositoryKeys: boolean,
): ReviewProgressPRFile[] {
  if (!primaryPRID) return [];
  const source = sourcesByPRID.get(primaryPRID);
  if (!source) return [];
  return source.files.map((file) => ({
    ...file,
    repository_name: useRepositoryKeys ? source.repositoryName : undefined,
  }));
}

function addReviewSource(
  winningDiffs: Map<string, string | undefined>,
  paths: Set<string>,
  source: {
    key: string;
    path: string;
    repositoryName?: string;
    diff?: string;
  },
): void {
  const collidesWithHigherPriority = source.repositoryName
    ? winningDiffs.has(source.path)
    : paths.has(source.path);
  if (winningDiffs.has(source.key) || collidesWithHigherPriority) return;
  winningDiffs.set(source.key, source.diff);
  paths.add(source.path);
}

function buildReviewProgressIndex(
  uncommittedFiles: FileInfo[],
  cumulativeDiffFiles: CumulativeDiffFiles | undefined,
  prFiles?: ReviewProgressPRFile[],
  useRepositoryKeys = true,
): Map<string, string | undefined> {
  const winningDiffs = new Map<string, string | undefined>();
  const paths = new Set<string>();
  for (const file of uncommittedFiles) {
    const repositoryName = useRepositoryKeys ? file.repository_name : undefined;
    const key = reviewFileKey({ path: file.path, repository_name: repositoryName });
    addReviewSource(winningDiffs, paths, {
      key,
      path: file.path,
      repositoryName,
      diff: file.diff,
    });
  }
  if (cumulativeDiffFiles) {
    for (const [mapKey, file] of Object.entries(cumulativeDiffFiles)) {
      const identity = cumulativeFileIdentity(mapKey, file, useRepositoryKeys);
      addReviewSource(winningDiffs, paths, {
        ...identity,
        diff: file.diff,
      });
    }
  }
  if (prFiles) {
    for (const file of prFiles) {
      const repositoryName = useRepositoryKeys ? file.repository_name : undefined;
      const key = reviewFileKey({
        path: file.filename,
        repository_name: repositoryName,
      });
      addReviewSource(winningDiffs, paths, {
        key,
        path: file.filename,
        repositoryName,
        diff: file.patch,
      });
    }
  }
  return winningDiffs;
}

export function computeReviewProgress(
  uncommittedFiles: FileInfo[],
  cumulativeDiff: { files?: CumulativeDiffFiles } | null,
  reviews: Map<string, { reviewed: boolean; diffHash?: string }>,
  prFiles?: ReviewProgressPRFile[],
  useRepositoryKeys = true,
) {
  const cumulativeDiffFiles = cumulativeDiff?.files;
  const winningDiffs = buildReviewProgressIndex(
    uncommittedFiles,
    cumulativeDiffFiles,
    prFiles,
    useRepositoryKeys,
  );
  let reviewed = 0;
  for (const [key, diff] of winningDiffs) {
    const state = reviews.get(key);
    if (!state?.reviewed) continue;
    const diffContent = normalizeDiffContent(diff ?? "");
    if (diffContent && state.diffHash && state.diffHash !== hashDiff(diffContent)) continue;
    reviewed++;
  }
  return { reviewedCount: reviewed, totalFileCount: winningDiffs.size };
}

export function computeStagedStats(stagedFiles: FileInfo[]) {
  const adds = stagedFiles.reduce((sum, f) => sum + (f.additions || 0), 0);
  const dels = stagedFiles.reduce((sum, f) => sum + (f.deletions || 0), 0);
  return { stagedFileCount: stagedFiles.length, stagedAdditions: adds, stagedDeletions: dels };
}

export function mapPRFilesToChangedFiles(
  files: PRDiffFile[],
  repositoryName?: string,
  prKey?: string,
): PRChangedFile[] {
  return files.map((file) => ({
    path: file.filename,
    prKey,
    status: normalizeFileChangeStatus(file.status),
    plus: file.additions,
    minus: file.deletions,
    oldPath: file.old_path,
    repository_name: repositoryName ?? "",
  }));
}

export function filterUnpushedCommits<T extends { commit_sha: string }>(
  localCommits: T[],
  prCommits: { sha: string }[],
): T[] {
  if (prCommits.length === 0) return localCommits;
  return localCommits.filter(
    (c) =>
      !prCommits.some((pr) => pr.sha.startsWith(c.commit_sha) || c.commit_sha.startsWith(pr.sha)),
  );
}

/** Which timeline section, if any, renders first (topmost) in the Changes panel. */
type FirstVisibleSection = "pr" | "unstaged" | "staged" | "commits" | null;

/** PR Changes auto-expands only when the diff is small enough to scan inline. */
export const PR_CHANGES_AUTO_EXPAND_MAX_FILES = 5;

/**
 * Computes the first (topmost) visible section so the panel can auto-expand it,
 * mirroring the render precedence in `ChangesPanelTimeline`:
 *   PR (review mode, no local changes) → Unstaged → Staged → Commits.
 *
 * The "review mode" PR section only sits at the top when there are no local
 * working-tree changes; with local changes the PR section moves below Staged and
 * is therefore never first. Large PR diffs (>5 files) skip auto-expanding PR
 * Changes and expand Commits instead. Returns null when nothing is shown.
 */
export function firstVisibleSection(flags: {
  hasPRFiles: boolean;
  hasUnstaged: boolean;
  hasStaged: boolean;
  showCommitsList: boolean;
  prFileCount?: number;
}): FirstVisibleSection {
  const { hasPRFiles, hasUnstaged, hasStaged, showCommitsList, prFileCount = 0 } = flags;
  if (hasPRFiles && !hasUnstaged && !hasStaged) {
    if (prFileCount <= PR_CHANGES_AUTO_EXPAND_MAX_FILES) return "pr";
    if (showCommitsList) return "commits";
    return null;
  }
  if (hasUnstaged) return "unstaged";
  if (hasStaged) return "staged";
  if (showCommitsList) return "commits";
  return null;
}

export type MergedCommit = {
  commit_sha: string;
  commit_message: string;
  insertions: number;
  deletions: number;
  pushed: boolean;
  statsAvailable: boolean;
  detailTarget: CommitDetailTarget;
  /** Multi-repo: name of the repo this commit was made in. Empty for single-repo. */
  repository_name?: string;
  committed_at?: string;
};

export type PRCommitForMerge = {
  sha: string;
  message: string;
  additions: number;
  deletions: number;
  author_date?: string;
  stats_available?: boolean;
  workspace_id: string;
  owner: string;
  repo: string;
  repository_name?: string;
};

function localCommitTarget(commit: {
  commit_sha: string;
  repository_name?: string;
}): CommitDetailTarget {
  return {
    source: "local",
    sha: commit.commit_sha,
    ...(commit.repository_name ? { repo: commit.repository_name } : {}),
  };
}

function githubCommitTarget(commit: PRCommitForMerge): CommitDetailTarget | null {
  if (!commit.workspace_id.trim() || !commit.owner.trim() || !commit.repo.trim()) return null;
  return {
    source: "github",
    sha: commit.sha,
    workspaceId: commit.workspace_id,
    owner: commit.owner,
    repo: commit.repo,
    ...(commit.repository_name ? { repositoryName: commit.repository_name } : {}),
  };
}

function prCommitKey(pr: PRCommitForMerge): string {
  return `${pr.repository_name ?? ""}\0${pr.sha}`;
}

function appendPROnlyCommits(
  prCommits: PRCommitForMerge[],
  matchedPRKeys: Set<string>,
  pushed: MergedCommit[],
): void {
  for (const pr of prCommits) {
    if (matchedPRKeys.has(prCommitKey(pr))) continue;
    const detailTarget = githubCommitTarget(pr);
    if (!detailTarget) continue;
    pushed.push({
      commit_sha: pr.sha,
      commit_message: pr.message,
      insertions: pr.additions,
      deletions: pr.deletions,
      pushed: true,
      statsAvailable: pr.stats_available === true,
      detailTarget,
      ...(pr.repository_name ? { repository_name: pr.repository_name } : {}),
      committed_at: pr.author_date,
    });
  }
}

/**
 * Merge local session commits and PR commits into a single list. A commit is
 * pushed when EITHER source confirms it: the backend's `pushed` field
 * (computed from the upstream tracking ref) or a SHA match in `prCommits` (PR
 * existence implies the commit is on the remote). The git signal fixes the
 * original bug — pushed commits showing as unpushed when no PR exists — while
 * keeping the PR signal so a commit visible only via a PR (e.g. authored by
 * another contributor, or a rebased SHA the local repo doesn't track yet)
 * still renders as pushed. PR commits not represented locally are appended
 * as pushed. Order: unpushed first, then pushed.
 */
export function mergeCommits(
  localCommits: {
    commit_sha: string;
    commit_message: string;
    insertions: number;
    deletions: number;
    /** Multi-repo: name of the repo this commit was made in. */
    repository_name?: string;
    pushed?: boolean;
    committed_at?: string;
  }[],
  prCommits: PRCommitForMerge[],
): MergedCommit[] {
  const shaMatches = (a: string, b: string) => a.startsWith(b) || b.startsWith(a);
  const repositoryMatches = (a?: string, b?: string) => (a ?? "") === (b ?? "");
  const commitsMatch = (
    local: { commit_sha: string; repository_name?: string },
    pr: PRCommitForMerge,
  ) =>
    shaMatches(pr.sha, local.commit_sha) &&
    repositoryMatches(local.repository_name, pr.repository_name);
  const unpushed: MergedCommit[] = [];
  const pushed: MergedCommit[] = [];
  const matchedPRKeys = new Set<string>();
  for (const c of localCommits) {
    const matchesPR = prCommits.some((pr) => commitsMatch(c, pr));
    const isPushed = c.pushed === true || matchesPR;
    if (isPushed) {
      pushed.push({
        ...c,
        pushed: true,
        statsAvailable: true,
        detailTarget: localCommitTarget(c),
      });
    } else {
      unpushed.push({
        ...c,
        pushed: false,
        statsAvailable: true,
        detailTarget: localCommitTarget(c),
      });
    }
    if (matchesPR) {
      for (const pr of prCommits) {
        if (commitsMatch(c, pr)) {
          matchedPRKeys.add(prCommitKey(pr));
        }
      }
    }
  }
  appendPROnlyCommits(prCommits, matchedPRKeys, pushed);
  return [...unpushed, ...pushed];
}

export function getBaseBranchDisplay(baseBranch: string | undefined): string {
  return baseBranch ? baseBranch.replace(/^origin\//, "") : "main";
}

export function isMultiRepoCommits(commits: { repository_name?: string }[]): boolean {
  return commits.some((c) => !!c.repository_name);
}

type TaskPRUrlInput = {
  pr_url?: string;
  repository_id?: string;
};

/** Workspace repository id → display name for multi-repo PR/commit grouping. */
export function buildRepoNameById(
  reposByWorkspace: Record<string, Array<{ id: string; name: string }>>,
): Record<string, string> {
  const out: Record<string, string> = {};
  for (const list of Object.values(reposByWorkspace)) {
    for (const r of list) {
      out[r.id] = r.name;
    }
  }
  return out;
}

/** Merge synced TaskPR URLs and client-only pending PR URLs by repo display name. */
export function buildPrByRepoMap(
  taskPRs: TaskPRUrlInput[] | undefined,
  repoNameById: Record<string, string>,
  pendingByRepo: Record<string, string> | undefined,
): Record<string, string | undefined> {
  const map: Record<string, string | undefined> = {};
  if (taskPRs) {
    for (const pr of taskPRs) {
      if (!pr.pr_url) continue;
      const repoKey = pr.repository_id ? (repoNameById[pr.repository_id] ?? "") : "";
      map[repoKey] = map[repoKey] ?? pr.pr_url;
    }
  }
  if (pendingByRepo) {
    for (const [repoKey, url] of Object.entries(pendingByRepo)) {
      if (url) map[repoKey] = map[repoKey] ?? url;
    }
  }
  const legacySingleRepo = taskPRs?.find((p) => p.pr_url && !p.repository_id);
  if (!map[""] && legacySingleRepo?.pr_url) {
    map[""] = legacySingleRepo.pr_url;
  }
  return map;
}