зеркало из https://github.com/mislav/hub.git
115 строки
2.6 KiB
Go
115 строки
2.6 KiB
Go
package commands
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"regexp"
|
|
|
|
"github.com/github/hub/v2/github"
|
|
"github.com/github/hub/v2/utils"
|
|
)
|
|
|
|
var cmdApply = &Command{
|
|
Run: apply,
|
|
GitExtension: true,
|
|
Usage: "apply <GITHUB-URL>",
|
|
Long: `Download a patch from GitHub and apply it locally.
|
|
|
|
## Options:
|
|
<GITHUB-URL>
|
|
A URL to a pull request or commit on GitHub.
|
|
|
|
## Examples:
|
|
$ hub apply https://github.com/jingweno/gh/pull/55
|
|
> curl https://github.com/jingweno/gh/pull/55.patch -o /tmp/55.patch
|
|
> git apply /tmp/55.patch
|
|
|
|
## See also:
|
|
|
|
hub-am(1), hub(1), git-apply(1)
|
|
`,
|
|
}
|
|
|
|
var cmdAm = &Command{
|
|
Run: apply,
|
|
GitExtension: true,
|
|
Usage: "am [-3] <GITHUB-URL>",
|
|
Long: `Replicate commits from a GitHub pull request locally.
|
|
|
|
## Options:
|
|
-3
|
|
(Recommended) See git-am(1).
|
|
|
|
<GITHUB-URL>
|
|
A URL to a pull request or commit on GitHub.
|
|
|
|
## Examples:
|
|
$ hub am -3 https://github.com/jingweno/gh/pull/55
|
|
> curl https://github.com/jingweno/gh/pull/55.patch -o /tmp/55.patch
|
|
> git am -3 /tmp/55.patch
|
|
|
|
## See also:
|
|
|
|
hub-apply(1), hub-cherry-pick(1), hub(1), git-am(1)
|
|
`,
|
|
}
|
|
|
|
func init() {
|
|
CmdRunner.Use(cmdApply)
|
|
CmdRunner.Use(cmdAm)
|
|
}
|
|
|
|
func apply(command *Command, args *Args) {
|
|
if !args.IsParamsEmpty() {
|
|
transformApplyArgs(args)
|
|
}
|
|
}
|
|
|
|
func transformApplyArgs(args *Args) {
|
|
gistRegexp := regexp.MustCompile("^https?://gist\\.github\\.com/([\\w.-]+/)?([a-f0-9]+)")
|
|
commitRegexp := regexp.MustCompile("^(commit|pull/[0-9]+/commits)/([0-9a-f]+)")
|
|
pullRegexp := regexp.MustCompile("^pull/([0-9]+)")
|
|
for idx, arg := range args.Params {
|
|
var (
|
|
patch io.ReadCloser
|
|
apiError error
|
|
)
|
|
projectURL, err := github.ParseURL(arg)
|
|
if err == nil {
|
|
gh := github.NewClient(projectURL.Project.Host)
|
|
if match := commitRegexp.FindStringSubmatch(projectURL.ProjectPath()); match != nil {
|
|
patch, apiError = gh.CommitPatch(projectURL.Project, match[2])
|
|
} else if match := pullRegexp.FindStringSubmatch(projectURL.ProjectPath()); match != nil {
|
|
patch, apiError = gh.PullRequestPatch(projectURL.Project, match[1])
|
|
}
|
|
} else {
|
|
match := gistRegexp.FindStringSubmatch(arg)
|
|
if match != nil {
|
|
// TODO: support Enterprise gist
|
|
gh := github.NewClient(github.GitHubHost)
|
|
patch, apiError = gh.GistPatch(match[2])
|
|
}
|
|
}
|
|
|
|
utils.Check(apiError)
|
|
if patch == nil {
|
|
continue
|
|
}
|
|
|
|
tempDir := os.TempDir()
|
|
err = os.MkdirAll(tempDir, 0775)
|
|
utils.Check(err)
|
|
patchFile, err := ioutil.TempFile(tempDir, "hub")
|
|
utils.Check(err)
|
|
|
|
_, err = io.Copy(patchFile, patch)
|
|
utils.Check(err)
|
|
|
|
patchFile.Close()
|
|
patch.Close()
|
|
|
|
args.ReplaceParam(idx, patchFile.Name())
|
|
}
|
|
}
|