feat: use gh to derive the namespace (#3)

Author: leafty

pkg/cmd/copykeycloakadminpassword.go

@@ -5,10 +5,11 @@ import (
 	"fmt"
 	"os"
 
+	"github.com/SwissDataScienceCenter/renku-dev-utils/pkg/github"
 	"github.com/SwissDataScienceCenter/renku-dev-utils/pkg/k8s"
 	"github.com/spf13/cobra"
 	"golang.design/x/clipboard"
-	"k8s.io/apimachinery/pkg/apis/meta/v1"
+	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
 // Flags
@@ -27,6 +28,38 @@ var copyKeycloakAdminPasswordCmd = &cobra.Command{
 func runCopyKeycloakAdminPassword(cmd *cobra.Command, args []string) {
 	ctx := context.Background()
 
+	if namespace == "" {
+		cli, err := github.NewGitHubCLI("")
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+
+		repo, err := cli.GetCurrentRepository()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		fmt.Printf("Repo: %s", repo)
+		fmt.Println()
+
+		prNumber, err := cli.GetCurrentPullRequest()
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		fmt.Printf("Pull request: %d", prNumber)
+		fmt.Println()
+
+		namespace, err = github.DeriveK8sNamespace(repo, prNumber)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+		fmt.Printf("Derived namespace: %s", namespace)
+		fmt.Println()
+	}
+
 	clients, err := k8s.GetClientset()
 	if err != nil {
 		fmt.Println(err)
@@ -56,7 +89,7 @@ func runCopyKeycloakAdminPassword(cmd *cobra.Command, args []string) {
 }
 
 func init() {
-	copyKeycloakAdminPasswordCmd.Flags().StringVarP(&namespace, "namespace", "n", "renku", "k8s namespace")
+	copyKeycloakAdminPasswordCmd.Flags().StringVarP(&namespace, "namespace", "n", "", "k8s namespace")
 	copyKeycloakAdminPasswordCmd.Flags().StringVar(&secretName, "secret-name", "keycloak-password-secret", "secret name")
 	copyKeycloakAdminPasswordCmd.Flags().StringVar(&secretKey, "secret-key", "KEYCLOAK_ADMIN_PASSWORD", "secret key")
 }

pkg/github/cli.go

@@ -0,0 +1,30 @@
+package github
+
+import (
+	"fmt"
+	"os/exec"
+)
+
+type GitHubCLI struct {
+	gh string
+}
+
+func NewGitHubCLI(gh string) (*GitHubCLI, error) {
+	if gh == "" {
+		gh = "gh"
+	}
+
+	path, err := exec.LookPath(gh)
+	if err != nil {
+		return nil, err
+	}
+
+	fmt.Printf("Found GitHub CLI: %s", path)
+	fmt.Println()
+	return &GitHubCLI{gh: gh}, nil
+}
+
+func (cli *GitHubCLI) RunCmd(arg ...string) ([]byte, error) {
+	cmd := exec.Command(cli.gh, arg...)
+	return cmd.Output()
+}

pkg/github/pullrequest.go

@@ -0,0 +1,22 @@
+package github
+
+import "encoding/json"
+
+func (cli *GitHubCLI) GetCurrentPullRequest() (int, error) {
+	out, err := cli.RunCmd("pr", "view", "--json", "number")
+	if err != nil {
+		return 0, err
+	}
+
+	var res gitHubPRViewOutput
+	err = json.Unmarshal(out, &res)
+	if err != nil {
+		return 0, err
+	}
+
+	return res.Number, nil
+}
+
+type gitHubPRViewOutput struct {
+	Number int `json:"number"`
+}

pkg/github/repository.go

@@ -0,0 +1,22 @@
+package github
+
+import "encoding/json"
+
+func (cli *GitHubCLI) GetCurrentRepository() (string, error) {
+	out, err := cli.RunCmd("repo", "view", "--json", "nameWithOwner")
+	if err != nil {
+		return "", err
+	}
+
+	var res gitHubRepoViewOutput
+	err = json.Unmarshal(out, &res)
+	if err != nil {
+		return "", err
+	}
+
+	return res.NameWithOwner, nil
+}
+
+type gitHubRepoViewOutput struct {
+	NameWithOwner string `json:"nameWithOwner"`
+}

pkg/github/utils.go

@@ -0,0 +1,26 @@
+package github
+
+import "fmt"
+
+// Maps repositories to namespace templates
+var repoToNamespaceTemplateMap map[string]string
+
+func DeriveK8sNamespace(repo string, pr int) (string, error) {
+	tpl, found := repoToNamespaceTemplateMap[repo]
+	if found {
+		return fmt.Sprintf(tpl, pr), nil
+	}
+	return "", fmt.Errorf("Could not derive namespace from repository: %s", repo)
+}
+
+func init() {
+	initRepoToNamespaceTemplateMap()
+}
+
+func initRepoToNamespaceTemplateMap() {
+	repoToNamespaceTemplateMap = map[string]string{
+		"SwissDataScienceCenter/renku":               "ci-renku-%d",
+		"SwissDataScienceCenter/renku-data-services": "renku-ci-ds-%d",
+		"SwissDataScienceCenter/renku-ui":            "renku-ci-ui-%d",
+	}
+}