2 // ========================LICENSE_START=================================
5 // Copyright (C) 2022: Nordix Foundation
7 // Licensed under the Apache License, Version 2.0 (the "License");
8 // you may not use this file except in compliance with the License.
9 // You may obtain a copy of the License at
11 // http://www.apache.org/licenses/LICENSE-2.0
13 // Unless required by applicable law or agreed to in writing, software
14 // distributed under the License is distributed on an "AS IS" BASIS,
15 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 // See the License for the specific language governing permissions and
17 // limitations under the License.
18 // ========================LICENSE_END===================================
28 "github.com/pkg/errors"
30 "helm.sh/helm/v3/pkg/action"
31 "helm.sh/helm/v3/pkg/chart"
32 "helm.sh/helm/v3/pkg/chart/loader"
33 "helm.sh/helm/v3/pkg/cli"
34 "helm.sh/helm/v3/pkg/getter"
35 "helm.sh/helm/v3/pkg/kube"
36 "helm.sh/helm/v3/pkg/repo"
38 metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
39 "k8s.io/cli-runtime/pkg/genericclioptions"
40 kubernetes "k8s.io/client-go/kubernetes"
41 "k8s.io/client-go/rest"
47 var settings *cli.EnvSettings
48 var chartRequested *chart.Chart
53 var releaseName string
56 type ChartInfo struct {
57 Name string `json:",omitempty"`
58 Namespace string `json:",omitempty"`
59 Revision int `json:",omitempty"`
60 Updated string `json:",omitempty"`
61 Status string `json:",omitempty"`
62 Chart string `json:",omitempty"`
63 AppVersion string `json:",omitempty"`
64 Values map[string]interface{} `json:"-"`
86 host = "postgres.default"
93 func runInstall(res http.ResponseWriter, req *http.Request) {
94 query := req.URL.Query()
95 chartName = query.Get("chart")
96 releaseName = chartName
97 fmt.Println("Installing ", chartName)
101 var install *action.Install
102 chartMuseumService, chartMuseumPort := findService("chartmuseum", "default")
103 fmt.Printf("Chart Museum service:%s, Port:%d\n", chartMuseumService, chartMuseumPort)
104 url := "http://" + chartMuseumService + ":" + fmt.Sprint(chartMuseumPort)
105 if !chartInstalled(chartName) {
107 fmt.Printf("Adding %s to Helm Repo\n", url)
108 _, err := addToRepo(url)
112 install, err = dryRun()
116 err := installSecurity(rapp)
120 fmt.Printf("Installing chart %s to %s namespace\n", chartName, namespace)
121 chart, err = installHelmChart(install)
123 msg = "Error occurred during installation " + err.Error()
125 msg = "Successfully installed release: " + chart
131 registrerRapp(chartName, rapp.Type)
133 msg = chartName + " has already been installed"
136 // create response binary data
137 data := []byte(msg) // slice of bytes
138 // write `data` to response
142 func installSecurity(rapp Rapp) error {
145 role := rapp.Roles[0].Role
146 grants := rapp.Roles[0].Grants[0]
148 client := rapp.Client
149 authenticator := rapp.Authenticator
151 if !rapp.SecurityEnabled {
154 // Different security requirements depending on the rapp type
155 if rapp.Type == "provider" {
156 // keycloak client setup
157 fmt.Println("Setting up keycloak")
158 url = "http://rapps-keycloak-mgr.default/create?"
159 params = "realm=" + realm + "&name=" + client + "&role=" + role + "&authType=" + authenticator
161 _, err := http.Get(url)
165 fmt.Println("Setting up istio")
166 url = "http://rapps-istio-mgr.default/create-policy?"
167 params = "name=" + chartName + "&realm=" + realm + "&role=" + role + "&method=" + grants
170 _, err := http.Get(url)
176 fmt.Println("Setting up istio")
177 url = "http://rapps-istio-mgr.default/create-filter?"
178 params = "name=" + chartName + "&realm=" + realm + "&client=" + client + "&authType=" + authenticator
180 _, err := http.Get(url)
190 func runUninstall(res http.ResponseWriter, req *http.Request) {
191 query := req.URL.Query()
192 chartName = query.Get("chart")
193 releaseName = chartName
194 fmt.Println("Uninstalling ", chartName)
198 if chartInstalled(chartName) {
199 err := getChartValues(chartName)
203 chart, err = uninstallHelmChart(chartName)
205 msg = "Error occurred during uninstall " + err.Error()
207 msg = "Successfully uninstalled release: " + chart
209 err := uninstallSecurity(rapp, chartName)
214 unregistrerRapp(chartName, rapp.Type)
216 msg = chartName + " is not installed"
219 // create response binary data
220 data := []byte(msg) // slice of bytes
221 // write `data` to response
225 func uninstallSecurity(rapp Rapp, chartName string) error {
228 role := rapp.Roles[0].Role
230 client := rapp.Client
231 authenticator := rapp.Authenticator
233 if !rapp.SecurityEnabled {
236 if rapp.Type == "provider" {
237 // Remove istio objects for rapp
238 fmt.Println("Removing istio services")
239 _, err := http.Get("http://rapps-istio-mgr.default/remove-policy?name=" + chartName)
243 // remove keycloak client
244 fmt.Println("Removing keycloak client")
245 url = "http://rapps-keycloak-mgr.default/remove?"
246 params = "name=" + client + "&realm=" + realm + "&role=" + role + "&authType=" + authenticator
248 _, err = http.Get(url)
253 if rapp.Type == "invoker" {
254 // Remove istio objects for rapp
255 fmt.Println("Removing istio services")
256 _, err := http.Get("http://rapps-istio-mgr.default/remove-filter?name=" + chartName)
264 func runList(res http.ResponseWriter, req *http.Request) {
266 // create response binary data
267 data, err := json.Marshal(chartInfo)
269 fmt.Printf("Error happened in JSON marshal. Err: %s\n", err)
271 // write `data` to response
276 //flag.StringVar(&url, "url", "http://chartmuseum:8080", "ChartMuseum url")
277 flag.StringVar(&repoName, "repoName", "local-dev", "Repository name")
278 flag.StringVar(&namespace, "namespace", "istio-nonrtric", "namespace for install")
282 runInstallHandler := http.HandlerFunc(runInstall)
283 http.Handle("/install", runInstallHandler)
284 runUninstallHandler := http.HandlerFunc(runUninstall)
285 http.Handle("/uninstall", runUninstallHandler)
286 runListHandler := http.HandlerFunc(runList)
287 http.Handle("/list", runListHandler)
288 http.ListenAndServe(":9000", nil)
291 func addToRepo(url string) (string, error) {
292 repoFile := settings.RepositoryConfig
294 //Ensure the file directory exists as it is required for file locking
295 err := os.MkdirAll(filepath.Dir(repoFile), os.ModePerm)
296 if err != nil && !os.IsExist(err) {
300 b, err := ioutil.ReadFile(repoFile)
301 if err != nil && !os.IsNotExist(err) {
306 if err := yaml.Unmarshal(b, &f); err != nil {
311 fmt.Printf("repository name (%s) already exists\n", repoName)
320 r, err := repo.NewChartRepository(&c, getter.All(settings))
325 if _, err := r.DownloadIndexFile(); err != nil {
326 err := errors.Wrapf(err, "looks like %q is not a valid chart repository or cannot be reached", url)
332 if err := f.WriteFile(repoFile, 0644); err != nil {
335 fmt.Printf("%q has been added to your repositories\n", repoName)
339 func dryRun() (*action.Install, error) {
340 actionConfig, err := getActionConfig(namespace)
342 install := action.NewInstall(actionConfig)
344 cp, err := install.ChartPathOptions.LocateChart(fmt.Sprintf("%s/%s", repoName, chartName), settings)
346 chartRequested, err = loader.Load(cp)
348 install.Namespace = namespace
349 install.ReleaseName = releaseName
350 install.DryRun = true
351 rel, err := install.Run(chartRequested, nil)
357 rappMap := rel.Chart.Values["rapp"]
358 // Convert map to json string
359 jsonStr, err := json.Marshal(rappMap)
365 if err := json.Unmarshal(jsonStr, &rapp); err != nil {
369 fmt.Printf("Keycloak key/value pairs in values.yaml - Realm: %s Client: %s Client Role: %s\n", rapp.Realm, rapp.Client, rapp.Roles)
373 func installHelmChart(install *action.Install) (string, error) {
375 install.DryRun = false
376 rel, err := install.Run(chartRequested, nil)
380 fmt.Println("Successfully installed release: ", rel.Name)
385 func getActionConfig(namespace string) (*action.Configuration, error) {
386 actionConfig := new(action.Configuration)
387 // Create the rest config instance with ServiceAccount values loaded in them
388 config, err := rest.InClusterConfig()
390 // fallback to kubeconfig
391 home, exists := os.LookupEnv("HOME")
395 kubeconfigPath := filepath.Join(home, ".kube", "config")
396 if envvar := os.Getenv("KUBECONFIG"); len(envvar) > 0 {
397 kubeconfigPath = envvar
399 if err := actionConfig.Init(kube.GetConfig(kubeconfigPath, "", namespace), namespace, os.Getenv("HELM_DRIVER"),
400 func(format string, v ...interface{}) {
401 fmt.Sprintf(format, v)
406 // Create the ConfigFlags struct instance with initialized values from ServiceAccount
407 var kubeConfig *genericclioptions.ConfigFlags
408 kubeConfig = genericclioptions.NewConfigFlags(false)
409 kubeConfig.APIServer = &config.Host
410 kubeConfig.BearerToken = &config.BearerToken
411 kubeConfig.CAFile = &config.CAFile
412 kubeConfig.Namespace = &namespace
413 if err := actionConfig.Init(kubeConfig, namespace, os.Getenv("HELM_DRIVER"), func(format string, v ...interface{}) {
414 fmt.Sprintf(format, v)
419 return actionConfig, err
422 func uninstallHelmChart(name string) (string, error) {
423 actionConfig, err := getActionConfig(namespace)
428 iCli := action.NewUninstall(actionConfig)
430 resp, err := iCli.Run(name)
434 fmt.Println("Successfully uninstalled release: ", resp.Release.Name)
435 return resp.Release.Name, err
438 func connectToK8s() *kubernetes.Clientset {
439 config, err := rest.InClusterConfig()
441 fmt.Println("failed to create K8s config")
444 clientset, err := kubernetes.NewForConfig(config)
446 fmt.Println("Failed to create K8s clientset")
452 func findService(serviceName, namespace string) (string, int32) {
453 clientset := connectToK8s()
454 svc, err := clientset.CoreV1().Services(namespace).Get(context.TODO(), serviceName, metav1.GetOptions{})
456 fmt.Println(err.Error())
458 return svc.Name, svc.Spec.Ports[0].Port
461 func list() []ChartInfo {
462 var charts = []ChartInfo{}
464 actionConfig, err := getActionConfig(namespace)
469 listAction := action.NewList(actionConfig)
470 releases, err := listAction.Run()
474 for _, release := range releases {
475 //fmt.Println("Release: " + release.Name + " Status: " + release.Info.Status.String())
476 chart.Name = release.Name
477 chart.Namespace = release.Namespace
478 chart.Revision = release.Version
479 chart.Updated = release.Info.LastDeployed.String()
480 chart.Status = release.Info.Status.String()
481 chart.Chart = release.Chart.Metadata.Name + "-" + release.Chart.Metadata.Version
482 chart.AppVersion = release.Chart.Metadata.AppVersion
483 chart.Values = release.Chart.Values
484 charts = append(charts, chart)
489 func chartInstalled(chartName string) bool {
491 for _, chart := range charts {
492 if chart.Name == chartName {
499 func getChartValues(chartName string) error {
501 for _, chart := range charts {
502 if chart.Name == chartName {
503 rappMap := chart.Values["rapp"]
504 fmt.Println("rappMap:", rappMap)
505 // Convert map to json string
506 jsonStr, err := json.Marshal(rappMap)
508 fmt.Println("Error:", err)
512 if err := json.Unmarshal(jsonStr, &rapp); err != nil {
513 fmt.Println("Error:", err)
519 return errors.New("Chart: cannot retrieve values")
522 func registrerRapp(chartName, chartType string) {
523 psqlconn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname)
525 db, err := sql.Open("postgres", psqlconn)
529 fmt.Println("Connected!")
536 createStmt := `CREATE TABLE IF NOT EXISTS services (
537 id serial PRIMARY KEY,
538 name VARCHAR ( 50 ) UNIQUE NOT NULL,
539 type VARCHAR ( 50 ) NOT NULL,
540 created_on TIMESTAMP DEFAULT NOW()
542 _, err = db.Exec(createStmt)
546 fmt.Println("Created table for service registry")
550 insertDynStmt := `insert into "services"("name", "type") values($1, $2)`
551 _, err = db.Exec(insertDynStmt, chartName, chartType)
555 fmt.Println("Inserted " + chartName + " into service registry")
559 func unregistrerRapp(chartName, chartType string) {
560 psqlconn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname)
562 db, err := sql.Open("postgres", psqlconn)
566 fmt.Println("Connected!")
572 deleteDynStmt := `delete from services where name=$1 and type=$2`
573 _, err = db.Exec(deleteDynStmt, chartName, chartType)
577 fmt.Println("Deleted " + chartName + " from service registry")