55 lines
1.8 KiB
TypeScript
55 lines
1.8 KiB
TypeScript
import * as vscode from "vscode";
|
|
import type { SfChangeTracker, Checkpoint } from "./change-tracker.js";
|
|
|
|
/**
|
|
* TreeDataProvider that shows agent checkpoints (one per agent turn).
|
|
* Each checkpoint can be restored to revert all file changes since that point.
|
|
*/
|
|
export class SfCheckpointProvider implements vscode.TreeDataProvider<Checkpoint>, vscode.Disposable {
|
|
public static readonly viewId = "sf-checkpoints";
|
|
|
|
private readonly _onDidChangeTreeData = new vscode.EventEmitter<void>();
|
|
readonly onDidChangeTreeData = this._onDidChangeTreeData.event;
|
|
|
|
private disposables: vscode.Disposable[] = [];
|
|
|
|
constructor(private readonly tracker: SfChangeTracker) {
|
|
this.disposables.push(
|
|
this._onDidChangeTreeData,
|
|
tracker.onCheckpointChange(() => this._onDidChangeTreeData.fire()),
|
|
);
|
|
}
|
|
|
|
getTreeItem(checkpoint: Checkpoint): vscode.TreeItem {
|
|
const fileCount = checkpoint.snapshots.size;
|
|
const time = new Date(checkpoint.timestamp);
|
|
const timeStr = time.toLocaleTimeString([], { hour: "2-digit", minute: "2-digit", second: "2-digit" });
|
|
|
|
const item = new vscode.TreeItem(
|
|
checkpoint.label,
|
|
vscode.TreeItemCollapsibleState.None,
|
|
);
|
|
item.description = `${timeStr} (${fileCount} file${fileCount !== 1 ? "s" : ""})`;
|
|
item.iconPath = new vscode.ThemeIcon("history");
|
|
item.tooltip = `Checkpoint: ${checkpoint.label}\nTime: ${time.toLocaleString()}\nFiles tracked: ${fileCount}\n\nClick to restore to this point`;
|
|
item.contextValue = "checkpoint";
|
|
item.command = {
|
|
command: "sf.restoreCheckpoint",
|
|
title: "Restore Checkpoint",
|
|
arguments: [checkpoint.id],
|
|
};
|
|
|
|
return item;
|
|
}
|
|
|
|
getChildren(): Checkpoint[] {
|
|
// Show newest first
|
|
return [...this.tracker.checkpoints].reverse();
|
|
}
|
|
|
|
dispose(): void {
|
|
for (const d of this.disposables) {
|
|
d.dispose();
|
|
}
|
|
}
|
|
}
|