hub/commands/cherry_pick.go

76 строки
2.0 KiB
Go

package commands
import (
"github.com/jingweno/gh/github"
"regexp"
)
var cmdCherryPick = &Command{
Run: cherryPick,
GitExtension: true,
Usage: "cherry-pick GITHUB-REF",
Short: "Apply the changes introduced by some existing commits",
Long: `Cherry-pick a commit from a fork using either full URL to the commit
or GitHub-flavored Markdown notation, which is user@sha. If the remote
doesn't yet exist, it will be added. A git-fetch(1) user is issued
prior to the cherry-pick attempt.
`,
}
/*
$ gh cherry-pick https://github.com/jingweno/gh/commit/a319d88#comments
> git remote add -f jingweno git://github.com/jingweno/gh.git
> git cherry-pick a319d88
$ gh cherry-pick jingweno@a319d88
> git remote add -f jingweno git://github.com/jingweno/gh.git
> git cherry-pick a319d88
$ gh cherry-pick jingweno@SHA
> git fetch jingweno
> git cherry-pick SHA
*/
func cherryPick(command *Command, args *Args) {
if args.IndexOfParam("-m") == -1 && args.IndexOfParam("--mainline") == -1 {
transformCherryPickArgs(args)
}
}
func transformCherryPickArgs(args *Args) {
ref := args.LastParam()
project, sha := parseCherryPickProjectAndSha(ref)
if project != nil {
args.ReplaceParam(args.IndexOfParam(ref), sha)
if hasGitRemote(project.Owner) {
args.Before("git", "fetch", project.Owner)
} else {
args.Before("git", "remote", "add", "-f", project.Owner, project.GitURL("", "", false))
}
}
}
func parseCherryPickProjectAndSha(ref string) (project *github.Project, sha string) {
url, err := github.ParseURL(ref)
if err == nil {
commitRegex := regexp.MustCompile("^commit\\/([a-f0-9]{7,40})")
projectPath := url.ProjectPath()
if commitRegex.MatchString(projectPath) {
sha = commitRegex.FindStringSubmatch(projectPath)[1]
project = &url.Project
return
}
}
ownerWithShaRegexp := regexp.MustCompile("^(%s)@([a-f0-9]{7,40})$")
if ownerWithShaRegexp.MatchString(ref) {
matches := ownerWithShaRegexp.FindStringSubmatch(ref)
sha = matches[2]
project = github.CurrentProject()
project.Owner = matches[1]
}
return
}