Skip to content

Commit fb1e991

Browse files
poteatpoteat
committed
diff: the pane notices a finished merge by itself, and stays quiet on an unusual branch name
1 parent 732e167 commit fb1e991

4 files changed

Lines changed: 232 additions & 47 deletions

File tree

‎mods/diff/hooks/git/probes/head-key-of/head-key-of.ts‎

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@ import { stampProbeOf } from '../stamp-probe-of'
99
* A string that changes when HEAD moves: HEAD's text and the timestamps
1010
* of HEAD, its ref file and packed-refs, each a listed real file.
1111
*
12-
* A HEAD that is not a plain file, or a `ref:` outside the safe refname
13-
* grammar, is read through `git rev-parse` instead and nothing is stat'ed;
14-
* every directory above a stat'ed file was listed as a real directory.
12+
* A `ref:` outside the safe refname grammar is dated by HEAD's own log, the
13+
* listed real file `logs/HEAD`; without one, or where HEAD is not a plain
14+
* file, `git rev-parse` answers instead.
1515
*
1616
* @param deps git, listings, timestamps, a file read
1717
* @param repository the repository's directories
@@ -54,13 +54,16 @@ export async function headKeyOf(
5454
const isSymbolic = ref !== ''
5555

5656
const isUnsafeRef = isSymbolic && !isSafeRefName(ref)
57+
const logged = isUnsafeRef ? await stampIfFile(own, 'logs/HEAD') : null
58+
const isUndated = isUnsafeRef && typeof logged !== 'number'
59+
const isRefDated = isSymbolic && !isUnsafeRef
5760

58-
return isUnsafeRef
61+
return isUndated
5962
? viaGit()
6063
: [
6164
head,
6265
await stampIfFile(own, 'HEAD'),
63-
isSymbolic ? await stampIfFile(common, ref) : null,
66+
isRefDated ? await stampIfFile(common, ref) : logged,
6467
await stampIfFile(common, 'packed-refs'),
6568
].join('|')
6669
}

‎mods/diff/hooks/register.ts‎

Lines changed: 45 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -33,12 +33,9 @@ import Views from './views'
3333
* Registers the diff pane: `/diff` once the built-in stands down, the
3434
* pane's drawing and refresh, its opening on Claude's first edit, the ask.
3535
*
36-
* Git runs when the built-in's would: `session.start` binds the host and
37-
* registers `/diff`, and off its dispatch reads the transcript, so a resumed
38-
* session whose turns already edited opens as its first edit would; `/diff`
39-
* or the main loop's first checkpointed edit with room pins the backend,
40-
* until `/clear`, which reads afresh under a pane it leaves open; a docked
41-
* pane fetches, then opens.
36+
* Git runs when the built-in's would: none at the start; `/diff` or the main
37+
* loop's first checkpointed edit with room pins the backend, until `/clear`,
38+
* which reads afresh under a pane it leaves open.
4239
*
4340
* @param on the engine's registrar
4441
*/
@@ -150,6 +147,8 @@ export function register(on: On) {
150147
return asked.isAnswered || backend !== null
151148
}
152149

150+
startPoll(engine, probed)
151+
153152
const stored = PaneState.baseModeOf(
154153
await engine
155154
.storeGet(Names.baseStoreKeyOf(probed.repository.toplevel))
@@ -282,6 +281,25 @@ export function register(on: On) {
282281
).includes(null)
283282
}
284283

284+
function keepBaseline() {
285+
const pinned = backend
286+
287+
if (!pinned || polled.headKey !== '') {
288+
return
289+
}
290+
291+
void pinned
292+
.headKeyOf()
293+
.catch(() => '')
294+
.then(key => {
295+
const isFirst = backend === pinned && polled.headKey === ''
296+
297+
if (isFirst) {
298+
polled.headKey = key
299+
}
300+
})
301+
}
302+
285303
function startPoll(engine: Host, pinned: Backend.Backend) {
286304
const readHeadKey = () => pinned.headKeyOf().catch(() => '')
287305

@@ -353,10 +371,6 @@ export function register(on: On) {
353371
case 'data':
354372
generation += 1
355373

356-
if (pinned) {
357-
startPoll(engine, pinned)
358-
}
359-
360374
break
361375
}
362376

@@ -400,6 +414,18 @@ export function register(on: On) {
400414
)
401415
}
402416

