diff options
-rw-r--r-- | pkg/commands/pull_request.go | 261 | ||||
-rw-r--r-- | pkg/commands/pull_request_default_test.go | 2 | ||||
-rw-r--r-- | pkg/commands/pull_request_test.go | 20 | ||||
-rw-r--r-- | pkg/config/user_config.go | 2 | ||||
-rw-r--r-- | pkg/gui/commits_panel.go | 16 | ||||
-rw-r--r-- | pkg/gui/keybindings.go | 7 | ||||
-rw-r--r-- | pkg/gui/pull_request_menu_panel.go | 2 | ||||
-rw-r--r-- | pkg/i18n/english.go | 4 |
8 files changed, 216 insertions, 98 deletions
diff --git a/pkg/commands/pull_request.go b/pkg/commands/pull_request.go index ed065bbec..2f2a9cd1d 100644 --- a/pkg/commands/pull_request.go +++ b/pkg/commands/pull_request.go @@ -16,61 +16,112 @@ var defaultUrlRegexStrings = []string{ `^git@.*:(?P<owner>.*)/(?P<repo>.*?)(?:\.git)?$`, } -// Service is a service that repository is on (Github, Bitbucket, ...) -type Service struct { - Name string - pullRequestURLIntoDefaultBranch func(owner string, repository string, from string) string - pullRequestURLIntoTargetBranch func(owner string, repository string, from string, to string) string - URLRegexStrings []string +type ServiceDefinition struct { + provider string + pullRequestURLIntoDefaultBranch string + pullRequestURLIntoTargetBranch string + commitURL string + regexStrings []string } -func NewGithubService(repositoryDomain string, siteDomain string) *Service { - return &Service{ - Name: repositoryDomain, - pullRequestURLIntoDefaultBranch: func(owner string, repository string, from string) string { - return fmt.Sprintf("https://%s/%s/%s/compare/%s?expand=1", siteDomain, owner, repository, from) - }, - pullRequestURLIntoTargetBranch: func(owner string, repository string, from string, to string) string { - return fmt.Sprintf("https://%s/%s/%s/compare/%s...%s?expand=1", siteDomain, owner, repository, to, from) - }, - URLRegexStrings: defaultUrlRegexStrings, +func (self ServiceDefinition) getRepoInfoFromURL(url string) (*RepoInformation, error) { + for _, regexStr := range self.regexStrings { + re := regexp.MustCompile(regexStr) + matches := utils.FindNamedMatches(re, url) + if matches != nil { + return &RepoInformation{ + Owner: matches["owner"], + Repository: matches["repo"], + }, nil + } } + + return nil, errors.New("Failed to parse repo information from url") } -func NewBitBucketService(repositoryDomain string, siteDomain string) *Service { - return &Service{ - Name: repositoryDomain, - pullRequestURLIntoDefaultBranch: func(owner string, repository string, from string) string { - return fmt.Sprintf("https://%s/%s/%s/pull-requests/new?source=%s&t=1", siteDomain, owner, repository, from) - }, - pullRequestURLIntoTargetBranch: func(owner string, repository string, from string, to string) string { - return fmt.Sprintf("https://%s/%s/%s/pull-requests/new?source=%s&dest=%s&t=1", siteDomain, owner, repository, from, to) - }, - URLRegexStrings: defaultUrlRegexStrings, - } +// a service domains pairs a service definition with the actual domain it's being served from. +// Sometimes the git service is hosted in a custom domains so although it'll use say +// the github service definition, it'll actually be served from e.g. my-custom-github.com +type ServiceDomain struct { + gitDomain string // the one that appears in the git remote url + webDomain string // the one that appears in the web url + serviceDefinition ServiceDefinition } -func NewGitLabService(repositoryDomain string, siteDomain string) *Service { - return &Service{ - Name: repositoryDomain, - pullRequestURLIntoDefaultBranch: func(owner string, repository string, from string) string { - return fmt.Sprintf("https://%s/%s/%s/merge_requests/new?merge_request[source_branch]=%s", siteDomain, owner, repository, from) - }, - pullRequestURLIntoTargetBranch: func(owner string, repository string, from string, to string) string { - return fmt.Sprintf("https://%s/%s/%s/merge_requests/new?merge_request[source_branch]=%s&merge_request[target_branch]=%s", siteDomain, owner, repository, from, to) - }, - URLRegexStrings: defaultUrlRegexStrings, +func (self ServiceDomain) getRootFromRepoURL(repoURL string) (string, error) { + // we may want to make this more specific to the service in future e.g. if + // some new service comes along which has a different root url structure. + repoInfo, err := self.serviceDefinition.getRepoInfoFromURL(repoURL) + if err != nil { + return "", err } + return fmt.Sprintf("https://%s/%s/%s", self.webDomain, repoInfo.Owner, repoInfo.Repository), nil } -func (s *Service) PullRequestURL(repoURL string, from string, to string) string { - repoInfo := s.getRepoInfoFromURL(repoURL) +// we've got less type safety using go templates but this lends itself better to +// users adding custom service definitions in their config +var GithubServiceDef = ServiceDefinition{ + provider: "github", + pullRequestURLIntoDefaultBranch: "/compare/{{.From}}?expand=1", + pullRequestURLIntoTargetBranch: "/compare/{{.To}}...{{.From}}?expand=1", + commitURL: "/commit/{{.CommitSha}}", + regexStrings: defaultUrlRegexStrings, +} - if to == "" { - return s.pullRequestURLIntoDefaultBranch(repoInfo.Owner, repoInfo.Repository, from) - } else { - return s.pullRequestURLIntoTargetBranch(repoInfo.Owner, repoInfo.Repository, from, to) - } +var BitbucketServiceDef = ServiceDefinition{ + provider: "bitbucket", + pullRequestURLIntoDefaultBranch: "/pull-requests/new?source={{.From}}&t=1", + pullRequestURLIntoTargetBranch: "/pull-requests/new?source={{.From}}&dest={{.To}}&t=1", + commitURL: "/commits/{{.CommitSha}}", + regexStrings: defaultUrlRegexStrings, +} + +var GitLabServiceDef = ServiceDefinition{ + provider: "gitlab", + pullRequestURLIntoDefaultBranch: "/merge_requests/new?merge_request[source_branch]={{.From}}", + pullRequestURLIntoTargetBranch: "/merge_requests/new?merge_request[source_branch]={{.From}}&merge_request[target_branch]={{.To}}", + commitURL: "/commit/{{.CommitSha}}", + regexStrings: defaultUrlRegexStrings, +} + +var serviceDefinitions = []ServiceDefinition{GithubServiceDef, BitbucketServiceDef, GitLabServiceDef} +var defaultServiceDomains = []ServiceDomain{ + { + serviceDefinition: GithubServiceDef, + gitDomain: "github.com", + webDomain: "github.com", + }, + { + serviceDefinition: BitbucketServiceDef, + gitDomain: "bitbucket.org", + webDomain: "bitbucket.org", + }, + { + serviceDefinition: GitLabServiceDef, + gitDomain: "gitlab.com", + webDomain: "gitlab.com", + }, +} + +type Service struct { + root string + ServiceDefinition +} + +func (self *Service) getPullRequestURLIntoDefaultBranch(from string) string { + return self.resolveUrl(self.pullRequestURLIntoDefaultBranch, map[string]string{"From": from}) +} + +func (self *Service) getPullRequestURLIntoTargetBranch(from string, to string) string { + return self.resolveUrl(self.pullRequestURLIntoTargetBranch, map[string]string{"From": from, "To": to}) +} + +func (self *Service) getCommitURL(commitSha string) string { + return self.resolveUrl(self.commitURL, map[string]string{"CommitSha": commitSha}) +} + +func (self *Service) resolveUrl(templateString string, args map[string]string) string { + return self.root + utils.ResolvePlaceholderString(templateString, args) } // PullRequest opens a link in browser to create new pull request @@ -92,48 +143,86 @@ func NewPullRequest(gitCommand *GitCommand) *PullRequest { } } -func (pr *PullRequest) getServices() []*Service { - services := []*Service{ - NewGithubService("github.com", "github.com"), - NewBitBucketService("bitbucket.org", "bitbucket.org"), - NewGitLabService("gitlab.com", "gitlab.com"), +func (pr *PullRequest) getService() (*Service, error) { + serviceDomain, err := pr.getServiceDomain() + if err != nil { + return nil, err } - configServices := pr.GitCommand.Config.GetUserConfig().Services + repoURL := pr.GitCommand.GetRemoteURL() - if len(configServices) > 0 { - serviceFuncMap := map[string]func(repositoryDomain string, siteDomain string) *Service{ - "github": NewGithubService, - "bitbucket": NewBitBucketService, - "gitlab": NewGitLabService, + root, err := serviceDomain.getRootFromRepoURL(repoURL) + if err != nil { + return nil, err + } + + return &Service{ + root: root, + ServiceDefinition: serviceDomain.serviceDefinition, + }, nil +} + +func (pr *PullRequest) getServiceDomain() (*ServiceDomain, error) { + candidateServiceDomains := pr.getCandidateServiceDomains() + + repoURL := pr.GitCommand.GetRemoteURL() + + for _, serviceDomain := range candidateServiceDomains { + // I feel like it makes more sense to see if the repo url contains the service domain's git domain, + // but I don't want to break anything by changing that right now. + if strings.Contains(repoURL, serviceDomain.serviceDefinition.provider) { + return &serviceDomain, nil } + } - for repoDomain, typeAndDomain := range configServices { + return nil, errors.New(pr.GitCommand.Tr.UnsupportedGitService) +} + +func (pr *PullRequest) getCandidateServiceDomains() []ServiceDomain { + serviceDefinitionByProvider := map[string]ServiceDefinition{} + for _, serviceDefinition := range serviceDefinitions { + serviceDefinitionByProvider[serviceDefinition.provider] = serviceDefinition + } + + var serviceDomains = make([]ServiceDomain, len(defaultServiceDomains)) + copy(serviceDomains, defaultServiceDomains) + + // see https://github.com/jesseduffield/lazygit/blob/master/docs/Config.md#custom-pull-request-urls + configServices := pr.GitCommand.Config.GetUserConfig().Services + if len(configServices) > 0 { + for gitDomain, typeAndDomain := range configServices { splitData := strings.Split(typeAndDomain, ":") if len(splitData) != 2 { pr.GitCommand.Log.Errorf("Unexpected format for git service: '%s'. Expected something like 'github.com:github.com'", typeAndDomain) continue } - serviceFunc := serviceFuncMap[splitData[0]] - if serviceFunc == nil { - serviceNames := []string{} - for serviceName := range serviceFuncMap { - serviceNames = append(serviceNames, serviceName) + provider := splitData[0] + webDomain := splitData[1] + + serviceDefinition, ok := serviceDefinitionByProvider[provider] + if !ok { + providerNames := []string{} + for _, serviceDefinition := range serviceDefinitions { + providerNames = append(providerNames, serviceDefinition.provider) } - pr.GitCommand.Log.Errorf("Unknown git service type: '%s'. Expected one of %s", splitData[0], strings.Join(serviceNames, ", ")) + pr.GitCommand.Log.Errorf("Unknown git service type: '%s'. Expected one of %s", provider, strings.Join(providerNames, ", ")) continue } - services = append(services, serviceFunc(repoDomain, splitData[1])) + serviceDomains = append(serviceDomains, ServiceDomain{ + gitDomain: gitDomain, + webDomain: webDomain, + serviceDefinition: serviceDefinition, + }) } } - return services + return serviceDomains } -// Create opens link to new pull request in browser -func (pr *PullRequest) Create(from string, to string) (string, error) { +// CreatePullRequest opens link to new pull request in browser +func (pr *PullRequest) CreatePullRequest(from string, to string) (string, error) { pullRequestURL, err := pr.getPullRequestURL(from, to) if err != nil { return "", err @@ -159,36 +248,34 @@ func (pr *PullRequest) getPullRequestURL(from string, to string) (string, error) return "", errors.New(pr.GitCommand.Tr.NoBranchOnRemote) } - repoURL := pr.GitCommand.GetRemoteURL() - var gitService *Service + gitService, err := pr.getService() + if err != nil { + return "", err + } - for _, service := range pr.getServices() { - if strings.Contains(repoURL, service.Name) { - gitService = service - break - } + if to == "" { + return gitService.getPullRequestURLIntoDefaultBranch(from), nil + } else { + return gitService.getPullRequestURLIntoTargetBranch(from, to), nil } +} - if gitService == nil { - return "", errors.New(pr.GitCommand.Tr.UnsupportedGitService) +func (pr *PullRequest) getCommitURL(commitSha string) (string, error) { + gitService, err := pr.getService() + if err != nil { + return "", err } - pullRequestURL := gitService.PullRequestURL(repoURL, from, to) + pullRequestURL := gitService.getCommitURL(commitSha) return pullRequestURL, nil } -func (s *Service) getRepoInfoFromURL(url string) *RepoInformation { - for _, regexStr := range s.URLRegexStrings { - re := regexp.MustCompile(regexStr) - matches := utils.FindNamedMatches(re, url) - if matches != nil { - return &RepoInformation{ - Owner: matches["owner"], - Repository: matches["repo"], - } - } +func (pr *PullRequest) OpenCommitInBrowser(commitSha string) (string, error) { + url, err := pr.getCommitURL(commitSha) + if err != nil { + return "", err } - return nil + return url, pr.GitCommand.OSCommand.OpenLink(url) } diff --git a/pkg/commands/pull_request_default_test.go b/pkg/commands/pull_request_default_test.go index e1ce61a5a..9a7d775ee 100644 --- a/pkg/commands/pull_request_default_test.go +++ b/pkg/commands/pull_request_default_test.go @@ -250,7 +250,7 @@ func TestCreatePullRequest(t *testing.T) { } gitCommand.GitConfig = git_config.NewFakeGitConfig(map[string]string{"remote.origin.url": s.remoteUrl}) dummyPullRequest := NewPullRequest(gitCommand) - s.test(dummyPullRequest.Create(s.from, s.to)) + s.test(dummyPullRequest.CreatePullRequest(s.from, s.to)) }) } } diff --git a/pkg/commands/pull_request_test.go b/pkg/commands/pull_request_test.go index 6195dff4c..844a15998 100644 --- a/pkg/commands/pull_request_test.go +++ b/pkg/commands/pull_request_test.go @@ -9,15 +9,15 @@ import ( // TestGetRepoInfoFromURL is a function. func TestGetRepoInfoFromURL(t *testing.T) { type scenario struct { - service *Service - testName string - repoURL string - test func(*RepoInformation) + serviceDefinition ServiceDefinition + testName string + repoURL string + test func(*RepoInformation) } scenarios := []scenario{ { - NewGithubService("github.com", "github.com"), + GithubServiceDef, "Returns repository information for git remote url", "git@github.com:petersmith/super_calculator", func(repoInfo *RepoInformation) { @@ -26,7 +26,7 @@ func TestGetRepoInfoFromURL(t *testing.T) { }, }, { - NewGithubService("github.com", "github.com"), + GithubServiceDef, "Returns repository information for git remote url, trimming trailing '.git'", "git@github.com:petersmith/super_calculator.git", func(repoInfo *RepoInformation) { @@ -35,7 +35,7 @@ func TestGetRepoInfoFromURL(t *testing.T) { }, }, { - NewGithubService("github.com", "github.com"), + GithubServiceDef, "Returns repository information for ssh remote url", "ssh://git@github.com/petersmith/super_calculator", func(repoInfo *RepoInformation) { @@ -44,7 +44,7 @@ func TestGetRepoInfoFromURL(t *testing.T) { }, }, { - NewGithubService("github.com", "github.com"), + GithubServiceDef, "Returns repository information for http remote url", "https://my_username@bitbucket.org/johndoe/social_network.git", func(repoInfo *RepoInformation) { @@ -56,7 +56,9 @@ func TestGetRepoInfoFromURL(t *testing.T) { for _, s := range scenarios { t.Run(s.testName, func(t *testing.T) { - s.test(s.service.getRepoInfoFromURL(s.repoURL)) + result, err := s.serviceDefinition.getRepoInfoFromURL(s.repoURL) + assert.NoError(t, err) + s.test(result) }) } } diff --git a/pkg/config/user_config.go b/pkg/config/user_config.go index bcf0374cd..8c1d90a0e 100644 --- a/pkg/config/user_config.go +++ b/pkg/config/user_config.go @@ -246,6 +246,7 @@ type KeybindingCommitsConfig struct { ResetCherryPick string `yaml:"resetCherryPick"` CopyCommitMessageToClipboard string `yaml:"copyCommitMessageToClipboard"` OpenLogMenu string `yaml:"openLogMenu"` + OpenInBrowser string `yaml:"openInBrowser"` } type KeybindingStashConfig struct { @@ -508,6 +509,7 @@ func GetDefaultConfig() *UserConfig { ResetCherryPick: "<c-R>", CopyCommitMessageToClipboard: "<c-y>", OpenLogMenu: "<c-l>", + OpenInBrowser: "o", }, Stash: KeybindingStashConfig{ PopStash: "g", diff --git a/pkg/gui/commits_panel.go b/pkg/gui/commits_panel.go index 01039e028..3beec3d28 100644 --- a/pkg/gui/commits_panel.go +++ b/pkg/gui/commits_panel.go @@ -797,3 +797,19 @@ func (gui *Gui) handleOpenLogMenu() error { }, }, createMenuOptions{showCancel: true}) } + +func (gui *Gui) handleOpenCommitInBrowser() error { + commit := gui.getSelectedLocalCommit() + if commit == nil { + return nil + } + + pullRequest := commands.NewPullRequest(gui.GitCommand) + url, err := pullRequest.OpenCommitInBrowser(commit.Sha) + if err != nil { + return gui.surfaceError(err) + } + gui.OnRunCommand(oscommands.NewCmdLogEntry(fmt.Sprintf(gui.Tr.OpeningCommitInBrowser, url), gui.Tr.CreatePullRequest, false)) + + return nil +} diff --git a/pkg/gui/keybindings.go b/pkg/gui/keybindings.go index d9bf2dd69..6a65e63fc 100644 --- a/pkg/gui/keybindings.go +++ b/pkg/gui/keybindings.go @@ -903,6 +903,13 @@ func (gui *Gui) GetInitialKeybindings() []*Binding { }, { ViewName: "commits", + Contexts: []string{string(BRANCH_COMMITS_CONTEXT_KEY)}, + Key: gui.getKey(config.Commits.OpenInBrowser), + Handler: gui.handleOpenCommitInBrowser, + Description: gui.Tr.LcOpenCommitInBrowser, + }, + { + ViewName: "commits", Contexts: []string{string(REFLOG_COMMITS_CONTEXT_KEY)}, Key: gui.getKey(config.Universal.GoInto), Handler: gui.handleViewReflogCommitFiles, diff --git a/pkg/gui/pull_request_menu_panel.go b/pkg/gui/pull_request_menu_panel.go index 60fdee50c..dd59f99b6 100644 --- a/pkg/gui/pull_request_menu_panel.go +++ b/pkg/gui/pull_request_menu_panel.go @@ -57,7 +57,7 @@ func (gui *Gui) createPullRequestMenu(selectedBranch *models.Branch, checkedOutB func (gui *Gui) createPullRequest(from string, to string) error { pullRequest := commands.NewPullRequest(gui.GitCommand) - url, err := pullRequest.Create(from, to) + url, err := pullRequest.CreatePullRequest(from, to) if err != nil { return gui.surfaceError(err) } diff --git a/pkg/i18n/english.go b/pkg/i18n/english.go index f08addff7..aa2af0a3a 100644 --- a/pkg/i18n/english.go +++ b/pkg/i18n/english.go @@ -439,6 +439,7 @@ type TranslationSet struct { LcSelectBranch string CreatePullRequest string CreatingPullRequestAtUrl string + OpeningCommitInBrowser string SelectConfigFile string NoConfigFileFoundErr string LcLoadingFileSuggestions string @@ -454,6 +455,7 @@ type TranslationSet struct { ShowGitGraph string SortCommits string CantChangeContextSizeError string + LcOpenCommitInBrowser string Spans Spans } @@ -987,6 +989,7 @@ func englishTranslationSet() TranslationSet { LcDefaultBranch: "default branch", LcSelectBranch: "select branch", CreatingPullRequestAtUrl: "Creating pull request at URL: %s", + OpeningCommitInBrowser: "Opening commit in browser at URL: %s", SelectConfigFile: "Select config file", NoConfigFileFoundErr: "No config file found", LcLoadingFileSuggestions: "loading file suggestions", @@ -1002,6 +1005,7 @@ func englishTranslationSet() TranslationSet { ShowGitGraph: "show git graph", SortCommits: "commit sort order", CantChangeContextSizeError: "Cannot change context while in patch building mode because we were too lazy to support it when releasing the feature. If you really want it, please let us know!", + LcOpenCommitInBrowser: "open commit in browser", Spans: Spans{ // TODO: combine this with the original keybinding descriptions (those are all in lowercase atm) CheckoutCommit: "Checkout commit", |