feat(rdb): add a boolean to reveal if a backup is exported (#2484)

scaleway · Sep 2, 2022 · 212374c · 212374c
1 parent 432cfad
commit 212374c
Show file tree

Hide file tree

Showing 5 changed files with 1,120 additions and 0 deletions.
diff --git a/internal/namespaces/rdb/v1/custom.go b/internal/namespaces/rdb/v1/custom.go
@@ -53,5 +53,7 @@ func GetCommands() *core.Commands {
 
 	cmds.MustFind("rdb", "user", "list").Override(userListBuilder)
 
+	cmds.MustFind("rdb", "backup", "list").Override(backupListBuilder)
+
 	return cmds
 }
diff --git a/internal/namespaces/rdb/v1/custom_backup.go b/internal/namespaces/rdb/v1/custom_backup.go
@@ -119,6 +119,72 @@ func backupRestoreBuilder(c *core.Command) *core.Command {
 	return c
 }
 
+func backupListBuilder(c *core.Command) *core.Command {
+	type customBackup struct {
+		ID         string                   `json:"ID"`
+		Name       string                   `json:"name"`
+		InstanceID string                   `json:"instance_ID"`
+		Exported   bool                     `json:"exported"`
+		Status     rdb.DatabaseBackupStatus `json:"status"`
+	}
+
+	c.View = &core.View{
+		Fields: []*core.ViewField{
+			{
+				Label:     "ID",
+				FieldName: "ID",
+			},
+			{
+				Label:     "Name",
+				FieldName: "Name",
+			},
+			{
+				Label:     "Status",
+				FieldName: "Status",
+			},
+			{
+				Label:     "Instance ID",
+				FieldName: "InstanceID",
+			},
+			{
+				Label:     "Exported",
+				FieldName: "Exported",
+			},
+		},
+	}
+
+	c.AddInterceptors(func(ctx context.Context, argsI interface{}, runner core.CommandRunner) (i interface{}, err error) {
+		listBackupResp, err := runner(ctx, argsI)
+		if err != nil {
+			return listBackupResp, err
+		}
+		backupList := listBackupResp.([]*rdb.DatabaseBackup)
+		var res []customBackup
+		for _, backup := range backupList {
+			res = append(res, customBackup{
+				ID:         backup.ID,
+				Name:       backup.Name,
+				Status:     backup.Status,
+				InstanceID: backup.InstanceID,
+				Exported:   isExported(backup.DownloadURLExpiresAt),
+			})
+		}
+		return res, nil
+	})
+
+	return c
+}
+
+func isExported(expirationDate *time.Time) bool {
+	var exported bool
+	if expirationDate == nil {
+		exported = false
+	} else {
+		exported = time.Now().Before(*expirationDate)
+	}
+	return exported
+}
+
 func getDefaultFileName(rawURL string) (string, error) {
 	u, err := url.Parse(rawURL)
 	if err != nil {

diff --git a/internal/namespaces/rdb/v1/custom_backup_test.go b/internal/namespaces/rdb/v1/custom_backup_test.go
@@ -111,3 +111,30 @@ func Test_DownloadBackup(t *testing.T) {
 		TmpHomeDir:    true,
 	}))
 }
+
+func Test_ListBackup(t *testing.T) {
+	t.Run("Simple", core.Test(&core.TestConfig{
+		Commands: GetCommands(),
+		BeforeFunc: core.BeforeFuncCombine(
+			createInstance(engine),
+			core.ExecStoreBeforeCmd(
+				"BackupA",
+				"scw rdb backup create name=will_be_exported expires-at=2999-01-02T15:04:05-07:00 instance-id={{ .Instance.ID }} database-name=rdb --wait",
+			),
+			core.ExecStoreBeforeCmd(
+				"BackupB",
+				"scw rdb backup create name=will_not_be_exported expires-at=2999-01-02T15:04:05-07:00 instance-id={{ .Instance.ID }} database-name=rdb --wait",
+			),
+			core.ExecStoreBeforeCmd(
+				"BackupExport",
+				"scw rdb backup export {{ .BackupA.ID }} --wait",
+			),
+		),
+		Cmd: "scw rdb backup list instance-id={{ .Instance.ID }}",
+		Check: core.TestCheckCombine(
+			core.TestCheckGolden(),
+			core.TestCheckExitCode(0),
+		),
+		AfterFunc: deleteInstance(),
+	}))
+}