Initial commit

Author: olevitt

.gitignore

@@ -0,0 +1,3 @@
+onyxia-cli
+go.sum
+.gitlab-ci.yml

README.md

@@ -0,0 +1,11 @@
+# Onyxia CLI
+
+## Work in progress  
+
+This is currently a work in progress.   
+
+## Enable autocompletion
+
+```
+PROG=onyxia source bash_autocomplete
+```

bash_autocomplete

@@ -0,0 +1,23 @@
+#! /bin/bash
+
+: ${PROG:=$(basename ${BASH_SOURCE})}
+
+_cli_bash_autocomplete() {
+  if [[ "${COMP_WORDS[0]}" != "source" ]]; then
+    local cur opts base
+    COMPREPLY=()
+    cur="${COMP_WORDS[COMP_CWORD]}"
+    if [[ "$cur" == "-"* ]]; then
+      opts=$( ${COMP_WORDS[@]:0:$COMP_CWORD} ${cur} --generate-bash-completion )
+    else
+      opts=$( ${COMP_WORDS[@]:0:$COMP_CWORD} --generate-bash-completion )
+      local IFS=$'\n'
+    fi
+    COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+    unset IFS
+    return 0
+  fi
+}
+
+complete -o bashdefault -o default -F _cli_bash_autocomplete $PROG
+unset PROG

config.yml

@@ -0,0 +1,2 @@
+API:
+  URL: "https://example.com"

go.mod

@@ -0,0 +1,9 @@
+module onyxia-cli
+
+go 1.13
+
+require (
+	github.com/kelseyhightower/envconfig v1.4.0
+	github.com/urfave/cli/v2 v2.1.1
+	gopkg.in/yaml.v2 v2.2.8
+)

main.go

