Show origin vs upstream location for in-flight branches in wl me

Queries dolt_remote_branches after fetch to determine where each
wl/* branch has been pushed: "origin", "upstream", "origin + upstream",
or "local only". Displayed alongside each in-flight branch item.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: julianknutsen

cmd/wl/cmd_me.go

@@ -102,7 +102,7 @@ func runMe(cmd *cobra.Command, stdout, _ io.Writer) error {
 	if len(branchItems) > 0 {
 		fmt.Fprintf(stdout, "\n%s\n", style.Bold.Render("In-flight branches (not yet on main):"))
 		for _, bi := range branchItems {
-			fmt.Fprintf(stdout, "  %-12s %-30s %s\n", bi.id, bi.title, style.Dim.Render(bi.branch))
+			fmt.Fprintf(stdout, "  %-12s %-30s %-18s %s\n", bi.id, bi.title, style.Dim.Render(bi.location), style.Dim.Render(bi.branch))
 		}
 		printed = true
 	}
@@ -181,9 +181,59 @@ func queryClaimedAsOf(dbDir, handle, ref string) [][]string {
 
 // branchItem represents a wanted item found on a wl/* branch.
 type branchItem struct {
-	id     string
-	title  string
-	branch string
+	id       string
+	title    string
+	branch   string
+	location string // "local only", "origin", "upstream", "origin + upstream"
+}
+
+// remoteBranchSet returns the set of branch names that exist on remotes,
+// keyed by the local branch name (e.g. "wl/alice/w-123" → {"origin": true}).
+func remoteBranchSet(dbDir string) map[string]map[string]bool {
+	csv, err := commons.DoltSQLQuery(dbDir, "SELECT name FROM dolt_remote_branches")
+	if err != nil {
+		return nil
+	}
+	rows := wlParseCSV(csv)
+	result := make(map[string]map[string]bool)
+	for i, row := range rows {
+		if i == 0 || len(row) < 1 { // skip header
+			continue
+		}
+		// name is "origin/wl/alice/w-123" or "upstream/wl/alice/w-123"
+		name := row[0]
+		slashIdx := strings.Index(name, "/")
+		if slashIdx < 0 {
+			continue
+		}
+		remote := name[:slashIdx]
+		localName := name[slashIdx+1:]
+		if result[localName] == nil {
+			result[localName] = make(map[string]bool)
+		}
+		result[localName][remote] = true
+	}
+	return result
+}
+
+// branchLocation returns a human-readable location string for a branch.
+func branchLocation(remotes map[string]map[string]bool, branch string) string {
+	if remotes == nil {
+		return "local only"
+	}
+	rm := remotes[branch]
+	onOrigin := rm["origin"]
+	onUpstream := rm["upstream"]
+	switch {
+	case onOrigin && onUpstream:
+		return "origin + upstream"
+	case onOrigin:
+		return "origin"
+	case onUpstream:
+		return "upstream"
+	default:
+		return "local only"
+	}
 }
 
 // listBranchItems scans wl/<handle>/* branches and returns items that differ
@@ -195,6 +245,8 @@ func listBranchItems(dbDir, handle string, skipIDs map[string]bool) []branchItem
 		return nil
 	}
 
+	remotes := remoteBranchSet(dbDir)
+
 	var items []branchItem
 	for _, branch := range branches {
 		wantedID := extractBranchWantedID(branch)
@@ -215,9 +267,10 @@ func listBranchItems(dbDir, handle string, skipIDs map[string]bool) []branchItem
 			continue
 		}
 		items = append(items, branchItem{
-			id:     rows[1][0],
-			title:  rows[1][1],
-			branch: branch,
+			id:       rows[1][0],
+			title:    rows[1][1],
+			branch:   branch,
+			location: branchLocation(remotes, branch),
 		})
 	}
 	return items