snapshot.go

package cmd

import (
	"encoding/json"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"

	"github.com/spf13/cobra"

	"github.com/SUSE/cf-plugin-backup/models"
	"github.com/SUSE/cf-plugin-backup/util"
	termuiprogressbar "github.com/SUSE/termui/termprogressbar"
)

// snapshotCmd represents the snapshot command
var snapshotCmd = &cobra.Command{
	Use:   "backup-snapshot",
	Short: "Create a new CloudFoundry backup snapshot",
	Long: `Create a new CloudFoundry backup snapshot to a local file.
`,
	Run: func(cmd *cobra.Command, args []string) {
		var currentIndex int

		orgQuotas, err := util.GetOrgQuotaDefinitions(&util.CliConnectionCCApi{CliConnection: CliConnection})
		util.FreakOut(err)
		log.Println("org quota definitions done")
		spaceQuotas, err := util.GetSpaceQuotaDefinitions(&util.CliConnectionCCApi{CliConnection: CliConnection})
		util.FreakOut(err)
		log.Println("space quota definitions done")
		backupResources, err := util.GetOrgsResourcesRecurively(&util.CliConnectionCCApi{CliConnection: CliConnection})
		util.FreakOut(err)
		log.Println("orgs done")
		sharedDomains, err := util.GetSharedDomains(&util.CliConnectionCCApi{CliConnection: CliConnection})
		util.FreakOut(err)
		log.Println("shared domains done")
		securityGroups, err := util.GetSecurityGroups(&util.CliConnectionCCApi{CliConnection: CliConnection})
		util.FreakOut(err)
		log.Println("groups done")
		featureFlags, err := util.GetFeatureFlags(&util.CliConnectionCCApi{CliConnection: CliConnection})
		util.FreakOut(err)
		log.Println("feature flags done")

		backupJSON, err := util.CreateBackupJSON(models.BackupModel{
			OrgQuotas:      orgQuotas,
			SpaceQuotas:    spaceQuotas,
			Organizations:  backupResources,
			SharedDomains:  sharedDomains,
			SecurityGroups: securityGroups,
			FeatureFlags:   featureFlags,
		})

		util.FreakOut(err)

		err = ioutil.WriteFile(backupFile, []byte(backupJSON), 0644)
		util.FreakOut(err)

		// Save app bits

		packager := &util.CFPackager{
			Cli:    CliConnection,
			Writer: new(util.CFFileWriter),
			Reader: new(util.CFFileReader),
		}
		appBits := util.NewCFDroplet(CliConnection, packager)

		backupModel := models.BackupModel{}
		err = json.Unmarshal([]byte(backupJSON), &backupModel)
		util.FreakOut(err)

		err = os.Mkdir(filepath.Join(backupDir, backupAppBitsDir), 0755)
		if err != nil && !os.IsExist(err) {
			util.FreakOut(err)
		}

		var appsToBackup []*models.ResourceModel

		resources := util.RestoreOrgResourceModels(backupModel.Organizations)
		if resources != nil {
			for _, org := range *resources {
				if org.Entity["spaces"] != nil {
					for _, space := range *(org.Entity["spaces"].(*[]*models.ResourceModel)) {
						if space.Entity["apps"] != nil {
							for _, app := range *(space.Entity["apps"].(*[]*models.ResourceModel)) {
								if dockerImg, hit := app.Entity["docker_image"]; !hit || dockerImg == nil {
									appsToBackup = append(appsToBackup, app)
								}
							}
						}
					}
				}
			}
		}

		if len(appsToBackup) > 0 {

			log.Printf("Saving bits for %d apps", len(appsToBackup))

			termuiPGBar := termuiprogressbar.NewProgressBar(len(appsToBackup), true)
			termuiPGBar.Start()

			for _, app := range appsToBackup {
				appGUID := app.Metadata["guid"].(string)
				if currentIndex < len(appsToBackup) {
					termuiPGBar.Increment()
					currentIndex++
				}
				appZipPath := filepath.Join(backupDir, backupAppBitsDir, appGUID+".zip")
				err := appBits.SaveDroplet(appGUID, appZipPath)
				if err != nil {
					log.Printf("Could not save bits for %v: %v", appGUID, err)
				}
			}
			termuiPGBar.FinishPrint("App bits saved")
		}
	},
}

func init() {
	RootCmd.AddCommand(snapshotCmd)

	// Here you will define your flags and configuration settings.

	// Cobra supports Persistent Flags which will work for this command
	// and all subcommands, e.g.:
	// snapshotCmd.PersistentFlags().String("foo", "", "A help for foo")

	// Cobra supports local flags which will only run when this command
	// is called directly, e.g.:
	// snapshotCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}