@@ -0,0 +1,122 @@
+package main
+
+import (
+	"crypto/tls"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+
+	// "strings"
+
+	"onyxia-cli/src/api"
+	. "onyxia-cli/src/configuration"
+	"onyxia-cli/src/oidc"
+
+	"github.com/urfave/cli/v2"
+)
+
+func main() {
+	ReadConfig()
+	// Disable SSL validation
+	http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
+
+	app := &cli.App{
+		Version:              "v0.1",
+		Name:                 "onyxia-cli",
+		Usage:                "CLI for Onyxia services",
+		EnableBashCompletion: true,
+		Flags: []cli.Flag{
+			&cli.StringFlag{
+				Name:     "accessToken",
+				Aliases:  []string{"token"},
+				EnvVars:  []string{"KC_ACCESS_TOKEN"},
+				Required: true,
+			},
+		},
+		Commands: []*cli.Command{
+			{
+				Name:  "whoami",
+				Usage: "Show user's informations",
+				Action: func(c *cli.Context) error {
+					tokenString := c.String("accessToken")
+					oidc.DisplayID(tokenString)
+					return nil
+				},
+			},
+			{
+				Name:  "kub",
+				Usage: "Setup kubernetes",
+				Subcommands: []*cli.Command{
+					{
+						Name:  "create",
+						Usage: "Create resources",
+						Subcommands: []*cli.Command{
+							{
+								Name:  "namespace",
+								Usage: "Create namespace",
+								Flags: []cli.Flag{
+									&cli.StringFlag{
+										Name:     "namespaceName",
+										Aliases:  []string{"name", "n"},
+										Usage:    "Name of the namespace to create",
+										Required: true,
+									},
+									&cli.StringFlag{
+										Name:        "namespaceOwnerType",
+										Aliases:     []string{"ownerType", "t"},
+										Required:    false,
+										Usage:       "The type of ownership. Can be either USER or GROUP.",
+										DefaultText: "USER",
+										Value:       "USER",
+									},
+									&cli.StringFlag{
+										Name:        "namespaceOwner",
+										Aliases:     []string{"owner", "o"},
+										Required:    false,
+										Usage:       "The owner of the namespace.",
+										DefaultText: "YOU",
+									},
+								},
+								Action: func(c *cli.Context) error {
+									namespaceName := c.String("namespaceName")
+									tokenString := c.String("accessToken")
+									ownerType := c.String("namespaceOwnerType")
+									ownerName := c.String("namespaceOwner")
+									if ownerType == "USER" {
+										api.CreateNamespace(tokenString, namespaceName, ownerType, "oidc"+oidc.GetID(tokenString).Preferred_username)
+									} else {
+										if ownerName == "" {
+											fmt.Println("Veuillez choisir un groupe parmi ceux-ci :")
+											oidc.DisplayGroups(oidc.GetID(tokenString), false)
+										} else {
+											api.CreateNamespace(tokenString, namespaceName, ownerType, "oidc"+ownerName)
+										}
+									}
+									return nil
+								},
+								BashComplete: func(c *cli.Context) {
+									if c.NArg() > 0 {
+										return
+									}
+									if os.Args[len(os.Args)-2] == "-o" || os.Args[len(os.Args)-2] == "--owner" {
+										oidc.DisplayGroups(oidc.GetID(c.String("accessToken")), true)
+									} else if os.Args[len(os.Args)-2] == "-t" || os.Args[len(os.Args)-2] == "--ownerType" {
+										for _, elem := range []string{"GROUP", "USER"} {
+											fmt.Println(elem)
+										}
+									}
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	err := app.Run(os.Args)
+	if err != nil {
+		log.Fatal(err)
+	}
+}

src/api/api.go

@@ -0,0 +1,43 @@
+package api
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+
+	"onyxia-cli/src/configuration"
+)
+
+func CreateNamespace(token string, namespaceName string, ownerType string, ownerId string) {
+	request := &CreateNamespaceDTO{
+		Namespace: Namespace{
+			Id: namespaceName,
+		},
+		Owner: Owner{
+			Type: ownerType,
+			Id:   ownerId,
+		},
+	}
+	fmt.Printf("Creating namespace %s as %s %s\n", namespaceName, request.Owner.Type, request.Owner.Id)
+	buf := new(bytes.Buffer)
+	json.NewEncoder(buf).Encode(request)
+	req, _ := http.NewRequest("POST", configuration.Configuration.API.URL+"/api/namespace", buf)
+
+	req.Header.Add("Authorization", "Bearer "+token)
+	req.Header.Set("Content-Type", "application/json")
+
+	client := &http.Client{}
+	res, e := client.Do(req)
+	if e != nil {
+		panic(e)
+	}
+
+	defer res.Body.Close()
+
+	fmt.Println("response Status:", res.Status)
+	// Print the body to the stdout
+	io.Copy(os.Stdout, res.Body)
+}

src/api/createNamespaceDTO.go

@@ -0,0 +1,15 @@
+package api
+
+type CreateNamespaceDTO struct {
+	Owner     Owner     `json:"owner"`
+	Namespace Namespace `json:"namespace"`
+}
+
+type Owner struct {
+	Id   string `json:"id"`
+	Type string `json:"type"`
+}
+
+type Namespace struct {
+	Id string `json:"id"`
+}

src/configuration/config.go

@@ -0,0 +1,7 @@
+package configuration
+
+type Config struct {
+	API struct {
+		URL string `yaml:"URL", envconfig:"API_URL"`
+	} `yaml:"API"`
+}

src/configuration/readConfig.go

@@ -0,0 +1,43 @@
+package configuration
+
+import (
+	"fmt"
+	"os"
+
+	"gopkg.in/yaml.v2"
+
+	"github.com/kelseyhightower/envconfig"
+)
+
+var Configuration Config
+
+func ReadConfig() {
+	readFile(&Configuration)
+	readEnv(&Configuration)
+}
+
+func processError(err error) {
+	fmt.Println(err)
+	os.Exit(2)
+}
+
+func readFile(cfg *Config) {
+	f, err := os.Open("config.yml")
+	if err != nil {
+		processError(err)
+	}
+	defer f.Close()
+
+	decoder := yaml.NewDecoder(f)
+	err = decoder.Decode(cfg)
+	if err != nil {
+		processError(err)
+	}
+}
+
+func readEnv(cfg *Config) {
+	err := envconfig.Process("", cfg)
+	if err != nil {
+		processError(err)
+	}
+}