persist migration data (#136)

Author: oren-codefresh

package.json

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

venonactl/VERSION

@@ -1 +1 @@
-1.2.12
+1.2.13

venonactl/cmd/migrate.go

@@ -35,7 +35,7 @@ var migrateCmd = &cobra.Command{
 	Use:   "migrate",
 	Short: "Migrate existing runtime-environment from 0.X to 1.X version",
 	Run: func(cmd *cobra.Command, args []string) {
-		lgr := createLogger("Migrate", true)
+		lgr := createLogger("Migrate", verbose)
 		builder := plugins.NewBuilder(lgr)
 		builder.Add(plugins.VenonaPluginType)
 		s := store.GetStore()
@@ -45,6 +45,9 @@ var migrateCmd = &cobra.Command{
 		extendStoreWithAgentAPI(lgr, "", "")
 		fillKubernetesAPI(lgr, migrateCmdOpt.kube.context, migrateCmdOpt.kube.namespace, false)
 		values := s.BuildValues()
+		spn := createSpinner("Migrating runtime (might take a few seconds)", "")
+		spn.Start()
+		defer spn.Stop()
 		for _, p := range builder.Get() {
 			err := p.Migrate( &plugins.MigrateOptions{
 				ClusterNamespace: migrateCmdOpt.kube.namespace,

venonactl/pkg/plugins/venona.go

@@ -18,8 +18,10 @@ package plugins
 
 import (
 	"encoding/base64"
+	"encoding/json"
 	"errors"
 	"fmt"
+	"io/ioutil"
 	"time"
 
 	"github.com/codefresh-io/go-sdk/pkg/codefresh"
@@ -28,13 +30,20 @@ import (
 	templates "github.com/codefresh-io/venona/venonactl/pkg/templates/kubernetes"
 	kerrors "k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/api/core/v1"
+	
 )
 
 // venonaPlugin installs assets on Kubernetes Dind runtimectl Env
 type venonaPlugin struct {
 	logger logger.Logger
 }
 
+type migrationData struct {
+	NodeSelector map[string]string `json:"nodeSelector,omitempty"`
+	Tolerations [] v1.Toleration `json:"tolerations,omitempty"`
+}
+
 const (
 	venonaFilesPattern = ".*.venona.yaml"
 )
@@ -248,6 +257,21 @@ func (u *venonaPlugin) Migrate(opt *MigrateOptions, v Values) error {
 		u.logger.Error(fmt.Sprintf("Cannot find agent pod: %v ", err))
 		return err
 	}
+	if (len(list.Items) == 0) {
+		u.logger.Debug("Runner pod not found , existing migration")
+		return nil
+	}
+	migrationData := migrationData{
+		Tolerations: list.Items[0].Spec.Tolerations,
+		NodeSelector: list.Items[0].Spec.NodeSelector,
+	}
+	var jsonData []byte
+	jsonData, err = json.Marshal(migrationData)
+	err = ioutil.WriteFile("migration.json", jsonData, 0644)
+	if (err != nil) {
+		u.logger.Error("Cannot write migration json")
+	}
+
 	podName := list.Items[0].ObjectMeta.Name
 	for fileName := range kubeObjects {
 		if _, ok := deletePriorUpgrade[fileName]; ok {