Coverage for src/workstack/cli/commands/tree.py: 87%

1"""Tree visualization command for workstack.""" 

2 

3from pathlib import Path 

4 

5import click 

6 

7from workstack.cli.core import discover_repo_context 

8from workstack.cli.tree import build_workstack_tree, render_tree 

9from workstack.core.context import WorkstackContext 

10 

11 

12@click.command("tree") 

13@click.pass_obj 

14def tree_cmd(ctx: WorkstackContext) -> None: 

15 """Display tree of worktrees with their dependencies. 

16 

17 Shows ONLY branches that have active worktrees, organized 

18 by their Graphite parent-child relationships. 

19 

20 Requires Graphite to be enabled and configured. 

21 

22 Example: 

23 $ workstack tree 

24 main [@root] 

25 ├─ feature-a [@feature-a] 

26 │ └─ feature-a-2 [@feature-a-2] 

27 └─ feature-b [@feature-b] 

28 

29 Legend: 

30 [@worktree-name] = worktree directory name 

31 Current worktree is highlighted in bright green 

32 """ 

33 repo = discover_repo_context(ctx, Path.cwd()) 

34 

35 # Build tree structure (will exit with error if Graphite cache missing) 

36 roots = build_workstack_tree(ctx, repo.root) 

37 

38 if not roots: 

39 click.echo("No worktrees found", err=True) 

40 raise SystemExit(1) 

41 

42 # Render and display 

43 tree_output = render_tree(roots) 

44 click.echo(tree_output)