Added new command

Author: italolelis

cmd/create_test_repo.go

@@ -0,0 +1,101 @@
+package cmd
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"os/user"
+
+	"github.com/deiwin/interact"
+	"github.com/hellofresh/github-cli/pkg/config"
+	"github.com/hellofresh/github-cli/pkg/repo"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"gopkg.in/src-d/go-git.v4"
+	"gopkg.in/src-d/go-git.v4/plumbing/transport/ssh"
+	"gopkg.in/src-d/go-git.v4/storage/memory"
+)
+
+// RunCreateTestRepo runs the command to create a new hiring test repository
+func RunCreateTestRepo(cmd *cobra.Command, args []string) {
+	var err error
+	actor := interact.NewActor(os.Stdin, os.Stdout)
+	org, err := actor.PromptOptional("Please enter the org name", globalConfig.Github.Organization, checkNotEmpty)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	candidate, err := actor.Prompt("GitHub username of the candidate", checkNotEmpty)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	testRepo, err := actor.Prompt("Name of the repo with the test", checkNotEmpty)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	target := fmt.Sprintf("%s-%s", candidate, testRepo)
+	opts := &repo.HelloFreshRepoOpt{
+		Name:    target,
+		Org:     org,
+		Private: true,
+		Collaborators: &repo.CollaboratorsRule{
+			Enabled: true,
+			Collaborators: []*config.Collaborator{
+				&config.Collaborator{
+					Username:   candidate,
+					Permission: "push",
+				},
+			},
+		},
+	}
+	creator := repo.NewGithub(githubClient)
+	err = creator.Create(opts)
+	if err != nil {
+		log.WithError(err).Fatal("Could not create github repo for candidate")
+	}
+
+	if globalConfig.PublicKeyPath == "" {
+		user, _ := user.Current()
+		globalConfig.PublicKeyPath = fmt.Sprintf("%s/.ssh/id_rsa", user.HomeDir)
+	}
+
+	sshKey, err := ioutil.ReadFile(globalConfig.PublicKeyPath)
+	if err != nil {
+		log.WithError(err).Fatal("Error reading public key")
+	}
+
+	authMethod, err := ssh.NewPublicKeys("git", []byte(sshKey), "")
+	if err != nil {
+		log.WithError(err).Fatal("Error when creating public keys")
+	}
+
+	log.Info("Cloning repository...")
+	r, err := git.Clone(memory.NewStorage(), nil, &git.CloneOptions{
+		Auth:     authMethod,
+		Progress: os.Stdout,
+		URL:      fmt.Sprintf("git@github.com:%s/%s", org, testRepo),
+	})
+	if err != nil {
+		log.WithError(err).Fatal("Error cloning to repository")
+	}
+
+	log.Info("Changing remote...")
+	remote, err := r.Remote(git.DefaultRemoteName)
+	if err != nil {
+		log.WithError(err).Fatal("Error changing remote for repository")
+	}
+
+	log.Info("Pushing changes...")
+	remote.Config().URLs = []string{fmt.Sprintf("git@github.com:%s/%s", org, target)}
+	err = remote.Push(&git.PushOptions{
+		RemoteName: git.DefaultRemoteName,
+		Progress:   os.Stdout,
+	})
+	if err != nil {
+		log.WithError(err).Fatal("Error pushing to repository")
+	}
+
+	log.Infof("Done! Test for %s is created", candidate)
+}

cmd/root.go

@@ -1,16 +1,20 @@
 package cmd
 
 import (
+	"context"
 	"os"
 
+	"github.com/google/go-github/github"
 	"github.com/hellofresh/github-cli/pkg/config"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
+	"golang.org/x/oauth2"
 )
 
 var (
 	cfgFile      string
 	globalConfig *config.Spec
+	githubClient *github.Client
 	version      string
 	// RootCmd is our main command
 	RootCmd = &cobra.Command{
@@ -41,14 +45,31 @@ var (
 
 			// Only log the warning severity or above.
 			log.SetLevel(lvl)
+
+			if globalConfig.Github.Token == "" {
+				log.Fatal("You must provide a github token")
+			}
+
+			ts := oauth2.StaticTokenSource(
+				&oauth2.Token{AccessToken: globalConfig.Github.Token},
+			)
+			tc := oauth2.NewClient(context.Background(), ts)
+			githubClient = github.NewClient(tc)
 		},
 	}
 
 	createRepoCmd = &cobra.Command{
-		Use:   "create",
+		Use:   "create-repo",
 		Short: "Creates a new github repository",
 		Long:  `Creates a new github repository based on the rules defined on your .github.toml`,
-		Run:   RunCreate,
+		Run:   RunCreateRepo,
+	}
+
+	createTestCmd = &cobra.Command{
+		Use:   "create-test",
+		Short: "Creates a new hellofresh hiring test",
+		Long:  `Creates a new hellofresh hiring test based on the rules defined on your .github.toml`,
+		Run:   RunCreateTestRepo,
 	}
 )
 
@@ -63,4 +84,5 @@ func init() {
 	createRepoCmd.Flags().BoolVar(&createRepoFlags.HasWebhooks, "add-webhooks", true, "Enables webhooks configurations")
 	createRepoCmd.Flags().BoolVar(&createRepoFlags.HasBranchProtections, "add-branch-protections", true, "Enables branch protections")
 	RootCmd.AddCommand(createRepoCmd)
+	RootCmd.AddCommand(createTestCmd)
 }

pkg/config/config.go

@@ -7,9 +7,10 @@ import (
 type (
 	// Spec represents the global app configuration
 	Spec struct {
-		Github      Github
-		PullApprove PullApprove
-		LogLevel    string
+		PublicKeyPath string
+		Github        Github
+		PullApprove   PullApprove
+		LogLevel      string
 	}
 
 	// PullApprove represents teh Pull Approve configurations
@@ -21,12 +22,13 @@ type (
 
 	// Github represents the github configurations
 	Github struct {
-		Organization string
-		Token        string
-		Teams        []*Team
-		Labels       []*Label
-		Webhooks     []*Webhook
-		Protections  map[string][]string
+		Organization  string
+		Token         string
+		Teams         []*Team
+		Collaborators []*Collaborator
+		Labels        []*Label
+		Webhooks      []*Webhook
+		Protections   map[string][]string
 		// RemoveDefaultLabels Remove GitHub's default labels?
 		RemoveDefaultLabels bool
 	}
@@ -37,6 +39,12 @@ type (
 		Permission string
 	}
 
+	// Collaborator represents a github collaborator
+	Collaborator struct {
+		Username   string
+		Permission string
+	}
+
 	// Label represents a github label
 	Label struct {
 		Name  string