feat(f6): DiffViewController — baseline lifecycle + content provider (SLICE-2)

F6 §6.2/§6.4, INV-18/INV-19. ensure/advance/pin capture buffer text, persist
via BaselineStore, refresh the open diff via onDidChange. Storage-unavailable
degrades to in-memory + one warning. toggle UX lands in SLICE-3.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Ben Stull
2026-06-11 07:13:30 -07:00
parent 281eb09883
commit 4f2fc6fa96
+154
View File
@@ -0,0 +1,154 @@
/**
* DiffViewController — F6 diff-view toggle (spec §6.2/§6.4). Owns the baseline
* lifecycle (initialize at first track / advance at every machine landing / pin
* on demand), serves the baseline as a readonly `cowriting-baseline:` virtual
* document, and toggles a native vscode.diff (baseline left, the LIVE document
* right). A pure view: never mutates the document, sidecar, or attribution
* state (INV-19). Baselines persist via the vscode-free BaselineStore; if
* storage is unavailable the controller degrades to in-memory baselines + one
* warning (reload survival is lost; the toggle still works) — §6.5 PUC-5.
*/
import * as path from "node:path";
import * as vscode from "vscode";
import { BaselineStore, type Baseline, type BaselineReason } from "./baselineStore";
export const BASELINE_SCHEME = "cowriting-baseline";
export class DiffViewController implements vscode.Disposable {
private readonly disposables: vscode.Disposable[] = [];
/** Source of truth for the content provider; mirrors what the store persists. */
private readonly baselines = new Map<string, Baseline>();
private readonly onDidChangeEmitter = new vscode.EventEmitter<vscode.Uri>();
private storageWarned = false;
constructor(
private readonly store: BaselineStore | null,
private readonly rootDir: string,
) {
const provider: vscode.TextDocumentContentProvider = {
onDidChange: this.onDidChangeEmitter.event,
provideTextDocumentContent: (uri) => {
const docPath = this.docPathFromBaselineUri(uri);
return this.baselines.get(docPath)?.text ?? "";
},
};
this.disposables.push(
this.onDidChangeEmitter,
vscode.workspace.registerTextDocumentContentProvider(BASELINE_SCHEME, provider),
vscode.commands.registerCommand("cowriting.toggleDiffView", () =>
this.toggle(vscode.window.activeTextEditor),
),
vscode.commands.registerCommand("cowriting.pinDiffBaseline", () =>
this.pinCommand(vscode.window.activeTextEditor),
),
);
}
// ---- tracking / uri helpers --------------------------------------------------------
private isTracked(document: vscode.TextDocument): boolean {
return document.uri.scheme === "file" && document.uri.fsPath.startsWith(this.rootDir);
}
private docPathOf(uri: vscode.Uri): string {
return vscode.workspace.asRelativePath(uri, false);
}
/** The readonly virtual-doc URI whose content the provider serves for this doc. */
private baselineUri(docPath: string): vscode.Uri {
return vscode.Uri.from({ scheme: BASELINE_SCHEME, path: "/" + docPath });
}
private docPathFromBaselineUri(uri: vscode.Uri): string {
return uri.path.replace(/^\//, "");
}
// ---- baseline lifecycle (§6.4) -----------------------------------------------------
/** First sight of a tracked doc: load the stored baseline, else capture `opened`. */
ensureBaseline(document: vscode.TextDocument): void {
if (!this.isTracked(document)) return;
const docPath = this.docPathOf(document.uri);
if (this.baselines.has(docPath)) return;
if (this.store) {
try {
const stored = this.store.load(docPath);
if (stored) {
this.baselines.set(docPath, stored);
return;
}
} catch {
this.warnStorageOnce();
}
}
this.capture(document, "opened");
}
/** Machine landing (INV-18): re-capture so landed text never shows as a change. */
advance(document: vscode.TextDocument): void {
if (!this.isTracked(document)) return;
this.capture(document, "machine-landing");
}
/** Human pin: baseline := now; the open diff visibly empties (left = right). */
pin(document: vscode.TextDocument): void {
if (!this.isTracked(document)) return;
this.capture(document, "pinned");
}
/** Capture buffer text at this epoch, persist, and refresh any open diff's left side. */
private capture(document: vscode.TextDocument, reason: BaselineReason): void {
const docPath = this.docPathOf(document.uri);
const baseline: Baseline = {
docPath,
text: document.getText(),
capturedAt: new Date().toISOString(),
reason,
};
this.baselines.set(docPath, baseline);
if (this.store) {
try {
this.store.save(docPath, baseline);
} catch {
this.warnStorageOnce();
}
}
// An open diff re-requests the left side when its baseline URI changes.
this.onDidChangeEmitter.fire(this.baselineUri(docPath));
}
private warnStorageOnce(): void {
if (this.storageWarned) return;
this.storageWarned = true;
void vscode.window.showWarningMessage(
"Cowriting: diff-view storage is unavailable — baselines are kept in memory only and won't survive a reload.",
);
}
// ---- commands (toggle implemented in SLICE-3, Task 4) ------------------------------
private pinCommand(editor: vscode.TextEditor | undefined): void {
if (!editor || !this.isTracked(editor.document)) {
void vscode.window.showWarningMessage("Cowriting: open a tracked workspace document to pin its diff baseline.");
return;
}
this.pin(editor.document);
}
// ---- test-facing surface (§6.4) ----------------------------------------------------
getBaseline(docPath: string): { text: string; reason: BaselineReason; capturedAt: string } | undefined {
const b = this.baselines.get(docPath);
return b ? { text: b.text, reason: b.reason, capturedAt: b.capturedAt } : undefined;
}
/** Absolute on-disk path of this doc's persisted baseline, or undefined if in-memory. */
baselineFilePath(docPath: string): string | undefined {
return this.store?.baselinePath(docPath);
}
// Replaced by the real implementation in Task 4 (toggle UX, SLICE-3).
private toggle(_editor: vscode.TextEditor | undefined): void {
/* stub */
}
dispose(): void {
for (const d of this.disposables) d.dispose();
}
}