417+
async function markShown(
418+
engine: Host,
419+
trigger: (typeof Record.SHOWN_TRIGGERS)[number],
420+
): Promise<void> {
421+
const sessionId = await engine.sessionId().catch(() => null)
422+
423+
if (sessionId !== null && sessionId !== shownSessionId) {
424+
shownSessionId = sessionId
425+
Record.recorderOf(engine).shown(trigger, Record.widthBucketOf(columns))
426+
}
427+
}
428+
403429
async function openPane(
404430
engine: Host,
405431
trigger: (typeof Record.SHOWN_TRIGGERS)[number],
@@ -436,13 +462,9 @@ export function register(on: On) {
436462
}
437463

438464
isPaneOpen = true
465+
keepBaseline()
439466

440-
const sessionId = await engine.sessionId().catch(() => null)
441-
442-
if (sessionId !== null && sessionId !== shownSessionId) {
443-
shownSessionId = sessionId
444-
Record.recorderOf(engine).shown(trigger, Record.widthBucketOf(columns))
445-
}
467+
await markShown(engine, trigger)
446468

447469
const isStale = isDialog || landed !== landedBefore
448470

@@ -506,9 +528,10 @@ export function register(on: On) {
506528
}
507529

508530
async function openOnRestore(engine: Host): Promise<void> {
509-
const messages = await engine.messages().catch((): SessionMessage[] => [])
510-
511-
hasRestoredEdits = Turns.turnDiffsOf(messages).length > 0
531+
hasRestoredEdits =
532+
Turns.turnDiffsOf(
533+
await engine.messages().catch((): SessionMessage[] => []),
534+
).length > 0
512535

513536
if (hasRestoredEdits) {
514537
await openOnFirstEdit(engine)
@@ -687,8 +710,7 @@ export function register(on: On) {
687710

688711
on('ui.render', { component: 'PromptHint' }, ($, e, next) => {
689712
if (isOnPaneSurface(e)) {
690-
const viewport: { columns?: number; isFullscreen?: boolean } | undefined =
691-
e.viewport
713+
const { viewport } = e
692714

693715
const isFirstMeasure = columns === null && viewport?.columns !== undefined
694716

@@ -903,7 +925,9 @@ export function register(on: On) {
903925
(isResume ? sessionStartMs : await host.now())
904926

905927
if (isKeptOpen) {
928+
await markShown(host, 'manual')
906929
await pinBackend(host)
930+
keepBaseline()
907931
void refresh(host)
908932
}
909933

‎mods/diff/tests/git/probes/head-key-of.test.ts‎

Lines changed: 54 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ describe('head-key-of', () => {
1616
function recorded(
1717
listings: Readonly<Record<string, ReadonlyMap<string, Git.EntryKind>>>,
1818
head: string,
19+
stampOf: (path: string) => number = () => Fixtures.STAMP,
1920
): Recorded {
2021
const statted: string[] = []
2122
const read: string[] = []
@@ -30,7 +31,7 @@ describe('head-key-of', () => {
3031
mtimeOf: path => {
3132
statted.push(path)
3233

33-
return Promise.resolve(Fixtures.STAMP)
34+
return Promise.resolve(stampOf(path))
3435
},
3536
readFile: path => {
3637
read.push(path)
@@ -57,6 +58,27 @@ describe('head-key-of', () => {
5758
'/r/.git/refs/heads': heads,
5859
}
5960

61+
const logged = {
62+
'/r/.git': new Map<string, Git.EntryKind>([
63+
...plainGitDir,
64+
['logs', 'dir'],
65+
]),
66+
'/r/.git/logs': new Map<string, Git.EntryKind>([['HEAD', 'file']]),
67+
}
68+
69+
async function idleOn(branch: string) {
70+
const probe = recorded(logged, `ref: refs/heads/${branch}\n`)
71+
72+
return {
73+
keys: new Set([
74+
await Git.headKeyOf(probe.deps, Fixtures.CHECKOUT),
75+
await Git.headKeyOf(probe.deps, Fixtures.CHECKOUT),
76+
await Git.headKeyOf(probe.deps, Fixtures.CHECKOUT),
77+
]).size,
78+
ran: probe.ran,
79+
}
80+
}
81+
6082
test('a plain checkout reads HEAD, its ref file, packed-refs', async () => {
6183
const probe = recorded(healthy, 'ref: refs/heads/main\n')
6284
const key = await Git.headKeyOf(probe.deps, Fixtures.CHECKOUT)
@@ -79,6 +101,37 @@ describe('head-key-of', () => {
79101
expect(key).toBe(`rev:${Fixtures.HEAD_SHA}`)
80102
})
81103

104+
test('a branch named with a # starts no git while idle', async () => {
105+
expect(await idleOn('fix/#12')).toEqual({ ran: [], keys: 1 })
106+
})
107+
108+
test('a branch named outside ASCII starts no git while idle', async () => {
109+
expect(await idleOn('höher')).toEqual({ ran: [], keys: 1 })
110+
})
111+
112+
test('on such a branch a moved HEAD is seen at the next look', async () => {
113+
let moves = 0
114+
115+
const probe = recorded(logged, 'ref: refs/heads/fix/#12\n', path =>
116+
path.endsWith('/logs/HEAD') ? Fixtures.STAMP + moves : Fixtures.STAMP,
117+
)
118+
119+
const before = await Git.headKeyOf(probe.deps, Fixtures.CHECKOUT)
120+
121+
moves += 1
122+
123+
expect(await Git.headKeyOf(probe.deps, Fixtures.CHECKOUT)).not.toBe(before)
124+
expect(probe.ran, 'dated by the log of HEAD, no git started').toEqual([])
125+
})
126+
127+
test('with no log of HEAD such a branch is read through git', async () => {
128+
const probe = recorded(healthy, 'ref: refs/heads/fix/#12\n')
129+
const key = await Git.headKeyOf(probe.deps, Fixtures.CHECKOUT)
130+
131+
expect(probe.statted).toEqual([])
132+
expect(key).toBe(`rev:${Fixtures.HEAD_SHA}`)
133+
})
134+
82135
test('a HEAD that is a symbolic link is never read or stat-ed', async () => {
83136
const probe = recorded(
84137
{

0 commit comments

Comments
 (0)