jesseduffield.lazygit/pkg/gui/controllers/helpers/host_helper.go

package helpers

import (
	"github.com/jesseduffield/lazygit/pkg/commands/hosting_service"
	"github.com/jesseduffield/lazygit/pkg/commands/models"
)

// this helper just wraps our hosting_service package

type HostHelper struct {
	c *HelperCommon
}

func NewHostHelper(
	c *HelperCommon,
) *HostHelper {
	return &HostHelper{
		c: c,
	}
}

func (self *HostHelper) GetPullRequestURL(from string, to string) (string, error) {
	mgr, err := self.getHostingServiceMgr()
	if err != nil {
		return "", err
	}
	return mgr.GetPullRequestURL(from, to)
}

func (self *HostHelper) GetCommitURL(commitHash string) (string, error) {
	mgr, err := self.getHostingServiceMgr()
	if err != nil {
		return "", err
	}
	return mgr.GetCommitURL(commitHash)
}

// getting this on every request rather than storing it in state in case our remoteURL changes
// from one invocation to the next.
func (self *HostHelper) getHostingServiceMgr() (*hosting_service.HostingServiceMgr, error) {
	remoteName := getPreferredRemoteName(
		self.c.Contexts().RemoteBranches.GetSelected(),
		self.c.Contexts().Remotes.GetSelected(),
	)

	remoteUrl, err := self.c.Git().Remote.GetRemoteURL(remoteName)
	if err != nil && remoteName != "origin" {
		remoteUrl, err = self.c.Git().Remote.GetRemoteURL("origin")
	}
	if err != nil {
		return nil, err
	}

	configServices := self.c.UserConfig().Services
	return hosting_service.NewHostingServiceMgr(self.c.Log, self.c.Tr, remoteUrl, configServices), nil
}

func getPreferredRemoteName(selectedRemoteBranch *models.RemoteBranch, selectedRemote *models.Remote) string {
	if selectedRemoteBranch != nil && selectedRemoteBranch.RemoteName != "" {
		return selectedRemoteBranch.RemoteName
	}

	if selectedRemote != nil && selectedRemote.Name != "" {
		return selectedRemote.Name
	}

	return "origin"
}