hub/commands/fork.go

141 строка
3.5 KiB
Go

package commands
import (
"fmt"
"github.com/github/hub/github"
"github.com/github/hub/ui"
"github.com/github/hub/utils"
)
var cmdFork = &Command{
Run: fork,
Usage: "fork [--no-remote] [--remote-name=<REMOTE>] [--org=<ORGANIZATION>]",
Long: `Fork the current project on GitHub and add a git remote for it.
## Options:
--no-remote
Skip adding a git remote for the fork.
--org=<ORGANIZATION>
Fork the repository within this organization.
## Examples:
$ hub fork
[ repo forked on GitHub ]
> git remote add -f USER git@github.com:USER/REPO.git
$ hub fork --org=ORGANIZATION
[ repo forked on Github into the ORGANIZATION organization]
> git remote add -f ORGANIZATION git@github.com:ORGANIZATION/REPO.git
## See also:
hub-clone(1), hub(1)
`,
}
var (
flagForkNoRemote bool
flagForkOrganization string
flagForkRemoteName string
)
func init() {
cmdFork.Flag.BoolVar(&flagForkNoRemote, "no-remote", false, "")
cmdFork.Flag.StringVarP(&flagForkRemoteName, "remote-name", "", "", "REMOTE")
cmdFork.Flag.StringVarP(&flagForkOrganization, "org", "", "", "ORGANIZATION")
CmdRunner.Use(cmdFork)
}
func fork(cmd *Command, args *Args) {
localRepo, err := github.LocalRepo()
utils.Check(err)
project, err := localRepo.MainProject()
utils.Check(err)
config := github.CurrentConfig()
host, err := config.PromptForHost(project.Host)
utils.Check(github.FormatError("forking repository", err))
originRemote, err := localRepo.MainRemote()
utils.Check(err)
params := map[string]interface{}{}
forkOwner := host.User
if flagForkOrganization != "" {
forkOwner = flagForkOrganization
params["organization"] = forkOwner
}
forkProject := github.NewProject(forkOwner, project.Name, project.Host)
var newRemoteName string
if flagForkRemoteName != "" {
newRemoteName = flagForkRemoteName
} else {
newRemoteName = forkProject.Owner
}
client := github.NewClient(project.Host)
existingRepo, err := client.Repository(forkProject)
if err == nil {
existingProject, err := github.NewProjectFromRepo(existingRepo)
if err == nil && !existingProject.SameAs(forkProject) {
existingRepo = nil
}
}
if err == nil && existingRepo != nil {
var parentURL *github.URL
if parent := existingRepo.Parent; parent != nil {
parentURL, _ = github.ParseURL(parent.HtmlUrl)
}
if parentURL == nil || !project.SameAs(parentURL.Project) {
err = fmt.Errorf("Error creating fork: %s already exists on %s",
forkProject, forkProject.Host)
utils.Check(err)
}
} else {
if !args.Noop {
newRepo, err := client.ForkRepository(project, params)
utils.Check(err)
forkProject.Owner = newRepo.Owner.Login
forkProject.Name = newRepo.Name
}
}
args.NoForward()
if !flagForkNoRemote {
originURL := originRemote.URL.String()
url := forkProject.GitURL("", "", true)
// Check to see if the remote already exists.
currentRemote, err := localRepo.RemoteByName(newRemoteName)
if err == nil {
currentProject, err := currentRemote.Project()
if err == nil {
if currentProject.SameAs(forkProject) {
ui.Printf("existing remote: %s\n", newRemoteName)
return
}
if newRemoteName == "origin" {
// Assume user wants to follow github guides for collaboration
ui.Printf("renaming existing \"origin\" remote to \"upstream\"\n")
args.Before("git", "remote", "rename", "origin", "upstream")
}
}
}
args.Before("git", "remote", "add", "-f", newRemoteName, originURL)
args.Before("git", "remote", "set-url", newRemoteName, url)
args.AfterFn(func() error {
ui.Printf("new remote: %s\n", newRemoteName)
return nil
})
}
}