feat(releaser): add support for custom krew index repo

Signed-off-by: Casale, Robert <robert.casale@fmr.com>

Author: Gearheads

README.md

@@ -70,10 +70,12 @@ $ docker run -v /path/to/your/template-file.yaml:/tmp/template-file.yaml ghcr.io
 
 # Inputs for the action
 
-| Key                | Default Value          | Description                                                                          |
-| ------------------ | ---------------------- | ------------------------------------------------------------------------------------ |
-| workdir            | `env.GITHUB_WORKSPACE` | Overrides the GitHub workspace directory path                                        |
-| krew_template_file | `.krew.yaml`           | The path to template file relative to $workdir. e.g. templates/misc/plugin-name.yaml |
+| Key                            | Default Value          | Description                                                                          |
+| ------------------------------ | ---------------------- | ------------------------------------------------------------------------------------ |
+| workdir                        | `env.GITHUB_WORKSPACE` | Overrides the GitHub workspace directory path                                        |
+| krew_template_file             | `.krew.yaml`           | The path to template file relative to $workdir. e.g. templates/misc/plugin-name.yaml |
+| upstream_krew_index_repo_owner | `kubernetes-sigs`      | Overrides the GitHub owner of the Krew index, defaults to kubernetes-sigs            |
+| upstream_krew_index_repo_name  | `krew-index`           | Overrides the GitHub repository of the Krew index, defaults to krew-index            |
 
 # Limitations of krew-release-bot
 

action.yml

@@ -9,3 +9,11 @@ inputs:
     description: "Working directory, defaults to env.GITHUB_WORKSPACE"
   krew_template_file:
     description: "the path to template file relative to $workdir. e.g. templates/misc/plugin-name.yaml. defaults to .krew.yaml"
+  upstream_krew_index_repo_owner:
+    description: 'The owner of the Krew index repository'
+    required: false
+    default: kubernetes-sigs
+  upstream_krew_index_repo_name:
+    description: 'The name of the Krew index repository'
+    required: false
+    default: krew-index

pkg/krew/repos.go

@@ -9,7 +9,7 @@ const (
 
 //GetKrewIndexRepoName returns the krew-index repo name
 func GetKrewIndexRepoName() string {
-	override := os.Getenv("UPSTREAM_KREW_INDEX_REPO_NAME")
+	override := os.Getenv("INPUT_UPSTREAM_KREW_INDEX_REPO_NAME")
 	if override != "" {
 		return override
 	}
@@ -19,7 +19,7 @@ func GetKrewIndexRepoName() string {
 
 //GetKrewIndexRepoOwner returns the krew-index repo owner
 func GetKrewIndexRepoOwner() string {
-	override := os.Getenv("UPSTREAM_KREW_INDEX_REPO_OWNER")
+	override := os.Getenv("INPUT_UPSTREAM_KREW_INDEX_REPO_OWNER")
 	if override != "" {
 		return override
 	}

pkg/releaser/git.go

@@ -30,10 +30,16 @@ const (
 // CloneRepos clones the repo
 func (r *Releaser) cloneRepos(dir string, request *source.ReleaseRequest) (*ugit.Repository, error) {
 	logrus.Infof("Cloning %s", r.UpstreamKrewIndexRepoCloneURL)
+	client := getGitHubClient(r.Token)
+	remoteRepo, err := r.getRepo(client, r.UpstreamKrewIndexRepoOwner, r.UpstreamKrewIndexRepo)
+	if err != nil {
+		return nil, err
+	}
+
 	repo, err := ugit.PlainClone(dir, false, &ugit.CloneOptions{
 		URL:           r.UpstreamKrewIndexRepoCloneURL,
 		Progress:      os.Stdout,
-		ReferenceName: plumbing.Master,
+		ReferenceName: plumbing.ReferenceName(*remoteRepo.DefaultBranch),
 		SingleBranch:  true,
 		Auth:          r.getAuth(),
 		RemoteName:    OriginNameUpstream,
@@ -123,23 +129,39 @@ func getPushRefSpec(branchName string) string {
 	return fmt.Sprintf("refs/heads/%s:refs/heads/%s", branchName, branchName)
 }
 
+func getGitHubClient(token string) *github.Client {
+	ts := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: token})
+	tc := oauth2.NewClient(context.TODO(), ts)
+	return github.NewClient(tc)
+}
+
+func (r *Releaser) getRepo(client *github.Client, owner string, repoName string) (*github.Repository, error) {
+	repo, _, err := client.Repositories.Get(context.TODO(), owner, repoName)
+    if err != nil {
+        return nil, err
+    }
+	return repo, nil
+}
+
 // SubmitPR submits the PR
 func (r *Releaser) submitPR(request *source.ReleaseRequest) (string, error) {
-	ts := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: r.Token})
-	tc := oauth2.NewClient(context.TODO(), ts)
-	client := github.NewClient(tc)
+	client := getGitHubClient(r.Token)
+	repo, err := r.getRepo(client, r.UpstreamKrewIndexRepoOwner, r.UpstreamKrewIndexRepo)
+	if err != nil {
+		return "", err
+	}
 
 	prr := &github.NewPullRequest{
 		Title: r.getTitle(request),
 		Head:  r.getHead(request),
-		Base:  github.String("master"),
+		Base:  github.String(*repo.DefaultBranch),
 		Body:  r.getPRBody(request),
 	}
 
 	logrus.Infof("creating pr with title %q, \nhead %q, \nbase %q, \nbody %q",
 		github.Stringify(r.getTitle(request)),
 		github.Stringify(r.getHead(request)),
-		"master",
+		github.Stringify(*repo.DefaultBranch),
 		github.Stringify(r.getPRBody(request)),
 	)