Also consider the default tmux server when generating completions.

Author: JeffFaer

cmd/update.go

@@ -100,19 +100,26 @@ func suggestWorkUnitNames(ctx context.Context, toComplete state.WorkUnitName) []
 }
 
 func discoverRepositories(ctx context.Context, vcs api.VersionControlSystems) (current api.Repository, all map[state.RepoName]api.Repository) {
-	repos := make(map[state.RepoName]api.Repository)
+	all = make(map[state.RepoName]api.Repository)
+	var srv tmux.Server
+	var curSesh tmux.Session
 	if sesh := tmux.MaybeCurrentSession(); sesh != nil {
-		st, err := state.New(ctx, sesh.Server(), vcs)
+		srv = sesh.Server()
+		curSesh = sesh
+	} else {
+		srv = tmux.DefaultServer()
+	}
+
+	st, err := state.New(ctx, srv, vcs)
+	if err != nil {
+		slog.Warn("Could not determine repositories from tmux server.", "server", srv, "error", err)
+	} else if curSesh != nil {
+		all = st.Repositories()
+		repo, _, err := st.WorkUnit(ctx, curSesh)
 		if err != nil {
-			slog.Warn("Could not determine repositories from tmux server.", "server", sesh.Server(), "error", err)
+			slog.Warn("Could not determine current repository from tmux.", "server", curSesh, "error", err)
 		} else {
-			repos = st.Repositories()
-			repo, _, err := st.WorkUnit(ctx, sesh)
-			if err != nil {
-				slog.Warn("Could not determine current repository from tmux.", "server", sesh.Server(), "error", err)
-			} else {
-				return repo, repos
-			}
+			return repo, all
 		}
 	}
 
@@ -121,10 +128,10 @@ func discoverRepositories(ctx context.Context, vcs api.VersionControlSystems) (c
 	cur, err := vcs.MaybeCurrentRepository(ctx)
 	if err != nil {
 		slog.Warn("Could not determine current repository.", "error", err)
-		return nil, repos
+		return nil, all
 	}
-	repos[state.NewRepoName(cur)] = cur
-	return cur, repos
+	all[state.NewRepoName(cur)] = cur
+	return cur, all
 }
 
 func update(ctx context.Context) error {