fix(worktrees): detect external merges, clean up empty dirs, prune in owning repo

- ContentInMain: HEAD ancestor of the default branch or squash-merged
  (merge-tree == target tree), against local origin/<default> without fetch;
  list_worktrees mergedIntoMain and the overview modal ("merged (external)")
  use it, "Cleanup finished" takes clean Active rows that are in main
- cleanup_task_worktree skips the dirty check for a non-git directory
- prune / worktree remove / branch -D run in the repo owning the worktree
  (git-common-dir), not the list's working dir; branch already gone reports
  branchDeleted=true
This commit is contained in:
mika kuns
2026-10-02 09:07:43 +02:00
parent 9c015f34cc
commit 0c0bb7f24c
12 changed files with 333 additions and 22 deletions
+7
View File
@@ -489,6 +489,13 @@ public sealed class GitService
return head.StartsWith("origin/", StringComparison.Ordinal) ? head["origin/".Length..] : head;
}
/// <summary>Absolute path of the <c>.git</c> dir shared by all worktrees of the repo owning <paramref name="dir"/>, or null.</summary>
public async Task<string?> GetCommonDirAsync(string dir, CancellationToken ct = default)
{
var (exitCode, stdout, _) = await RunGitAsync(dir, ["rev-parse", "--path-format=absolute", "--git-common-dir"], ct);
return exitCode == 0 && !string.IsNullOrWhiteSpace(stdout) ? stdout.Trim() : null;
}
/// <summary><c>git merge --ff-only</c>: moves the checked-out branch to <paramref name="sourceRef"/> or fails without touching it.</summary>
public async Task<(int ExitCode, string Stderr)> MergeFastForwardOnlyAsync(
string worktreePath, string sourceRef, CancellationToken ct = default)
+2 -1
View File
@@ -88,7 +88,8 @@ public record WorktreeOverviewDto(
string? DiffStat,
DateTime CreatedAt,
bool PathExistsOnDisk,
int TaskNumber = 0);
int TaskNumber = 0,
bool ContentInMain = false);
public record ForceRemoveResultDto(bool Removed, string? Reason);
@@ -555,6 +555,8 @@
"resolve": "Lösen",
"phantom": "Phantom",
"phantomTooltip": "Verzeichnis fehlt auf der Festplatte",
"mergedExternal": "gemergt (extern)",
"mergedExternalTooltip": "Die Commits sind schon im Default-Branch (lokaler origin-Stand, ohne Fetch), z.B. per Squash-PR. \"Abgeschlossene aufräumen\" entfernt ihn.",
"ctxShowDiff": "Diff anzeigen",
"ctxOpenInExplorer": "Im Explorer öffnen",
"ctxJumpToTask": "Zur Aufgabe springen",
@@ -555,6 +555,8 @@
"resolve": "Resolve",
"phantom": "phantom",
"phantomTooltip": "Directory missing on disk",
"mergedExternal": "merged (external)",
"mergedExternalTooltip": "Its commits are already in the default branch (local origin ref, not fetched) — e.g. a squash PR. \"Cleanup finished\" removes it.",
"ctxShowDiff": "Show diff",
"ctxOpenInExplorer": "Open in Explorer",
"ctxJumpToTask": "Jump to task",
@@ -33,6 +33,8 @@ public sealed partial class WorktreeOverviewRowViewModel : ViewModelBase
[ObservableProperty] private string? _diffStat;
[ObservableProperty][NotifyPropertyChangedFor(nameof(AgeText))] private DateTime _createdAt;
[ObservableProperty] private bool _pathExistsOnDisk;
/// <summary>Not Merged in ClaudeDo, but the branch's commits are already in the default branch (e.g. a squash PR).</summary>
[ObservableProperty] private bool _contentInMain;
[ObservableProperty] private bool _isSelected;
[ObservableProperty] private bool _isChecked;
[ObservableProperty]
@@ -374,6 +376,7 @@ public sealed partial class WorktreesOverviewModalViewModel : ViewModelBase
ListId = d.ListId, ListName = d.ListName,
Path = d.Path, BranchName = d.BranchName, BaseCommit = d.BaseCommit, State = d.State,
DiffStat = d.DiffStat, CreatedAt = d.CreatedAt, PathExistsOnDisk = d.PathExistsOnDisk,
ContentInMain = d.ContentInMain,
};
public IEnumerable<WorktreeOverviewRowViewModel> AllRows =>
@@ -77,6 +77,11 @@
<TextBlock Classes="meta" Text="{loc:Tr modals.worktreesOverview.phantom}" Foreground="{DynamicResource StatusErrorBrush}"
IsVisible="{Binding !PathExistsOnDisk}"
ToolTip.Tip="{loc:Tr modals.worktreesOverview.phantomTooltip}"/>
<TextBlock Classes="meta" Text="•"
IsVisible="{Binding ContentInMain}"/>
<TextBlock Classes="meta" Text="{loc:Tr modals.worktreesOverview.mergedExternal}" Foreground="{DynamicResource StatusDoneBrush}"
IsVisible="{Binding ContentInMain}"
ToolTip.Tip="{loc:Tr modals.worktreesOverview.mergedExternalTooltip}"/>
</StackPanel>
</StackPanel>
<TextBlock Grid.Column="2" Classes="meta" VerticalAlignment="Center"
+13 -8
View File
@@ -23,7 +23,7 @@ public sealed record TaskDiffDto(
public sealed record WorktreeListItemDto(
string? TaskId, int? Number, string Path, string Branch,
string HeadCommit, bool IsDirty, bool MergedIntoMain);
string HeadCommit, bool IsDirty, bool MergedIntoMain, bool ContentInMain = false);
public sealed record CleanupWorktreeResult(
bool Removed, string WorktreePath, bool BranchDeleted, int? Number = null);
@@ -152,27 +152,31 @@ public sealed class WorktreeMcpTools
[McpServerTool, Description(
"Survey every worktree ClaudeDo tracks — use it to find leftovers to clean up. Only worktrees recorded in " +
"the ClaudeDo database appear here, and headCommit is empty when the path is missing from disk." +
"the ClaudeDo database appear here, and headCommit is empty when the path is missing from disk or is not a " +
"git worktree. contentInMain: the branch's own commits are already in the default branch (merged, " +
"fast-forwarded or squash-merged outside ClaudeDo), checked against the LOCAL origin/<default> (else " +
"main/master) without fetching -- fetch first for a current answer. mergedIntoMain = merged via ClaudeDo " +
"or contentInMain." +
McpToolDocs.ProgressHint)]
public async Task<IReadOnlyList<WorktreeListItemDto>> ListWorktrees(
CancellationToken cancellationToken = default,
IProgress<ProgressNotificationValue>? progress = null)
{
var rows = await _maintenance.GetOverviewAsync(null, cancellationToken);
// One git status + one rev-parse per row, run concurrently -- with many tracked worktrees
// One git status + one rev-parse per row (on top of GetOverviewAsync's content-in-main check), run concurrently -- with many tracked worktrees
// (this tool's whole point is surfacing leftovers nobody cleaned up) that can still take a
// while, so this reports on the same elapsed-time cadence as a single long git call rather
// than per-row (rows finish out of order under Task.WhenAll, so there's no natural i/n).
var results = await ProgressReporter.RunAsync(
Task.WhenAll(rows.Select(async row =>
{
var isDirty = row.PathExistsOnDisk && await TryGetIsDirtyAsync(row.Path, cancellationToken);
var headCommit = row.PathExistsOnDisk
var isDirty = row.IsGitWorktree && await TryGetIsDirtyAsync(row.Path, cancellationToken);
var headCommit = row.IsGitWorktree
? (await TryRunGitAsync(row.Path, ["rev-parse", "HEAD"], cancellationToken) ?? "")
: "";
return new WorktreeListItemDto(
row.TaskId, row.TaskNumber, row.Path, row.BranchName, headCommit,
isDirty, row.State == WorktreeState.Merged);
isDirty, row.State == WorktreeState.Merged || row.ContentInMain, row.ContentInMain);
})),
ProgressReportInterval, progress, "surveying worktrees");
return results;
@@ -180,7 +184,7 @@ public sealed class WorktreeMcpTools
[McpServerTool, Description(
"Remove a task's worktree directory and delete its git branch. branchDeleted reports whether the branch " +
"went too." + McpToolDocs.NotWhileRunning)]
"is gone afterwards (also true when it was already deleted)." + McpToolDocs.NotWhileRunning)]
public async Task<CleanupWorktreeResult> CleanupTaskWorktree(
string taskId,
[Description("false (default): refuse a worktree with uncommitted changes. true: remove it anyway, losing " +
@@ -200,7 +204,8 @@ public sealed class WorktreeMcpTools
if (task.Status == TaskStatus.Running)
throw new InvalidOperationException("Cannot remove worktree of a running task.");
if (!force && Directory.Exists(wt.Path))
// Only a real checkout can be dirty; git status in an empty leftover dir fails (or walks up into an enclosing repo).
if (!force && WorktreeMaintenanceService.IsGitWorktree(wt.Path))
{
var isDirty = await ProgressReporter.RunAsync(
_git.HasChangesAsync(wt.Path, cancellationToken), ProgressReportInterval, progress, "checking worktree for changes");
+1 -1
View File
@@ -467,7 +467,7 @@ public sealed class WorkerHub : Microsoft.AspNetCore.SignalR.Hub
return rows.Select(r => new WorktreeOverviewDto(
r.TaskId, r.TaskTitle, r.TaskStatus, r.ListId, r.ListName,
r.Path, r.BranchName, r.BaseCommit, r.State, r.DiffStat, r.CreatedAt, r.PathExistsOnDisk,
r.TaskNumber)).ToList();
r.TaskNumber, r.ContentInMain)).ToList();
}
public async Task<bool> SetWorktreeState(string taskId, WorktreeState newState)
@@ -1,3 +1,4 @@
using System.Collections.Concurrent;
using ClaudeDo.Data;
using ClaudeDo.Data.Git;
using ClaudeDo.Data.Models;
@@ -44,12 +45,28 @@ public sealed class WorktreeMaintenanceService
join t in context.Tasks on w.TaskId equals t.Id
join l in context.Lists on t.ListId equals l.Id
where w.State == WorktreeState.Merged || w.State == WorktreeState.Discarded
select new { Row = new WorktreeRow(w.TaskId, w.Path, w.BranchName, l.WorkingDir), ListId = t.ListId };
|| (w.State == WorktreeState.Active && t.Status != ClaudeDo.Data.Models.TaskStatus.Running)
select new
{
Row = new WorktreeRow(w.TaskId, w.Path, w.BranchName, l.WorkingDir),
ListId = t.ListId, w.State, w.BaseCommit,
};
if (!string.IsNullOrEmpty(listId))
query = query.Where(x => x.ListId == listId);
var rows = await query.AsNoTracking().Select(x => x.Row).ToListAsync(ct);
var candidates = await query.AsNoTracking().ToListAsync(ct);
var rows = new List<WorktreeRow>();
foreach (var c in candidates)
{
// An Active row only counts as finished when it was merged outside ClaudeDo (e.g. a
// squash PR) and holds nothing uncommitted -- TryRemoveAsync would delete that too.
if (c.State != WorktreeState.Active
|| (IsGitWorktree(c.Row.Path)
&& await IsContentInMainAsync(c.Row.Path, c.BaseCommit, ct)
&& !await TryHasChangesAsync(c.Row.Path, ct)))
rows.Add(c.Row);
}
int removed = 0;
var removedTaskIds = new List<string>();
@@ -114,10 +131,100 @@ public sealed class WorktreeMaintenanceService
var rows = await query.AsNoTracking().ToListAsync(ct);
return rows.Select(x => new WorktreeOverviewRow(
return await Task.WhenAll(rows.Select(async x =>
{
var isGitWorktree = IsGitWorktree(x.Path);
var contentInMain = isGitWorktree
&& (x.State is WorktreeState.Active or WorktreeState.Kept)
&& await IsContentInMainAsync(x.Path, x.BaseCommit, ct);
return new WorktreeOverviewRow(
x.TaskId, x.Number, x.Title, x.Status, x.ListId, x.ListName,
x.Path, x.BranchName, x.BaseCommit ?? "", x.State, x.DiffStat, x.CreatedAt,
PathExistsOnDisk: !string.IsNullOrWhiteSpace(x.Path) && Directory.Exists(x.Path))).ToList();
PathExistsOnDisk: !string.IsNullOrWhiteSpace(x.Path) && Directory.Exists(x.Path),
isGitWorktree, contentInMain);
}));
}
/// <summary>True when <paramref name="path"/> holds a git checkout (<c>.git</c> file or dir), not just a directory.</summary>
// A bare directory check isn't enough: git walks up from an empty dir into whatever repo encloses it.
internal static bool IsGitWorktree(string? path) =>
!string.IsNullOrWhiteSpace(path)
&& (File.Exists(Path.Combine(path, ".git")) || Directory.Exists(Path.Combine(path, ".git")));
// ponytail: 30s TTL per path -- the overview modal reloads every 4s and this costs ~5 git
// processes per row; key on (path, HEAD, target sha) if a stale answer ever matters.
private static readonly TimeSpan ContentInMainTtl = TimeSpan.FromSeconds(30);
private readonly ConcurrentDictionary<string, (DateTime At, bool Result)> _contentInMainCache = new();
/// <summary>
/// The worktree's own commits are already in the default branch -- HEAD is an ancestor of it, or
/// merging HEAD would not change its tree (squash PR). Checks local refs only, never fetches.
/// </summary>
private async Task<bool> IsContentInMainAsync(string path, string? baseCommit, CancellationToken ct)
{
if (string.IsNullOrWhiteSpace(baseCommit)) return false;
if (_contentInMainCache.TryGetValue(path, out var hit) && DateTime.UtcNow - hit.At < ContentInMainTtl)
return hit.Result;
bool result;
try
{
// No commits beyond the base: trivially "in main", but there is nothing merged.
if (await _git.IsAncestorAsync(path, "HEAD", baseCommit, ct) != false) result = false;
else if (await ResolveDefaultRefAsync(path, ct) is not { } target) result = false;
else if (await _git.IsAncestorAsync(path, "HEAD", target, ct) == true) result = true;
else
{
var preview = await _git.PreviewMergeAsync(path, target, "HEAD", ct);
result = preview.Clean
&& preview.TreeOid == await _git.RevParseAsync(path, $"{target}^{{tree}}", ct);
}
}
catch (OperationCanceledException) { throw; }
catch (Exception ex)
{
_logger.LogDebug(ex, "content-in-main check failed for {Path}", path);
result = false;
}
_contentInMainCache[path] = (DateTime.UtcNow, result);
return result;
}
/// <summary>origin/HEAD's branch, else the first existing of origin/main, origin/master, main, master.</summary>
private async Task<string?> ResolveDefaultRefAsync(string dir, CancellationToken ct)
{
var remoteHead = await _git.GetRemoteHeadBranchAsync(dir, ct);
string[] candidates = remoteHead is null
? ["origin/main", "origin/master", "main", "master"]
: [$"origin/{remoteHead}"];
foreach (var c in candidates)
{
try { await _git.RevParseAsync(dir, $"{c}^{{commit}}", ct); return c; }
catch (InvalidOperationException) { }
}
return null;
}
private async Task<bool> TryHasChangesAsync(string path, CancellationToken ct)
{
// Can't tell -> treat as dirty, so a bulk cleanup never deletes it.
try { return await _git.HasChangesAsync(path, ct); }
catch (OperationCanceledException) { throw; }
catch { return true; }
}
/// <summary>The repo owning the worktree (parent of its git-common-dir); the list's working dir when the path is no git checkout anymore.</summary>
private async Task<string?> ResolveOwnerRepoAsync(WorktreeRow row, CancellationToken ct)
{
if (!IsGitWorktree(row.Path)) return row.WorkingDir;
var commonDir = await _git.GetCommonDirAsync(row.Path, ct);
if (commonDir is null) return row.WorkingDir;
var full = Path.TrimEndingDirectorySeparator(Path.GetFullPath(commonDir));
// Non-bare repo: .../<repo>/.git -> <repo>. A bare repo's common dir is the repo itself.
return Path.GetFileName(full).Equals(".git", StringComparison.OrdinalIgnoreCase)
? Path.GetDirectoryName(full)
: full;
}
public async Task<ForceRemoveResult> ForceRemoveAsync(string taskId, CancellationToken ct = default)
@@ -145,7 +252,10 @@ public sealed class WorktreeMaintenanceService
private async Task<(bool Removed, bool BranchDeleted)> TryRemoveAsync(WorktreeRow row, bool force, CancellationToken ct)
{
var repoDirExists = !string.IsNullOrWhiteSpace(row.WorkingDir) && Directory.Exists(row.WorkingDir);
// The list's working dir isn't necessarily the repo that owns the worktree (a list can point
// at another repo than the task ran in) -- prune/branch -D there would silently miss.
var repoDir = await ResolveOwnerRepoAsync(row, ct);
var repoDirExists = !string.IsNullOrWhiteSpace(repoDir) && Directory.Exists(repoDir);
bool dirRemoved;
if (repoDirExists)
@@ -153,7 +263,7 @@ public sealed class WorktreeMaintenanceService
dirRemoved = true;
try
{
await _git.WorktreeRemoveAsync(row.WorkingDir!, row.Path, force, ct);
await _git.WorktreeRemoveAsync(repoDir!, row.Path, force, ct);
}
catch (Exception ex)
{
@@ -195,21 +305,24 @@ public sealed class WorktreeMaintenanceService
bool branchDeleted = false;
if (repoDirExists)
{
try { await _git.WorktreePruneAsync(row.WorkingDir!, ct); }
catch (Exception ex) { _logger.LogWarning(ex, "git worktree prune failed for {Repo}", row.WorkingDir); }
try { await _git.WorktreePruneAsync(repoDir!, ct); }
catch (Exception ex) { _logger.LogWarning(ex, "git worktree prune failed for {Repo}", repoDir); }
if (!string.IsNullOrWhiteSpace(row.BranchName))
{
try
{
await _git.BranchDeleteAsync(row.WorkingDir!, row.BranchName, force: true, ct);
await _git.BranchDeleteAsync(repoDir!, row.BranchName, force: true, ct);
branchDeleted = true;
}
catch (Exception ex)
{
// Already deleted -- the normal case once the branch has been merged.
if (IsBranchAlreadyGone(ex))
{
_logger.LogDebug(ex, "branch {Branch} was already deleted", row.BranchName);
branchDeleted = true;
}
else
_logger.LogWarning(ex, "Failed to delete branch {Branch} for worktree {Path}",
row.BranchName, row.Path);
@@ -3,6 +3,9 @@ using TaskStatus = ClaudeDo.Data.Models.TaskStatus;
namespace ClaudeDo.Worker.Worktrees;
/// <summary>One tracked worktree with its task/list context and on-disk facts.</summary>
/// <param name="IsGitWorktree">The path holds a git checkout (<c>.git</c> present), not just a directory.</param>
/// <param name="ContentInMain">Own commits are already in the local default branch (merged, fast-forwarded or squashed) — independent of <paramref name="State"/>.</param>
public sealed record WorktreeOverviewRow(
string TaskId,
int TaskNumber,
@@ -16,4 +19,6 @@ public sealed record WorktreeOverviewRow(
WorktreeState State,
string? DiffStat,
DateTime CreatedAt,
bool PathExistsOnDisk);
bool PathExistsOnDisk,
bool IsGitWorktree = false,
bool ContentInMain = false);
@@ -489,4 +489,45 @@ public sealed class WorktreeMcpToolsTests : IDisposable
WorktreeMcpTools.ProgressReportInterval = original;
}
}
[Fact]
public async Task CleanupTaskWorktree_EmptyDirectory_RemovesDbRow()
{
if (!GitAvailable) { Assert.True(true, "git not available -- skipping"); return; }
var (task, list, wt) = await SeedWorktreeAsync(TaskStatus.Done);
// Leave an empty, non-git directory behind where the worktree was.
GitRepoFixture.RunGit(list.WorkingDir!, "worktree", "remove", "--force", wt.WorktreePath);
Directory.CreateDirectory(wt.WorktreePath);
var sut = BuildSut(CreateQueue());
var rows = await sut.ListWorktrees(CancellationToken.None);
Assert.Equal("", rows.Single(r => r.TaskId == task.Id).HeadCommit);
var result = await sut.CleanupTaskWorktree(task.Id, false, CancellationToken.None);
Assert.True(result.Removed);
Assert.False(Directory.Exists(wt.WorktreePath));
using var ctx = _db.CreateContext();
Assert.Null(await new WorktreeRepository(ctx).GetByTaskIdAsync(task.Id));
}
[Fact]
public async Task ListWorktrees_SquashMergedBranch_ReportsMergedIntoMain()
{
if (!GitAvailable) { Assert.True(true, "git not available -- skipping"); return; }
var (task, list, wt) = await SeedWorktreeAsync(TaskStatus.Done);
File.WriteAllText(Path.Combine(wt.WorktreePath, "feature.txt"), "x");
GitRepoFixture.RunGit(wt.WorktreePath, "add", "-A");
GitRepoFixture.RunGit(wt.WorktreePath, "commit", "-m", "feature");
GitRepoFixture.RunGit(list.WorkingDir!, "merge", "--squash", wt.BranchName);
GitRepoFixture.RunGit(list.WorkingDir!, "commit", "-m", "squash");
var sut = BuildSut(CreateQueue());
var row = (await sut.ListWorktrees(CancellationToken.None)).Single(r => r.TaskId == task.Id);
Assert.True(row.ContentInMain);
Assert.True(row.MergedIntoMain);
}
}
@@ -531,4 +531,131 @@ public class WorktreeMaintenanceServiceTests : IDisposable
Assert.Equal(1, progressCalls[0].Args[2]);
Assert.Equal(2, progressCalls[1].Args[2]);
}
private static async Task SeedRowAsync(DbFixture db, ListEntity? list, TaskEntity task, string path, string branch,
string baseCommit, WorktreeState state = WorktreeState.Active)
{
using var ctx = db.CreateContext();
if (list is not null) await new ListRepository(ctx).AddAsync(list);
await new TaskRepository(ctx).AddAsync(task);
await new WorktreeRepository(ctx).AddAsync(new WorktreeEntity
{
TaskId = task.Id, Path = path, BranchName = branch,
BaseCommit = baseCommit, State = state, CreatedAt = DateTime.UtcNow,
});
}
private static void CommitFile(string dir, string file)
{
File.WriteAllText(Path.Combine(dir, file), file);
GitRepoFixture.RunGit(dir, "add", "-A");
GitRepoFixture.RunGit(dir, "commit", "-m", $"add {file}");
}
private static void SquashMerge(string repoDir, string branch)
{
GitRepoFixture.RunGit(repoDir, "merge", "--squash", branch);
GitRepoFixture.RunGit(repoDir, "commit", "-m", $"squash {branch}");
}
[Fact]
public async Task GetOverview_ContentInMain_True_For_SquashMerged_False_For_Unmerged_Commit()
{
if (!GitAvailable) { Assert.True(true, "git not available -- skipping"); return; }
var repo = NewRepo();
var git = new GitService();
var db = NewDb();
var (list, squashed) = MakeEntities(repo.RepoDir);
var unmerged = MakeTaskForList(list.Id, ClaudeDo.Data.Models.TaskStatus.Done);
var squashedWt = await CreateWorktreeAsync(git, repo.RepoDir, squashed.Id);
var unmergedWt = await CreateWorktreeAsync(git, repo.RepoDir, unmerged.Id);
CommitFile(squashedWt, "squashed.txt");
CommitFile(unmergedWt, "unmerged.txt");
SquashMerge(repo.RepoDir, $"test/{squashed.Id}");
await SeedRowAsync(db, list, squashed, squashedWt, $"test/{squashed.Id}", repo.BaseCommit);
await SeedRowAsync(db, null, unmerged, unmergedWt, $"test/{unmerged.Id}", repo.BaseCommit);
var svc = new WorktreeMaintenanceService(db.CreateFactory(), git, NullLogger<WorktreeMaintenanceService>.Instance);
var rows = await svc.GetOverviewAsync(null);
Assert.True(rows.Single(r => r.TaskId == squashed.Id).ContentInMain);
Assert.False(rows.Single(r => r.TaskId == unmerged.Id).ContentInMain);
try { await git.WorktreeRemoveAsync(repo.RepoDir, squashedWt, force: true); } catch { }
try { await git.WorktreeRemoveAsync(repo.RepoDir, unmergedWt, force: true); } catch { }
}
[Fact]
public async Task CleanupFinished_Removes_Active_Worktree_Squash_Merged_Outside_ClaudeDo()
{
if (!GitAvailable) { Assert.True(true, "git not available -- skipping"); return; }
var repo = NewRepo();
var git = new GitService();
var db = NewDb();
var (list, squashed) = MakeEntities(repo.RepoDir);
var unmerged = MakeTaskForList(list.Id, ClaudeDo.Data.Models.TaskStatus.Done);
var squashedWt = await CreateWorktreeAsync(git, repo.RepoDir, squashed.Id);
var unmergedWt = await CreateWorktreeAsync(git, repo.RepoDir, unmerged.Id);
CommitFile(squashedWt, "squashed.txt");
CommitFile(unmergedWt, "unmerged.txt");
SquashMerge(repo.RepoDir, $"test/{squashed.Id}");
await SeedRowAsync(db, list, squashed, squashedWt, $"test/{squashed.Id}", repo.BaseCommit);
await SeedRowAsync(db, null, unmerged, unmergedWt, $"test/{unmerged.Id}", repo.BaseCommit);
var svc = new WorktreeMaintenanceService(db.CreateFactory(), git, NullLogger<WorktreeMaintenanceService>.Instance);
var result = await svc.CleanupFinishedAsync();
Assert.Equal(new[] { squashed.Id }, result.RemovedTaskIds);
Assert.False(Directory.Exists(squashedWt));
Assert.True(Directory.Exists(unmergedWt));
try { await git.WorktreeRemoveAsync(repo.RepoDir, unmergedWt, force: true); } catch { }
}
[Fact]
public async Task ForceRemove_Prunes_And_Deletes_Branch_In_Owning_Repo_Not_List_WorkingDir()
{
if (!GitAvailable) { Assert.True(true, "git not available -- skipping"); return; }
var owner = NewRepo();
var other = NewRepo();
var git = new GitService();
var db = NewDb();
// The list points at a different repo than the one the worktree was created in.
var (list, task) = MakeEntities(other.RepoDir);
var wt = await CreateWorktreeAsync(git, owner.RepoDir, task.Id);
await SeedRowAsync(db, list, task, wt, $"test/{task.Id}", owner.BaseCommit);
var svc = new WorktreeMaintenanceService(db.CreateFactory(), git, NullLogger<WorktreeMaintenanceService>.Instance);
var result = await svc.ForceRemoveAsync(task.Id);
Assert.True(result.Removed);
Assert.True(result.BranchDeleted);
Assert.DoesNotContain($"test/{task.Id}", GitRepoFixture.RunGit(owner.RepoDir, "branch", "--list"));
Assert.DoesNotContain(Path.GetFileName(wt), GitRepoFixture.RunGit(owner.RepoDir, "worktree", "list"));
}
[Fact]
public async Task ForceRemove_Reports_BranchDeleted_When_Branch_Was_Already_Gone()
{
if (!GitAvailable) { Assert.True(true, "git not available -- skipping"); return; }
var repo = NewRepo();
var db = NewDb();
var (list, task) = MakeEntities(repo.RepoDir);
var phantomPath = Path.Combine(Path.GetTempPath(), $"wt_phantom_{Guid.NewGuid():N}");
await SeedRowAsync(db, list, task, phantomPath, $"test/{task.Id}-never-existed", repo.BaseCommit);
var svc = new WorktreeMaintenanceService(db.CreateFactory(), new GitService(), NullLogger<WorktreeMaintenanceService>.Instance);
var result = await svc.ForceRemoveAsync(task.Id);
Assert.True(result.Removed);
Assert.True(result.BranchDeleted);
}
}