worktrees

1import type { EngineInterface, Register, Timer } from 'claude-code';
2
3import { countChanges, parseWorktrees, shorten, type Worktree } from './parse';
4
5const LIMIT = 10;
6const REFRESH_MS = 60_000;
7
8let trees: Worktree[] = [];
9let timer: Timer | undefined;
10let current = '';
11
12function git($: EngineInterface, cwd: string, args: readonly string[]) {
13  return $.process.run(['git', '--no-optional-locks', '-C', cwd, ...args], { timeoutMs: 5_000 });
14}
15
16async function refresh($: EngineInterface): Promise<void> {
17  const cwd = await $.session.cwd();
18  const list = await git($, cwd, ['worktree', 'list', '--porcelain']).catch(() => null);
19  if (list === null || list.exitCode !== 0) {
20    trees = [];
21    $.ui.invalidate('ui.render');
22    return;
23  }
24  const top = await git($, cwd, ['rev-parse', '--show-toplevel']);
25  current = top.stdout.trim();
26  trees = await Promise.all(
27    parseWorktrees(list.stdout).map(async tree => {
28      const status = await git($, tree.path, ['status', '--porcelain']).catch(() => null);
29      return { ...tree, changes: status === null ? 0 : countChanges(status.stdout) };
30    }),
31  );
32  $.ui.invalidate('ui.render');
33}
34
35export const register: Register = on => {
36  on('session.start', async ($, e, next) => {
37    const started = await next(e);
38    await refresh($).catch(() => undefined);
39    timer?.cancel();
40    timer = $.clock.every(REFRESH_MS, () => {
41      refresh($).catch(() => undefined);
42    });
43    return started;
44  });
45
46  on('turn.complete', async ($, e, next) => {
47    const done = await next(e);
48    // Started on a timer so the turn is not held up by one git call per worktree.
49    if (e.agentId === undefined) $.clock.after(1, () => refresh($).catch(() => undefined));
50    return done;
51  });
52
53  on('ui.render', { component: 'AbovePrompt' }, async ($, e, next) => {
54    const below = await next(e);
55    if (e.props.hasSurvey) {
56      return below;
57    }
58    // One worktree is just the repo: nothing to show.
59    if (trees.length < 2) {
60      return below;
61    }
62
63    const { Box, Text } = $.ui.resolve(e);
64    const full = trees.length >= LIMIT;
65    // Narrow: only the tree this session is in, and a count of the rest.
66    const narrow = (e.viewport?.columns ?? Number.POSITIVE_INFINITY) < 80;
67    const shown = narrow ? trees.filter(tree => tree.path === current) : trees;
68    const others = trees.length - shown.length;
69    const dirtyOthers = trees.filter(tree => tree.path !== current && tree.changes > 0).length;
70    return (
71      <Box flexDirection="column">
72        <Box gap={1} flexWrap="wrap">
73          <Text color={full ? '#ef4444' : '#a78bfa'} bold>
74            ⑂ {trees.length}/{LIMIT}
75          </Text>
76          {shown.map(tree => (
77            <Text>
78              <Text color={tree.path === current ? '#38bdf8' : undefined} bold={tree.path === current}>
79                {shorten(tree.branch, narrow ? 24 : 32)}
80              </Text>
81              <Text color={tree.changes > 0 ? '#f59e0b' : '#22c55e'}>{tree.changes > 0 ? ` ●${tree.changes}` : ' ✓'}</Text>
82            </Text>
83          ))}
84          {others > 0 ? <Text dimColor>{`+${others}${dirtyOthers > 0 ? ` (${dirtyOthers} with changes)` : ''}`}</Text> : null}
85        </Box>
86        {below}
87      </Box>
88    );
89  });
90};
91

1export type Worktree = { path: string; branch: string; changes: number };
2
3/** Parses `git worktree list --porcelain`. Detached heads show their short commit. */
4export function parseWorktrees(porcelain: string): Omit<Worktree, 'changes'>[] {
5  const trees: Omit<Worktree, 'changes'>[] = [];
6  for (const block of porcelain.split('\n\n')) {
7    const lines = block.split('\n');
8    const path = lines.find(line => line.startsWith('worktree '))?.slice('worktree '.length);
9    if (path === undefined || lines.includes('bare')) {
10      continue;
11    }
12    const branch = lines.find(line => line.startsWith('branch '))?.slice('branch refs/heads/'.length);
13    const head = lines.find(line => line.startsWith('HEAD '))?.slice('HEAD '.length, 'HEAD '.length + 7);
14    trees.push({ path, branch: branch ?? `(${head ?? 'detached'})` });
15  }
16  return trees;
17}
18
19/** Lines of `git status --porcelain`: one per changed or untracked file. */
20export const countChanges = (porcelain: string): number => porcelain.split('\n').filter(line => line.trim() !== '').length;
21
22/** Shortens a long name in the middle so both ends stay readable: feature/rob…ch-319677 */
23export function shorten(name: string, max: number): string {
24  if (name.length <= max) return name;
25  const keep = max - 1;
26  const head = Math.ceil(keep / 2);
27  return `${name.slice(0, head)}…${name.slice(name.length - (keep - head))}`;
28}
29