There's gotta be a better way for initial content

1 files changed, 15 insertions, 5 deletions

diff --git a/pkg/gui/controllers/helpers/upstream_helper.go b/pkg/gui/controllers/helpers/upstream_helper.go
index a0307a9d4..a3ece704e 100644
--- a/pkg/gui/controllers/helpers/upstream_helper.go
+++ b/pkg/gui/controllers/helpers/upstream_helper.go
@@ -17,7 +17,8 @@ type UpstreamHelper struct {
 
 type IUpstreamHelper interface {
 	ParseUpstream(string) (string, string, error)
-	PromptForUpstream(*models.Branch, func(string) error) error
+	PromptForUpstreamWithInitialContent(*models.Branch, func(string) error) error
+	PromptForUpstreamWithoutInitialContent(*models.Branch, func(string) error) error
 	GetSuggestedRemote() string
 }
 
@@ -48,17 +49,26 @@ func (self *UpstreamHelper) ParseUpstream(upstream string) (string, string, erro
 	return upstreamRemote, upstreamBranch, nil
 }
 
-func (self *UpstreamHelper) PromptForUpstream(currentBranch *models.Branch, onConfirm func(string) error) error {
-	suggestedRemote := self.GetSuggestedRemote()
-
+func (self *UpstreamHelper) promptForUpstream(currentBranch *models.Branch, initialContent string, onConfirm func(string) error) error {
 	return self.c.Prompt(types.PromptOpts{
 		Title:               self.c.Tr.EnterUpstream,
-		InitialContent:      suggestedRemote + " " + currentBranch.Name,
+		InitialContent:      initialContent,
 		FindSuggestionsFunc: self.getRemoteBranchesSuggestionsFunc(" "),
 		HandleConfirm:       onConfirm,
 	})
 }
 
+func (self *UpstreamHelper) PromptForUpstreamWithInitialContent(currentBranch *models.Branch, onConfirm func(string) error) error {
+	suggestedRemote := self.GetSuggestedRemote()
+	initialContent := suggestedRemote + " " + currentBranch.Name
+
+	return self.promptForUpstream(currentBranch, initialContent, onConfirm)
+}
+
+func (self *UpstreamHelper) PromptForUpstreamWithoutInitialContent(currentBranch *models.Branch, onConfirm func(string) error) error {
+	return self.promptForUpstream(currentBranch, "", onConfirm)
+}
+
 func (self *UpstreamHelper) GetSuggestedRemote() string {
 	return getSuggestedRemote(self.model.Remotes)
 }