Storage functions for the migration phases

api/external/infra_proxy/response/orgs.proto

@@ -29,8 +29,8 @@ message GetOrg {
 }
 
 message GetInfraServerOrgs {
-    // Chef organization list from chef server
-    repeated Org orgs = 1;
+    // id of organisation migration
+    string migration_id = 1;
 }
 
 message Org {

api/interservice/infra_proxy/response/orgs.proto

@@ -29,8 +29,8 @@ message GetOrg {
 }
 
 message GetInfraServerOrgs {
-    // Chef organization list from chef server
-    repeated Org orgs = 1;
+    // id of organisation migration
+    string migration_id = 1;
 }
 
 message Org {

components/automate-gateway/handler/infra_proxy/orgs.go

@@ -118,7 +118,7 @@ func (c *InfraProxyServer) GetInfraServerOrgs(ctx context.Context, r *gwreq.GetI
 		return nil, err
 	}
 	return &gwres.GetInfraServerOrgs{
-		Orgs: fromUpstreamOrgs(res.Orgs),
+		MigrationId: res.MigrationId,
 	}, nil
 }
 func fromUpstreamOrg(t *infra_res.Org) *gwres.Org {

components/infra-proxy-service/constants/constants.go

@@ -1,6 +1,48 @@
 package constants
 
+// MigrationPhase: enum of migration phases
+type MigrationPhase int64
+
+// Migration phases id should be between 100 & 5000
+const (
+	StartMigration    = 100
+	CompleteMigration = 5000
+)
+
+// MigrationStatus: enum of migration status
+type MigrationStatus int64
+
 // IAM default project ID
 const (
 	UnassignedProjectID = "(unassigned)"
 )
+
+const (
+	StartFileUpload MigrationPhase = iota + 1000
+	CompleteFileUpload
+	FailedFileUpload
+	StartUnzip
+	ComplteUnzip
+	FailedUnzip
+	StartZipParsing
+	CompleteZipParsing
+	FailedZipParsing
+	StartOrgMigration
+	CompleteOrgMigration
+	FailedOrgMigration
+	StartUserMigration
+	CompleteUserMigration
+	FailedUserMigration
+	StartAssciation
+	CompleteAssciation
+	FailedAssciation
+	StartPermissionMigration
+	CompletePermissionMigration
+	FailedPermissionMigration
+)
+
+const (
+	InProgress MigrationStatus = iota + 100
+	Completed
+	Failed
+)

components/infra-proxy-service/server/migration.go

@@ -0,0 +1,38 @@
+package server
+
+import (
+	"context"
+	"errors"
+
+	secrets "github.com/chef/automate/api/external/secrets"
+	"github.com/chef/automate/components/infra-proxy-service/service"
+)
+
+var IsMigrationAlreadyRunning bool
+
+func setMigrationStatus(status bool) {
+	IsMigrationAlreadyRunning = status
+}
+
+// getChefClient: creates the chef client
+func (s *Server) getChefClient(ctx context.Context, serverId string) (*ChefClient, error) {
+	// Get the credential ID from servers table
+	server, err := s.service.Storage.GetServer(ctx, serverId)
+	if err != nil {
+		return nil, service.ParseStorageError(err, serverId, "server")
+	}
+	if server.CredentialID == "" {
+		return nil, errors.New("webui key is not available with server")
+	}
+	// Get web ui key from secrets service
+	secret, err := s.service.Secrets.Read(ctx, &secrets.Id{Id: server.CredentialID})
+	if err != nil {
+		return nil, err
+	}
+
+	c, err := s.createChefServerClient(ctx, serverId, GetAdminKeyFrom(secret), "pivotal", true)
+	if err != nil {
+		return nil, err
+	}
+	return c, nil
+}