From d87e10a63298953e621a5f611fbe76c4e449bae7 Mon Sep 17 00:00:00 2001
From: Sergei Shmakov <sergei.shmakov@gitkraken.com>
Date: Tue, 19 Aug 2025 16:32:36 +0200
Subject: [PATCH] Fixes commands to change or set branch upstream in graph

(#4498, #4539)
---
 src/webviews/plus/graph/graphWebview.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/webviews/plus/graph/graphWebview.ts b/src/webviews/plus/graph/graphWebview.ts
index 4b4c0c1260b79..03dfd01a2aa2a 100644
--- a/src/webviews/plus/graph/graphWebview.ts
+++ b/src/webviews/plus/graph/graphWebview.ts
@@ -526,6 +526,8 @@ export class GraphWebviewProvider implements WebviewProvider<State, State, Graph
 			this.host.registerWebviewCommand('gitlens.graph.rebaseOntoUpstream', this.rebaseToRemote),
 			this.host.registerWebviewCommand('gitlens.graph.renameBranch', this.renameBranch),
 			this.host.registerWebviewCommand('gitlens.graph.associateIssueWithBranch', this.associateIssueWithBranch),
+			this.host.registerWebviewCommand('gitlens.changeUpstream:graph', this.changeUpstreamBranch),
+			this.host.registerWebviewCommand('gitlens.setUpstream:graph', this.changeUpstreamBranch),
 
 			this.host.registerWebviewCommand('gitlens.graph.switchToBranch', this.switchTo),
 
@@ -3819,6 +3821,13 @@ export class GraphWebviewProvider implements WebviewProvider<State, State, Graph
 		return Promise.resolve();
 	}
 
+	@log()
+	private changeUpstreamBranch(item?: GraphItemContext) {
+		if (!isGraphItemRefContext(item, 'branch')) return Promise.resolve();
+		const { ref } = item.webviewItemValue;
+		return BranchActions.changeUpstream(ref.repoPath, ref);
+	}
+
 	@log()
 	private compareWorkingWith(item?: GraphItemContext) {
 		const ref = this.getGraphItemRef(item);