implement upgrade

* update version to 0.9.0

* fix lisence

Author: Oleg Sucharevich

package.json

@@ -1,6 +1,6 @@
 {
   "name": "venona",
-  "version": "0.8.1",
+  "version": "0.9.0",
   "description": "Codefresh agent to run on Codefresh's runtime environment and execute pipeline",
   "main": "index.js",
   "scripts": {

venonactl/cmd/upgrade.go

@@ -0,0 +1,57 @@
+package cmd
+
+/*
+Copyright 2019 The Codefresh Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+import (
+	"errors"
+
+	"github.com/codefresh-io/venona/venonactl/pkg/operators"
+	"github.com/codefresh-io/venona/venonactl/pkg/store"
+	"github.com/spf13/cobra"
+)
+
+// upgradeCmd represents the upgrade command
+var upgradeCmd = &cobra.Command{
+	Use:   "upgrade [name]",
+	Short: "Upgrade existing runtime-environment",
+	Args: func(cmd *cobra.Command, args []string) error {
+		if len(args) < 1 {
+			return errors.New("requires name of the runtime-environment")
+		}
+
+		if len(args) > 1 {
+			return errors.New("Cannot upgrade multiple runtimes once")
+		}
+		return nil
+	},
+	Run: func(cmd *cobra.Command, args []string) {
+		s := store.GetStore()
+		kubeContextFlag := cmd.Flag("kube-context-name")
+		re, _ := s.CodefreshAPI.Client.RuntimeEnvironments().Get(args[0])
+		contextName := re.RuntimeScheduler.Cluster.ClusterProvider.Selector
+		if kubeContextFlag != nil {
+			contextName = kubeContextFlag.Value.String()
+		}
+		s.KubernetesAPI.ContextName = contextName
+		s.KubernetesAPI.Namespace = re.RuntimeScheduler.Cluster.Namespace
+		operators.GetOperator(operators.VenonaOperatorType).Upgrade()
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(upgradeCmd)
+}

venonactl/pkg/obj/kubeobj/kubeobj.go

@@ -60,7 +60,7 @@ var functionsMap map[string]string = map[string]string{
 	"rbacv1.Role":               "RbacV1().Roles(namespace)",
 	"rbacv1.RoleBinding":        "RbacV1().RoleBindings(namespace)",
 
-	"storagev1.StorageClass":    "StorageV1().StorageClasses()",
+	"storagev1.StorageClass": "StorageV1().StorageClasses()",
 }
 
 var packageTemplate = template.Must(template.New("").Parse(
@@ -141,6 +141,23 @@ func DeleteObject(clientset *kubernetes.Clientset, obj runtime.Object, namespace
     return name, kind, err
 }
 
+// ReplaceObject - replaces kubernetes object from *runtime.Object. Returns object name, kind and creation error
+func ReplaceObject(clientset *kubernetes.Clientset, obj runtime.Object, namespace string) (string, string, error){
+	var name, kind string
+	var err error
+	switch objT := obj.(type) {
+    {{ range $key, $value := .FunctionsMap }}
+    case *{{ $key }}:
+        name = objT.ObjectMeta.Name
+		kind = objT.TypeMeta.Kind
+        _, err = clientset.{{ $value }}.Update(objT)
+    {{ end }}
+    default:
+        return "", "", fmt.Errorf("Unknown object type %T\n ", objT)
+    }
+    return name, kind, err
+}
+
 `))
 
 type tempateData struct {

venonactl/pkg/obj/kubeobj_generator.go

@@ -24,6 +24,7 @@ import (
 	"github.com/sirupsen/logrus"
 
 	"github.com/codefresh-io/venona/venonactl/pkg/store"
+	templates "github.com/codefresh-io/venona/venonactl/pkg/templates/kubernetes"
 
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/client-go/kubernetes"
@@ -125,3 +126,8 @@ func NewKubeClientset(s *store.Values) (*kubernetes.Clientset, error) {
 	}
 	return kubernetes.NewForConfig(kubeClientConfig)
 }
+
+func getKubeObjectsFromTempalte(values map[string]interface{}) (map[string]runtime.Object, error) {
+	templatesMap := templates.TemplatesMap()
+	return KubeObjectsFromTemplates(templatesMap, values)
+}

venonactl/pkg/operators/helper.go

@@ -10,6 +10,7 @@ type (
 		Install() error
 		Status() ([][]string, error)
 		Delete() error
+		Upgrade() error
 	}
 )
 

venonactl/pkg/operators/operator.go

@@ -174,3 +174,7 @@ func (u *RuntimeEnvironmentOperator) Delete() error {
 	}
 	return nil
 }
+
+func (u *RuntimeEnvironmentOperator) Upgrade() error {
+	return nil
+}

venonactl/pkg/operators/runtime-environment.go

@@ -207,3 +207,63 @@ func (u *venonaOperator) Delete() error {
 	}
 	return nil
 }
+
+func (u *venonaOperator) Upgrade() error {
+
+	// replace of sa creates new secert with sa creds
+	// avoid it till patch fully implemented
+	var skipUpgradeFor = map[string]interface{}{
+		"service-account.venona.yaml": nil,
+	}
+
+	var err error
+	s := store.GetStore()
+
+	kubeClientset, err := NewKubeClientset(s)
+	if err != nil {
+		logrus.Errorf("Cannot create kubernetes clientset: %v\n ", err)
+		return err
+	}
+
+	namespace := s.KubernetesAPI.Namespace
+
+	// special case when we need to get the token from the remote to no regenrate it
+	// whole flow should be more like kubectl apply that build a patch
+	// based on remote object and candidate object
+	secret, err := kubeClientset.CoreV1().Secrets(namespace).Get(s.AppName, metav1.GetOptions{})
+	if err != nil {
+		return err
+	}
+	token := secret.Data["codefresh.token"]
+	s.AgentToken = string(token)
+
+	kubeObjects, err := getKubeObjectsFromTempalte(s.BuildValues())
+	if err != nil {
+		return err
+	}
+
+	for fileName, local := range kubeObjects {
+		match, _ := regexp.MatchString(venonaInstallPattern, fileName)
+		if match != true {
+			logrus.WithFields(logrus.Fields{
+				"Operator": VenonaOperatorType,
+				"Pattern":  venonaInstallPattern,
+			}).Debugf("Skipping upgrade of %s: pattern not match", fileName)
+			continue
+		}
+
+		if _, ok := skipUpgradeFor[fileName]; ok {
+			logrus.WithFields(logrus.Fields{
+				"Operator": VenonaOperatorType,
+			}).Debugf("Skipping upgrade of %s: should be ignored", fileName)
+			continue
+		}
+
+		_, _, err := kubeobj.ReplaceObject(kubeClientset, local, namespace)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}